unique_orchestrator 0.0.2__py3-none-any.whl → 0.0.3__py3-none-any.whl

unique_orchestrator/config.py

@@ -20,13 +20,7 @@ from unique_toolkit.evals.schemas import EvaluationMetricName
 from unique_toolkit.history_manager.history_manager import (
     UploadedContentConfig,
 )
-from unique_toolkit.language_model import LanguageModelName
-from unique_toolkit.language_model.infos import (
-    LanguageModelInfo,
-)
 from unique_toolkit.tools.config import get_configuration_dict
-from unique_toolkit.tools.factory import ToolFactory
-from unique_toolkit.tools.schemas import BaseToolConfig
 from unique_toolkit.tools.tool import ToolBuildConfig
 from unique_web_search.config import WebSearchConfig
 from unique_web_search.service import WebSearchTool
@@ -104,7 +98,6 @@ class UniqueAISpaceConfig(SpaceConfigBase):
 
 UniqueAISpaceConfig.model_rebuild()
 
-LIMIT_LOOP_ITERATIONS = 50
 LIMIT_MAX_TOOL_CALLS_PER_ITERATION = 50
 
 
@@ -123,135 +116,6 @@ class EvaluationConfig(BaseModel):
     hallucination_config: HallucinationConfig = HallucinationConfig()
 
 
-class LoopAgentTokenLimitsConfig(BaseModel):
-    model_config = get_configuration_dict()
-
-    language_model: LMI = LanguageModelInfo.from_name(
-        LanguageModelName.AZURE_GPT_4o_2024_1120
-    )
-
-    percent_of_max_tokens_for_history: float = Field(
-        default=0.2,
-        ge=0.0,
-        lt=1.0,
-        description="The fraction of the max input tokens that will be reserved for the history.",
-    )
-
-    @property
-    def max_history_tokens(self) -> int:
-        return int(
-            self.language_model.token_limits.token_limit_input
-            * self.percent_of_max_tokens_for_history,
-        )
-
-
-class SearchAgentConfig(BaseModel):
-    """Configure the search agent."""
-
-    model_config = get_configuration_dict(frozen=True)
-
-    language_model: LMI = LanguageModelInfo.from_name(DEFAULT_GPT_4o)
-
-    token_limits: LoopAgentTokenLimitsConfig = Field(
-        default=LoopAgentTokenLimitsConfig(percent_of_max_tokens_for_history=0.6)
-    )
-    temperature: float = 0.0
-    additional_llm_options: dict[str, Any] = Field(
-        default={},
-        description="Additional options to pass to the language model.",
-    )
-
-    # Space 2.0
-    project_name: str = Field(
-        default="Unique AI",
-        description="The project name as optained from spaces 2.0",
-    )
-
-    # Space 2.0
-    custom_instructions: str = Field(
-        default="",
-        description="A custom instruction provided by the system admin.",
-    )
-
-    thinking_steps_display: bool = False
-
-    ##############################
-    ### General Configurations
-    ##############################
-    max_loop_iterations: Annotated[
-        int, *ClipInt(min_value=1, max_value=LIMIT_LOOP_ITERATIONS)
-    ] = 8
-
-    loop_configuration: LoopConfiguration = LoopConfiguration()
-
-    tools: list[ToolBuildConfig] = Field(
-        default=[
-            ToolBuildConfig(
-                name=InternalSearchTool.name,
-                configuration=InternalSearchConfig(
-                    exclude_uploaded_files=True,
-                ),
-            ),
-            ToolBuildConfig(
-                name=WebSearchTool.name,
-                configuration=WebSearchConfig(),
-            ),
-            ToolBuildConfig(
-                name=DeepResearchTool.name,
-                configuration=DeepResearchToolConfig(),
-            ),
-        ],
-    )
-
-    system_prompt_template: str = Field(
-        default_factory=lambda: (
-            Path(__file__).parent / "prompts" / "system_prompt.jinja2"
-        ).read_text(),
-        description="The system prompt template as a Jinja2 template string.",
-    )
-
-    user_message_prompt_template: str = Field(
-        default_factory=lambda: (
-            Path(__file__).parent / "prompts" / "user_message_prompt.jinja2"
-        ).read_text(),
-        description="The user message prompt template as a Jinja2 template string.",
-    )
-
-    ##############################
-    ### Follow-up Questions
-    ##############################
-    follow_up_questions_config: FollowUpQuestionsConfig = FollowUpQuestionsConfig()
-
-    ##############################
-    ### Evaluation
-    ##############################
-    evaluation_config: EvaluationConfig = EvaluationConfig(
-        hallucination_config=HallucinationConfig(),
-        max_review_steps=0,
-    )
-
-    ##############################
-    ### Stock Ticker
-    ##############################
-    stock_ticker_config: StockTickerConfig = StockTickerConfig()
-
-    # TODO: generalize this there should only be 1 point in the code where we do the tool check.
-    def get_tool_config(self, tool: str) -> BaseToolConfig:
-        """Get the tool configuration by name."""
-        return ToolFactory.build_tool_config(tool)
-
-    # TODO: @gustavhartz, the Hallucination check should be triggered if enabled and the answer contains references.
-    force_checks_on_stream_response_references: list[EvaluationMetricName] = Field(
-        default=[EvaluationMetricName.HALLUCINATION],
-        description="A list of checks to force on references. This is used to add hallucination check to references without new tool calls.",
-    )
-
-    uploaded_content_config: UploadedContentConfig = Field(
-        default_factory=UploadedContentConfig,
-        description="The uploaded content config.",
-    )
-
-
 # ------------------------------------------------------------
 # Space 2.0 Config
 # ------------------------------------------------------------
