unique_deep_research 3.2.0__tar.gz → 3.2.2__tar.gz

{unique_deep_research-3.2.0 → unique_deep_research-3.2.2}/CHANGELOG.md

@@ -5,6 +5,12 @@ 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).
 
+## [3.2.2] - 2026-02-05
+- Fix bug where deep research tool was not using the correct headers causing an authentication error
+
+## [3.2.1] - 2026-02-05
+- Use deep research logger instead of tool logger
+
 ## [3.2.0] - 2026-02-03
 - Use a backwards compatible config style
 

{unique_deep_research-3.2.0 → unique_deep_research-3.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_deep_research
-Version: 3.2.0
+Version: 3.2.2
 Summary: Deep Research Tool for complex research tasks
 License: Proprietary
 Author: Martin Fadler
@@ -36,6 +36,12 @@ 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).
 
+## [3.2.2] - 2026-02-05
+- Fix bug where deep research tool was not using the correct headers causing an authentication error
+
+## [3.2.1] - 2026-02-05
+- Use deep research logger instead of tool logger
+
 ## [3.2.0] - 2026-02-03
 - Use a backwards compatible config style
 

{unique_deep_research-3.2.0 → unique_deep_research-3.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_deep_research"
-version = "3.2.0"
+version = "3.2.2"
 description = "Deep Research Tool for complex research tasks"
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",

{unique_deep_research-3.2.0 → unique_deep_research-3.2.2}/unique_deep_research/service.py

@@ -1,3 +1,4 @@
+import logging
 from typing import Any, Optional
 
 from httpx import AsyncClient
@@ -65,6 +66,8 @@ from .unique_custom.utils import (
     get_next_message_order,
 )
 
+_LOGGER = logging.getLogger(__name__)
+
 
 class DeepResearchToolInput(BaseModel):
     model_config = ConfigDict(extra="forbid")
@@ -107,9 +110,16 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
         self.company_id = event.company_id
         self.user_id = event.user_id
 
-        self.client = get_async_openai_client()
+        self.client = get_async_openai_client(
+            additional_headers={
+                "x-company-id": self.company_id,
+                "x-user-id": self.user_id,
+                "x-assistant-id": self.event.payload.assistant_id,
+                "x-chat-id": self.chat_id,
+            }
+        )
 
-        self.logger.info(f"Using async OpenAI client pointed to {self.client.base_url}")
+        _LOGGER.info(f"Using async OpenAI client pointed to {self.client.base_url}")
 
         self.content_service = ContentService(
             company_id=self.company_id,
@@ -207,7 +217,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
         except Exception as e:
             if self.is_message_execution():
                 await self._update_execution_status(MessageExecutionUpdateStatus.FAILED)
-            self.logger.exception(f"Deep Research tool run failed: {e}")
+            _LOGGER.exception(f"Deep Research tool run failed: {e}")
             await self.chat_service.modify_assistant_message_async(
                 content="Deep Research failed to complete for an unknown reason",
                 set_completed_at=True,
@@ -220,13 +230,13 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
         )
 
     async def _run(self, tool_call: LanguageModelFunction) -> ToolCallResponse:
-        self.logger.info("Starting Deep Research tool run")
+        _LOGGER.info("Starting Deep Research tool run")
 
         await self._clear_original_message()
 
         # Question answer and message execution will have the same message id, so we need to check if it is a message execution
         if await self.is_followup_question_answer() and not self.is_message_execution():
-            self.logger.info("This is a follow-up question answer")
+            _LOGGER.info("This is a follow-up question answer")
             self.chat_service.create_message_execution(
                 message_id=self.event.payload.assistant_message.id,
                 type=MessageExecutionType.DEEP_RESEARCH,
@@ -240,7 +250,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                 content="",
             )
         if self.is_message_execution():
-            self.logger.info("Starting research")
+            _LOGGER.info("Starting research")
             # Run research
             self.write_message_log_text_message("**Generating research plan**")
             research_brief = await self.generate_research_brief_from_dict(
@@ -386,15 +396,15 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             result = "", []
             match self.config.engine.get_type():
                 case DeepResearchEngine.OPENAI:
-                    self.logger.info("Running OpenAI research")
+                    _LOGGER.info("Running OpenAI research")
                     result = await self.openai_research(research_brief)
                 case DeepResearchEngine.UNIQUE:
-                    self.logger.info("Running Custom research")
+                    _LOGGER.info("Running Custom research")
                     result = await self.custom_research(research_brief)
             self.write_message_log_text_message("**Research done**")
             return result
         except Exception as e:
-            self.logger.exception(f"Research failed: {e}")
+            _LOGGER.exception(f"Research failed: {e}")
             return "", []
 
     async def custom_research(self, research_brief: str) -> tuple[str, list[Any]]:
@@ -464,14 +474,14 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                 content=processed_result,
                 references=references,
             )
-            self.logger.info(
+            _LOGGER.info(
                 f"Custom research completed with {len(references)} validated citations"
             )
             return processed_result, []
 
         except Exception as e:
             error_msg = f"Custom research failed: {str(e)}"
-            self.logger.exception(error_msg)
+            _LOGGER.exception(error_msg)
             return error_msg, []
 
     async def openai_research(self, research_brief: str) -> tuple[str, list[Any]]:
@@ -553,14 +563,14 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                 match event.type:
                     case "response.completed":
                         if event.response.usage:
-                            self.logger.info(
+                            _LOGGER.info(
                                 f"OpenAI research token usage: {event.response.usage}"
                             )
                         # Extract the final output with annotations
                         if event.response.output and len(event.response.output) > 0:
                             final_output = event.response.output[-1]
                             if not isinstance(final_output, ResponseOutputMessage):
-                                self.logger.warning(
+                                _LOGGER.warning(
                                     f"Unexpected output type: {type(final_output)}"
                                 )
                                 continue
@@ -573,9 +583,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                             # Extract final report and references
                             report_text = content_item.text
                             annotations = content_item.annotations or []
-                            self.logger.info(
-                                "Final report extracted from OpenAI stream"
-                            )
+                            _LOGGER.info("Final report extracted from OpenAI stream")
                             return report_text, annotations
                         return event.response.output_text or "", []
                     case "response.incomplete":
@@ -607,7 +615,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                             if isinstance(
                                 event.item.action, ActionSearch
                             ) and isinstance(event.item.action.query, str):
-                                self.logger.info("OpenAI web search")
+                                _LOGGER.info("OpenAI web search")
                                 self.chat_service.create_message_log(
                                     message_id=self.event.payload.assistant_message.id,
                                     text="**Searching the web**",
@@ -630,13 +638,13 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                             elif isinstance(
                                 event.item.action, ActionOpenPage
                             ) or isinstance(event.item.action, ActionFind):
-                                self.logger.info("OpenAI reading web page")
+                                _LOGGER.info("OpenAI reading web page")
                                 if (
                                     not event.item.action.url
                                     or not isinstance(event.item.action.url, str)
                                     or "https://" not in event.item.action.url
                                 ):
-                                    self.logger.warning(
+                                    _LOGGER.warning(
                                         f"Invalid URL from OpenAI: {event.item.action}"
                                     )
                                     continue
@@ -646,12 +654,12 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                                     event.item.action.url,
                                 )
                                 if not success:
-                                    self.logger.info(
+                                    _LOGGER.info(
                                         f"Failed to crawl URL: {event.item.action.url} but openai still opened the page"
                                     )
                                     continue
                                 if not title:
-                                    self.logger.info(
+                                    _LOGGER.info(
                                         f"No title found for URL: {event.item.action.url}"
                                     )
                                     continue
@@ -678,7 +686,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                                     ),
                                 )
                             else:
-                                self.logger.info(
+                                _LOGGER.info(
                                     f"OpenAI web action unexpected type: {type(event.item)}"
                                 )
                     case "response.failed":
@@ -699,9 +707,9 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                         if event.response.error:
                             return event.response.error.message, []
             except Exception as e:
-                self.logger.exception(f"Error processing research stream event: {e}")
+                _LOGGER.exception(f"Error processing research stream event: {e}")
 
-        self.logger.error("Stream ended without completion")
+        _LOGGER.error("Stream ended without completion")
         return "", []
 
     async def _postprocess_report_with_gpt(self, research_result: str) -> str:
@@ -737,10 +745,10 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
 
         formatted_result = response.choices[0].message.content
         if formatted_result:
-            self.logger.info("Successfully post-processed research report")
+            _LOGGER.info("Successfully post-processed research report")
             return formatted_result
         else:
-            self.logger.warning("Post-processing returned empty result, using original")
+            _LOGGER.warning("Post-processing returned empty result, using original")
             return research_result
 
     def get_tool_call_result_for_loop_history(