unique_toolkit 0.8.30__tar.gz → 0.8.31__tar.gz

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/CHANGELOG.md

@@ -5,6 +5,18 @@ 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).
 
+
+## [0.8.31] - 2025-08-29
+- Add various openai models to supported model list
+  - o1
+  - o3
+  - o3-deep-research
+  - o3-pro
+  - o4-mini
+  - o4-mini-deep-research
+  - gpt-4-1-mini
+  - gpt-4-1-nano
+
 ## [0.8.30] - 2025-08-28
 - Added A2A manager
 

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.30
+Version: 0.8.31
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -115,6 +115,18 @@ 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).
 
+
+## [0.8.31] - 2025-08-29
+- Add various openai models to supported model list
+  - o1
+  - o3
+  - o3-deep-research
+  - o3-pro
+  - o4-mini
+  - o4-mini-deep-research
+  - gpt-4-1-mini
+  - gpt-4-1-nano
+
 ## [0.8.30] - 2025-08-28
 - Added A2A manager
 

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.8.30"
+version = "0.8.31"
 description = ""
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/language_model/infos.py

@@ -48,6 +48,14 @@ class LanguageModelName(StrEnum):
     LITELLM_OPENAI_GPT_5_MINI = "litellm:openai-gpt-5-mini"
     LITELLM_OPENAI_GPT_5_NANO = "litellm:openai-gpt-5-nano"
     LITELLM_OPENAI_GPT_5_CHAT = "litellm:openai-gpt-5-chat"
+    LITELLM_OPENAI_O1 = "litellm:openai-o1"
+    LITELLM_OPENAI_O3 = "litellm:openai-o3"
+    LITELLM_OPENAI_O3_DEEP_RESEARCH = "litellm:openai-o3-deep-research"
+    LITELLM_OPENAI_O3_PRO = "litellm:openai-o3-pro"
+    LITELLM_OPENAI_O4_MINI = "litellm:openai-o4-mini"
+    LITELLM_OPENAI_O4_MINI_DEEP_RESEARCH = "litellm:openai-o4-mini-deep-research"
+    LITELLM_OPENAI_GPT_4_1_MINI = "litellm:openai-gpt-4-1-mini"
+    LITELLM_OPENAI_GPT_4_1_NANO = "litellm:openai-gpt-4-1-nano"
     LITELLM_DEEPSEEK_R1 = "litellm:deepseek-r1"
     LITELLM_DEEPSEEK_V3 = "litellm:deepseek-v3-1"
     LITELLM_QWEN_3 = "litellm:qwen-3-235B-A22B"
@@ -83,6 +91,14 @@ def get_encoder_name(model_name: LanguageModelName) -> EncoderName:
             | LMN.LITELLM_OPENAI_GPT_5_MINI
             | LMN.LITELLM_OPENAI_GPT_5_NANO
             | LMN.LITELLM_OPENAI_GPT_5_CHAT
+            | LMN.LITELLM_OPENAI_O1
+            | LMN.LITELLM_OPENAI_O3
+            | LMN.LITELLM_OPENAI_O3_DEEP_RESEARCH
+            | LMN.LITELLM_OPENAI_O4_MINI
+            | LMN.LITELLM_OPENAI_O4_MINI_DEEP_RESEARCH
+            | LMN.LITELLM_OPENAI_GPT_4_1_MINI
+            | LMN.LITELLM_OPENAI_GPT_4_1_NANO
+            | LMN.LITELLM_OPENAI_O3_PRO
         ):
             return EncoderName.O200K_BASE
         case _:
@@ -879,6 +895,146 @@ class LanguageModelInfo(BaseModel):
                     deprecated_at=date(2026, 8, 7),
                     retirement_at=date(2026, 8, 7),
                 )
