unique_deep_research 3.0.27__py3-none-any.whl → 3.1.0__py3-none-any.whl

unique_deep_research/config.py

@@ -1,13 +1,19 @@
 from dataclasses import dataclass
 from enum import StrEnum
 from pathlib import Path
-from typing import Literal
+from typing import Any, Generic, Literal, TypeVar
 
 from jinja2 import Environment, FileSystemLoader
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, field_validator
 from unique_toolkit._common.validators import LMI, get_LMI_default_field
+from unique_toolkit.agentic.tools.config import get_configuration_dict
 from unique_toolkit.agentic.tools.schemas import BaseToolConfig
 from unique_toolkit.language_model.infos import LanguageModelName
+from unique_web_search.config import (
+    ActivatedSearchEngine,
+    DefaultSearchEngine,
+)
+from unique_web_search.services.search_engine import GoogleConfig
 
 # Global template environment for the deep research tool
 TEMPLATE_DIR = Path(__file__).parent / "templates"
@@ -32,10 +38,14 @@ class UniqueCustomEngineConfig:
 RESPONSES_API_TIMEOUT_SECONDS = 3600
 
 
-class BaseEngine(BaseModel):
-    engine_type: Literal[DeepResearchEngine.UNIQUE, DeepResearchEngine.OPENAI] = Field(
-        description="The type of engine to use for deep research"
-    )
+T = TypeVar("T", bound=DeepResearchEngine)
+
+
+class BaseEngine(BaseModel, Generic[T]):
+    model_config = get_configuration_dict()
+
+    engine_type: T = Field(description="The type of engine to use for deep research")
+
     small_model: LMI = get_LMI_default_field(
         LanguageModelName.AZURE_GPT_4o_2024_1120,
         description="A smaller fast model for less demanding tasks",
@@ -55,7 +65,9 @@ class BaseEngine(BaseModel):
         return DeepResearchEngine(self.engine_type)
 
 
-class OpenAIEngine(BaseEngine):
+class OpenAIEngine(BaseEngine[Literal[DeepResearchEngine.OPENAI]]):
+    model_config = get_configuration_dict()
+
     engine_type: Literal[DeepResearchEngine.OPENAI] = Field(
         default=DeepResearchEngine.OPENAI
     )
@@ -65,18 +77,51 @@ class OpenAIEngine(BaseEngine):
     )
 
 
-class Tools(BaseModel):
-    web_tools: bool = Field(
+class WebToolsConfig(BaseModel):
+    model_config = get_configuration_dict()
+
+    enable: bool = Field(
         default=True,
         description="Allow agent to use web search tools to access the web",
     )
+
+    search_engine: ActivatedSearchEngine = Field(  # pyright: ignore[reportInvalidTypeForm]
+        default_factory=DefaultSearchEngine,  # pyright: ignore[reportArgumentType]
+        description="Search Engine Configuration",
+        discriminator="search_engine_name",
+        title="Search Engine Configuration",
+    )
+
+
+class Tools(BaseModel):
+    model_config = get_configuration_dict()
+
+    web_tools: WebToolsConfig = Field(
+        default=WebToolsConfig(),
+        description="Configuration for web search tools",
+    )
     internal_tools: bool = Field(
         default=True,
         description="Allow agent to use internal search tools access information from the knowledge base and uploaded documents",
     )
 
+    @field_validator("web_tools", mode="before")
+    @classmethod
+    def handle_bool_case(cls, v: Any) -> Any:
+        if isinstance(v, bool):
+            if v:
+                # Backward compatibility with old config behaviour
+                return WebToolsConfig(
+                    enable=True, search_engine=GoogleConfig(fetch_size=10)
+                )
+            else:
+                return WebToolsConfig(enable=False)
+        return v
+
+
+class UniqueEngine(BaseEngine[Literal[DeepResearchEngine.UNIQUE]]):
+    model_config = get_configuration_dict()
 
-class UniqueEngine(BaseEngine):
     engine_type: Literal[DeepResearchEngine.UNIQUE] = Field(
         default=DeepResearchEngine.UNIQUE
     )

unique_deep_research/service.py

@@ -207,7 +207,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
         except Exception as e:
             if self.is_message_execution():
                 await self._update_execution_status(MessageExecutionUpdateStatus.FAILED)
-            self.logger.error(f"Deep Research tool run failed: {e}")
+            self.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,
@@ -394,7 +394,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             self.write_message_log_text_message("**Research done**")
             return result
         except Exception as e:
-            self.logger.error(f"Research failed: {e}")
+            self.logger.exception(f"Research failed: {e}")
             return "", []
 
     async def custom_research(self, research_brief: str) -> tuple[str, list[Any]]:
@@ -427,12 +427,13 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             enable_web_tools = True
             enable_internal_tools = True
             if isinstance(self.config.engine, UniqueEngine):
-                enable_web_tools = self.config.engine.tools.web_tools
+                enable_web_tools = self.config.engine.tools.web_tools.enable
                 enable_internal_tools = self.config.engine.tools.internal_tools
 
             config = {
                 "configurable": {
                     "engine_config": self.config.engine,
+                    "language_model_service": self.language_model_service,
                     "openai_client": self.client,
                     "chat_service": self.chat_service,
                     "content_service": self.content_service,
@@ -470,7 +471,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
 
         except Exception as e:
             error_msg = f"Custom research failed: {str(e)}"
-            self.logger.error(error_msg, exc_info=True)
+            self.logger.exception(error_msg)
             return error_msg, []
 
     async def openai_research(self, research_brief: str) -> tuple[str, list[Any]]:
@@ -698,7 +699,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                         if event.response.error:
                             return event.response.error.message, []
             except Exception as e:
-                self.logger.error(f"Error processing research stream event: {e}")
+                self.logger.exception(f"Error processing research stream event: {e}")
 
         self.logger.error("Stream ended without completion")
         return "", []

unique_deep_research/unique_custom/agents.py

@@ -253,7 +253,7 @@ async def supervisor_tools(
         all_tool_messages.extend(research_tool_messages)
 
     except Exception as e:
-        _LOGGER.error(f"Research execution failed: {e}")
+        _LOGGER.exception(f"Research execution failed: {e}")
         return Command(
             goto="__end__",
             update={
@@ -596,7 +596,9 @@ async def _handle_conduct_research_batch(
     tool_messages = []
     for observation, tool_call in zip(tool_results, allowed_calls):
         if isinstance(observation, Exception):
-            _LOGGER.error(f"Research task failed: {str(observation)}")
+            _LOGGER.exception(
+                f"Research task failed: {str(observation)}", exc_info=observation
+            )
             error_content = f"Research task failed: {str(observation)}"
             tool_messages.append(
                 ToolMessage(

unique_deep_research/unique_custom/tools.py

@@ -22,8 +22,7 @@ from unique_toolkit.chat.schemas import (
 )
 from unique_toolkit.content import ContentReference
 from unique_toolkit.content.schemas import ContentChunk, ContentSearchType
-from unique_web_search.client_settings import get_google_search_settings
-from unique_web_search.services.search_engine.google import GoogleConfig, GoogleSearch
+from unique_web_search.services.search_engine import get_search_engine_service
 
 from .utils import (
     get_citation_manager,
@@ -153,18 +152,18 @@ async def web_search(query: str, config: RunnableConfig) -> str:
     You can search again with a new query if you need more results.
     Should be followed up by web_fetch to get the complete content of the results.
     """
-    google_settings = get_google_search_settings()
+    if "configurable" not in config:
+        raise ValueError("RunnableConfig missing 'configurable' section")
 
-    if not google_settings.is_configured:
-        _LOGGER.error("Google Search not configured")
-        raise ValueError("Google Search not configured")
+    configurable = config["configurable"]
+    engine_config = configurable["engine_config"].tools.web_tools.search_engine
 
-    # Create Google search configuration and service
-    google_config = GoogleConfig(fetch_size=10)
-    google_search = GoogleSearch(google_config)
+    search_engine_service = get_search_engine_service(
+        engine_config, configurable["language_model_service"]
+    )
 
     # Perform the search
-    search_results = await google_search.search(query)
+    search_results = await search_engine_service.search(query)
     write_tool_message_log(
         config,
         "**Searching the web**",

unique_deep_research/unique_custom/utils.py

@@ -355,7 +355,7 @@ async def execute_tool_safely(
             _LOGGER.warning(f"Token limit exceeded in tool {tool_name}: {str(e)}")
             return f"Tool {tool_name} failed due to token limit: {str(e)}"
         else:
-            _LOGGER.error(f"Error executing tool {tool_name}: {str(e)}")
+            _LOGGER.exception(f"Error executing tool {tool_name}: {str(e)}")
             return f"Error executing tool {tool_name}: {str(e)}"
 
 
@@ -588,7 +588,7 @@ async def ainvoke_with_token_handling(
                 )
                 await asyncio.sleep(delay)
             else:
-                _LOGGER.error(
+                _LOGGER.exception(
                     f"Model invocation failed after {max_retries + 1} attempts: {str(e)}"
                 )
                 raise e

{unique_deep_research-3.0.27.dist-info → unique_deep_research-3.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_deep_research
-Version: 3.0.27
+Version: 3.1.0
 Summary: Deep Research Tool for complex research tasks
 License: Proprietary
 Author: Martin Fadler
@@ -13,8 +13,8 @@ Requires-Dist: beautifulsoup4 (>=4.12.0,<5.0.0)
 Requires-Dist: fake-useragent (>=2.0.0,<3.0.0)
 Requires-Dist: httpx (>=0.28.0,<0.29.0)
 Requires-Dist: jinja2 (>=3.1.2,<4.0.0)
-Requires-Dist: langchain (>=1.1.0,<2.0.0)
 Requires-Dist: langchain-core (>=1.0.0,<2.0.0)
+Requires-Dist: langchain[openai] (>=1.1.0,<2.0.0)
 Requires-Dist: langgraph (>=1.0.0,<2.0.0)
 Requires-Dist: markdownify (>=0.14.0,<0.15.0)
 Requires-Dist: openai (>=1.99.0,<2.0.0)
@@ -22,8 +22,8 @@ Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: tiktoken (>=0.12.0,<0.13.0)
 Requires-Dist: timeout-decorator (>=0.5.0,<0.6.0)
 Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
-Requires-Dist: unique-toolkit (>=1.38.3,<2.0.0)
 Requires-Dist: unique-web-search (>=1.7.0,<2.0.0)
+Requires-Dist: unique_toolkit (>=1.38.3,<2.0.0)
 Description-Content-Type: text/markdown
 
 # Deep Research Tool
@@ -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.1.00] - 2026-01-30
+- Support other search engines than Google
+
+## [3.0.28] - 2026-01-26
+- Specify `langchain-openai` version
+
 ## [3.0.27] - 2026-01-23
 - Fix bug where deep research model usage analytics were not fully exported
 

{unique_deep_research-3.0.27.dist-info → unique_deep_research-3.1.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
 unique_deep_research/__init__.py,sha256=HMksc03A7au88XB0jjeIR3zGt3YGYTA7ZIAkR8hmQZs,396
-unique_deep_research/config.py,sha256=j-Rdlp1W5RDsiJBwbgtdPHd6G9k_dRCxh2m0XroH6qw,3173
+unique_deep_research/config.py,sha256=i6mFGQ7m0D0IYhcKsEotLX1kxl3xrHZ6cSsGITUNuug,4665
 unique_deep_research/markdown_utils.py,sha256=zyKO4c9T_KnZzOYlYZEDEvC1NHRR9koCzcWmOJbrcok,11024
-unique_deep_research/service.py,sha256=mlgT8FpD-IpxehRIGmXRaYG75EteVYvablgV9wFjwJI,36305
+unique_deep_research/service.py,sha256=M_dEBleP_5vqJp37dAg5VPQ5MONmfJoPO10BTogaaIg,36388
 unique_deep_research/templates/clarifying_agent.j2,sha256=TOgD5ezrlRyurpquBGT902oa7PpkdHxsvrkIFWBFa2A,3938
 unique_deep_research/templates/openai/oai_research_system_message.j2,sha256=2YipVh0V4eEikg9iXqIaBVtYwS8ycHMDu3xiFDoJ0vI,1362
 unique_deep_research/templates/report_cleanup_prompt.j2,sha256=9mjj3GX1HEQ0kkxzqvB3D_2wFgAJ-drMbTYDiSc2riI,2049
@@ -11,12 +11,12 @@ unique_deep_research/templates/unique/lead_agent_system.j2,sha256=tWxljNKiQd5i8P
 unique_deep_research/templates/unique/report_writer_system_open_deep_research.j2,sha256=7Ojg144Z0RfH8xLalA068oTV_jAUwxAGEjm3YtAdQac,6365
 unique_deep_research/templates/unique/research_agent_system.j2,sha256=D4mPHVUaPRPo-ElbE_FR0P3zS6BGNsoSohz4FkCdnx8,7981
 unique_deep_research/unique_custom/__init__.py,sha256=S1Rw16tPDfqOIZwAoJ6lAJT00Pk8uh36Aa-otM8vST0,1370
-unique_deep_research/unique_custom/agents.py,sha256=ynTMDDxyWApuK5TZ-g_wxuKbNoULLNytmGcN-t1bbfM,23979
+unique_deep_research/unique_custom/agents.py,sha256=zMJ8zCz-0714bMPZTzIlMqG79AdZAf5orEIJbPLIO3U,24039
 unique_deep_research/unique_custom/citation.py,sha256=dfElOqBTvbXlHXxshT6heNzeHVVKTlgSlIEFTuoNv20,4315
 unique_deep_research/unique_custom/state.py,sha256=tn4pIKN-ZCH6Tqu9XL31lziyrwFkhTVf_lBiUpETVyo,3515
-unique_deep_research/unique_custom/tools.py,sha256=NpWJtgxqPsG7VbUdTmjHV3OWhoVpw5RAexgZVIceIiw,25494
-unique_deep_research/unique_custom/utils.py,sha256=MX57fqVq2-uyBFNu0fgVE-MVPhvaMY78TkUNxOKVFMg,20061
-unique_deep_research-3.0.27.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_deep_research-3.0.27.dist-info/METADATA,sha256=qwxzDFjjt0aoRvjMtefuFbzroITysy6UOuudvY_Vomg,7028
-unique_deep_research-3.0.27.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_deep_research-3.0.27.dist-info/RECORD,,
+unique_deep_research/unique_custom/tools.py,sha256=QEjdKc44hMGpZ8SiF25nrgTvqeu9d_0zd2t27V7LBjY,25424
+unique_deep_research/unique_custom/utils.py,sha256=ZOKABCVKODawfsig-72CcpXCMveg7K_y_rzwlkaJVz8,20069
+unique_deep_research-3.1.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_deep_research-3.1.0.dist-info/METADATA,sha256=Mu0KA77LolMjv99raWziiejX8Y1A7XVr-CjnfC63dXI,7167
+unique_deep_research-3.1.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_deep_research-3.1.0.dist-info/RECORD,,