@@ -386,120 +250,3 @@ class UniqueAIConfig(BaseModel):
     space: UniqueAISpaceConfig = UniqueAISpaceConfig()
 
     agent: UniqueAIAgentConfig = UniqueAIAgentConfig()
-
-
-# ---
-# Configuration adapter SearchAgentConfig -> UniqueAISpaceConfig
-# --
-
-
-def search_agent_config_to_unique_ai_space_config(
-    search_agent_config: SearchAgentConfig,
-) -> UniqueAIConfig:
-    space = UniqueAISpaceConfig(
-        project_name=search_agent_config.project_name,
-        custom_instructions=search_agent_config.custom_instructions,
-        tools=search_agent_config.tools,
-        language_model=search_agent_config.language_model,
-        type=SpaceType.UNIQUE_AI,
-    )
-
-    prompt_config = UniqueAIPromptConfig(
-        system_prompt_template=search_agent_config.system_prompt_template,
-        user_message_prompt_template=search_agent_config.user_message_prompt_template,
-    )
-
-    services = UniqueAIServices(
-        follow_up_questions_config=search_agent_config.follow_up_questions_config,
-        evaluation_config=search_agent_config.evaluation_config,
-        stock_ticker_config=search_agent_config.stock_ticker_config,
-        uploaded_content_config=search_agent_config.uploaded_content_config,
-    )
-
-    experimental = ExperimentalConfig(
-        thinking_steps_display=search_agent_config.thinking_steps_display,
-        force_checks_on_stream_response_references=search_agent_config.force_checks_on_stream_response_references,
-        temperature=search_agent_config.temperature,
-        additional_llm_options=search_agent_config.additional_llm_options,
-        loop_configuration=search_agent_config.loop_configuration,
-    )
-
-    # Calculate remaining token percentages based on history percentage
-
-    history_percent = search_agent_config.token_limits.percent_of_max_tokens_for_history
-
-    agent = UniqueAIAgentConfig(
-        max_loop_iterations=search_agent_config.max_loop_iterations,
-        input_token_distribution=InputTokenDistributionConfig(
-            percent_for_history=history_percent,
-        ),
-        prompt_config=prompt_config,
-        services=services,
-        experimental=experimental,
-    )
-
-    return UniqueAIConfig(
-        space=space,
-        agent=agent,
-    )
-
-
-def needs_conversion_to_unique_ai_space_config(
-    configuration: dict[str, Any],
-) -> bool:
-    """Check if the configuration needs to be converted to the new UniqueAISpaceConfig."""
-    if (
-        "space_two_point_zero" in configuration
-        or "SpaceTwoPointZeroConfig" in configuration
-        or ("space" in configuration and "agent" in configuration)
-    ):
-        return False
-
-    return True
-
-
-if __name__ == "__main__":
-    import json
-
-    from unique_toolkit._common.utils.write_configuration import (
-        write_service_configuration,
-    )
-
-    write_service_configuration(
-        service_folderpath=Path(__file__).parent.parent,
-        write_folderpath=Path(__file__).parent,
-        config=UniqueAIConfig(),
-        sub_name="unique_ai_config",
-    )
-
-    # TODO: @cdkl Delete these models
-    # This model is only used to have the old and new models in the same json
-    # schema for the data migration in the node chat backend
-
-    # The types can be generated with quicktype.io with the following command:
-    # quicktype unique_ai_old_and_new_config.json \
-    #   --src-lang schema --lang typescript \
-    #   --just-types --prefer-types --explicit-unions \
-    #   -o unique_ai_old_new_configuration.ts \
-    #   --top-level UniqueAIOldAndNewConfig \
-    #   --raw-type any
-
-    # You will need to replace the `any` type with `unknown` in the generated file.
-    # On the branch `feat/unique-ai-configuration-migration-node-chat-part`.
-    # I you further update the types you will need to adapt both branches
-    # - feat/unique-ai-configuration-migration-next-admin-part
-    # - feat/unique-ai-configuration-migration-node-chat-part
-
-    class UniqueAIOldAndNewConfig(BaseModel):
-        new: UniqueAIConfig = UniqueAIConfig()
-        old: SearchAgentConfig = SearchAgentConfig()
-
-    with open(
-        Path(__file__).parent / "unique_ai_old_and_new_config.json",
-        "w",
-    ) as f:
-        json.dump(
-            UniqueAIOldAndNewConfig().model_json_schema(by_alias=True),
-            f,
-            indent=4,
-        )

