unique_orchestrator 1.2.4__py3-none-any.whl → 1.4.0__py3-none-any.whl

unique_orchestrator/config.py

@@ -3,6 +3,7 @@ from pathlib import Path
 from typing import Annotated, Any, Generic, Literal, TypeVar
 
 from pydantic import BaseModel, Field, ValidationInfo, field_validator, model_validator
+from pydantic.json_schema import SkipJsonSchema
 from unique_deep_research.config import DeepResearchToolConfig
 from unique_deep_research.service import DeepResearchTool
 from unique_follow_up_questions.config import FollowUpQuestionsConfig
@@ -21,6 +22,7 @@ from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
 from unique_toolkit.agentic.history_manager.history_manager import (
     UploadedContentConfig,
 )
+from unique_toolkit.agentic.responses_api import ShowExecutedCodePostprocessorConfig
 from unique_toolkit.agentic.tools.a2a import (
     REFERENCING_INSTRUCTIONS_FOR_SYSTEM_PROMPT,
     REFERENCING_INSTRUCTIONS_FOR_USER_PROMPT,
@@ -32,6 +34,11 @@ from unique_toolkit.language_model.default_language_model import DEFAULT_GPT_4o
 from unique_web_search.config import WebSearchConfig
 from unique_web_search.service import WebSearchTool
 
+DeactivatedNone = Annotated[
+    None,
+    Field(title="Deactivated", description="None"),
+]
+
 
 class SpaceType(StrEnum):
     UNIQUE_CUSTOM = "unique_custom"
@@ -121,9 +128,6 @@ class EvaluationConfig(BaseModel):
     model_config = get_configuration_dict()
     max_review_steps: int = 3
     hallucination_config: HallucinationConfig = HallucinationConfig()
-    sub_agents_config: SubAgentEvaluationServiceConfig | None = (
-        SubAgentEvaluationServiceConfig()
-    )
 
 
 # ------------------------------------------------------------
@@ -149,12 +153,6 @@ class UniqueAIPromptConfig(BaseModel):
     )
 
 
