浏览代码

Merge pull request #4342 from michaelpoluektov/fix-non-streaming-function-text

fix: function returning "str"
Timothy Jaeryang Baek 9 月之前
父节点
当前提交
05977bd1c7
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      backend/utils/misc.py

+ 3 - 2
backend/utils/misc.py

@@ -97,18 +97,19 @@ def openai_chat_message_template(model: str):
     }
 
 
-def openai_chat_chunk_message_template(model: str, message: str):
+def openai_chat_chunk_message_template(model: str, message: str) -> dict:
     template = openai_chat_message_template(model)
     template["object"] = "chat.completion.chunk"
     template["choices"][0]["delta"] = {"content": message}
     return template
 
 
-def openai_chat_completion_message_template(model: str, message: str):
+def openai_chat_completion_message_template(model: str, message: str) -> dict:
     template = openai_chat_message_template(model)
     template["object"] = "chat.completion"
     template["choices"][0]["message"] = {"content": message, "role": "assistant"}
     template["choices"][0]["finish_reason"] = "stop"
+    return template
 
 
 def get_gravatar_url(email):