Fixed magics

jupyterlab · Oct 5, 2023 · f1c50e3 · f1c50e3
1 parent 14f426e
commit f1c50e3
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
@@ -556,7 +556,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