unique_orchestrator 2026.30.0.dev0__tar.gz → 2026.30.0.dev1__tar.gz

{unique_orchestrator-2026.30.0.dev0 → unique_orchestrator-2026.30.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-orchestrator
-Version: 2026.30.0.dev0
+Version: 2026.30.0.dev1
 Summary: 
 Author: Andreas Hauri
 Author-email: Andreas Hauri <andreas.hauri@unique.ai>
@@ -9,7 +9,7 @@ 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.30.0.dev0,<2026.30.0rc0
+Requires-Dist: unique-toolkit>=2026.30.0.dev3,<2026.30.0rc0
 Requires-Dist: unique-stock-ticker>=2026.30.0.dev0,<2026.30.0rc0
 Requires-Dist: unique-follow-up-questions>=2026.30.0.dev0,<2026.30.0rc0
 Requires-Dist: unique-internal-search>=2026.30.0.dev0,<2026.30.0rc0

{unique_orchestrator-2026.30.0.dev0 → unique_orchestrator-2026.30.0.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_orchestrator"
-version = "2026.30.0.dev0"
+version = "2026.30.0.dev1"
 description = ""
 readme = "README.md"
 license = { text = "Proprietary" }
@@ -13,7 +13,7 @@ dependencies = [
     "pydantic-settings>=2.10.1,<3",
     "typing-extensions>=4.9.0,<5",
     "jinja2>=3.1.0,<4",
-    "unique-toolkit>=2026.30.0.dev0,<2026.30.0rc0",
+    "unique-toolkit>=2026.30.0.dev3,<2026.30.0rc0",
     "unique-stock-ticker>=2026.30.0.dev0,<2026.30.0rc0",
     "unique-follow-up-questions>=2026.30.0.dev0,<2026.30.0rc0",
     "unique-internal-search>=2026.30.0.dev0,<2026.30.0rc0",

{unique_orchestrator-2026.30.0.dev0 → unique_orchestrator-2026.30.0.dev1}/unique_orchestrator/config.py

@@ -457,6 +457,10 @@ class ExperimentalConfig(BaseToolConfig):
         description="If set, the main agent will use the Responses API from OpenAI",
     )
 
+    use_experimental_python_streaming: bool = Field(
+        default=False,
+    )
+
 
 class UniqueAIAgentConfig(BaseToolConfig):
     max_loop_iterations: Annotated[

{unique_orchestrator-2026.30.0.dev0 → unique_orchestrator-2026.30.0.dev1}/unique_orchestrator/tests/test_unique_ai_plan_or_execute_include.py

@@ -9,7 +9,10 @@ from unittest.mock import AsyncMock, MagicMock
 import pytest
 from unique_toolkit.agentic.tools.tool_manager import ResponsesApiToolManager
 
-from unique_orchestrator.unique_ai_builder import ResponsesStreamingHandler
+from unique_orchestrator.unique_ai_builder import (
+    ResponsesStreamingHandler,
+    _ChatServiceResponsesStreaming,
+)
 
 
 def _make_handler(
@@ -17,15 +20,11 @@ def _make_handler(
 ) -> ResponsesStreamingHandler:
     tool_manager = MagicMock(spec=ResponsesApiToolManager)
     tool_manager.get_required_include_params.return_value = include_params
-    chat_service = MagicMock()
-    chat_service.complete_responses_with_references_async = AsyncMock(
-        return_value=MagicMock()
-    )
-    chat_service.complete_responses_with_references = MagicMock(
-        return_value=MagicMock()
-    )
+    inner = MagicMock()
+    inner.complete_with_references_async = AsyncMock(return_value=MagicMock())
+    inner.complete_with_references = MagicMock(return_value=MagicMock())
     return ResponsesStreamingHandler(
-        chat_service=chat_service,
+        inner=inner,
         tool_manager=tool_manager,
     )
 
@@ -45,9 +44,7 @@ async def test_responses_streaming_handler__injects_include__when_tool_manager_r
 
     await handler.complete_with_references_async(messages=[], model_name="model")
 
-    call_kwargs = (
-        handler._chat_service.complete_responses_with_references_async.await_args.kwargs
-    )
+    call_kwargs = handler._inner.complete_with_references_async.await_args.kwargs
     assert call_kwargs["include"] == ["code_interpreter_call.outputs"]
     handler._tool_manager.get_required_include_params.assert_called_once()
 
@@ -66,9 +63,7 @@ async def test_responses_streaming_handler__omits_include__when_tool_manager_ret
 
     await handler.complete_with_references_async(messages=[], model_name="model")
 
-    call_kwargs = (
-        handler._chat_service.complete_responses_with_references_async.await_args.kwargs
-    )
+    call_kwargs = handler._inner.complete_with_references_async.await_args.kwargs
     assert "include" not in call_kwargs
 
 
@@ -81,9 +76,7 @@ def test_responses_streaming_handler__injects_include__sync_path() -> None:
 
     handler.complete_with_references(messages=[], model_name="model")
 
-    call_kwargs = (
-        handler._chat_service.complete_responses_with_references.call_args.kwargs
-    )
+    call_kwargs = handler._inner.complete_with_references.call_args.kwargs
     assert call_kwargs["include"] == ["code_interpreter_call.outputs"]
 
 
@@ -94,7 +87,34 @@ def test_responses_streaming_handler__omits_include__sync_path_when_empty() -> N
 
     handler.complete_with_references(messages=[], model_name="model")
 
-    call_kwargs = (
-        handler._chat_service.complete_responses_with_references.call_args.kwargs
-    )
+    call_kwargs = handler._inner.complete_with_references.call_args.kwargs
     assert "include" not in call_kwargs
+
+
+@pytest.mark.ai
+@pytest.mark.asyncio
+async def test_chat_service_responses_streaming__forwards_to_responses_methods() -> (
+    None
+):
+    """
+    Purpose: The legacy adapter must route the protocol's
+    `complete_with_references[_async]` to ChatService's Responses-specific
+    methods, not the Chat Completions ones.
+    Why this matters: ChatService exposes both; routing to the wrong method
+    would silently stream through the Chat Completions API.
+    """
+    chat_service = MagicMock()
+    chat_service.complete_responses_with_references = MagicMock(
+        return_value=MagicMock()
+    )
+    chat_service.complete_responses_with_references_async = AsyncMock(
+        return_value=MagicMock()
+    )
+    adapter = _ChatServiceResponsesStreaming(chat_service)
+
+    adapter.complete_with_references(messages=[], model_name="model")
+    await adapter.complete_with_references_async(messages=[], model_name="model")
+
+    chat_service.complete_responses_with_references.assert_called_once()
+    chat_service.complete_responses_with_references_async.assert_awaited_once()
+    chat_service.complete_with_references.assert_not_called()

{unique_orchestrator-2026.30.0.dev0 → unique_orchestrator-2026.30.0.dev1}/unique_orchestrator/unique_ai_builder.py

@@ -65,9 +65,13 @@ from unique_toolkit.agentic.tools.tool_manager import (
 )
 from unique_toolkit.agentic.tools.tool_progress_reporter import ToolProgressReporter
 from unique_toolkit.app.schemas import ChatEvent, McpServer
+from unique_toolkit.app.unique_settings import UniqueSettings
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content import Content
 from unique_toolkit.content.service import ContentService
+from unique_toolkit.experimental.integrations.openai.streaming.event_routing import (
+    ResponsesCompleteWithReferences,
+)
 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
@@ -99,14 +103,19 @@ class ResponsesStreamingHandler(ResponsesSupportCompleteWithReferences):
 
     Injects `include` params from the tool manager on every call so the
     orchestrator loop stays generic (no isinstance checks in UniqueAI).
+
+    Delegates the actual streaming to an injected
+    `ResponsesSupportCompleteWithReferences` (the event-routing
+    `ResponsesCompleteWithReferences` orchestrator), so this class only owns
+    the `include` injection.
     """
 
     def __init__(
         self,
-        chat_service: ChatService,
+        inner: ResponsesSupportCompleteWithReferences,
         tool_manager: ResponsesApiToolManager,
     ) -> None:
-        self._chat_service: ChatService = chat_service
+        self._inner: ResponsesSupportCompleteWithReferences = inner
         self._tool_manager: ResponsesApiToolManager = tool_manager
 
     @override
@@ -114,13 +123,26 @@ class ResponsesStreamingHandler(ResponsesSupportCompleteWithReferences):
         include = self._tool_manager.get_required_include_params()
         if include:
             kwargs["include"] = include
-        return self._chat_service.complete_responses_with_references(*args, **kwargs)
+        return self._inner.complete_with_references(*args, **kwargs)
 
     @override
     async def complete_with_references_async(self, *args: Any, **kwargs: Any) -> Any:
         include = self._tool_manager.get_required_include_params()
         if include:
             kwargs["include"] = include
+        return await self._inner.complete_with_references_async(*args, **kwargs)
+
+
+class _ChatServiceResponsesStreaming(ResponsesSupportCompleteWithReferences):
+    def __init__(self, chat_service: ChatService) -> None:
+        self._chat_service: ChatService = chat_service
+
+    @override
+    def complete_with_references(self, *args: Any, **kwargs: Any) -> Any:
+        return self._chat_service.complete_responses_with_references(*args, **kwargs)
+
+    @override
+    async def complete_with_references_async(self, *args: Any, **kwargs: Any) -> Any:
         return await self._chat_service.complete_responses_with_references_async(
             *args, **kwargs
         )
@@ -572,8 +594,22 @@ async def _build_responses(
         openai_client=client,
     )
 
+    if config.agent.experimental.use_experimental_python_streaming:
+        inner: ResponsesSupportCompleteWithReferences = ResponsesCompleteWithReferences(
+            UniqueSettings.from_chat_event(event),
+            additional_headers={
+                "x-model": config.space.language_model.name,
+                "x-user-id": event.user_id,
+                "x-company-id": event.company_id,
+                "x-assistant-id": event.payload.assistant_id,
+                "x-chat-id": event.payload.chat_id,
+            },
+        )
+    else:
+        inner = _ChatServiceResponsesStreaming(common_components.chat_service)
+
     streaming_handler = ResponsesStreamingHandler(
-        chat_service=common_components.chat_service,
+        inner=inner,
         tool_manager=tool_manager,
     )