unique_orchestrator 0.0.1__py3-none-any.whl

unique_orchestrator/unique_ai_builder.py

@@ -0,0 +1,178 @@
+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.app.schemas import ChatEvent
+from unique_toolkit.chat.service import ChatService
+from unique_toolkit.content.service import ContentService
+from unique_toolkit.debug_info_manager.debug_info_manager import (
+    DebugInfoManager,
+)
+from unique_toolkit.evals.evaluation_manager import EvaluationManager
+from unique_toolkit.evals.hallucination.hallucination_evaluation import (
+    HallucinationEvaluation,
+)
+from unique_toolkit.history_manager import (
+    history_manager as history_manager_module,
+)
+from unique_toolkit.history_manager.history_manager import (
+    HistoryManager,
+    HistoryManagerConfig,
+)
+from unique_toolkit.postprocessor.postprocessor_manager import (
+    PostprocessorManager,
+)
+from unique_toolkit.reference_manager.reference_manager import ReferenceManager
+from unique_toolkit.thinking_manager.thinking_manager import (
+    ThinkingManager,
+    ThinkingManagerConfig,
+)
+from unique_toolkit.tools.a2a.manager import A2AManager
+from unique_toolkit.tools.config import ToolBuildConfig
+from unique_toolkit.tools.mcp.manager import MCPManager
+from unique_toolkit.tools.tool_manager import ToolManager, ToolManagerConfig
+from unique_toolkit.tools.tool_progress_reporter import ToolProgressReporter
+
+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:
+        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),
+            )
+        )
+
+    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,
+    )

unique_orchestrator-0.0.1.dist-info/LICENSE

@@ -0,0 +1 @@
+`unique_toolkit` is covered by the [`Unique License v1`](https://github.com/Unique-AG/license/releases/tag/unique-license.v1), unless the/a header or a nested LICENSE specifies another license.

unique_orchestrator-0.0.1.dist-info/METADATA

@@ -0,0 +1,39 @@
+Metadata-Version: 2.1
+Name: unique_orchestrator
+Version: 0.0.1
+Summary: 
+License: Proprietary
+Author: Martin Fadler
+Author-email: martin.fadler@unique.ch
+Requires-Python: >=3.11,<4.0
+Classifier: License :: Other/Proprietary License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: pydantic (>=2.8.2,<3.0.0)
+Requires-Dist: pydantic-settings (>=2.10.1,<3.0.0)
+Requires-Dist: pytest (>=8.4.1,<9.0.0)
+Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
+Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
+Requires-Dist: unique-deep-research (>=0.0.10,<0.0.11)
+Requires-Dist: unique-follow-up-questions (>=0.0.4,<0.0.5)
+Requires-Dist: unique-internal-search (==0.0.5)
+Requires-Dist: unique-sdk (>=0.10.0,<0.11.0)
+Requires-Dist: unique-stock-ticker (>=0.0.5,<0.0.6)
+Requires-Dist: unique-toolkit (>=0.8.57,<0.9.0)
+Requires-Dist: unique-web-search (>=0.1.0,<0.2.0)
+Description-Content-Type: text/markdown
+
+# Internal Search Tool
+
+Internal Search Tool to find documents in the Knowledge Base
+# Changelog
+
+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).
+
+
+## [0.0.1] - 2025-08-18
+- Initial release of `orchestrator`.

unique_orchestrator-0.0.1.dist-info/RECORD

@@ -0,0 +1,12 @@
+unique_orchestrator/config.py,sha256=y5keSY9hDrtinV6LnMxpIwEOQIJ1zEfwKfr8EQvX-wY,16652
+unique_orchestrator/prompts/generic_reference_prompt.jinja2,sha256=fYPaiE-N1gSoOqu85OeEBa_ttAim8grOhHuOHJjSHNU,2663
+unique_orchestrator/prompts/system_prompt.jinja2,sha256=2Dy8GHrT018Sj9UkYGxSfjt7fvTTDvD53BxjDMZRPVY,6897
+unique_orchestrator/prompts/user_message_prompt.jinja2,sha256=KLCfbxMu5R7_V8-AHkdXEmVorcSHNej_xM_7xJyNkl0,692
+unique_orchestrator/tests/test_config.py,sha256=P7QkVlVv0ppiK77s5i4rE4IOzdtdBCZZkeIA3KRc82k,4070
+unique_orchestrator/tests/test_unique_ai_reference_order.py,sha256=8mZeVP1k8neH4qrFW3oa3zwIdaq2c7R1VvurC7kjBU8,4445
+unique_orchestrator/unique_ai.py,sha256=xGQogDbxA7qjfk8R85t5m8PI_XDY7tk-s4tkvp8on_Q,15574
+unique_orchestrator/unique_ai_builder.py,sha256=J73_Lg3XjI9JKirUmC6juWTefk7gPkG67NPLdUL4zr0,5991
+unique_orchestrator-0.0.1.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_orchestrator-0.0.1.dist-info/METADATA,sha256=z1CJYqVYKbGHsQe8ScjDIf4jXHuBkAM9wWiDPBUaRlE,1408
+unique_orchestrator-0.0.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_orchestrator-0.0.1.dist-info/RECORD,,

unique_orchestrator-0.0.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.9.0
+Root-Is-Purelib: true
+Tag: py3-none-any