Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] authored and JasonWeill committed Jan 25, 2024
1 parent e34d76c commit 177cb31
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
Original file line number Diff line number Diff line change
Expand Up @@ -484,7 +484,7 @@ def run_ai_cell(self, args: CellArgs, prompt: str):
# Get the output, either as raw text or as the contents of the 'text' key of a dict
invoke_output = self.custom_model_registry[args.model_id].invoke(prompt)
if isinstance(invoke_output, dict):
invoke_output = invoke_output.get('text')
invoke_output = invoke_output.get("text")

return self.display_output(
invoke_output,
Expand Down

0 comments on commit 177cb31

Please sign in to comment.