Skip to content

Commit 28d403f

Browse files
committed
fix: update message handling in image.py to support multiple message types
1 parent 79fce87 commit 28d403f

File tree

1 file changed

+19
-6
lines changed
  • apps/models_provider/impl/aliyun_bai_lian_model_provider/model

1 file changed

+19
-6
lines changed

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/image.py

Lines changed: 19 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -104,15 +104,28 @@ def stream(
104104
"Content-Type": "application/json",
105105
"X-DashScope-OssResourceResolve": "enable"
106106
}
107+
# 遍历input 获取所有的content 构造新的消息体
108+
messages = []
109+
for message in input:
110+
if message.type == "human":
111+
messages.append({
112+
"role": "user",
113+
"content": message.content
114+
})
115+
elif message.type == "ai":
116+
messages.append({
117+
"role": "assistant",
118+
"content": message.content
119+
})
120+
elif message.type == "system":
121+
messages.append({
122+
"role": "system",
123+
"content": message.content
124+
})
107125

108126
data = {
109127
"model": self.model_name,
110-
"messages": [
111-
{
112-
"role": "user",
113-
"content": input[0].content
114-
}
115-
],
128+
"messages": messages,
116129
**self.extra_body,
117130
"stream": True,
118131
}

0 commit comments

Comments
 (0)