Browse Source

fix: hunyuan assistant_prompt_message pydantic error (#7062)

Joe 8 tháng trước cách đây
mục cha
commit
d7bb422a5c
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  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,