unique_deep_research 2026.28.0.dev4__tar.gz → 2026.28.0.dev5__tar.gz

{unique_deep_research-2026.28.0.dev4 → unique_deep_research-2026.28.0.dev5}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.3
 Name: unique-deep-research
-Version: 2026.28.0.dev4
+Version: 2026.28.0.dev5
 Summary: Deep Research Tool for complex research tasks
 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.dev13,<2026.28.0rc0
+Requires-Dist: unique-toolkit>=2026.28.0.dev18,<2026.28.0rc0
 Requires-Dist: jinja2>=3.1.2,<4
 Requires-Dist: openai>=1.99.0,<3
 Requires-Dist: langgraph>=1.0.0,<2

{unique_deep_research-2026.28.0.dev4 → unique_deep_research-2026.28.0.dev5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_deep_research"
-version = "2026.28.0.dev4"
+version = "2026.28.0.dev5"
 description = "Deep Research Tool for complex research tasks"
 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.dev13,<2026.28.0rc0",
+    "unique-toolkit>=2026.28.0.dev18,<2026.28.0rc0",
     "jinja2>=3.1.2,<4",
     "openai>=1.99.0,<3",
     "langgraph>=1.0.0,<2",

{unique_deep_research-2026.28.0.dev4 → unique_deep_research-2026.28.0.dev5}/unique_deep_research/service.py

@@ -28,6 +28,7 @@ from unique_toolkit.agentic.tools.agent_chunks_hanlder import AgentChunksHandler
 from unique_toolkit.agentic.tools.factory import ToolFactory
 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 ToolProgressReporter
 from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.chat.cancellation import CancellationEvent
 from unique_toolkit.chat.schemas import (
@@ -38,10 +39,11 @@ from unique_toolkit.chat.schemas import (
     MessageLogStatus,
     MessageLogUncitedReferences,
 )
-from unique_toolkit.chat.service import LanguageModelToolDescription
+from unique_toolkit.chat.service import ChatService, LanguageModelToolDescription
 from unique_toolkit.content.schemas import ContentReference
 from unique_toolkit.content.service import ContentService
 from unique_toolkit.framework_utilities.openai.client import get_async_openai_client
+from unique_toolkit.language_model import LanguageModelService
 from unique_toolkit.language_model.schemas import (
     LanguageModelFunction,
     LanguageModelMessage,
@@ -107,9 +109,21 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
         self,
         configuration: DeepResearchToolConfig,
         event: ChatEvent,
-        tool_progress_reporter,
+        tool_progress_reporter: ToolProgressReporter | None = None,
+        *,
+        chat_service: ChatService | None = None,
+        language_model_service: LanguageModelService | None = None,
     ):
-        super().__init__(configuration, event, tool_progress_reporter)
+        if chat_service is not None and language_model_service is not None:
+            super().__init__(
+                configuration,
+                event,
+                tool_progress_reporter,
+                chat_service=chat_service,
+                language_model_service=language_model_service,
+            )
+        else:
+            super().__init__(configuration, event, tool_progress_reporter)
         self.chat_id = event.payload.chat_id
         self.company_id = event.company_id
         self.user_id = event.user_id
@@ -290,7 +304,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
         if await self.is_followup_question_answer() and not self.is_message_execution():
             _LOGGER.info("This is a follow-up question answer")
             self.chat_service.create_message_execution(
-                message_id=self.event.payload.assistant_message.id,
+                message_id=self.chat_service.assistant_message_id,
                 type=MessageExecutionType.DEEP_RESEARCH,
             )
             self.write_message_log_text_message(
@@ -347,7 +361,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
 
         # put message in short term memory to remember that we asked the followup questions
         await self.memory_service.save_async(
-            MemorySchema(message_id=self.event.payload.assistant_message.id),
+            MemorySchema(message_id=self.chat_service.assistant_message_id),
         )
         return ToolCallResponse(
             id=tool_call.id or "",
@@ -369,7 +383,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             percentage = 100 if status == MessageExecutionUpdateStatus.COMPLETED else 0
 
         await self.chat_service.update_message_execution_async(
-            message_id=self.event.payload.assistant_message.id,
+            message_id=self.chat_service.assistant_message_id,
             status=status,
             percentage_completed=percentage,
         )
@@ -405,7 +419,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
     def write_message_log_text_message(self, text: str):
         create_message_log_entry(
             self.chat_service,
-            self.event.payload.assistant_message.id,
+            self.chat_service.assistant_message_id,
             text,
             MessageLogStatus.COMPLETED,
             details=MessageLogDetails(data=[]),
@@ -468,7 +482,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             "notes": [],
             "final_report": "",
             "chat_service": self.chat_service,
-            "message_id": self.event.payload.assistant_message.id,
+            "message_id": self.chat_service.assistant_message_id,
         }
 
         # Prepare configuration for LangGraph
@@ -486,7 +500,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                 "openai_client": self.client,
                 "chat_service": self.chat_service,
                 "content_service": self.content_service,
-                "message_id": self.event.payload.assistant_message.id,
+                "message_id": self.chat_service.assistant_message_id,
                 "citation_manager": citation_manager,
                 "additional_openai_proxy_headers": additional_openai_proxy_headers,
             },
@@ -652,11 +666,11 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                         if isinstance(event.item, ResponseReasoningItem):
                             for summary in event.item.summary:
                                 self.chat_service.create_message_log(
-                                    message_id=self.event.payload.assistant_message.id,
+                                    message_id=self.chat_service.assistant_message_id,
                                     text=summary.text,
                                     status=MessageLogStatus.COMPLETED,
                                     order=get_next_message_order(
-                                        self.event.payload.assistant_message.id
+                                        self.chat_service.assistant_message_id
                                     ),
                                     uncited_references=MessageLogUncitedReferences(
                                         data=[],
@@ -671,11 +685,11 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                             ) and isinstance(event.item.action.query, str):
                                 _LOGGER.info("OpenAI web search")
                                 self.chat_service.create_message_log(
-                                    message_id=self.event.payload.assistant_message.id,
+                                    message_id=self.chat_service.assistant_message_id,
                                     text="**Searching the web**",
                                     status=MessageLogStatus.COMPLETED,
                                     order=get_next_message_order(
-                                        self.event.payload.assistant_message.id
+                                        self.chat_service.assistant_message_id
                                     ),
                                     uncited_references=MessageLogUncitedReferences(
                                         data=[],
@@ -718,11 +732,11 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                                     )
                                     continue
                                 self.chat_service.create_message_log(
-                                    message_id=self.event.payload.assistant_message.id,
+                                    message_id=self.chat_service.assistant_message_id,
                                     text="**Reading website**",
                                     status=MessageLogStatus.COMPLETED,
                                     order=get_next_message_order(
-                                        self.event.payload.assistant_message.id
+                                        self.chat_service.assistant_message_id
                                     ),
                                     uncited_references=MessageLogUncitedReferences(
                                         data=[
@@ -745,11 +759,11 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                                 )
                     case "response.failed":
                         self.chat_service.create_message_log(
-                            message_id=self.event.payload.assistant_message.id,
+                            message_id=self.chat_service.assistant_message_id,
                             text=f"Failed to complete research: {event.response.error}",
                             status=MessageLogStatus.FAILED,
                             order=get_next_message_order(
-                                self.event.payload.assistant_message.id
+                                self.chat_service.assistant_message_id
                             ),
                             uncited_references=MessageLogUncitedReferences(
                                 data=[],