Преглед на файлове

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,