mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 01:33:05 +00:00
fix: update message handling in image.py to support multiple message types
This commit is contained in:
parent
79fce87e93
commit
28d403f237
|
|
@ -104,15 +104,28 @@ class QwenVLChatModel(MaxKBBaseModel, BaseChatOpenAI):
|
|||
"Content-Type": "application/json",
|
||||
"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 = {
|
||||
"model": self.model_name,
|
||||
"messages": [
|
||||
{
|
||||
"role": "user",
|
||||
"content": input[0].content
|
||||
}
|
||||
],
|
||||
"messages": messages,
|
||||
**self.extra_body,
|
||||
"stream": True,
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue