🐛 Bug: Fix the bug of extra request body fields when all function cal…

…ls are turned off.
yym68686 · May 20, 2024 · 8de5c9f · 8de5c9f
1 parent 7c11fce
commit 8de5c9f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 
 setup(
     name="modelmerge",
-    version="0.5.0",
+    version="0.5.1",
     description="modelmerge is a multi-large language model API aggregator.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

diff --git a/src/ModelMerge/models/chatgpt.py b/src/ModelMerge/models/chatgpt.py
@@ -353,8 +353,11 @@ def ask_stream(
                 return
             # print("response.text", response.text)
             if response.status_code == 400:
-                del json_post["function_call"]
-                del json_post["functions"]
+                print("response.text", response.text)
+                if "function_call" in json_post:
+                    del json_post["function_call"]
+                if "functions" in json_post:
+                    del json_post["functions"]
                 continue
             if response.status_code == 200:
                 break