unique_internal_search 2026.28.0.dev2__tar.gz → 2026.28.0.dev3__tar.gz

{unique_internal_search-2026.28.0.dev2 → unique_internal_search-2026.28.0.dev3}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.3
 Name: unique-internal-search
-Version: 2026.28.0.dev2
+Version: 2026.28.0.dev3
 Summary: 
 Author: Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri
 Author-email: Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>
 License: Proprietary
 Requires-Dist: pydantic>=2.8.2,<3
 Requires-Dist: typing-extensions>=4.9.0,<5
-Requires-Dist: unique-toolkit>=2026.28.0.dev5,<2026.28.0rc0
+Requires-Dist: unique-toolkit>=2026.28.0.dev7,<2026.28.0rc0
 Requires-Python: >=3.12, <4
 Description-Content-Type: text/markdown
 

{unique_internal_search-2026.28.0.dev2 → unique_internal_search-2026.28.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_internal_search"
-version = "2026.28.0.dev2"
+version = "2026.28.0.dev3"
 description = ""
 readme = "README.md"
 license = { text = "Proprietary" }
@@ -14,7 +14,7 @@ requires-python = ">=3.12,<4"
 dependencies = [
     "pydantic>=2.8.2,<3",
     "typing-extensions>=4.9.0,<5",
-    "unique-toolkit>=2026.28.0.dev5,<2026.28.0rc0",
+    "unique-toolkit>=2026.28.0.dev7,<2026.28.0rc0",
 ]
 
 [build-system]

{unique_internal_search-2026.28.0.dev2 → unique_internal_search-2026.28.0.dev3}/unique_internal_search/service.py

@@ -21,7 +21,7 @@ from unique_toolkit.agentic.tools.names import INTERNAL_SEARCH_TOOL_NAME
 from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 from unique_toolkit.agentic.tools.tool import Tool
 from unique_toolkit.agentic.tools.tool_progress_reporter import ProgressState
-from unique_toolkit.app.schemas import BaseEvent, ChatEvent, Event
+from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.chat.service import LanguageModelToolDescription
 from unique_toolkit.content.schemas import Content, ContentChunk
 from unique_toolkit.content.service import ContentService
@@ -361,7 +361,7 @@ class InternalSearchTool(Tool[InternalSearchConfig], InternalSearchService):
     def __init__(
         self,
         configuration: InternalSearchConfig,
-        event: BaseEvent,
+        event: ChatEvent,
         *args,
         **kwargs,
     ):
@@ -370,13 +370,10 @@ class InternalSearchTool(Tool[InternalSearchConfig], InternalSearchService):
         content_service = ContentService.from_event(self.event)
         chunk_relevancy_sorter = ChunkRelevancySorter.from_event(self.event)
         selected_uploaded_file_ids = extract_selected_uploaded_file_ids(self.event)
-        if isinstance(self.event, (ChatEvent, Event)):
-            if self.event.payload.correlation:
-                chat_id = self.event.payload.correlation.parent_chat_id
-            else:
-                chat_id = self.event.payload.chat_id
+        if self.event.payload.correlation:
+            chat_id = self.event.payload.correlation.parent_chat_id
         else:
-            chat_id = None
+            chat_id = self.event.payload.chat_id
         self._display_name = kwargs.get("display_name", "Internal Search")
         InternalSearchService.__init__(
             self,

{unique_internal_search-2026.28.0.dev2 → unique_internal_search-2026.28.0.dev3}/unique_internal_search/uploaded_search/service.py

@@ -14,7 +14,7 @@ from unique_toolkit.agentic.tools.tool_progress_reporter import (
     ProgressState,
     ToolProgressReporter,
 )
-from unique_toolkit.app.schemas import BaseEvent, ChatEvent
+from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.chat.service import LanguageModelToolDescription
 from unique_toolkit.content import Content
 from unique_toolkit.language_model.schemas import (
@@ -33,7 +33,7 @@ class UploadedSearchTool(Tool[UploadedSearchConfig]):
     def __init__(
         self,
         config: UploadedSearchConfig,
-        event: BaseEvent,
+        event: ChatEvent,
         tool_progress_reporter: ToolProgressReporter,
         *args,
         **kwargs,
@@ -48,10 +48,7 @@ class UploadedSearchTool(Tool[UploadedSearchConfig]):
         )
         self._internal_search_tool._display_name = self._display_name
         self._selected_uploaded_files = extract_selected_uploaded_file_ids(event)
-        if isinstance(event, ChatEvent):
-            self._user_query = event.payload.user_message.text
-        else:
-            self._user_query = None
+        self._user_query = event.payload.user_message.text
 
         # This blocking API call should be avoided.
         # However, we don't have an easy way to pass user uploaded files to the tool currently