From 177cb31f729d395ab07427f10aee2f833de19f0d Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Thu, 25 Jan 2024 00:00:40 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 packages/jupyter-ai-magics/jupyter_ai_magics/magics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
index da635021a..51b1f3bb9 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
@@ -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,