Update code for tool call

2026-02-10 16:28:20 +08:00
parent 539cf2fda2
commit 4b8da32787
7 changed files with 676 additions and 91 deletions
--- a/engine/models/ws_v1.py
+++ b/engine/models/ws_v1.py
@@ -39,12 +39,18 @@ class ResponseCancelMessage(BaseModel):
    graceful: bool = False


+class ToolCallResultsMessage(BaseModel):
+    type: Literal["tool_call.results"]
+    results: list[Dict[str, Any]] = Field(default_factory=list)
+
+
 CLIENT_MESSAGE_TYPES = {
    "hello": HelloMessage,
    "session.start": SessionStartMessage,
    "session.stop": SessionStopMessage,
    "input.text": InputTextMessage,
    "response.cancel": ResponseCancelMessage,
+    "tool_call.results": ToolCallResultsMessage,
 }