Forráskód Böngészése

fix: hunyuan assistant_prompt_message pydantic error (#7062)

Joe 8 hónapja
szülő
commit
d7bb422a5c

+ 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,