qzl
1c02503d1d
- 移除冗余的 user_token 参数传递 - 重构 tool.result 事件使用 ToolAgentOutput 模型 - 重构 text.end 事件使用 WorkerAgentOutput 模型 - 简化 store 模块的 tool result 处理逻辑 - 更新 router/service 适配新事件结构 - 清理废弃的测试文件与设计文档 - 新增 AgentRuns 多模态存储设计文档
30 lines
837 B
Python
30 lines
837 B
Python
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
from uuid import uuid4
|
|
|
|
from schemas.messages.chat_message import AgentChatMessage
|
|
|
|
|
|
def test_agent_chat_message_schema_matches_messages_columns() -> None:
|
|
now = datetime.now(UTC)
|
|
payload = {
|
|
"id": uuid4(),
|
|
"seq": 3,
|
|
"role": "assistant",
|
|
"content": "hello",
|
|
"metadata": {"run_id": "run-1"},
|
|
"timestamp": now,
|
|
}
|
|
|
|
message = AgentChatMessage.model_validate(payload)
|
|
|
|
assert message.seq == 3
|
|
assert message.role == "assistant"
|
|
assert message.content == "hello"
|
|
assert message.metadata is not None
|
|
if isinstance(message.metadata, dict):
|
|
assert message.metadata == {"run_id": "run-1"}
|
|
else:
|
|
assert message.metadata.model_dump(exclude_none=True) == {"run_id": "run-1"}
|