unique_orchestrator 1.3.0__tar.gz → 1.4.0__tar.gz

{unique_orchestrator-1.3.0 → unique_orchestrator-1.4.0}/CHANGELOG.md

@@ -5,6 +5,9 @@ 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.
 

{unique_orchestrator-1.3.0 → unique_orchestrator-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.3.0
+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,9 @@ 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.
 

{unique_orchestrator-1.3.0 → unique_orchestrator-1.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_orchestrator"
-version = "1.3.0"
+version = "1.4.0"
 description = ""
 authors = ["Andreas Hauri <andreas.hauri@unique.ai>"]
 readme = ["README.md", "CHANGELOG.md"]
@@ -15,7 +15,7 @@ python-dotenv = "^1.0.1"
 pytest = "^8.4.1"
 unique-sdk = "^0.10.24"
 
-unique-toolkit = "^1.14.5"
+unique-toolkit = "^1.16.0"
 unique-stock-ticker = "^1.0.2"
 unique-follow-up-questions = "^1.1.2"
 unique-internal-search = "^1.0.1"

{unique_orchestrator-1.3.0 → unique_orchestrator-1.4.0}/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,
@@ -226,6 +228,32 @@ class SubAgentsConfig(BaseModel):
     ) = 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"""
 
@@ -259,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)

{unique_orchestrator-1.3.0 → unique_orchestrator-1.4.0}/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(),
@@ -351,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
@@ -404,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-1.4.0/unique_orchestrator/unique_ai_builder.py

@@ -0,0 +1,447 @@
+import os
+from logging import Logger
+from typing import NamedTuple
+
+from openai import AsyncOpenAI
+from unique_follow_up_questions.follow_up_postprocessor import (
+    FollowUpPostprocessor,
+)
+from unique_internal_search.uploaded_search.config import (
+    UploadedSearchConfig,
+)
+from unique_internal_search.uploaded_search.service import (
+    UploadedSearchTool,
+)
+from unique_stock_ticker.stock_ticker_postprocessor import (
+    StockTickerPostprocessor,
+)
+from unique_toolkit import LanguageModelService, get_async_openai_client
+from unique_toolkit.agentic.debug_info_manager.debug_info_manager import (
+    DebugInfoManager,
+)
+from unique_toolkit.agentic.evaluation.evaluation_manager import EvaluationManager
+from unique_toolkit.agentic.evaluation.hallucination.hallucination_evaluation import (
+    HallucinationEvaluation,
+)
+from unique_toolkit.agentic.history_manager import (
+    history_manager as history_manager_module,
+)
+from unique_toolkit.agentic.history_manager.history_manager import (
+    HistoryManager,
+    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,
+)
+from unique_toolkit.agentic.tools.a2a import (
+    A2AManager,
+    ExtendedSubAgentToolConfig,
+    SubAgentEvaluationService,
+    SubAgentResponsesPostprocessor,
+)
+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 (
+    OpenAIBuiltInToolManager,
+    ResponsesApiToolManager,
+    ToolManager,
+    ToolManagerConfig,
+)
+from unique_toolkit.agentic.tools.tool_progress_reporter import ToolProgressReporter
+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, UniqueAIResponsesApi
+
+
+async def build_unique_ai(
+    event: ChatEvent,
+    logger: Logger,
+    config: UniqueAIConfig,
+    debug_info_manager: DebugInfoManager,
+) -> 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)
+    thinking_manager_config = ThinkingManagerConfig(
+        thinking_steps_display=config.agent.experimental.thinking_steps_display
+    )
+    thinking_manager = ThinkingManager(
+        logger=logger,
+        config=thinking_manager_config,
+        tool_progress_reporter=tool_progress_reporter,
+        chat_service=chat_service,
+    )
+
+    reference_manager = ReferenceManager()
+
+    history_manager_config = HistoryManagerConfig(
+        experimental_features=history_manager_module.ExperimentalFeatures(
+            full_sources_serialize_dump=False,
+        ),
+        percent_of_max_tokens_for_history=config.agent.input_token_distribution.percent_for_history,
+        language_model=config.space.language_model,
+        uploaded_content_config=config.agent.services.uploaded_content_config,
+    )
+    history_manager = HistoryManager(
+        logger,
+        event,
+        history_manager_config,
+        config.space.language_model,
+        reference_manager,
+    )
+
+    evaluation_manager = EvaluationManager(logger=logger, chat_service=chat_service)
+    if config.agent.services.evaluation_config:
+        evaluation_manager.add_evaluation(
+            HallucinationEvaluation(
+                config.agent.services.evaluation_config.hallucination_config,
+                event,
+                reference_manager,
+            )
+        )
+
+    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_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:
+        postprocessors.append(
+            StockTickerPostprocessor(
+                config=config.agent.services.stock_ticker_config,
+                event=event,
+            )
+        )
+
+    if (
+        config.agent.services.follow_up_questions_config
+        and config.agent.services.follow_up_questions_config.number_of_questions > 0
+    ):
+        postprocessors.append(
+            FollowUpPostprocessor(
+                logger=logger,
+                config=config.agent.services.follow_up_questions_config,
+                event=event,
+                historyManager=history_manager,
+                llm_service=LanguageModelService.from_event(event),
+            )
+        )
+
+    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
+        )
+
+
+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
+            )
+        )
+
+    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
+            )
+
+        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=common_components.chat_service,
+        content_service=common_components.content_service,
+        tool_manager=tool_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.3.0/unique_orchestrator/unique_ai_builder.py

@@ -1,212 +0,0 @@
-from logging import Logger
-
-from unique_follow_up_questions.follow_up_postprocessor import (
-    FollowUpPostprocessor,
-)
-from unique_internal_search.uploaded_search.config import (
-    UploadedSearchConfig,
-)
-from unique_internal_search.uploaded_search.service import (
-    UploadedSearchTool,
-)
-from unique_stock_ticker.stock_ticker_postprocessor import (
-    StockTickerPostprocessor,
-)
-from unique_toolkit import LanguageModelService
-from unique_toolkit.agentic.debug_info_manager.debug_info_manager import (
-    DebugInfoManager,
-)
-from unique_toolkit.agentic.evaluation.evaluation_manager import EvaluationManager
-from unique_toolkit.agentic.evaluation.hallucination.hallucination_evaluation import (
-    HallucinationEvaluation,
-)
-from unique_toolkit.agentic.history_manager import (
-    history_manager as history_manager_module,
-)
-from unique_toolkit.agentic.history_manager.history_manager import (
-    HistoryManager,
-    HistoryManagerConfig,
-)
-from unique_toolkit.agentic.postprocessor.postprocessor_manager import (
-    PostprocessorManager,
-)
-from unique_toolkit.agentic.reference_manager.reference_manager import ReferenceManager
-from unique_toolkit.agentic.thinking_manager.thinking_manager import (
-    ThinkingManager,
-    ThinkingManagerConfig,
-)
-from unique_toolkit.agentic.tools.a2a import (
-    A2AManager,
-    ExtendedSubAgentToolConfig,
-    SubAgentEvaluationService,
-    SubAgentResponsesPostprocessor,
-)
-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_progress_reporter import ToolProgressReporter
-from unique_toolkit.app.schemas import ChatEvent
-from unique_toolkit.chat.service import ChatService
-from unique_toolkit.content.service import ContentService
-
-from unique_orchestrator.config import UniqueAIConfig
-from unique_orchestrator.unique_ai import UniqueAI
-
-
-def build_unique_ai(
-    event: ChatEvent,
-    logger: Logger,
-    config: UniqueAIConfig,
-    debug_info_manager: DebugInfoManager,
-) -> UniqueAI:
-    chat_service = ChatService(event)
-
-    content_service = ContentService.from_event(event)
-    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,
-        tool_progress_reporter=tool_progress_reporter,
-        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,
-    )
-
-    history_manager_config = HistoryManagerConfig(
-        experimental_features=history_manager_module.ExperimentalFeatures(
-            full_sources_serialize_dump=False,
-        ),
-        percent_of_max_tokens_for_history=config.agent.input_token_distribution.percent_for_history,
-        language_model=config.space.language_model,
-        uploaded_content_config=config.agent.services.uploaded_content_config,
-    )
-
-    history_manager = HistoryManager(
-        logger,
-        event,
-        history_manager_config,
-        config.space.language_model,
-        reference_manager,
-    )
-
-    evaluation_manager = EvaluationManager(logger=logger, chat_service=chat_service)
-
-    if config.agent.services.evaluation_config:
-        evaluation_manager.add_evaluation(
-            HallucinationEvaluation(
-                config.agent.services.evaluation_config.hallucination_config,
-                event,
-                reference_manager,
-            )
-        )
-
-    postprocessor_manager = PostprocessorManager(
-        logger=logger,
-        chat_service=chat_service,
-    )
-
-    if config.agent.services.stock_ticker_config:
-        postprocessor_manager.add_postprocessor(
-            StockTickerPostprocessor(
-                config=config.agent.services.stock_ticker_config,
-                event=event,
-            )
-        )
-
-    if (
-        config.agent.services.follow_up_questions_config
-        and config.agent.services.follow_up_questions_config.number_of_questions > 0
-    ):
-        postprocessor_manager.add_postprocessor(
-            FollowUpPostprocessor(
-                logger=logger,
-                config=config.agent.services.follow_up_questions_config,
-                event=event,
-                historyManager=history_manager,
-                llm_service=LanguageModelService.from_event(event),
-            )
-        )
-
-    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,
-        )
-        postprocessor_manager.add_postprocessor(sub_agent_responses_postprocessor)
-
-        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_responses_postprocessor.register_sub_agent_tool(
-                tool, tool.config.response_display_config
-            )
-            if sub_agent_evaluation is not None:
-                sub_agent_evaluation.register_sub_agent_tool(
-                    tool, tool.config.evaluation_config
-                )
-
-    return UniqueAI(
-        event=event,
-        config=config,
-        logger=logger,
-        chat_service=chat_service,
-        content_service=content_service,
-        tool_manager=tool_manager,
-        thinking_manager=thinking_manager,
-        history_manager=history_manager,
-        reference_manager=reference_manager,
-        evaluation_manager=evaluation_manager,
-        postprocessor_manager=postprocessor_manager,
-        debug_info_manager=debug_info_manager,
-        mcp_servers=event.payload.mcp_servers,
-    )