unique_toolkit 1.21.1__tar.gz → 1.21.2__tar.gz

{unique_toolkit-1.21.1 → unique_toolkit-1.21.2}/CHANGELOG.md

@@ -5,6 +5,8 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.21.2] - 2025-10-30
+- Fixing that system format info is only appended to system prompt if tool is called
 
 ## [1.21.1] - 2025-10-30
 - Improve Spaces 2.0 display of tool progress reporter configuration.

{unique_toolkit-1.21.1 → unique_toolkit-1.21.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.21.1
+Version: 1.21.2
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,8 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.21.2] - 2025-10-30
+- Fixing that system format info is only appended to system prompt if tool is called
 
 ## [1.21.1] - 2025-10-30
 - Improve Spaces 2.0 display of tool progress reporter configuration.

{unique_toolkit-1.21.1 → unique_toolkit-1.21.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "1.21.1"
+version = "1.21.2"
 description = ""
 authors = [
     "Cedric Klinkert <cedric.klinkert@unique.ch>",

{unique_toolkit-1.21.1 → unique_toolkit-1.21.2}/unique_toolkit/agentic/history_manager/history_manager.py

@@ -120,9 +120,16 @@ class HistoryManager:
             reference_manager=reference_manager,
         )
         self._tool_call_result_history: list[ToolCallResponse] = []
+        self._tool_calls: list[LanguageModelFunction] = []
         self._loop_history: list[LanguageModelMessage] = []
         self._source_enumerator = 0
 
+    def add_tool_call(self, tool_call: LanguageModelFunction) -> None:
+        self._tool_calls.append(tool_call)
+
+    def get_tool_calls(self) -> list[LanguageModelFunction]:
+        return self._tool_calls
+
     def has_no_loop_messages(self) -> bool:
         return len(self._loop_history) == 0
 

{unique_toolkit-1.21.1 → unique_toolkit-1.21.2}/unique_toolkit/agentic/tools/tool_manager.py

@@ -1,7 +1,7 @@
 import asyncio
 from abc import ABC, abstractmethod
 from logging import Logger, getLogger
-from typing import override
+from typing import Literal, override
 
 from openai.types.chat import (
     ChatCompletionNamedToolChoiceParam,
@@ -70,6 +70,24 @@ class BaseToolManager(ABC):
     def get_exclusive_tools(self) -> list[str]:
         raise NotImplementedError()
 
+    @abstractmethod
+    def filter_tool_calls(
+        self,
+        tool_calls: list[LanguageModelFunction],
+        tool_types: list[Literal["mcp", "internal", "subagent"]],
+    ) -> list[LanguageModelFunction]:
+        """
+        Filter tool calls by their types.
+
+        Args:
+            tool_calls: List of tool calls to filter
+            tool_types: List of tool types to include (e.g., ["mcp", "internal", "subagent"])
+
+        Returns:
+            Filtered list of tool calls matching the specified types
+        """
+        raise NotImplementedError()
+
     def does_a_tool_take_control(self, tool_calls: list[LanguageModelFunction]) -> bool:
         for tool_call in tool_calls:
             tool_instance = self.get_tool_by_name(tool_call.name)
@@ -266,7 +284,7 @@ class ToolManager(BaseToolManager):
         )
 
         # Build internal tools from configurations
-        internal_tools = [
+        self._internal_tools = [
             ToolFactory.build_tool_with_settings(
                 t.name,
                 t,
@@ -278,9 +296,9 @@ class ToolManager(BaseToolManager):
         ]
 
         # Get MCP tools (these are already properly instantiated)
-        mcp_tools = self._mcp_manager.get_all_mcp_tools()
+        self._mcp_tools = self._mcp_manager.get_all_mcp_tools()
         # Combine both types of tools
-        self.available_tools = internal_tools + mcp_tools + sub_agents
+        self.available_tools = self._internal_tools + self._mcp_tools + sub_agents
         self._sub_agents = sub_agents
 
         for t in self.available_tools:
@@ -301,6 +319,29 @@ class ToolManager(BaseToolManager):
 
             self._tools.append(t)
 
+    @override
+    def filter_tool_calls(
+        self,
+        tool_calls: list[LanguageModelFunction],
+        tool_types: list[Literal["mcp", "internal", "subagent"]],
+    ) -> list[LanguageModelFunction]:
+        filtered_calls = []
+
+        # Build sets for efficient lookup
+        internal_tool_names = {tool.name for tool in self._internal_tools}
+        mcp_tool_names = {tool.name for tool in self._mcp_tools}
+        sub_agent_names = {tool.name for tool in self._sub_agents}
+
+        for tool_call in tool_calls:
+            if "internal" in tool_types and tool_call.name in internal_tool_names:
+                filtered_calls.append(tool_call)
+            elif "mcp" in tool_types and tool_call.name in mcp_tool_names:
+                filtered_calls.append(tool_call)
+            elif "subagent" in tool_types and tool_call.name in sub_agent_names:
+                filtered_calls.append(tool_call)
+
+        return filtered_calls
+
     @property
     def sub_agents(self) -> list[SubAgentTool]:
         return self._sub_agents
@@ -416,6 +457,15 @@ class ResponsesApiToolManager(BaseToolManager):
             builtin_tools=builtin_tools,
         )
 
+    @override
+    def filter_tool_calls(
+        self,
+        tool_calls: list[LanguageModelFunction],
+        tool_types: list[Literal["mcp", "internal", "subagent"]],
+    ) -> list[LanguageModelFunction]:
+        """Delegate filtering to the underlying tool manager."""
+        return self._tool_manager.filter_tool_calls(tool_calls, tool_types)
+
     @override
     def get_tool_by_name(self, name: str) -> Tool | None:
         return self._tool_manager.get_tool_by_name(name)