unique_toolkit 0.8.30__py3-none-any.whl → 0.8.32__py3-none-any.whl

unique_toolkit/tools/agent_chunks_hanlder.py

@@ -0,0 +1,65 @@
+from typing_extensions import deprecated
+
+from unique_toolkit.content.schemas import ContentChunk, ContentReference
+
+
+@deprecated("do not use this its only used in old tools")
+class AgentChunksHandler:
+    def __init__(self):
+        self._tool_chunks = {}
+        self._chunks: list[ContentChunk] = []
+        self._references: list[list[ContentReference]] = []
+
+    @property
+    def chunks(self) -> list[ContentChunk]:
+        return self._chunks
+
+    @property
+    def tool_chunks(self) -> dict:
+        return self._tool_chunks
+
+    def extend(self, chunks: list[ContentChunk]):
+        self._chunks.extend(chunks)
+
+    def replace(self, chunks: list[ContentChunk]):
+        self._chunks = chunks
+
+    def add_references(
+        self,
+        references: list[ContentReference],
+    ):
+        self._references.append(references)
+
+    @property
+    def all_references(
+        self,
+    ) -> list[list[ContentReference]]:
+        return self._references
+
+    @property
+    def latest_references(
+        self,
+    ) -> list[ContentReference]:
+        if not self._references:
+            return []
+        return self._references[-1]
+
+    @property
+    def latest_referenced_chunks(self) -> list[ContentChunk]:
+        if not self._references:
+            return []
+        return self._get_referenced_chunks_from_references(self._references[-1])
+
+    def _get_referenced_chunks_from_references(
+        self,
+        references: list[ContentReference],
+    ) -> list[ContentChunk]:
+        """
+        Get _referenced_chunks by matching sourceId from _references with merged id and chunk_id from _chunks.
+        """
+        referenced_chunks: list[ContentChunk] = []
+        for ref in references:
+            for chunk in self._chunks:
+                if ref.source_id == str(chunk.id) + "_" + str(chunk.chunk_id):
+                    referenced_chunks.append(chunk)
+        return referenced_chunks

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/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/tools/tool.py

@@ -15,6 +15,7 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelMessage,
 )
 from unique_toolkit.language_model.service import LanguageModelService
+from unique_toolkit.tools.agent_chunks_hanlder import AgentChunksHandler
 from unique_toolkit.tools.config import ToolBuildConfig, ToolSelectionPolicy
 from unique_toolkit.tools.schemas import BaseToolConfig, ToolCallResponse, ToolPrompts
 from unique_toolkit.tools.tool_progress_reporter import ToolProgressReporter
@@ -80,6 +81,7 @@ class Tool(ABC, Generic[ConfigType]):
     def get_tool_call_result_for_loop_history(
         self,
         tool_response: ToolCallResponse,
+        agent_chunks_handler: AgentChunksHandler,
     ) -> LanguageModelMessage:
         raise NotImplementedError
 

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

