bug fix: fixed a bug that will cause system prompt message failed

Maplemx · Apr 3, 2024 · 1f9d275 · 1f9d275
1 parent 9ea6fa6
commit 1f9d275
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/plugins/request/OAIClient.py b/src/plugins/request/OAIClient.py
@@ -86,7 +86,9 @@ def format_request_messages(self, request_messages):
             if message["role"] == "system":
                 # no_multi_system_messages=True
                 if self.model_settings.get_trace_back("message_rules.no_multi_system_messages"):
-                    system_prompt += f"{ message['content'] }\n"
+                    for content in message["content"]:
+                        if content["type"] == "text":
+                            system_prompt += f"{ content['text'] }\n"
                 # no_multi_system_messages=False
                 else:
                     system_messages.append(message)
@@ -107,7 +109,13 @@ def format_request_messages(self, request_messages):
                     chat_messages.append(message)
         # no_multi_system_messages=True
         if self.model_settings.get_trace_back("message_rules.no_multi_system_messages") and system_prompt != "":
-            system_messages.append({ "role": "system", "content": system_prompt })
+            system_messages.append({
+                "role": "system",
+                "content": [{
+                    "type": "text",
+                    "text": system_prompt
+                }]
+            })
         formatted_messages = system_messages.copy()
         formatted_messages.extend(chat_messages)
         # no_multi_type_messages=True