unique_orchestrator 2026.44.0.dev0__tar.gz → 2026.44.0.dev2__tar.gz

unique_orchestrator-2026.44.0.dev2/PKG-INFO

@@ -0,0 +1,27 @@
+Metadata-Version: 2.3
+Name: unique-orchestrator
+Version: 2026.44.0.dev2
+Summary: 
+Author: Andreas Hauri
+Author-email: Andreas Hauri <andreas.hauri@unique.ai>
+License: Proprietary
+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.44.0.dev7,<2026.44.0rc0
+Requires-Dist: unique-stock-ticker>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: unique-follow-up-questions>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: unique-internal-search>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: unique-deep-research>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: unique-web-search>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: unique-swot>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: unique-skill-tool>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: unique-user-memory>=2026.44.0.dev1,<2026.44.0rc0
+Requires-Dist: openai>=1.109.1,<3
+Requires-Python: >=3.12, <4
+Description-Content-Type: text/markdown
+
+# Unique Orchestrator
+
+Unique AI orchestrator agent to struct and direct subagents, tools and services

{unique_orchestrator-2026.44.0.dev0 → unique_orchestrator-2026.44.0.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_orchestrator"
-version = "2026.44.0.dev0"
+version = "2026.44.0.dev2"
 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.44.0.dev0,<2026.44.0rc0",
-    "unique-stock-ticker>=2026.44.0.dev0,<2026.44.0rc0",
-    "unique-follow-up-questions>=2026.44.0.dev0,<2026.44.0rc0",
-    "unique-internal-search>=2026.44.0.dev0,<2026.44.0rc0",
-    "unique-deep-research>=2026.44.0.dev0,<2026.44.0rc0",
-    "unique-web-search>=2026.44.0.dev0,<2026.44.0rc0",
-    "unique-swot>=2026.44.0.dev0,<2026.44.0rc0",
-    "unique-skill-tool>=2026.44.0.dev0,<2026.44.0rc0",
-    "unique-user-memory>=2026.44.0.dev0,<2026.44.0rc0",
+    "unique-toolkit>=2026.44.0.dev7,<2026.44.0rc0",
+    "unique-stock-ticker>=2026.44.0.dev1,<2026.44.0rc0",
+    "unique-follow-up-questions>=2026.44.0.dev1,<2026.44.0rc0",
+    "unique-internal-search>=2026.44.0.dev1,<2026.44.0rc0",
+    "unique-deep-research>=2026.44.0.dev1,<2026.44.0rc0",
+    "unique-web-search>=2026.44.0.dev1,<2026.44.0rc0",
+    "unique-swot>=2026.44.0.dev1,<2026.44.0rc0",
+    "unique-skill-tool>=2026.44.0.dev1,<2026.44.0rc0",
+    "unique-user-memory>=2026.44.0.dev1,<2026.44.0rc0",
     "openai>=1.109.1,<3",
 ]
 

{unique_orchestrator-2026.44.0.dev0 → unique_orchestrator-2026.44.0.dev2}/unique_orchestrator/unique_ai.py

@@ -13,12 +13,17 @@ from unique_toolkit.agentic.debug_info_manager.debug_info_manager import (
 )
 from unique_toolkit.agentic.evaluation.evaluation_manager import EvaluationManager
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
+from unique_toolkit.agentic.history_manager.context_overflow import (
+    ContextOverflowGuard,
+    is_context_window_exceeded_error,
+)
 from unique_toolkit.agentic.history_manager.history_manager import HistoryManager
 from unique_toolkit.agentic.loop_runner import (
     LoopIterationRunner,
     ResponsesLoopIterationRunner,
     SupportsInvocationStats,
 )
+from unique_toolkit.agentic.loop_runner.base import _LoopIterationRunnerKwargs
 from unique_toolkit.agentic.message_log_manager.service import MessageStepLogger
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
     PostprocessorManager,
@@ -631,7 +636,7 @@ class UniqueAI:
         )
         self._record_loop_debug_params(other_options)
 
-        kwargs: dict = dict(
+        kwargs: _LoopIterationRunnerKwargs = dict(
             messages=messages,
             iteration_index=self.current_iteration_index,
             streaming_handler=self._streaming_handler,  # type: ignore (constructor accepts only compatible arguments)
@@ -645,10 +650,36 @@ class UniqueAI:
             other_options=other_options,
         )
 
+        overflow_guard = ContextOverflowGuard(
+            logger=self._logger,
+            history_manager=self._history_manager,
+            token_counter=self._chat_service,
+            model=self._config.space.language_model,
+        )
+
+        async def _recompose() -> None:
+            await self._recompose_messages(kwargs)
+
+        await overflow_guard.preflight_token_budget(kwargs, _recompose)
+
+        try:
+            return await self._run_loop_iteration(kwargs)
+        except Exception as exc:
+            if not is_context_window_exceeded_error(exc):
+                raise
+            return await overflow_guard.retry_after_overflow(
+                exc,
+                recompose=_recompose,
+                run_iteration=lambda: self._run_loop_iteration(kwargs),
+            )
+
+    async def _run_loop_iteration(
+        self, kwargs: _LoopIterationRunnerKwargs
+    ) -> LanguageModelStreamResponse:
         # Experimental Feature UN-17905
         if self._config.agent.experimental.open_file_tool_config.enabled:
             try:
-                return await self._loop_iteration_runner(**kwargs)
+                return await self._loop_iteration_runner(**kwargs)  # type: ignore[arg-type]
             except Exception as exc:
                 if not self._open_file_runtime.should_retry_without_files(exc):
                     raise
@@ -658,13 +689,19 @@ class UniqueAI:
                 )
                 self._file_fallback_occurred = True
                 kwargs["messages"] = self._open_file_runtime.prepare_retry_messages(
-                    messages=messages
+                    messages=kwargs["messages"]
                 )
                 kwargs["tools"] = self._tool_manager.get_tool_definitions()  # type: ignore (as above)
                 kwargs["tool_choices"] = self._tool_manager.get_forced_tools()  # type: ignore (as above)
-                return await self._loop_iteration_runner(**kwargs)
+                return await self._loop_iteration_runner(**kwargs)  # type: ignore[arg-type]
+
+        return await self._loop_iteration_runner(**kwargs)  # type: ignore[arg-type]
 
-        return await self._loop_iteration_runner(**kwargs)
+    async def _recompose_messages(self, kwargs: _LoopIterationRunnerKwargs) -> None:
+        kwargs["messages"] = await self._compose_message_plan_execution()
+        kwargs["content_chunks"] = (
+            self._history_manager.get_content_chunks_for_backend()
+        )
 
     async def _process_plan(self, loop_response: LanguageModelStreamResponse) -> bool:
         self._logger.info(

unique_orchestrator-2026.44.0.dev0/PKG-INFO

@@ -1,27 +0,0 @@
-Metadata-Version: 2.3
-Name: unique-orchestrator
-Version: 2026.44.0.dev0
-Summary: 
-Author: Andreas Hauri
-Author-email: Andreas Hauri <andreas.hauri@unique.ai>
-License: Proprietary
-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.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-stock-ticker>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-follow-up-questions>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-internal-search>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-deep-research>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-web-search>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-swot>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-skill-tool>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: unique-user-memory>=2026.44.0.dev0,<2026.44.0rc0
-Requires-Dist: openai>=1.109.1,<3
-Requires-Python: >=3.12, <4
-Description-Content-Type: text/markdown
-
-# Unique Orchestrator
-
-Unique AI orchestrator agent to struct and direct subagents, tools and services