Fix Assistant schema

artitw · Oct 7, 2024 · fcd9fcc · fcd9fcc
1 parent b838be8
commit fcd9fcc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.6.9",
+  version="1.7.0",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",

diff --git a/text2text/assistant.py b/text2text/assistant.py
@@ -84,12 +84,12 @@ def chat_completion(self, messages=[{"role": "user", "content": "hello"}], strea
       if not result or not result.get("models"):
         warnings.warn("No model loaded. Retrying...")
         self.load_model()
-        return self.chat_completion(messages=messages, stream=stream, **kwargs)
+        return self.chat_completion(messages=messages, stream=stream, schema=schema, **kwargs)
     except Exception as e:
       warnings.warn(str(e))
       warnings.warn("Retrying...")
       self.load_model()
-      return self.chat_completion(messages=messages, stream=stream, **kwargs)
+      return self.chat_completion(messages=messages, stream=stream, schema=schema, **kwargs)
 
     if schema:
       msgs = [ChatMessage(**m) for m in messages]