From f1c50e3b176a5108b31c387ae93f25b700f7b6ae Mon Sep 17 00:00:00 2001 From: Piyush Jain Date: Wed, 4 Oct 2023 23:07:08 -0700 Subject: [PATCH] Fixed magics --- 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 025d32dba..6f1d29546 100644 --- 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