unique_toolkit 1.60.0__tar.gz → 1.61.0__tar.gz

{unique_toolkit-1.60.0 → unique_toolkit-1.61.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.60.0
+Version: 1.61.0
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-1.60.0 → unique_toolkit-1.61.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.60.0"
+version = "1.61.0"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

unique_toolkit-1.61.0/unique_toolkit/agentic/debug_info_manager/debug_info_manager.py

@@ -0,0 +1,82 @@
+from typing import Any
+
+from unique_toolkit.agentic.tools.openai_builtin import OpenAICodeInterpreterTool
+from unique_toolkit.agentic.tools.openai_builtin.base import OpenAIBuiltInToolName
+from unique_toolkit.agentic.tools.schemas import ToolCallResponse
+from unique_toolkit.language_model.schemas import (
+    LanguageModelStreamResponse,
+    ResponsesLanguageModelStreamResponse,
+)
+
+
+class DebugInfoManager:
+    def __init__(self):
+        self.debug_info = {"tools": []}
+
+    def extract_tool_debug_info(
+        self,
+        tool_call_responses: list[ToolCallResponse],
+        loop_iteration_index: int | None = None,
+    ) -> None:
+        for tool_call_response in tool_call_responses:
+            debug_info = (
+                tool_call_response.debug_info.copy()
+                if tool_call_response.debug_info
+                else {}
+            )
+            tool_info: dict = {
+                "name": tool_call_response.name,
+                "info": debug_info,
+            }
+            if debug_info.get("mcp_server"):
+                tool_info["mcp_server"] = debug_info["mcp_server"]
+            if loop_iteration_index is not None:
+                tool_info["info"]["loop_iteration"] = loop_iteration_index
+            self.debug_info["tools"].append(tool_info)
+
+    def extract_builtin_tool_debug_info(
+        self,
+        stream_response: LanguageModelStreamResponse,
+        loop_iteration_index: int | None = None,
+    ) -> None:
+        self.debug_info["tools"].extend(
+            _extract_tool_calls_from_stream_response(
+                stream_response, loop_iteration_index
+            )
+        )
+
+    def add(self, key: str, value: Any) -> None:
+        self.debug_info = self.debug_info | {key: value}
+
+    def get(self) -> dict[str, Any]:
+        return self.debug_info
+
+
+def _extract_tool_calls_from_stream_response(
+    stream_response: LanguageModelStreamResponse,
+    loop_iteration_index: int | None = None,
+) -> list[dict[str, Any]]:
+    if not isinstance(stream_response, ResponsesLanguageModelStreamResponse):
+        return []
+
+    seen = set()
+    tool_infos = []
+
+    for code_interpreter_call in stream_response.code_interpreter_calls:
+        if code_interpreter_call.id in seen:
+            continue
+
+        seen.add(code_interpreter_call.id)
+
+        debug_info = OpenAICodeInterpreterTool.get_debug_info(code_interpreter_call)
+        if loop_iteration_index is not None:
+            debug_info["loop_iteration"] = loop_iteration_index
+
+        tool_infos.append(
+            {
+                "name": OpenAIBuiltInToolName.CODE_INTERPRETER,
+                "info": debug_info,
+            }
+        )
+
+    return tool_infos

{unique_toolkit-1.60.0 → unique_toolkit-1.61.0}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/service.py

@@ -1,7 +1,8 @@
 import logging
-from typing import override
+from typing import Any, override
 
 from openai import AsyncOpenAI, BaseModel, NotFoundError
+from openai.types.responses import ResponseCodeInterpreterToolCall
 from openai.types.responses.tool_param import CodeInterpreter
 
 from unique_toolkit import ContentService, ShortTermMemoryService
@@ -273,3 +274,10 @@ class OpenAICodeInterpreterTool(OpenAIBuiltInTool[CodeInterpreter]):
     @override
     def display_name(self) -> str:
         return self.DISPLAY_NAME
+
+    @classmethod
+    def get_debug_info(cls, call: ResponseCodeInterpreterToolCall) -> dict[str, Any]:
+        return {
+            "id": call.id,
+            "container_id": call.container_id,
+        }

unique_toolkit-1.60.0/unique_toolkit/agentic/debug_info_manager/debug_info_manager.py

@@ -1,35 +0,0 @@
-from typing import Any
-
-from unique_toolkit.agentic.tools.schemas import ToolCallResponse
-
-
-class DebugInfoManager:
-    def __init__(self):
-        self.debug_info = {"tools": []}
-
-    def extract_tool_debug_info(
-        self,
-        tool_call_responses: list[ToolCallResponse],
-        loop_iteration_index: int | None = None,
-    ):
-        for tool_call_response in tool_call_responses:
-            debug_info = (
-                tool_call_response.debug_info.copy()
-                if tool_call_response.debug_info
-                else {}
-            )
-            tool_info: dict = {
-                "name": tool_call_response.name,
-                "info": debug_info,
-            }
-            if debug_info.get("mcp_server"):
-                tool_info["mcp_server"] = debug_info["mcp_server"]
-            if loop_iteration_index is not None:
-                tool_info["info"]["loop_iteration"] = loop_iteration_index
-            self.debug_info["tools"].append(tool_info)
-
-    def add(self, key: str, value: Any) -> None:
-        self.debug_info = self.debug_info | {key: value}
-
-    def get(self) -> dict[str, Any]:
-        return self.debug_info