feat(agent): 增强多模态链路与工具调用能力

2026-03-12 00:18:45 +08:00
parent 18db6c50e7
commit 21ba8e4a44
35 changed files with 2057 additions and 829 deletions
@@ -23,10 +23,12 @@ class MessageRepository:
        role: AgentChatMessageRole,
        content: str,
        model_code: str | None = None,
+        tool_name: str | None = None,
        metadata: dict[str, object] | None = None,
        input_tokens: int = 0,
        output_tokens: int = 0,
        cost: Decimal = Decimal("0"),
+        latency_ms: int | None = None,
    ) -> AgentChatMessage:
        message = AgentChatMessage(
            session_id=session_id,
@@ -34,10 +36,12 @@ class MessageRepository:
            role=role,
            content=content,
            model_code=model_code,
+            tool_name=tool_name,
            metadata_json=metadata,
            input_tokens=input_tokens,
            output_tokens=output_tokens,
            cost=cost,
+            latency_ms=latency_ms,
        )
        self._session.add(message)
        await self._session.flush()