Fix the bug where Azure is not compatible with function role

yym68686 · May 27, 2024 · 71a8e2c · 71a8e2c
1 parent 0d4ddae
commit 71a8e2c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ModelMerge/models/chatgpt.py b/src/ModelMerge/models/chatgpt.py
@@ -360,7 +360,7 @@ def ask_stream(
                                 "role": mess["role"],
                                 "content": mess["content"][0]["text"]
                             }
-                if "'function' is not an allowed role" in response.text:
+                elif "'function' is not an allowed role" in response.text:
                     if json_post["messages"][-1]["role"] == "function":
                         mess = json_post["messages"][-1]
                         json_post["messages"][-1] = {