diff --git a/api/app/controllers/memory_working_controller.py b/api/app/controllers/memory_working_controller.py index e5de3c04..8aab039a 100644 --- a/api/app/controllers/memory_working_controller.py +++ b/api/app/controllers/memory_working_controller.py @@ -8,6 +8,7 @@ from app.core.response_utils import success from app.db import get_db from app.dependencies import get_current_user from app.models import User +from app.schemas import conversation_schema from app.schemas.response_schema import ApiResponse from app.services.conversation_service import ConversationService @@ -90,11 +91,7 @@ def get_messages( conversation_id, ) messages = [ - { - "role": message.role, - "content": message.content, - "created_at": int(message.created_at.timestamp() * 1000), - } + conversation_schema.Message.model_validate(message) for message in messages_obj ] return success(data=messages, msg="get conversation history success") diff --git a/api/app/schemas/conversation_schema.py b/api/app/schemas/conversation_schema.py index 13766ef6..98715612 100644 --- a/api/app/schemas/conversation_schema.py +++ b/api/app/schemas/conversation_schema.py @@ -51,6 +51,10 @@ class Message(BaseModel): def _serialize_created_at(self, dt: datetime.datetime): return int(dt.timestamp() * 1000) if dt else None + @field_serializer("meta_data", when_used="json") + def _serialize_meta_data(self, data: Optional[Dict[str, Any]]): + return data or {} + class Conversation(BaseModel): """会话输出""" diff --git a/api/app/services/workflow_service.py b/api/app/services/workflow_service.py index 9f421976..6503fe25 100644 --- a/api/app/services/workflow_service.py +++ b/api/app/services/workflow_service.py @@ -802,30 +802,33 @@ class WorkflowService: final_messages = event.get("data", {}).get("messages", [])[init_message_length:] human_message = "" assistant_message = "" + human_meta = { + "files": [] + } for message in final_messages: if message["role"] == "user": if isinstance(message["content"], str): human_message += message["content"] elif isinstance(message["content"], list): for file in message["content"]: - if file.get("type") == FileType.IMAGE: - human_message += f"![image]({file.get('url', '')})" - else: - human_message += f"[{file.get('type')}]({file.get('url', '')})" + human_meta["files"].append({ + "type": file.get("type"), + "url": file.get("url") + }) if message["role"] == "assistant": assistant_message = message["content"] self.conversation_service.add_message( conversation_id=conversation_id_uuid, role="user", content=human_message, - meta_data=None + meta_data=human_meta ) self.conversation_service.add_message( message_id=message_id, conversation_id=conversation_id_uuid, role="assistant", content=assistant_message, - meta_data={"usage": token_usage} + meta_data={"usage": token_usage, "audio_url": None} ) self.update_execution_status( execution.execution_id,