Fixed magics

jupyterlab · Oct 5, 2023 · e4bb7e4 · e4bb7e4
1 parent f8a0ab2
commit e4bb7e4
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