Add Role Check to Prevent Validation Error

RockChinQ · Sep 23, 2024 · 81c2c3c · 81c2c3c
1 parent 3c2db50
commit 81c2c3c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/provider/modelmgr/apis/chatcmpl.py b/pkg/provider/modelmgr/apis/chatcmpl.py
@@ -55,6 +55,10 @@ async def _make_msg(
     ) -> llm_entities.Message:
         chatcmpl_message = chat_completion.choices[0].message.dict()
 
+        # 确保 role 字段存在且不为 None
+        if 'role' not in chatcmpl_message or chatcmpl_message['role'] is None:
+            chatcmpl_message['role'] = 'assistant'
+
         message = llm_entities.Message(**chatcmpl_message)
 
         return message