fix: hunyuan assistant_prompt_message pydantic error (#7062)

2 years ago · d7bb422a5c
parent 40c6f3c724
commit d7bb422a5c
1 changed files with 1 additions and 1 deletions
--- a/api/core/model_runtime/model_providers/hunyuan/llm/llm.py
+++ b/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,