unique_toolkit 1.61.0__tar.gz → 1.62.0__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.61.0
+Version: 1.62.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.61.0 → unique_toolkit-1.62.0}/pyproject.toml

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

{unique_toolkit-1.61.0 → unique_toolkit-1.62.0}/unique_toolkit/agentic/debug_info_manager/debug_info_manager.py

@@ -3,6 +3,7 @@ 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.agentic.tools.tool_manager import ToolManager
 from unique_toolkit.language_model.schemas import (
     LanguageModelStreamResponse,
     ResponsesLanguageModelStreamResponse,
@@ -37,11 +38,12 @@ class DebugInfoManager:
     def extract_builtin_tool_debug_info(
         self,
         stream_response: LanguageModelStreamResponse,
+        tool_manager: ToolManager,
         loop_iteration_index: int | None = None,
     ) -> None:
         self.debug_info["tools"].extend(
             _extract_tool_calls_from_stream_response(
-                stream_response, loop_iteration_index
+                stream_response, tool_manager, loop_iteration_index
             )
         )
 
@@ -54,6 +56,7 @@ class DebugInfoManager:
 
 def _extract_tool_calls_from_stream_response(
     stream_response: LanguageModelStreamResponse,
+    tool_manager: ToolManager,
     loop_iteration_index: int | None = None,
 ) -> list[dict[str, Any]]:
     if not isinstance(stream_response, ResponsesLanguageModelStreamResponse):
@@ -67,14 +70,21 @@ def _extract_tool_calls_from_stream_response(
             continue
 
         seen.add(code_interpreter_call.id)
+        tool_name = OpenAIBuiltInToolName.CODE_INTERPRETER
+
+        is_exclusive = tool_name in tool_manager.get_exclusive_tools()
+        is_forced = tool_name in tool_manager.get_tool_choices()
 
         debug_info = OpenAICodeInterpreterTool.get_debug_info(code_interpreter_call)
+
         if loop_iteration_index is not None:
             debug_info["loop_iteration"] = loop_iteration_index
+        debug_info["is_exclusive"] = is_exclusive
+        debug_info["is_forced"] = is_forced
 
         tool_infos.append(
             {
-                "name": OpenAIBuiltInToolName.CODE_INTERPRETER,
+                "name": tool_name,
                 "info": debug_info,
             }
         )

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

@@ -280,4 +280,5 @@ class OpenAICodeInterpreterTool(OpenAIBuiltInTool[CodeInterpreter]):
         return {
             "id": call.id,
             "container_id": call.container_id,
+            "code": call.code,
         }