瀏覽代碼

fix: hunyuan assistant_prompt_message pydantic error (#7062)

Joe 8 月之前
父節點
當前提交
d7bb422a5c
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      api/core/model_runtime/model_providers/hunyuan/llm/llm.py

+ 1 - 1
api/core/model_runtime/model_providers/hunyuan/llm/llm.py

@@ -214,7 +214,7 @@ class HunyuanLargeLanguageModel(LargeLanguageModel):
     def _handle_chat_response(self, credentials, model, prompt_messages, response):
         usage = self._calc_response_usage(model, credentials, response.Usage.PromptTokens,
                                           response.Usage.CompletionTokens)
-        assistant_prompt_message = PromptMessage(role="assistant")
+        assistant_prompt_message = AssistantPromptMessage()
         assistant_prompt_message.content = response.Choices[0].Message.Content
         result = LLMResult(
             model=model,