|
@@ -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}
|