Fixed magics

jupyterlab · Oct 5, 2023 · 5f62aee · 5f62aee
1 parent 971c2e5
commit 5f62aee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
@@ -536,7 +536,7 @@ def run_ai_cell(self, args: CellArgs, prompt: str):
         ip = get_ipython()
         prompt = prompt.format_map(FormatDict(ip.user_ns))
 
-        if provider.is_chat_provider(provider):
+        if provider.is_chat_provider:
             result = provider.generate([[HumanMessage(content=prompt)]])
         else:
             # generate output from model via provider