From d7bb422a5cc89941e5eb8a8c9bb1bc5630a67274 Mon Sep 17 00:00:00 2001
From: Joe <79627742+ZhouhaoJiang@users.noreply.github.com>
Date: Wed, 7 Aug 2024 18:31:40 +0800
Subject: [PATCH] fix: hunyuan assistant_prompt_message pydantic error (#7062)

---
 api/core/model_runtime/model_providers/hunyuan/llm/llm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/core/model_runtime/model_providers/hunyuan/llm/llm.py b/api/core/model_runtime/model_providers/hunyuan/llm/llm.py
index 6d22f9d2d6..0bdf6ec005 100644
--- 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,