{unique_toolkit-0.8.30.dist-info → unique_toolkit-0.8.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.30
+Version: 0.8.32
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -115,6 +115,22 @@ 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-30
+
+moved over general packages for `web_search`
+
+## [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.dist-info → unique_toolkit-0.8.32.dist-info}/RECORD

@@ -1,11 +1,18 @@
 unique_toolkit/__init__.py,sha256=nbOYPIKERt-ITsgifrnJhatn1YNR38Ntumw-dCn_tsA,714
 unique_toolkit/_common/_base_service.py,sha256=S8H0rAebx7GsOldA7xInLp3aQJt9yEPDQdsGSFRJsGg,276
 unique_toolkit/_common/_time_utils.py,sha256=ztmTovTvr-3w71Ns2VwXC65OKUUh-sQlzbHdKTQWm-w,135
+unique_toolkit/_common/chunk_relevancy_sorter/config.py,sha256=v6Ljo-WIZCtYJgfaPfpzZegCV0DEw_nNhTzNtw0Jg7c,1744
+unique_toolkit/_common/chunk_relevancy_sorter/exception.py,sha256=1mY4zjbvnXsd5oIxwiVsma09bS2XRnHrxW8KJBGtgCM,126
+unique_toolkit/_common/chunk_relevancy_sorter/schemas.py,sha256=doAWPPx8d0zIqHMXmnJy47Z5_NlblJBhMqo8KE7fyyc,1329
+unique_toolkit/_common/chunk_relevancy_sorter/service.py,sha256=UxYn4xJMNEXQ1afMiT2sMwXgdmlFNPaglVhx6CRRtiM,13864
+unique_toolkit/_common/chunk_relevancy_sorter/tests/test_service.py,sha256=JRvLtJXPmz6bm1tFSSqt01HHVeanSD8zk70XVLJHOEM,8878
 unique_toolkit/_common/default_language_model.py,sha256=tmHSqg6e8G7RmKqmdE_tmLxkSN0x-aGoyUdy6Pl2oAE,334
 unique_toolkit/_common/endpoint_builder.py,sha256=oM6uDmxUqTAJut6MuJQj3bIX4yOccyErWD5bJ1d1lcY,4526
 unique_toolkit/_common/exception.py,sha256=caQIE1btsQnpKCHqL2cgWUSbHup06enQu_Pt7uGUTTE,727
+unique_toolkit/_common/feature_flags/schema.py,sha256=3JpTuld8kK-UQ5B0sbYTu0yqhyFPnChXG2Iv4BNqHdg,539
 unique_toolkit/_common/token/image_token_counting.py,sha256=VpFfZyY0GIH27q_Wy4YNjk2algqvbCtJyzuuROoFQPw,2189
 unique_toolkit/_common/token/token_counting.py,sha256=Jo5B11GLlnaZDi4u7xCzIboLl8zn5cY_dmrogHixVdk,6132
+unique_toolkit/_common/utils/structured_output/schema.py,sha256=Tp7kDYcmKtnUhcuRkH86TSYhylRff0ZZJYb2dLkISts,131
 unique_toolkit/_common/validate_required_values.py,sha256=Y_M1ub9gIKP9qZ45F6Zq3ZHtuIqhmOjl8Z2Vd3avg8w,588
 unique_toolkit/_common/validators.py,sha256=aZwbMho7XszN7lT5RtemaiXgC0WJ4u40oeVgsNGhF4U,2803
 unique_toolkit/app/__init__.py,sha256=ETxYDpEizg_PKmi4JPX_P76ySq-us-xypfAIdKQ1QZU,1284
@@ -39,8 +46,8 @@ unique_toolkit/embedding/service.py,sha256=2KjYlUKxeh-Je8S1mOPiQ735pqAmdbeztycVN
 unique_toolkit/embedding/utils.py,sha256=v86lo__bCJbxZBQ3OcLu5SuwT6NbFfWlcq8iyk6BuzQ,279
 unique_toolkit/evals/config.py,sha256=ywHIrJs5SFdKr1WXfrofWuFfzb0iPQw8iZDpq5oEug4,953
 unique_toolkit/evals/context_relevancy/prompts.py,sha256=EdHFUOB581yVxcOL8482KUv_LzaRjuiem71EF8udYMc,1331
-unique_toolkit/evals/context_relevancy/schema.py,sha256=lm9x0jExOinUk9itqC8ZpgReC7yj1VDwEMppxlZGqpY,2923
-unique_toolkit/evals/context_relevancy/service.py,sha256=txTWIhV65QGFhxG1jCb5TTqZc_c7K9I8pi2HVDTAfm8,8384
+unique_toolkit/evals/context_relevancy/schema.py,sha256=ILA0ClTBFJbtZavkVIrqPz2vj6oql8U50e0G4bh8jEI,2911
+unique_toolkit/evals/context_relevancy/service.py,sha256=xoe0ezA4xy-KaPjSO22xhiqo_WKMuYjlYFKEYJWl25A,9601
 unique_toolkit/evals/evaluation_manager.py,sha256=g-8qa_6_p53C9Okx8iNkuoIXYSJrf-6sQ-xku7bo9kI,7895
 unique_toolkit/evals/exception.py,sha256=7lcVbCyoN4Md1chNJDFxpUYyWbVrcr9dcc3TxWykJTc,115
 unique_toolkit/evals/hallucination/constants.py,sha256=FLcXl5XU07jCvS8YPX9l6UjTaqyQ8YvnSKpx4Z6wZ2Y,1997
@@ -81,7 +88,7 @@ unique_toolkit/language_model/__init__.py,sha256=lRQyLlbwHbNFf4-0foBU13UGb09lwEe
 unique_toolkit/language_model/builder.py,sha256=4OKfwJfj3TrgO1ezc_ewIue6W7BCQ2ZYQXUckWVPPTA,3369
 unique_toolkit/language_model/constants.py,sha256=B-topqW0r83dkC_25DeQfnPk3n53qzIHUCBS7YJ0-1U,119
 unique_toolkit/language_model/functions.py,sha256=4-zOzLsdjcfeTy6alqkYEBl-oVWptz9xLi8C5vdbWEg,16769
-unique_toolkit/language_model/infos.py,sha256=2roY4jqVd5p11Cavc4JLqh8uqpynOKjs2_eb5OJhqkk,48810
+unique_toolkit/language_model/infos.py,sha256=fKmpTEvrHotGFo-PHkLpdlmk57UyFLCkuPeTic-Zohg,56304
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
 unique_toolkit/language_model/reference.py,sha256=nkX2VFz-IrUz8yqyc3G5jUMNwrNpxITBrMEKkbqqYoI,8583
 unique_toolkit/language_model/schemas.py,sha256=EOgy-p1GRcS46Sq0qEsN8MfOMl-KCcvEd9aCmqm9d08,16497
@@ -100,28 +107,29 @@ unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
 unique_toolkit/thinking_manager/thinking_manager.py,sha256=AJfmrTXTr-DxBnJ2_zYYpYo57kr5deqT0LiZb8UdaDQ,4175
 unique_toolkit/tools/a2a/__init__.py,sha256=BIPxaqnfTD9IywxrqTLbGnkqNQrfLnKbdzF5ELYzCxM,171
-unique_toolkit/tools/a2a/config.py,sha256=NjLBt8Bmxmh0XSU6W20_c-0jZoj4x8EX7Uodgyg1F7Y,866
-unique_toolkit/tools/a2a/manager.py,sha256=nUEhjQmBh49rHWXAMDiRxR62vaPQdbSaBwaa_upUSEA,1667
-unique_toolkit/tools/a2a/memory.py,sha256=cP4vaVAtL0IrlBT8QlYl72M6HCrds_Q7QRCIrwwP71A,1005
+unique_toolkit/tools/a2a/config.py,sha256=DKerB7gJ0oUFfqdvVN4I174EZYfYzt7ET2NYGkMLd0Q,864
+unique_toolkit/tools/a2a/manager.py,sha256=-Vt0pL886zMaaG4sGQhXmbwOhOL928WbZzm4FRZuZMM,1666
+unique_toolkit/tools/a2a/memory.py,sha256=F18kUA3m3NqoKdKAJSwDv8JQneHvZTGOkcZTLNMXAYs,1004
 unique_toolkit/tools/a2a/schema.py,sha256=1R7qIu2l2qnUJDGRuUnZLqPPkHnT3x4d3d4PALoBzcY,296
-unique_toolkit/tools/a2a/service.py,sha256=bQv6gooQyP_wvwEV_7Bhkvq4kHcLXtQfMQWov5UPT9A,5178
+unique_toolkit/tools/a2a/service.py,sha256=8yiG2zYqHva4fQJX4dfnE9Bm4vDnbK0k6YRyCSQy9RQ,5147
+unique_toolkit/tools/agent_chunks_hanlder.py,sha256=x32Dp1Z8cVW5i-XzXbaMwX2KHPcNGmqEU-FB4AV9ZGo,1909
 unique_toolkit/tools/config.py,sha256=nYwglwUSb3fxhqYxI83k-qoecuF5Zcol5FMOxHvTNeE,3827
 unique_toolkit/tools/factory.py,sha256=w3uNHuYBIJ330Xi8PTdAkr8G3OMbQH2cBgvk5UT16oE,1253
 unique_toolkit/tools/mcp/__init__.py,sha256=RLF_p-LDRC7GhiB3fdCi4u3bh6V9PY_w26fg61BLyco,122
 unique_toolkit/tools/mcp/manager.py,sha256=lQcCsfCMqW2j6uCwNmgYoQDwjm8YosvCh4-apw3KZKs,2683
-unique_toolkit/tools/mcp/models.py,sha256=6-qFq3ip2OhqRC5H6YzhsC27w5CYrWln75dWmeIovRk,713
+unique_toolkit/tools/mcp/models.py,sha256=f9OXcGadBx2tDlhYFu-7oEkl5p9ppoDKTqZbca_NblQ,714
 unique_toolkit/tools/mcp/tool_wrapper.py,sha256=w7Fbo4FSMYvtgSq7Sqt1dmAPvqHjoBQS-TwnSp0p6j0,9927
 unique_toolkit/tools/schemas.py,sha256=rArQccbfIv7CWcozClAZ-BVlOwAsjpgL8KUab_WeO3k,4817
-unique_toolkit/tools/test/test_mcp_manager.py,sha256=kzEq4cdhV4vI8FvqKBCvMVXSyJRMrUdedg14Ps5g8ko,15678
+unique_toolkit/tools/test/test_mcp_manager.py,sha256=dySiytBUfRjTOzwY_oGKi_jT6BNVgzZuh1du-EvbcJ4,15627
 unique_toolkit/tools/test/test_tool_progress_reporter.py,sha256=GTtmBqOUo0-4fh_q0lRgxDhwKeankc3FHFD5ULZAm4Y,6299
-unique_toolkit/tools/tool.py,sha256=E0qORpQBrdksAylf_RuPaEbXKi_AlswVGv-67t3DNZI,5879
-unique_toolkit/tools/tool_manager.py,sha256=9gncHvsv_8bHAeufV99u6V_lskoT5QG8vT_RP5AA_-0,10208
+unique_toolkit/tools/tool.py,sha256=QvsNsiJ6bes4ezKin3r039oGCilvqtlthZm1nsfCjKk,6002
+unique_toolkit/tools/tool_manager.py,sha256=uW1uf8mYvbnBN_TVa4L79p_8Sf9HyY4V_nXZqrh3fPM,10206
 unique_toolkit/tools/tool_progress_reporter.py,sha256=ixud9VoHey1vlU1t86cW0-WTvyTwMxNSWBon8I11SUk,7955
 unique_toolkit/tools/utils/execution/execution.py,sha256=vjG2Y6awsGNtlvyQAGCTthQ5thWHYnn-vzZXaYLb3QE,7922
 unique_toolkit/tools/utils/source_handling/schema.py,sha256=vzAyf6ZWNexjMO0OrnB8y2glGkvAilmGGQXd6zcDaKw,870
 unique_toolkit/tools/utils/source_handling/source_formatting.py,sha256=C7uayNbdkNVJdEARA5CENnHtNY1SU6etlaqbgHNyxaQ,9152
 unique_toolkit/tools/utils/source_handling/tests/test_source_formatting.py,sha256=oM5ZxEgzROrnX1229KViCAFjRxl9wCTzWZoinYSHleM,6979
-unique_toolkit-0.8.30.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.8.30.dist-info/METADATA,sha256=3HA-gaW3UOJjBHHeFK9DyuMzHzFi529HIABflY9Y5CI,29342
-unique_toolkit-0.8.30.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.8.30.dist-info/RECORD,,
+unique_toolkit-0.8.32.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.8.32.dist-info/METADATA,sha256=zGeIqWysPTGI5qrDXXz2JqvpgaXSXXXbz-NXS8LHWeM,29611
+unique_toolkit-0.8.32.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.8.32.dist-info/RECORD,,