unique_deep_research 2026.36.0.dev0__tar.gz → 2026.36.0.dev2__tar.gz

{unique_deep_research-2026.36.0.dev0 → unique_deep_research-2026.36.0.dev2}/PKG-INFO

@@ -1,18 +1,18 @@
 Metadata-Version: 2.3
 Name: unique-deep-research
-Version: 2026.36.0.dev0
+Version: 2026.36.0.dev2
 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.36.0.dev0,<2026.36.0rc0
+Requires-Dist: unique-toolkit>=2026.36.0.dev6,<2026.36.0rc0
 Requires-Dist: jinja2>=3.1.2,<4
 Requires-Dist: openai>=1.99.0,<3
 Requires-Dist: langgraph>=1.0.0,<2
 Requires-Dist: langchain-core>=1.0.0,<2
-Requires-Dist: unique-web-search>=2026.36.0.dev0,<2026.36.0rc0
+Requires-Dist: unique-web-search>=2026.36.0.dev3,<2026.36.0rc0
 Requires-Dist: httpx>=0.28.0,<0.29
 Requires-Dist: timeout-decorator>=0.5.0,<0.6
 Requires-Dist: beautifulsoup4>=4.12.0,<5

{unique_deep_research-2026.36.0.dev0 → unique_deep_research-2026.36.0.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_deep_research"
-version = "2026.36.0.dev0"
+version = "2026.36.0.dev2"
 description = "Deep Research Tool for complex research tasks"
 readme = "README.md"
 license = { text = "Proprietary" }
@@ -14,12 +14,12 @@ requires-python = ">=3.12,<4"
 dependencies = [
     "pydantic>=2.8.2,<3",
     "typing-extensions>=4.9.0,<5",
-    "unique-toolkit>=2026.36.0.dev0,<2026.36.0rc0",
+    "unique-toolkit>=2026.36.0.dev6,<2026.36.0rc0",
     "jinja2>=3.1.2,<4",
     "openai>=1.99.0,<3",
     "langgraph>=1.0.0,<2",
     "langchain-core>=1.0.0,<2",
-    "unique-web-search>=2026.36.0.dev0,<2026.36.0rc0",
+    "unique-web-search>=2026.36.0.dev3,<2026.36.0rc0",
     "httpx>=0.28.0,<0.29",
     "timeout-decorator>=0.5.0,<0.6",
     "beautifulsoup4>=4.12.0,<5",

unique_deep_research-2026.36.0.dev2/unique_deep_research/invocation_stats.py

@@ -0,0 +1,5 @@
+"""Run-scoped collection of Deep Research language-model usage."""
+
+from unique_toolkit.language_model.invocation_stats import InvocationStatsCollector
+
+collector = InvocationStatsCollector("deep_research")

{unique_deep_research-2026.36.0.dev0 → unique_deep_research-2026.36.0.dev2}/unique_deep_research/service.py

@@ -50,9 +50,7 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelMessageRole,
 )
 from unique_toolkit.short_term_memory.service import ShortTermMemoryService
-from unique_web_search.invocation_stats import (
-    invocation_stats_scope as web_search_invocation_stats_scope,
-)
+from unique_web_search.invocation_stats import collector as web_search_collector
 
 from unique_deep_research.unique_custom.tools import crawl_url, get_today_str
 
@@ -63,12 +61,7 @@ from .config import (
     DeepResearchToolConfig,
     UniqueEngine,
 )
-from .invocation_stats import (
-    invocation_stats_scope,
-    record_invocation_stats,
-    record_language_model_response,
-    record_token_usage,
-)
+from .invocation_stats import collector
 from .markdown_utils import (
     export_report_to_docx,
     postprocess_research_result_with_chunks,
@@ -270,7 +263,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
         )
 
     async def run(self, tool_call: LanguageModelFunction) -> ToolCallResponse:
-        with invocation_stats_scope() as invocation_stats:
+        with collector.scope() as invocation_stats:
             sub = self.chat_service.cancellation.on_cancellation.subscribe(
                 self._on_cancellation
             )
@@ -555,7 +548,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             },
         }
 
