qzl
19e273a9e6
- calendar: split write → create/read/update/delete/share - contacts: rename lookup → read - memory: merge write+forget → update (unified action field in operations) - Remove all alias/normalization logic from adapter and handlers - Update tool_postprocessor ui_hints builders to canonical keys - Remove frontend legacy TOOL_CALL_START/ARGS/END events and ToolCallItem - Update SKILL.md files and protocol docs - Update tests and settings screens
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
from uuid import uuid4
|
|
|
|
from v1.agent.utils import convert_message_to_history
|
|
|
|
|
|
class _FakeMessage:
|
|
def __init__(self, *, role: str, metadata: dict[str, object] | None) -> None:
|
|
self.id = uuid4()
|
|
self.seq = 1
|
|
self.role = role
|
|
self.content = "content"
|
|
self.metadata = metadata
|
|
self.timestamp = datetime.now(timezone.utc)
|
|
|
|
|
|
def test_convert_message_to_history_attaches_ui_schema_for_tool_message() -> None:
|
|
message = _FakeMessage(
|
|
role="tool",
|
|
metadata={
|
|
"tool_agent_output": {
|
|
"result": {"status": "success"},
|
|
"ui_hints": {
|
|
"intent": "status",
|
|
"status": "success",
|
|
"title": "完成",
|
|
"items": [],
|
|
"listItems": [],
|
|
"sections": [],
|
|
"actions": [],
|
|
},
|
|
}
|
|
},
|
|
)
|
|
|
|
result = convert_message_to_history(message) # type: ignore[arg-type]
|
|
|
|
assert "ui_schema" in result
|
|
|
|
|
|
def test_convert_message_to_history_adds_suggested_actions_for_assistant_message() -> None:
|
|
message = _FakeMessage(
|
|
role="assistant",
|
|
metadata={
|
|
"agent_output": {
|
|
"suggested_actions": ["查今天日程", "创建会议"]
|
|
}
|
|
},
|
|
)
|
|
|
|
result = convert_message_to_history(message) # type: ignore[arg-type]
|
|
|
|
assert result["suggestedActions"] == ["查今天日程", "创建会议"]
|
|
|
|
|
|
def test_convert_message_to_history_returns_multiple_user_attachments() -> None:
|
|
message = _FakeMessage(
|
|
role="user",
|
|
metadata={
|
|
"user_message_attachments": [
|
|
{"bucket": "bucket-a", "path": "path/a.png", "mime_type": "image/png"},
|
|
{"bucket": "bucket-a", "path": "path/b.jpg", "mime_type": "image/jpeg"},
|
|
]
|
|
},
|
|
)
|
|
|
|
def _signed(payload: dict[str, str]) -> str:
|
|
return f"https://signed.example/{payload['bucket']}/{payload['path']}"
|
|
|
|
result = convert_message_to_history(message, _signed) # type: ignore[arg-type]
|
|
|
|
attachments = result.get("attachments")
|
|
assert isinstance(attachments, list)
|
|
assert len(attachments) == 2
|
|
assert attachments[0]["mimeType"] == "image/png"
|
|
assert attachments[1]["mimeType"] == "image/jpeg"
|