{unique_orchestrator-0.0.2.dist-info → unique_orchestrator-0.0.3.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 0.0.2
+Version: 0.0.3
 Summary: 
 License: Proprietary
-Author: Martin Fadler
-Author-email: martin.fadler@unique.ch
+Author: Andreas Hauri
+Author-email: andreas.hauri@unique.ai
 Requires-Python: >=3.11,<4.0
 Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
@@ -34,6 +34,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).
 
+## [0.0.3] - 2025-09-16
+- Cleaned configuration
+
 ## [0.0.2] - 2025-09-15
 - Resolve dependency bug
 

{unique_orchestrator-0.0.2.dist-info → unique_orchestrator-0.0.3.dist-info}/RECORD

@@ -1,4 +1,4 @@
-unique_orchestrator/config.py,sha256=y5keSY9hDrtinV6LnMxpIwEOQIJ1zEfwKfr8EQvX-wY,16652
+unique_orchestrator/config.py,sha256=SV5jBkGlRg4BrFiDlAIMXVSRy0k_EMnZE8dmhvBtfIQ,8004
 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
@@ -6,7 +6,7 @@ unique_orchestrator/tests/test_config.py,sha256=P7QkVlVv0ppiK77s5i4rE4IOzdtdBCZZ
 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.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_orchestrator-0.0.2.dist-info/METADATA,sha256=9ywF7T7AkbTZ2TSsgpCafIWeK26qhlb4EvFyh_fr5G4,1464
-unique_orchestrator-0.0.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_orchestrator-0.0.2.dist-info/RECORD,,
+unique_orchestrator-0.0.3.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_orchestrator-0.0.3.dist-info/METADATA,sha256=gFcyRvlvxIjS_8Id75q-fTcxqGp829PMGH6GDj-q76s,1513
+unique_orchestrator-0.0.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_orchestrator-0.0.3.dist-info/RECORD,,