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/core/session.py
+++ b/engine/core/session.py
@@ -28,6 +28,7 @@ from models.ws_v1 import (
    SessionStopMessage,
    InputTextMessage,
    ResponseCancelMessage,
+    ToolCallResultsMessage,
 )


@@ -174,6 +175,8 @@ class Session:
                logger.info(f"Session {self.id} graceful response.cancel")
            else:
                await self.pipeline.interrupt()
+        elif isinstance(message, ToolCallResultsMessage):
+            await self.pipeline.handle_tool_call_results(message.results)
        elif isinstance(message, SessionStopMessage):
            await self._handle_session_stop(message.reason)
        else: