unique_deep_research 3.0.28__tar.gz → 3.1.0__tar.gz

{unique_deep_research-3.0.28 → unique_deep_research-3.1.0}/CHANGELOG.md

@@ -5,6 +5,9 @@ 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
 

{unique_deep_research-3.0.28 → unique_deep_research-3.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_deep_research
-Version: 3.0.28
+Version: 3.1.0
 Summary: Deep Research Tool for complex research tasks
 License: Proprietary
 Author: Martin Fadler
@@ -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,9 @@ 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
 

{unique_deep_research-3.0.28 → unique_deep_research-3.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_deep_research"
-version = "3.0.28"
+version = "3.1.00"
 description = "Deep Research Tool for complex research tasks"
 authors = [
     "Martin Fadler <martin.fadler@unique.ch>",
@@ -15,7 +15,7 @@ license = "Proprietary"
 python = "^3.12"
 pydantic = "^2.8.2"
 typing-extensions = "^4.9.0"
-unique-toolkit = "^1.38.3"
+unique_toolkit = "^1.38.3"
 jinja2 = "^3.1.2"
 openai = "^1.99.0"
 langgraph = "^1.0.0"
@@ -31,9 +31,9 @@ langchain = {version = "^1.1.0", extras = ["openai"]}
 
 
 [tool.poetry.group.dev.dependencies]
-unique-sdk = { path = "../../unique_sdk" }
-unique-toolkit = { path = "../../unique_toolkit" }
-unique-web-search = { path = "../unique_web_search" }
+unique-sdk = { path = "../../unique_sdk" , develop = true}
+unique-toolkit = { path = "../../unique_toolkit" , develop = true}
+unique-web-search = { path = "../unique_web_search" , develop = true}
 pytest = "^8.4.1"
 pytest-asyncio = "^0.25.0"
 pytest-cov = "^7.0.0"

{unique_deep_research-3.0.28 → unique_deep_research-3.1.0}/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-3.0.28 → unique_deep_research-3.1.0}/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-3.0.28 → unique_deep_research-3.1.0}/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-3.0.28 → unique_deep_research-3.1.0}/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-3.0.28 → unique_deep_research-3.1.0}/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