-DeactivatedNone = Annotated[
-    None,
-    Field(title="Deactivated", description="None"),
-]
-
-
 class UniqueAIServices(BaseModel):
     """Determine the services the agent is using
 
@@ -205,12 +203,9 @@ class InputTokenDistributionConfig(BaseModel):
         return int(self.percent_for_history * max_input_token)
 
 
-class SubAgentsConfig(BaseModel):
+class SubAgentsReferencingConfig(BaseModel):
     model_config = get_configuration_dict()
-    use_sub_agent_references: bool = Field(
-        default=True,
-        description="Whether to use sub agent references in the main agent's response. Only has an effect if sub agents are used.",
-    )
+
     referencing_instructions_for_system_prompt: str = Field(
         default=REFERENCING_INSTRUCTIONS_FOR_SYSTEM_PROMPT,
         description="Referencing instructions for the main agent's system prompt.",
@@ -221,6 +216,44 @@ class SubAgentsConfig(BaseModel):
     )
 
 
+class SubAgentsConfig(BaseModel):
+    model_config = get_configuration_dict()
+
+    referencing_config: (
+        Annotated[SubAgentsReferencingConfig, Field(title="Active")] | DeactivatedNone
+    ) = SubAgentsReferencingConfig()
+    evaluation_config: (
+        Annotated[SubAgentEvaluationServiceConfig, Field(title="Active")]
+        | DeactivatedNone
+    ) = SubAgentEvaluationServiceConfig()
+
+
+class ResponsesApiConfig(BaseModel):
+    model_config = get_configuration_dict(frozen=True)
+
+    use_responses_api: bool = Field(
+        default=False,
+        description="Whether to use the responses API instead of the completions API.",
+    )
+    code_interpreter_display_config: (
+        Annotated[
+            ShowExecutedCodePostprocessorConfig,
+            Field(title="Active"),
+        ]
+        | DeactivatedNone
+    ) = ShowExecutedCodePostprocessorConfig()
+
+    use_direct_azure_client: SkipJsonSchema[bool] = Field(
+        default=True,
+        description="Temporary",
+    )
+
+    generated_files_scope_id: str = Field(
+        default="<SCOPE_ID_PLACEHOLDER>",
+        description="Scope ID for the responses API.",
+    )
+
+
 class ExperimentalConfig(BaseModel):
     """Experimental features this part of the configuration might evolve in the future continuously"""
 
@@ -254,6 +287,8 @@ class ExperimentalConfig(BaseModel):
 
     sub_agents_config: SubAgentsConfig = SubAgentsConfig()
 
+    responses_api_config: ResponsesApiConfig = ResponsesApiConfig()
+
 
 class UniqueAIAgentConfig(BaseModel):
     model_config = get_configuration_dict(frozen=True)
@@ -282,5 +317,5 @@ class UniqueAIConfig(BaseModel):
     @model_validator(mode="after")
     def disable_sub_agent_referencing_if_not_used(self) -> "UniqueAIConfig":
         if not any(tool.is_sub_agent for tool in self.space.tools):
-            self.agent.experimental.sub_agents_config.use_sub_agent_references = False
+            self.agent.experimental.sub_agents_config.referencing_config = None
         return self

unique_orchestrator/unique_ai.py

@@ -12,15 +12,22 @@ from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
 )
 from unique_toolkit.agentic.reference_manager.reference_manager import ReferenceManager
 from unique_toolkit.agentic.thinking_manager.thinking_manager import ThinkingManager
-from unique_toolkit.agentic.tools.tool_manager import ToolManager
+from unique_toolkit.agentic.tools.tool_manager import (
+    ResponsesApiToolManager,
+    ToolManager,
+)
 from unique_toolkit.app.schemas import ChatEvent, McpServer
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content.service import ContentService
+from unique_toolkit.language_model import LanguageModelAssistantMessage
 from unique_toolkit.language_model.schemas import (
-    LanguageModelAssistantMessage,
     LanguageModelMessages,
     LanguageModelStreamResponse,
 )
+from unique_toolkit.protocols.support import (
+    ResponsesSupportCompleteWithReferences,
+    SupportCompleteWithReferences,
+)
 
 from unique_orchestrator.config import UniqueAIConfig
 
@@ -45,6 +52,7 @@ class UniqueAI:
         chat_service: ChatService,
         content_service: ContentService,
         debug_info_manager: DebugInfoManager,
+        streaming_handler: SupportCompleteWithReferences,
         reference_manager: ReferenceManager,
         thinking_manager: ThinkingManager,
         tool_manager: ToolManager,
@@ -70,6 +78,7 @@ class UniqueAI:
         self._postprocessor_manager = postprocessor_manager
         self._latest_assistant_id: str = event.payload.assistant_message.id
         self._mcp_servers = mcp_servers
+        self._streaming_handler = streaming_handler
 
         # Helper variable to support control loop
         self._tool_took_control = False
@@ -148,7 +157,7 @@ class UniqueAI:
             self._logger.info("Its needs forced tool calls.")
             self._logger.info(f"Forced tools: {self._tool_manager.get_forced_tools()}")
             responses = [
-                await self._chat_service.complete_with_references_async(
+                await self._streaming_handler.complete_with_references_async(
                     messages=messages,
                     model_name=self._config.space.language_model.name,
                     tools=self._tool_manager.get_tool_definitions(),
@@ -156,8 +165,8 @@ class UniqueAI:
                     start_text=self.start_text,
                     debug_info=self._debug_info_manager.get(),
                     temperature=self._config.agent.experimental.temperature,
-                    other_options=self._config.agent.experimental.additional_llm_options
-                    | {"toolChoice": opt},
+                    tool_choice=opt,
+                    other_options=self._config.agent.experimental.additional_llm_options,
                 )
                 for opt in self._tool_manager.get_forced_tools()
             ]
@@ -178,7 +187,7 @@ class UniqueAI:
                 "we are in the last iteration we need to produce an answer now"
             )
             # No tool calls in last iteration
-            stream_response = await self._chat_service.complete_with_references_async(
+            stream_response = await self._streaming_handler.complete_with_references_async(
                 messages=messages,
                 model_name=self._config.space.language_model.name,
                 content_chunks=self._reference_manager.get_chunks(),
@@ -192,7 +201,7 @@ class UniqueAI:
             self._logger.info(
                 f"we are in the iteration {self.current_iteration_index} asking the model to tell if we should use tools or if it will just stream"
             )
-            stream_response = await self._chat_service.complete_with_references_async(
+            stream_response = await self._streaming_handler.complete_with_references_async(
                 messages=messages,
                 model_name=self._config.space.language_model.name,
                 tools=self._tool_manager.get_tool_definitions(),
@@ -260,10 +269,15 @@ class UniqueAI:
 
         query = self._event.payload.user_message.text
 
-        use_sub_agent_references = (
-            self._config.agent.experimental.sub_agents_config.use_sub_agent_references
-        )
-        sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_instructions_for_user_prompt
+        if (
+            self._config.agent.experimental.sub_agents_config.referencing_config
+            is not None
+        ):
+            use_sub_agent_references = True
+            sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_config.referencing_instructions_for_user_prompt
+        else:
+            use_sub_agent_references = False
+            sub_agent_referencing_instructions = None
 
         user_msg = user_message_template.render(
             query=query,
@@ -294,10 +308,15 @@ class UniqueAI:
             mcp_server.system_prompt for mcp_server in self._mcp_servers
         ]
 
-        use_sub_agent_references = (
-            self._config.agent.experimental.sub_agents_config.use_sub_agent_references
-        )
-        sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_instructions_for_system_prompt
+        if (
+            self._config.agent.experimental.sub_agents_config.referencing_config
+            is not None
+        ):
+            use_sub_agent_references = True
+            sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_config.referencing_instructions_for_system_prompt
+        else:
+            use_sub_agent_references = False
+            sub_agent_referencing_instructions = None
 
         system_message = system_prompt_template.render(
             model_info=self._config.space.language_model.model_dump(mode="json"),
@@ -341,7 +360,7 @@ class UniqueAI:
 
         tool_calls = loop_response.tool_calls or []
 
-        # Append function call to history
+        # Append function calls to history
         self._history_manager._append_tool_calls_to_history(tool_calls)
 
         # Execute tool calls
@@ -394,3 +413,39 @@ class UniqueAI:
                 content=loop_response.message.original_text or "",
             )
         )
+
+
+class UniqueAIResponsesApi(UniqueAI):
+    def __init__(
+        self,
+        logger: Logger,
+        event: ChatEvent,
+        config: UniqueAIConfig,
+        chat_service: ChatService,
+        content_service: ContentService,
+        debug_info_manager: DebugInfoManager,
+        streaming_handler: ResponsesSupportCompleteWithReferences,
+        reference_manager: ReferenceManager,
+        thinking_manager: ThinkingManager,
+        tool_manager: ResponsesApiToolManager,
+        history_manager: HistoryManager,
+        evaluation_manager: EvaluationManager,
+        postprocessor_manager: PostprocessorManager,
+        mcp_servers: list[McpServer],
+    ) -> None:
+        super().__init__(
+            logger,
+            event=event,
+            config=config,
+            chat_service=chat_service,
+            content_service=content_service,
+            debug_info_manager=debug_info_manager,
+            streaming_handler=streaming_handler,  # type: ignore
+            reference_manager=reference_manager,
+            thinking_manager=thinking_manager,
+            tool_manager=tool_manager,  # type: ignore
+            history_manager=history_manager,
+            evaluation_manager=evaluation_manager,
+            postprocessor_manager=postprocessor_manager,
+            mcp_servers=mcp_servers,
+        )

unique_orchestrator/unique_ai_builder.py

@@ -1,5 +1,8 @@
+import os
 from logging import Logger
+from typing import NamedTuple
 
+from openai import AsyncOpenAI
 from unique_follow_up_questions.follow_up_postprocessor import (
     FollowUpPostprocessor,
 )
@@ -12,7 +15,7 @@ from unique_internal_search.uploaded_search.service import (
 from unique_stock_ticker.stock_ticker_postprocessor import (
     StockTickerPostprocessor,
 )
-from unique_toolkit import LanguageModelService
+from unique_toolkit import LanguageModelService, get_async_openai_client
 from unique_toolkit.agentic.debug_info_manager.debug_info_manager import (
     DebugInfoManager,
 )
@@ -28,9 +31,15 @@ from unique_toolkit.agentic.history_manager.history_manager import (
     HistoryManagerConfig,
 )
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
+    Postprocessor,
     PostprocessorManager,
 )
 from unique_toolkit.agentic.reference_manager.reference_manager import ReferenceManager
+from unique_toolkit.agentic.responses_api import (
+    DisplayCodeInterpreterFilesPostProcessor,
+    DisplayCodeInterpreterFilesPostProcessorConfig,
+    ShowExecutedCodePostprocessor,
+)
 from unique_toolkit.agentic.thinking_manager.thinking_manager import (
     ThinkingManager,
     ThinkingManagerConfig,
@@ -43,31 +52,81 @@ from unique_toolkit.agentic.tools.a2a import (
 )
 from unique_toolkit.agentic.tools.config import ToolBuildConfig
 from unique_toolkit.agentic.tools.mcp.manager import MCPManager
-from unique_toolkit.agentic.tools.tool_manager import ToolManager, ToolManagerConfig
+from unique_toolkit.agentic.tools.tool_manager import (
+    OpenAIBuiltInToolManager,
+    ResponsesApiToolManager,
+    ToolManager,
+    ToolManagerConfig,
+)
 from unique_toolkit.agentic.tools.tool_progress_reporter import ToolProgressReporter
-from unique_toolkit.app.schemas import ChatEvent
+from unique_toolkit.app.schemas import ChatEvent, McpServer
 from unique_toolkit.chat.service import ChatService
+from unique_toolkit.content import Content
 from unique_toolkit.content.service import ContentService
+from unique_toolkit.protocols.support import ResponsesSupportCompleteWithReferences
 
 from unique_orchestrator.config import UniqueAIConfig
-from unique_orchestrator.unique_ai import UniqueAI
+from unique_orchestrator.unique_ai import UniqueAI, UniqueAIResponsesApi
 
 
-def build_unique_ai(
+async def build_unique_ai(
     event: ChatEvent,
     logger: Logger,
     config: UniqueAIConfig,
     debug_info_manager: DebugInfoManager,
-) -> UniqueAI:
+) -> UniqueAI | UniqueAIResponsesApi:
+    common_components = _build_common(event, logger, config)
+
+    if config.agent.experimental.responses_api_config.use_responses_api:
+        return await _build_responses(
+            event=event,
+            logger=logger,
+            config=config,
+            debug_info_manager=debug_info_manager,
+            common_components=common_components,
+        )
+    else:
+        return _build_completions(
+            event=event,
+            logger=logger,
+            config=config,
+            debug_info_manager=debug_info_manager,
+            common_components=common_components,
+        )
+
+
+class _CommonComponents(NamedTuple):
+    chat_service: ChatService
+    content_service: ContentService
+    uploaded_documents: list[Content]
+    thinking_manager: ThinkingManager
+    reference_manager: ReferenceManager
+    history_manager: HistoryManager
+    evaluation_manager: EvaluationManager
+    # Tool Manager Components
+    tool_progress_reporter: ToolProgressReporter
+    tool_manager_config: ToolManagerConfig
+    mcp_manager: MCPManager
+    a2a_manager: A2AManager
+    mcp_servers: list[McpServer]
+    postprocessors: list[Postprocessor]
+
+
+def _build_common(
+    event: ChatEvent,
+    logger: Logger,
+    config: UniqueAIConfig,
+) -> _CommonComponents:
     chat_service = ChatService(event)
 
     content_service = ContentService.from_event(event)
+
+    uploaded_documents = content_service.get_documents_uploaded_to_chat()
+
     tool_progress_reporter = ToolProgressReporter(chat_service=chat_service)
-    reference_manager = ReferenceManager()
     thinking_manager_config = ThinkingManagerConfig(
         thinking_steps_display=config.agent.experimental.thinking_steps_display
     )
-
     thinking_manager = ThinkingManager(
         logger=logger,
         config=thinking_manager_config,
@@ -75,44 +134,7 @@ def build_unique_ai(
         chat_service=chat_service,
     )
 
-    uploaded_documents = content_service.get_documents_uploaded_to_chat()
-    if len(uploaded_documents) > 0:
-        logger.info(
-            f"Adding UploadedSearchTool with {len(uploaded_documents)} documents"
-        )
-        config.space.tools.append(
-            ToolBuildConfig(
-                name=UploadedSearchTool.name,
-                display_name=UploadedSearchTool.name,
-                configuration=UploadedSearchConfig(),
-            ),
-        )
-        event.payload.tool_choices.append(str(UploadedSearchTool.name))
-
-    mcp_manager = MCPManager(
-        mcp_servers=event.payload.mcp_servers,
-        event=event,
-        tool_progress_reporter=tool_progress_reporter,
-    )
-
-    a2a_manager = A2AManager(
-        logger=logger,
-        tool_progress_reporter=tool_progress_reporter,
-    )
-
-    tool_config = ToolManagerConfig(
-        tools=config.space.tools,
-        max_tool_calls=config.agent.experimental.loop_configuration.max_tool_calls_per_iteration,
-    )
-
-    tool_manager = ToolManager(
-        logger=logger,
-        config=tool_config,
-        event=event,
-        tool_progress_reporter=tool_progress_reporter,
-        mcp_manager=mcp_manager,
-        a2a_manager=a2a_manager,
-    )
+    reference_manager = ReferenceManager()
 
     history_manager_config = HistoryManagerConfig(
         experimental_features=history_manager_module.ExperimentalFeatures(
@@ -122,7 +144,6 @@ def build_unique_ai(
         language_model=config.space.language_model,
         uploaded_content_config=config.agent.services.uploaded_content_config,
     )
-
     history_manager = HistoryManager(
         logger,
         event,
@@ -132,7 +153,6 @@ def build_unique_ai(
     )
 
     evaluation_manager = EvaluationManager(logger=logger, chat_service=chat_service)
-
     if config.agent.services.evaluation_config:
         evaluation_manager.add_evaluation(
             HallucinationEvaluation(
@@ -142,13 +162,24 @@ def build_unique_ai(
             )
         )
 
-    postprocessor_manager = PostprocessorManager(
+    mcp_manager = MCPManager(
+        mcp_servers=event.payload.mcp_servers,
+        event=event,
+        tool_progress_reporter=tool_progress_reporter,
+    )
+    a2a_manager = A2AManager(
         logger=logger,
-        chat_service=chat_service,
+        tool_progress_reporter=tool_progress_reporter,
     )
+    tool_manager_config = ToolManagerConfig(
+        tools=config.space.tools,
+        max_tool_calls=config.agent.experimental.loop_configuration.max_tool_calls_per_iteration,
+    )
+
+    postprocessors = []
 
     if config.agent.services.stock_ticker_config:
-        postprocessor_manager.add_postprocessor(
+        postprocessors.append(
             StockTickerPostprocessor(
                 config=config.agent.services.stock_ticker_config,
                 event=event,
@@ -159,7 +190,7 @@ def build_unique_ai(
         config.agent.services.follow_up_questions_config
         and config.agent.services.follow_up_questions_config.number_of_questions > 0
     ):
-        postprocessor_manager.add_postprocessor(
+        postprocessors.append(
             FollowUpPostprocessor(
                 logger=logger,
                 config=config.agent.services.follow_up_questions_config,
@@ -169,47 +200,248 @@ def build_unique_ai(
             )
         )
 
-    if len(tool_manager.sub_agents) > 0:
-        sub_agent_responses_postprocessor = SubAgentResponsesPostprocessor(
-            user_id=event.user_id,
-            main_agent_chat_id=event.payload.chat_id,
-            company_id=event.company_id,
+    return _CommonComponents(
+        chat_service=chat_service,
+        content_service=content_service,
+        uploaded_documents=uploaded_documents,
+        thinking_manager=thinking_manager,
+        reference_manager=reference_manager,
+        history_manager=history_manager,
+        evaluation_manager=evaluation_manager,
+        tool_progress_reporter=tool_progress_reporter,
+        mcp_manager=mcp_manager,
+        a2a_manager=a2a_manager,
+        tool_manager_config=tool_manager_config,
+        mcp_servers=event.payload.mcp_servers,
+        postprocessors=postprocessors,
+    )
+
+
+def _get_openai_client_from_env(config: UniqueAIConfig) -> AsyncOpenAI:
+    use_direct_azure_client = (
+        config.agent.experimental.responses_api_config.use_direct_azure_client
+    )
+    if use_direct_azure_client:
+        # TODO: (for testing only), remove when v1 endpoint is working
+        return AsyncOpenAI(
+            api_key=os.environ["OPENAI_API_KEY"],
+            base_url=os.environ["OPENAI_BASE_URL"],
+        )
+    else:
+        return get_async_openai_client().copy(
+            default_headers={
+                "x-model": config.space.language_model.name
+            }  # Backend requires a model name
         )
-        postprocessor_manager.add_postprocessor(sub_agent_responses_postprocessor)
 
-        sub_agent_evaluation = None
-        if (
-            config.agent.services.evaluation_config is not None
-            and config.agent.services.evaluation_config.sub_agents_config is not None
-        ):
-            sub_agent_evaluation = SubAgentEvaluationService(
-                config=config.agent.services.evaluation_config.sub_agents_config,
-                language_model_service=LanguageModelService.from_event(event),
+
+async def _build_responses(
+    event: ChatEvent,
+    logger: Logger,
+    config: UniqueAIConfig,
+    common_components: _CommonComponents,
+    debug_info_manager: DebugInfoManager,
+) -> UniqueAIResponsesApi:
+    client = _get_openai_client_from_env(config)
+    builtin_tool_manager = OpenAIBuiltInToolManager(
+        uploaded_files=common_components.uploaded_documents,
+        chat_id=event.payload.chat_id,
+        content_service=common_components.content_service,
+        user_id=event.user_id,
+        company_id=event.company_id,
+        client=client,
+    )
+
+    tool_manager = await ResponsesApiToolManager.build_manager(
+        logger=logger,
+        config=common_components.tool_manager_config,
+        event=event,
+        tool_progress_reporter=common_components.tool_progress_reporter,
+        mcp_manager=common_components.mcp_manager,
+        a2a_manager=common_components.a2a_manager,
+        builtin_tool_manager=builtin_tool_manager,
+    )
+
+    postprocessor_manager = PostprocessorManager(
+        logger=logger,
+        chat_service=common_components.chat_service,
+    )
+    for postprocessor in common_components.postprocessors:
+        postprocessor_manager.add_postprocessor(postprocessor)
+
+    if (
+        config.agent.experimental.responses_api_config.code_interpreter_display_config
+        is not None
+    ):
+        postprocessor_manager.add_postprocessor(
+            ShowExecutedCodePostprocessor(
+                config=config.agent.experimental.responses_api_config.code_interpreter_display_config
             )
-            evaluation_manager.add_evaluation(sub_agent_evaluation)
+        )
 
-        for tool in tool_manager.sub_agents:
-            assert isinstance(tool.config, ExtendedSubAgentToolConfig)
-            sub_agent_responses_postprocessor.register_sub_agent_tool(
-                tool, tool.config.response_display_config
+    postprocessor_manager.add_postprocessor(
+        DisplayCodeInterpreterFilesPostProcessor(
+            client=client,
+            content_service=common_components.content_service,
+            config=DisplayCodeInterpreterFilesPostProcessorConfig(
+                upload_scope_id=config.agent.experimental.responses_api_config.generated_files_scope_id,
+            ),
+        )
+    )
+
+    class ResponsesStreamingHandler(ResponsesSupportCompleteWithReferences):
+        def complete_with_references(self, *args, **kwargs):
+            return common_components.chat_service.complete_responses_with_references(
+                *args, **kwargs
             )
-            if sub_agent_evaluation is not None:
-                sub_agent_evaluation.register_sub_agent_tool(
-                    tool, tool.config.evaluation_config
-                )
+
+        async def complete_with_references_async(self, *args, **kwargs):
+            return await common_components.chat_service.complete_responses_with_references_async(
+                *args, **kwargs
+            )
+
+    streaming_handler = ResponsesStreamingHandler()
+
+    _add_sub_agents_postprocessor(
+        postprocessor_manager=postprocessor_manager,
+        tool_manager=tool_manager,
+        user_id=event.user_id,
+        company_id=event.company_id,
+        chat_id=event.payload.chat_id,
+    )
+    _add_sub_agents_evaluation(
+        evaluation_manager=common_components.evaluation_manager,
+        tool_manager=tool_manager,
+        config=config,
+        event=event,
+    )
+
+    return UniqueAIResponsesApi(
+        event=event,
+        config=config,
+        logger=logger,
+        chat_service=common_components.chat_service,
+        content_service=common_components.content_service,
+        tool_manager=tool_manager,
+        thinking_manager=common_components.thinking_manager,
+        streaming_handler=streaming_handler,
+        history_manager=common_components.history_manager,
+        reference_manager=common_components.reference_manager,
+        evaluation_manager=common_components.evaluation_manager,
+        postprocessor_manager=postprocessor_manager,
+        debug_info_manager=debug_info_manager,
+        mcp_servers=event.payload.mcp_servers,
+    )
+
+
+def _build_completions(
+    event: ChatEvent,
+    logger: Logger,
+    config: UniqueAIConfig,
+    common_components: _CommonComponents,
+    debug_info_manager: DebugInfoManager,
+) -> UniqueAI:
+    if len(common_components.uploaded_documents) > 0:
+        logger.info(
+            f"Adding UploadedSearchTool with {len(common_components.uploaded_documents)} documents"
+        )
+        config.space.tools.append(
+            ToolBuildConfig(
+                name=UploadedSearchTool.name,
+                display_name=UploadedSearchTool.name,
+                configuration=UploadedSearchConfig(),
+            ),
+        )
+        event.payload.tool_choices.append(str(UploadedSearchTool.name))
+
+    tool_manager = ToolManager(
+        logger=logger,
+        config=common_components.tool_manager_config,
+        event=event,
+        tool_progress_reporter=common_components.tool_progress_reporter,
+        mcp_manager=common_components.mcp_manager,
+        a2a_manager=common_components.a2a_manager,
+    )
+
+    postprocessor_manager = PostprocessorManager(
+        logger=logger,
+        chat_service=common_components.chat_service,
+    )
+    for postprocessor in common_components.postprocessors:
+        postprocessor_manager.add_postprocessor(postprocessor)
+
+    _add_sub_agents_postprocessor(
+        postprocessor_manager=postprocessor_manager,
+        tool_manager=tool_manager,
+        user_id=event.user_id,
+        company_id=event.company_id,
+        chat_id=event.payload.chat_id,
+    )
+    _add_sub_agents_evaluation(
+        evaluation_manager=common_components.evaluation_manager,
+        tool_manager=tool_manager,
+        config=config,
+        event=event,
+    )
 
     return UniqueAI(
         event=event,
         config=config,
         logger=logger,
-        chat_service=chat_service,
-        content_service=content_service,
+        chat_service=common_components.chat_service,
+        content_service=common_components.content_service,
         tool_manager=tool_manager,
-        thinking_manager=thinking_manager,
-        history_manager=history_manager,
-        reference_manager=reference_manager,
-        evaluation_manager=evaluation_manager,
+        thinking_manager=common_components.thinking_manager,
+        history_manager=common_components.history_manager,
+        reference_manager=common_components.reference_manager,
+        streaming_handler=common_components.chat_service,
+        evaluation_manager=common_components.evaluation_manager,
         postprocessor_manager=postprocessor_manager,
         debug_info_manager=debug_info_manager,
         mcp_servers=event.payload.mcp_servers,
     )
+
+
+def _add_sub_agents_postprocessor(
+    postprocessor_manager: PostprocessorManager,
+    tool_manager: ToolManager | ResponsesApiToolManager,
+    user_id: str,
+    company_id: str,
+    chat_id: str,
+) -> None:
+    sub_agents = tool_manager.sub_agents
+    if len(sub_agents) > 0:
+        sub_agent_responses_postprocessor = SubAgentResponsesPostprocessor(
+            user_id=user_id,
+            main_agent_chat_id=chat_id,
+            company_id=company_id,
+        )
+        postprocessor_manager.add_postprocessor(sub_agent_responses_postprocessor)
+
+        for tool in tool_manager.sub_agents:
+            assert isinstance(tool.config, ExtendedSubAgentToolConfig)
+            sub_agent_responses_postprocessor.register_sub_agent_tool(
+                tool, tool.config.response_display_config
+            )
+
+
+def _add_sub_agents_evaluation(
+    evaluation_manager: EvaluationManager,
+    tool_manager: ToolManager | ResponsesApiToolManager,
+    config: UniqueAIConfig,
+    event: ChatEvent,
+) -> None:
+    sub_agents = tool_manager.sub_agents
+    if len(sub_agents) > 0:
+        sub_agent_evaluation = None
+        if config.agent.experimental.sub_agents_config.evaluation_config is not None:
+            sub_agent_evaluation = SubAgentEvaluationService(
+                config=config.agent.experimental.sub_agents_config.evaluation_config,
+                language_model_service=LanguageModelService.from_event(event),
+            )
+            evaluation_manager.add_evaluation(sub_agent_evaluation)
+            for tool in tool_manager.sub_agents:
+                assert isinstance(tool.config, ExtendedSubAgentToolConfig)
+                sub_agent_evaluation.register_sub_agent_tool(
+                    tool, tool.config.evaluation_config
+                )

{unique_orchestrator-1.2.4.dist-info → unique_orchestrator-1.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.2.4
+Version: 1.4.0
 Summary: 
 License: Proprietary
 Author: Andreas Hauri
@@ -19,7 +19,7 @@ Requires-Dist: unique-follow-up-questions (>=1.1.2,<2.0.0)
 Requires-Dist: unique-internal-search (>=1.0.1,<2.0.0)
 Requires-Dist: unique-sdk (>=0.10.24,<0.11.0)
 Requires-Dist: unique-stock-ticker (>=1.0.2,<2.0.0)
-Requires-Dist: unique-toolkit (>=1.14.5,<2.0.0)
+Requires-Dist: unique-toolkit (>=1.16.0,<2.0.0)
 Requires-Dist: unique-web-search (>=1.3.1,<2.0.0)
 Description-Content-Type: text/markdown
 
@@ -33,6 +33,12 @@ 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).
 
+## [1.4.0] - 2025-10-14
+- Add responses api and code execution support.
+
+## [1.3.0] - 2025-10-14
+- Re-organize sub-agents configuration for clarity.
+
 ## [1.2.4] - 2025-10-14
 - Let control taking tool itself set the message state to completed
 

unique_orchestrator-1.4.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+unique_orchestrator/config.py,sha256=szJtEPLAOzpBV955boJYmQguu_bRc1gXQd-6lMysTNs,10546
+unique_orchestrator/prompts/generic_reference_prompt.jinja2,sha256=fYPaiE-N1gSoOqu85OeEBa_ttAim8grOhHuOHJjSHNU,2663
+unique_orchestrator/prompts/system_prompt.jinja2,sha256=YXFdx3PG2p4TKfjEpz7guIw2GaKoY-4zRMEzXaKhHXE,7213
+unique_orchestrator/prompts/user_message_prompt.jinja2,sha256=BQokpBh3H2J-rFk8i-PRph3jy4T1gAJPPb1mxxRWNuM,878
+unique_orchestrator/tests/test_unique_ai_reference_order.py,sha256=8mZeVP1k8neH4qrFW3oa3zwIdaq2c7R1VvurC7kjBU8,4445
+unique_orchestrator/unique_ai.py,sha256=7jnFkWbjm4musN7LTK03Ewwny9de_HrQ1kpcXzLeFfo,18793
+unique_orchestrator/unique_ai_builder.py,sha256=wfZeF9pqbxLqroKWJugJldHaZ3S4WQCot7uBECB4tic,15763
+unique_orchestrator-1.4.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_orchestrator-1.4.0.dist-info/METADATA,sha256=k1Y61MlrfhwKQL9TU4d9r6Izj8tC06Z8kBBIgskfz6A,2629
+unique_orchestrator-1.4.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_orchestrator-1.4.0.dist-info/RECORD,,

unique_orchestrator-1.2.4.dist-info/RECORD

@@ -1,11 +0,0 @@
-unique_orchestrator/config.py,sha256=miZXSKSyYkYahk4VOvZ5v0u93QQX-z0RO6aC0mtJu4w,9519
-unique_orchestrator/prompts/generic_reference_prompt.jinja2,sha256=fYPaiE-N1gSoOqu85OeEBa_ttAim8grOhHuOHJjSHNU,2663
-unique_orchestrator/prompts/system_prompt.jinja2,sha256=YXFdx3PG2p4TKfjEpz7guIw2GaKoY-4zRMEzXaKhHXE,7213
-unique_orchestrator/prompts/user_message_prompt.jinja2,sha256=BQokpBh3H2J-rFk8i-PRph3jy4T1gAJPPb1mxxRWNuM,878
-unique_orchestrator/tests/test_unique_ai_reference_order.py,sha256=8mZeVP1k8neH4qrFW3oa3zwIdaq2c7R1VvurC7kjBU8,4445
-unique_orchestrator/unique_ai.py,sha256=CBOd1eQuasvt-A1azsGZJbHlExm9UGsgy19_l613PoQ,16798
-unique_orchestrator/unique_ai_builder.py,sha256=FUIJAVv0lPTWDgWeMx-KuCyrIc8w8lWyEciFs3_7a3Q,7608
-unique_orchestrator-1.2.4.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_orchestrator-1.2.4.dist-info/METADATA,sha256=QkwdL4PgwDsEX2c-rSanDPDoN9AaIV11FVnmEkacK50,2479
-unique_orchestrator-1.2.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_orchestrator-1.2.4.dist-info/RECORD,,