fix: update message handling in image.py to support multiple message types

v3.2
wxg0103 2025-10-28 16:24:21 +08:00
parent 79fce87e93
commit 28d403f237
1 changed files with 19 additions and 6 deletions

View File

@ -104,15 +104,28 @@ class QwenVLChatModel(MaxKBBaseModel, BaseChatOpenAI):
"Content-Type": "application/json", "Content-Type": "application/json",
"X-DashScope-OssResourceResolve": "enable" "X-DashScope-OssResourceResolve": "enable"
} }
# 遍历input 获取所有的content 构造新的消息体
messages = []
for message in input:
if message.type == "human":
messages.append({
"role": "user",
"content": message.content
})
elif message.type == "ai":
messages.append({
"role": "assistant",
"content": message.content
})
elif message.type == "system":
messages.append({
"role": "system",
"content": message.content
})
data = { data = {
"model": self.model_name, "model": self.model_name,
"messages": [ "messages": messages,
{
"role": "user",
"content": input[0].content
}
],
**self.extra_body, **self.extra_body,
"stream": True, "stream": True,
} }