unique_orchestrator 2026.32.0.dev1__tar.gz → 2026.32.0.dev3__tar.gz

{unique_orchestrator-2026.32.0.dev1 → unique_orchestrator-2026.32.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-orchestrator
-Version: 2026.32.0.dev1
+Version: 2026.32.0.dev3
 Summary: 
 Author: Andreas Hauri
 Author-email: Andreas Hauri <andreas.hauri@unique.ai>
@@ -9,15 +9,15 @@ Requires-Dist: pydantic>=2.8.2,<3
 Requires-Dist: pydantic-settings>=2.10.1,<3
 Requires-Dist: typing-extensions>=4.9.0,<5
 Requires-Dist: jinja2>=3.1.0,<4
-Requires-Dist: unique-toolkit>=2026.32.0.dev4,<2026.32.0rc0
+Requires-Dist: unique-toolkit>=2026.32.0.dev8,<2026.32.0rc0
 Requires-Dist: unique-stock-ticker>=2026.32.0.dev1,<2026.32.0rc0
 Requires-Dist: unique-follow-up-questions>=2026.32.0.dev0,<2026.32.0rc0
 Requires-Dist: unique-internal-search>=2026.32.0.dev1,<2026.32.0rc0
 Requires-Dist: unique-deep-research>=2026.32.0.dev1,<2026.32.0rc0
-Requires-Dist: unique-web-search>=2026.32.0.dev1,<2026.32.0rc0
+Requires-Dist: unique-web-search>=2026.32.0.dev2,<2026.32.0rc0
 Requires-Dist: unique-swot>=2026.32.0.dev0,<2026.32.0rc0
 Requires-Dist: unique-skill-tool>=2026.32.0.dev0,<2026.32.0rc0
-Requires-Dist: unique-user-memory>=2026.32.0.dev2,<2026.32.0rc0
+Requires-Dist: unique-user-memory>=2026.32.0.dev3,<2026.32.0rc0
 Requires-Dist: openai>=1.109.1,<3
 Requires-Python: >=3.12, <4
 Description-Content-Type: text/markdown

{unique_orchestrator-2026.32.0.dev1 → unique_orchestrator-2026.32.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_orchestrator"
-version = "2026.32.0.dev1"
+version = "2026.32.0.dev3"
 description = ""
 readme = "README.md"
 license = { text = "Proprietary" }
@@ -13,15 +13,15 @@ dependencies = [
     "pydantic-settings>=2.10.1,<3",
     "typing-extensions>=4.9.0,<5",
     "jinja2>=3.1.0,<4",
-    "unique-toolkit>=2026.32.0.dev4,<2026.32.0rc0",
+    "unique-toolkit>=2026.32.0.dev8,<2026.32.0rc0",
     "unique-stock-ticker>=2026.32.0.dev1,<2026.32.0rc0",
     "unique-follow-up-questions>=2026.32.0.dev0,<2026.32.0rc0",
     "unique-internal-search>=2026.32.0.dev1,<2026.32.0rc0",
     "unique-deep-research>=2026.32.0.dev1,<2026.32.0rc0",
-    "unique-web-search>=2026.32.0.dev1,<2026.32.0rc0",
+    "unique-web-search>=2026.32.0.dev2,<2026.32.0rc0",
     "unique-swot>=2026.32.0.dev0,<2026.32.0rc0",
     "unique-skill-tool>=2026.32.0.dev0,<2026.32.0rc0",
-    "unique-user-memory>=2026.32.0.dev2,<2026.32.0rc0",
+    "unique-user-memory>=2026.32.0.dev3,<2026.32.0rc0",
     "openai>=1.109.1,<3",
 ]
 

{unique_orchestrator-2026.32.0.dev1 → unique_orchestrator-2026.32.0.dev3}/unique_orchestrator/tests/test_unique_ai_token_usage.py

@@ -1,3 +1,4 @@
+from pathlib import Path
 from unittest.mock import AsyncMock, MagicMock
 
 import pytest
@@ -7,6 +8,7 @@ from unique_toolkit.agentic.debug_info_manager.debug_info_manager import (
 from unique_toolkit.agentic.loop_runner import SupportsInvocationStats
 from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 from unique_toolkit.language_model.invocation_stats import LanguageModelInvocationStats
+from unique_toolkit.language_model.model_costs import MODEL_COSTS_FILE_ENV
 from unique_toolkit.language_model.schemas import LanguageModelTokenUsage
 
 from unique_orchestrator.unique_ai import UniqueAI
@@ -211,9 +213,25 @@ class TestAccumulateUsage:
 class TestRunTokenUsageIntegration:
     @pytest.mark.ai
     @pytest.mark.asyncio
-    async def test_run__debug_info_manager__receives_llm_invocations_key(self):
+    async def test_run__debug_info_manager__receives_llm_invocations_key(
+        self,
+        tmp_path: Path,
+        monkeypatch: pytest.MonkeyPatch,
+    ):
         """A single-iteration run with real usage on the loop response must
         surface it under debug_info's 'llm_invocations' key."""
+        cost_file = tmp_path / "costs.yaml"
+        cost_file.write_text(
+            f"""
+costSchemaVersion: 1
+models:
+  {MODEL_NAME}:
+    input: 2
+    completion: 8
+""",
+            encoding="utf-8",
+        )
+        monkeypatch.setenv(MODEL_COSTS_FILE_ENV, str(cost_file))
         loop_response = _make_loop_response(
             LanguageModelTokenUsage(
                 completion_tokens=12, prompt_tokens=34, total_tokens=46
@@ -231,6 +249,7 @@ class TestRunTokenUsageIntegration:
         assert len(payload) == 1
         assert payload[0]["modelName"] == MODEL_NAME
         assert payload[0]["source"] == "main_loop[1]"
+        assert payload[0]["costUsd"] == pytest.approx(0.000164)
         assert payload[0]["tokenUsage"] == {
             "completionTokens": 12,
             "promptTokens": 34,
@@ -712,7 +731,8 @@ class TestRunTokenUsageIntegration:
         assert [
             invocation["source"] for invocation in debug_info["llm_invocations"]
         ] == ["deep_research.clarification", "main_loop[1]"]
-        assert debug_info["analytics"]["tokens"][0]["total_tokens"] == 35
+        assert debug_info["analytics"]["consumption_by_llm"][0]["total_tokens"] == 35
+        assert debug_info["analytics"]["consumption"]["total_tokens"] == 35
         assert debug_info["llm_invocations_complete"] is True
 
     @pytest.mark.ai

{unique_orchestrator-2026.32.0.dev1 → unique_orchestrator-2026.32.0.dev3}/unique_orchestrator/unique_ai_builder.py

@@ -70,6 +70,7 @@ from unique_toolkit.experimental.integrations.openai.streaming.event_routing imp
 from unique_toolkit.language_model.infos import LanguageModelInfo, ModelCapabilities
 from unique_toolkit.protocols.support import ResponsesSupportCompleteWithReferences
 from unique_user_memory.user_memory import load_user_memory
+from unique_user_memory.user_memory_message_log import UserMemoryMessageLogger
 from unique_user_memory.user_memory_postprocessor import UserMemoryPostprocessor
 
 from unique_orchestrator._builders import (
@@ -298,6 +299,7 @@ async def _build_common(
     config: UniqueAIConfig,
 ) -> _CommonComponents:
     chat_service = ChatService(event)
+    message_step_logger = MessageStepLogger(chat_service)
 
     llm_service = LanguageModelService.from_event(event)
 
@@ -393,13 +395,37 @@ async def _build_common(
             if user_memory_config.use_orchestrator_language_model
             else user_memory_config.language_model
         )
-        user_memory_state = await load_user_memory(
-            event=event,
-            config=user_memory_config,
-            language_model=memory_language_model,
+        memory_message_step_logger = UserMemoryMessageLogger(
+            message_step_logger,
             logger=logger,
         )
-        if user_memory_state is not None:
+        await memory_message_step_logger.log_loading_start()
+        user_memory_state = None
+        load_succeeded = False
+        try:
+            user_memory_state = await load_user_memory(
+                event=event,
+                config=user_memory_config,
+                language_model=memory_language_model,
+                logger=logger,
+            )
+            load_succeeded = True
+        except Exception as exc:
+            # Soft-fail like a None return: keep the turn running without
+            # memory, but always close the RUNNING loading Step first.
+            logger.warning(
+                "[user-memory] load raised - running without memory: [%s] %s",
+                type(exc).__name__,
+                exc,
+            )
+        finally:
+            if not load_succeeded:
+                await memory_message_step_logger.log_loading_failed()
+
+        if load_succeeded and user_memory_state is not None:
+            await memory_message_step_logger.log_loading_complete(
+                with_settings_entry=True
+            )
             user_memory_text = user_memory_state.text
             postprocessor_manager.add_postprocessor(
                 UserMemoryPostprocessor(
@@ -408,9 +434,13 @@ async def _build_common(
                     event=event,
                     state=user_memory_state,
                     logger=logger,
-                    chat_service=chat_service,
+                    message_step_logger=memory_message_step_logger,
                 )
             )
+        elif load_succeeded:
+            await memory_message_step_logger.log_loading_complete(
+                with_settings_entry=False
+            )
 
     return _CommonComponents(
         chat_service=chat_service,
@@ -429,7 +459,7 @@ async def _build_common(
         user_memory_text=user_memory_text,
         postprocessor_manager=postprocessor_manager,
         response_watcher=response_watcher,
-        message_step_logger=MessageStepLogger(chat_service),
+        message_step_logger=message_step_logger,
     )