Explorar el Código

fix: volcengine_maas and baichuan message error (#11625)

Co-authored-by: zhongliliu <liuzlx@digitalchina.com>
tags/0.14.0
zhongliliu-butterfly hace 10 meses
padre
commit
daccb10d8c
No account linked to committer's email address

+ 6
- 1
api/core/model_runtime/model_providers/baichuan/llm/llm.py Ver fichero

@@ -10,6 +10,7 @@ from core.model_runtime.entities.llm_entities import (
from core.model_runtime.entities.message_entities import (
AssistantPromptMessage,
PromptMessage,
PromptMessageContentType,
PromptMessageTool,
SystemPromptMessage,
ToolPromptMessage,
@@ -105,7 +106,11 @@ class BaichuanLanguageModel(LargeLanguageModel):
if isinstance(message.content, str):
message_dict = {"role": "user", "content": message.content}
else:
raise ValueError("User message content must be str")
for message_content in message.content:
if message_content.type == PromptMessageContentType.TEXT:
message_dict = {"role": "user", "content": message_content.data}
elif message_content.type == PromptMessageContentType.IMAGE:
raise ValueError("Content object type not support image_url")
elif isinstance(message, AssistantPromptMessage):
message = cast(AssistantPromptMessage, message)
message_dict = {"role": "assistant", "content": message.content}

+ 6
- 1
api/core/model_runtime/model_providers/volcengine_maas/legacy/client.py Ver fichero

@@ -68,7 +68,12 @@ class MaaSClient(MaasService):
content = []
for message_content in message.content:
if message_content.type == PromptMessageContentType.TEXT:
raise ValueError("Content object type only support image_url")
content.append(
{
"type": "text",
"text": message_content.data,
}
)
elif message_content.type == PromptMessageContentType.IMAGE:
message_content = cast(ImagePromptMessageContent, message_content)
image_data = re.sub(r"^data:image\/[a-zA-Z]+;base64,", "", message_content.data)

Cargando…
Cancelar
Guardar