+            case LanguageModelName.LITELLM_OPENAI_O1:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2024-12-17",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                        ModelCapabilities.REASONING,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000, token_limit_output=100_000
+                    ),
+                    info_cutoff_at=date(2023, 10, 1),
+                    published_at=date(2024, 12, 17),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
+                )
+            case LanguageModelName.LITELLM_OPENAI_O3:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2025-04-16",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.REASONING,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000, token_limit_output=100_000
+                    ),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
+                    published_at=date(2025, 4, 16),
+                    info_cutoff_at=date(2024, 6, 1),
+                )
+            case LanguageModelName.LITELLM_OPENAI_O3_DEEP_RESEARCH:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2025-06-26",
+                    encoder_name=EncoderName.O200K_BASE,
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000, token_limit_output=100_000
+                    ),
+                    published_at=date(2025, 4, 16),
+                    capabilities=[ModelCapabilities.STREAMING],
+                    info_cutoff_at=date(2024, 6, 1),
+                )
+            case LanguageModelName.LITELLM_OPENAI_O3_PRO:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2025-06-10",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.REASONING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000, token_limit_output=100_000
+                    ),
+                    published_at=date(2025, 6, 10),
+                    info_cutoff_at=date(2024, 6, 1),
+                )
+            case LanguageModelName.LITELLM_OPENAI_O4_MINI:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2025-04-16",
+                    encoder_name=EncoderName.O200K_BASE,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                    ],
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000, token_limit_output=100_000
+                    ),
+                    published_at=date(2025, 4, 16),
+                    info_cutoff_at=date(2024, 6, 1),
+                    temperature_bounds=TemperatureBounds(
+                        min_temperature=1.0, max_temperature=1.0
+                    ),
+                )
+            case LanguageModelName.LITELLM_OPENAI_O4_MINI_DEEP_RESEARCH:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2025-06-26",
+                    encoder_name=EncoderName.O200K_BASE,
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000, token_limit_output=100_000
+                    ),
+                    published_at=date(2025, 4, 16),
+                    capabilities=[ModelCapabilities.STREAMING],
+                    info_cutoff_at=date(2024, 6, 1),
+                )
+            case LanguageModelName.LITELLM_OPENAI_GPT_4_1_MINI:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2025-04-14",
+                    encoder_name=EncoderName.O200K_BASE,
+                    published_at=date(2025, 4, 14),
+                    info_cutoff_at=date(2024, 6, 1),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=1_047_576, token_limit_output=32_768
+                    ),
+                    capabilities=[
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                    ],
+                )
+            case LanguageModelName.LITELLM_OPENAI_GPT_4_1_NANO:
+                return cls(
+                    name=model_name,
+                    provider=LanguageModelProvider.LITELLM,
+                    version="2025-04-14",
+                    encoder_name=EncoderName.O200K_BASE,
+                    published_at=date(2025, 4, 14),
+                    info_cutoff_at=date(2024, 6, 1),
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=1_047_576, token_limit_output=32_768
+                    ),
+                    capabilities=[
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STRUCTURED_OUTPUT,
+                    ],
+                )
             case LanguageModelName.LITELLM_DEEPSEEK_R1:
                 return cls(
                     name=model_name,

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/tools/a2a/config.py

@@ -1,7 +1,5 @@
-from unique_toolkit.tools.schemas import BaseToolConfig
-
 from unique_toolkit.tools.config import get_configuration_dict
-
+from unique_toolkit.tools.schemas import BaseToolConfig
 
 DEFAULT_PARAM_DESCRIPTION_SUB_AGENT_USER_MESSAGE = """
 This is the message that will be sent to the sub-agent.

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/tools/a2a/manager.py

@@ -1,9 +1,8 @@
 from logging import Logger
 
 from unique_toolkit.app.schemas import ChatEvent
-from unique_toolkit.tools.a2a.service import SubAgentTool, ToolProgressReporter
-
 from unique_toolkit.tools.a2a.config import SubAgentToolConfig
+from unique_toolkit.tools.a2a.service import SubAgentTool, ToolProgressReporter
 from unique_toolkit.tools.config import ToolBuildConfig
 from unique_toolkit.tools.schemas import BaseToolConfig
 from unique_toolkit.tools.tool import Tool

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/tools/a2a/memory.py

@@ -1,5 +1,4 @@
 from unique_toolkit import ShortTermMemoryService
-
 from unique_toolkit.short_term_memory.persistent_short_term_memory_manager import (
     PersistentShortMemoryManager,
 )

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/tools/a2a/service.py

@@ -1,25 +1,23 @@
 from pydantic import Field, create_model
 from unique_sdk.utils.chat_in_space import send_message_and_wait_for_completion
+
 from unique_toolkit.app import ChatEvent
+from unique_toolkit.evaluators.schemas import EvaluationMetricName
 from unique_toolkit.language_model import (
     LanguageModelFunction,
     LanguageModelMessage,
     LanguageModelToolDescription,
 )
-
-from unique_toolkit.evaluators.schemas import EvaluationMetricName
-from unique_toolkit.tools.schemas import ToolCallResponse
 from unique_toolkit.tools.a2a.config import SubAgentToolConfig
 from unique_toolkit.tools.a2a.memory import (
     get_sub_agent_short_term_memory_manager,
 )
 from unique_toolkit.tools.a2a.schema import (
     SubAgentShortTermMemorySchema,
-    SubAgentToolCallResponse,
     SubAgentToolInput,
 )
+from unique_toolkit.tools.schemas import ToolCallResponse
 from unique_toolkit.tools.tool import Tool
-
 from unique_toolkit.tools.tool_progress_reporter import (
     ProgressState,
     ToolProgressReporter,
@@ -148,7 +146,7 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
         tool_response: ToolCallResponse,
     ) -> LanguageModelMessage:
         return ToolCallResponse(
-            id=tool_call.id,  # type: ignore
-            name=tool_call.name,
-            content=response["text"],
+            id=tool_response.id,
+            name=tool_response.name,
+            content=tool_response["content"],
         )

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/tools/mcp/models.py

@@ -1,4 +1,5 @@
 from typing import Any, Dict, Optional
+
 from unique_toolkit.tools.schemas import BaseToolConfig
 
 

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/tools/test/test_mcp_manager.py

@@ -7,6 +7,7 @@ from pydantic import BaseModel
 from tests.test_obj_factory import get_event_obj
 from unique_toolkit.app.schemas import McpServer, McpTool
 from unique_toolkit.chat.service import ChatService
+from unique_toolkit.tools.a2a.manager import A2AManager
 from unique_toolkit.tools.config import ToolBuildConfig, ToolIcon, ToolSelectionPolicy
 from unique_toolkit.tools.factory import ToolFactory
 from unique_toolkit.tools.mcp.manager import MCPManager
@@ -15,8 +16,6 @@ from unique_toolkit.tools.tool import Tool
 from unique_toolkit.tools.tool_manager import ToolManager, ToolManagerConfig
 from unique_toolkit.tools.tool_progress_reporter import ToolProgressReporter
 
-from unique_toolkit.tools.a2a.manager import A2AManager
-
 
 class MockParameters(BaseModel):
     pass
@@ -163,9 +162,9 @@ class TestMCPManager:
             event=self.event,
             tool_progress_reporter=tool_progress_reporter,
         )
-    
+
     @pytest.fixture
-    def a2a_manager(self,tool_progress_reporter):
+    def a2a_manager(self, tool_progress_reporter):
         """Create MCP manager fixture"""
         return A2AManager(
             logger=self.logger,
@@ -178,17 +177,18 @@ class TestMCPManager:
         return ToolManagerConfig(tools=internal_tools, max_tool_calls=10)
 
     @pytest.fixture
-    def tool_manager(self, tool_manager_config, mcp_manager, a2a_manager,tool_progress_reporter):
+    def tool_manager(
+        self, tool_manager_config, mcp_manager, a2a_manager, tool_progress_reporter
+    ):
         """Create tool manager fixture"""
 
-
         return ToolManager(
             logger=self.logger,
             config=tool_manager_config,
             event=self.event,
             tool_progress_reporter=tool_progress_reporter,
             mcp_manager=mcp_manager,
-            a2a_manager=a2a_manager
+            a2a_manager=a2a_manager,
         )
 
     def test_tool_manager_initialization(self, tool_manager):
@@ -270,7 +270,6 @@ class TestMCPManager:
 
         # Test 1: Normal initialization with both tool types
 
-        
         a2a_manager = A2AManager(
             logger=self.logger,
             tool_progress_reporter=tool_progress_reporter,
@@ -282,7 +281,7 @@ class TestMCPManager:
             event=self.event,
             tool_progress_reporter=tool_progress_reporter,
             mcp_manager=mcp_manager,
-            a2a_manager=a2a_manager
+            a2a_manager=a2a_manager,
         )
 
         # Verify both tools are loaded
@@ -307,8 +306,6 @@ class TestMCPManager:
         event_with_disabled.payload.tool_choices = ["internal_search", "mcp_test_tool"]
         event_with_disabled.payload.disabled_tools = ["internal_search"]
 
-        
-        
         a2a_manager = A2AManager(
             logger=self.logger,
             tool_progress_reporter=tool_progress_reporter,
@@ -345,8 +342,6 @@ class TestMCPManager:
         event_with_limited_choices.payload.tool_choices = ["internal_search"]
         event_with_limited_choices.payload.disabled_tools = []
 
-
-        
         a2a_manager = A2AManager(
             logger=self.logger,
             tool_progress_reporter=tool_progress_reporter,
@@ -386,8 +381,6 @@ class TestMCPManager:
             tools=[exclusive_tool_config], max_tool_calls=10
         )
 
-        
-        
         a2a_manager = A2AManager(
             logger=self.logger,
             tool_progress_reporter=tool_progress_reporter,
@@ -399,7 +392,7 @@ class TestMCPManager:
             event=self.event,
             tool_progress_reporter=tool_progress_reporter,
             mcp_manager=mcp_manager,
-            a2a_manager=a2a_manager
+            a2a_manager=a2a_manager,
         )
 
         # Should only have the exclusive tool, MCP tools should be ignored
@@ -429,8 +422,6 @@ class TestMCPManager:
             tools=[disabled_tool_config], max_tool_calls=10
         )
 
-
-        
         a2a_manager = A2AManager(
             logger=self.logger,
             tool_progress_reporter=tool_progress_reporter,
@@ -442,7 +433,7 @@ class TestMCPManager:
             event=self.event,
             tool_progress_reporter=tool_progress_reporter,
             mcp_manager=mcp_manager,
-            a2a_manager=a2a_manager
+            a2a_manager=a2a_manager,
         )
 
         # Should only have MCP tool, disabled internal tool should be filtered out

{unique_toolkit-0.8.30 → unique_toolkit-0.8.31}/unique_toolkit/tools/tool_manager.py

@@ -2,10 +2,7 @@ import asyncio
 from logging import Logger, getLogger
 from typing import Any
 
-
 from pydantic import BaseModel, Field
-from unique_toolkit.tools.a2a.manager import A2AManager
-
 
 from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.evals.schemas import EvaluationMetricName
@@ -14,6 +11,7 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelTool,
     LanguageModelToolDescription,
 )
+from unique_toolkit.tools.a2a.manager import A2AManager
 from unique_toolkit.tools.config import ToolBuildConfig, _rebuild_config_model
 from unique_toolkit.tools.factory import ToolFactory
 from unique_toolkit.tools.mcp.manager import MCPManager