unique_orchestrator 1.6.0__py3-none-any.whl → 1.7.0__py3-none-any.whl

unique_orchestrator/config.py

@@ -33,6 +33,9 @@ from unique_toolkit.agentic.tools.openai_builtin.manager import (
     OpenAICodeInterpreterConfig,
 )
 from unique_toolkit.agentic.tools.tool import ToolBuildConfig
+from unique_toolkit.agentic.tools.tool_progress_reporter import (
+    ToolProgressReporterConfig,
+)
 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
@@ -191,6 +194,10 @@ class UniqueAIServices(BaseModel):
 
     uploaded_content_config: UploadedContentConfig = UploadedContentConfig()
 
+    tool_progress_reporter_config: ToolProgressReporterConfig = (
+        ToolProgressReporterConfig()
+    )
+
 
 class InputTokenDistributionConfig(BaseModel):
     model_config = get_configuration_dict(frozen=True)

unique_orchestrator/unique_ai_builder.py

@@ -31,7 +31,6 @@ 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
@@ -107,13 +106,13 @@ class _CommonComponents(NamedTuple):
     reference_manager: ReferenceManager
     history_manager: HistoryManager
     evaluation_manager: EvaluationManager
+    postprocessor_manager: PostprocessorManager
     # 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(
@@ -127,7 +126,10 @@ def _build_common(
 
     uploaded_documents = content_service.get_documents_uploaded_to_chat()
 
-    tool_progress_reporter = ToolProgressReporter(chat_service=chat_service)
+    tool_progress_reporter = ToolProgressReporter(
+        chat_service=chat_service,
+        config=config.agent.services.tool_progress_reporter_config,
+    )
     thinking_manager_config = ThinkingManagerConfig(
         thinking_steps_display=config.agent.experimental.thinking_steps_display
     )
@@ -141,9 +143,7 @@ def _build_common(
     reference_manager = ReferenceManager()
 
     history_manager_config = HistoryManagerConfig(
-        experimental_features=history_manager_module.ExperimentalFeatures(
-            full_sources_serialize_dump=False,
-        ),
+        experimental_features=history_manager_module.ExperimentalFeatures(),
         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,
@@ -180,10 +180,13 @@ def _build_common(
         max_tool_calls=config.agent.experimental.loop_configuration.max_tool_calls_per_iteration,
     )
 
-    postprocessors = []
+    postprocessor_manager = PostprocessorManager(
+        logger=logger,
+        chat_service=chat_service,
+    )
 
-    if config.agent.services.stock_ticker_config:
-        postprocessors.append(
+    if config.agent.services.stock_ticker_config is not None:
+        postprocessor_manager.add_postprocessor(
             StockTickerPostprocessor(
                 config=config.agent.services.stock_ticker_config,
                 event=event,
@@ -194,7 +197,8 @@ def _build_common(
         config.agent.services.follow_up_questions_config
         and config.agent.services.follow_up_questions_config.number_of_questions > 0
     ):
-        postprocessors.append(
+        # Should run last to make sure the follow up questions are displayed last.
+        postprocessor_manager.set_last_postprocessor(
             FollowUpPostprocessor(
                 logger=logger,
                 config=config.agent.services.follow_up_questions_config,
@@ -217,7 +221,7 @@ def _build_common(
         a2a_manager=a2a_manager,
         tool_manager_config=tool_manager_config,
         mcp_servers=event.payload.mcp_servers,
-        postprocessors=postprocessors,
+        postprocessor_manager=postprocessor_manager,
     )
 
 
@@ -299,12 +303,7 @@ async def _build_responses(
         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)
+    postprocessor_manager = common_components.postprocessor_manager
 
     if (
         config.agent.experimental.responses_api_config.code_interpreter_display_config
@@ -410,12 +409,7 @@ def _build_completions(
     if not TOOL_CHOICES and UPLOADED_DOCUMENTS:
         tool_manager.add_forced_tool(UploadedSearchTool.name)
 
-    postprocessor_manager = PostprocessorManager(
-        logger=logger,
-        chat_service=common_components.chat_service,
-    )
-    for postprocessor in common_components.postprocessors:
-        postprocessor_manager.add_postprocessor(postprocessor)
+    postprocessor_manager = common_components.postprocessor_manager
 
     _add_sub_agents_postprocessor(
         postprocessor_manager=postprocessor_manager,

{unique_orchestrator-1.6.0.dist-info → unique_orchestrator-1.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.6.0
+Version: 1.7.0
 Summary: 
 License: Proprietary
 Author: Andreas Hauri
@@ -33,6 +33,13 @@ 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.7.0] - 2025-10-30
+- Add option to customize the display of tool progress statuses.
+- Make follow-questions postprocessor run last to make sure the follow up questions are displayed last.
+
+## [1.6.1] - 2025-10-28
+- Removing unused experimental config `full_sources_serialize_dump` in `history_manager`
+
 ## [1.6.0] - 2025-10-27
 - Add temporary config option `sleep_time_before_update` to avoid rendering issues with sub agent responses`
 

{unique_orchestrator-1.6.0.dist-info → unique_orchestrator-1.7.0.dist-info}/RECORD

@@ -1,11 +1,11 @@
-unique_orchestrator/config.py,sha256=iSNjiXacox0ZCfDO8whWgCjWzRLa1WKK4SLZF_uzpwk,11587
+unique_orchestrator/config.py,sha256=NNBPNiLWZN0pFemOtUSxVbEcCEFoUbmu1p8i-hbxlRY,11797
 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=OO-85-Qvw5tf5A227_ur34p3qn8RKnU4EPJ21x7jdQE,19324
-unique_orchestrator/unique_ai_builder.py,sha256=SCk7dP3cRgD4hNnlhgHTH_79Dx1Og2brOhhe8gZtC04,17993
-unique_orchestrator-1.6.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_orchestrator-1.6.0.dist-info/METADATA,sha256=ll6s_8fzkumdRgcZwTusCO1MX2UaKnhjjGF6OZs6bXs,3370
-unique_orchestrator-1.6.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_orchestrator-1.6.0.dist-info/RECORD,,
+unique_orchestrator/unique_ai_builder.py,sha256=p7cXCdXN36_CAgTiq-YXuPlaTjWI2D-QTwMSe0S7Chg,17877
+unique_orchestrator-1.7.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_orchestrator-1.7.0.dist-info/METADATA,sha256=bZx2lv3lGnGuH5bSz-tFudCURZ_0519DBTeUxyKhGn4,3678
+unique_orchestrator-1.7.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_orchestrator-1.7.0.dist-info/RECORD,,