-        with web_search_invocation_stats_scope() as web_search_invocation_stats:
+        with web_search_collector.scope() as web_search_invocation_stats:
             try:
                 result = await self.chat_service.cancellation.run_with_cancellation(
                     custom_agent.ainvoke(initial_state, config=config),  # type: ignore[arg-type]
@@ -566,7 +559,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                 # calls happen during ainvoke(); merge it into the outer Deep Research
                 # scope even if the graph raises, so tokens already spent before the
                 # failure aren't dropped from the failure ToolCallResponse.
-                record_invocation_stats(web_search_invocation_stats)
+                collector.record_invocation_stats(web_search_invocation_stats)
 
         research_result = result.get("final_report", "")
 
@@ -691,7 +684,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
                             _LOGGER.info(
                                 f"OpenAI research token usage: {event.response.usage}"
                             )
-                            record_token_usage(
+                            collector.record_token_usage(
                                 model_name=self.config.engine.research_model.name,
                                 usage=event.response.usage,
                                 source="deep_research.openai_research",
@@ -872,7 +865,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             temperature=0.1,
             max_tokens=5000,
         )
-        record_language_model_response(
+        collector.record_language_model_response(
             model_name=self.config.engine.large_model.name,
             response=response,
             source="deep_research.report_cleanup",
@@ -937,7 +930,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             model_name=self.config.engine.small_model.name,
             content_chunks=None,
         )
-        record_language_model_response(
+        collector.record_language_model_response(
             model_name=self.config.engine.small_model.name,
             response=response,
             source="deep_research.clarification",
@@ -969,7 +962,7 @@ class DeepResearchTool(Tool[DeepResearchToolConfig]):
             model=self.config.engine.large_model.name,
             messages=chat_messages,
         )
-        record_language_model_response(
+        collector.record_language_model_response(
             model_name=self.config.engine.large_model.name,
             response=research_response,
             source="deep_research.research_brief",

{unique_deep_research-2026.36.0.dev0 → unique_deep_research-2026.36.0.dev2}/unique_deep_research/unique_custom/utils.py

@@ -29,7 +29,7 @@ from unique_toolkit.content.service import ContentService
 from unique_toolkit.language_model.infos import LanguageModelInfo
 
 from ..config import UniqueEngine
-from ..invocation_stats import record_language_model_response
+from ..invocation_stats import collector
 from .citation import GlobalCitationManager
 from .state import AgentState, ResearcherState, SupervisorState
 
@@ -586,7 +586,7 @@ async def ainvoke_with_token_handling(
                 )
                 raise e
         else:
-            record_language_model_response(
+            collector.record_language_model_response(
                 model_name=model_info.name,
                 response=response,
                 source=source,

unique_deep_research-2026.36.0.dev0/unique_deep_research/invocation_stats.py

@@ -1,84 +0,0 @@
-"""Run-scoped collection of Deep Research language-model usage."""
-
-import logging
-from collections.abc import Iterator
-from contextlib import contextmanager
-from contextvars import ContextVar, Token
-from typing import Any
-
-from unique_toolkit.language_model.invocation_stats import LanguageModelInvocationStats
-from unique_toolkit.language_model.schemas import LanguageModelTokenUsage
-
-_LOGGER = logging.getLogger(__name__)
-
-_CURRENT_INVOCATION_STATS: ContextVar[list[LanguageModelInvocationStats] | None] = (
-    ContextVar("deep_research_invocation_stats", default=None)
-)
-
-
-@contextmanager
-def invocation_stats_scope() -> Iterator[list[LanguageModelInvocationStats]]:
-    """Collect stats for one Deep Research run and isolate concurrent runs."""
-    invocation_stats: list[LanguageModelInvocationStats] = []
-    token: Token[list[LanguageModelInvocationStats] | None] = (
-        _CURRENT_INVOCATION_STATS.set(invocation_stats)
-    )
-    try:
-        yield invocation_stats
-    finally:
-        _CURRENT_INVOCATION_STATS.reset(token)
-
-
-def record_token_usage(
-    *,
-    model_name: str,
-    usage: Any,
-    source: str,
-) -> None:
-    """Record provider usage when both a run scope and usage are available."""
-    invocation_stats = _CURRENT_INVOCATION_STATS.get()
-    if invocation_stats is None or usage is None:
-        return
-
-    try:
-        token_usage = LanguageModelTokenUsage.model_validate(usage)
-        invocation_stats.append(
-            LanguageModelInvocationStats.from_usage(
-                model_name=model_name,
-                token_usage=token_usage,
-                source=source,
-            )
-        )
-    except Exception:
-        _LOGGER.warning(
-            "Unable to parse Deep Research token usage for %s",
-            source,
-            exc_info=True,
-        )
-
-
-def record_invocation_stats(
-    invocations: list[LanguageModelInvocationStats],
-) -> None:
-    """Add invocation stats collected by a nested Deep Research dependency."""
-    invocation_stats = _CURRENT_INVOCATION_STATS.get()
-    if invocation_stats is not None:
-        invocation_stats.extend(invocations)
-
-
-def record_language_model_response(
-    *,
-    model_name: str,
-    response: Any,
-    source: str,
-) -> None:
-    """Record usage from OpenAI SDK, toolkit, or LangChain responses."""
-    usage = getattr(response, "usage", None)
-    if usage is None:
-        usage = getattr(response, "usage_metadata", None)
-    if usage is None:
-        response_metadata = getattr(response, "response_metadata", None)
-        if isinstance(response_metadata, dict):
-            usage = response_metadata.get("token_usage")
-
-    record_token_usage(model_name=model_name, usage=usage, source=source)