unique_toolkit 0.8.56__tar.gz → 0.9.0__tar.gz

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/CHANGELOG.md

@@ -5,6 +5,20 @@ 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.9.0] - 2025-09-14
+- Moved agentic code into the `agentic` folder. This breaks imports of
+  - `debug_info_amanager`
+  - `evals`
+  - `history_manager`
+  - `post_processor`
+  - `reference-manager`
+  - `short_term_memory_manager`
+  - `thinking_manager`
+  - `tools`
+
+## [0.8.57] - 2025-09-14
+- Added more utils to commons
+
 ## [0.8.56] - 2025-09-12
 - Fixed token counter in utils
 

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.8.56
+Version: 0.9.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -119,6 +119,20 @@ 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.9.0] - 2025-09-14
+- Moved agentic code into the `agentic` folder. This breaks imports of
+  - `debug_info_amanager`
+  - `evals`
+  - `history_manager`
+  - `post_processor`
+  - `reference-manager`
+  - `short_term_memory_manager`
+  - `thinking_manager`
+  - `tools`
+
+## [0.8.57] - 2025-09-14
+- Added more utils to commons
+
 ## [0.8.56] - 2025-09-12
 - Fixed token counter in utils
 

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "0.8.56"
+version = "0.9.0"
 description = ""
 authors = [
     "Cedric Klinkert <cedric.klinkert@unique.ch>",
@@ -8,7 +8,7 @@ authors = [
     "Sadique Sheik <sadique@unique.ch>",
     "Fabian Schläpfer <fabian@unique.ch>",
     "Pascal Hauri <pascal@unique.ch>",
-    "Ejiroghene Oki-Peter<oki-peter.ejiroghene@unique.ai>"
+    "Ejiroghene Oki-Peter<oki-peter.ejiroghene@unique.ai>",
 ]
 readme = ["README.md", "CHANGELOG.md"]
 license = "Proprietary"

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/unique_toolkit/_common/chunk_relevancy_sorter/config.py

@@ -4,8 +4,10 @@ from pydantic import BaseModel, Field
 
 from unique_toolkit._common.default_language_model import DEFAULT_GPT_35_TURBO
 from unique_toolkit._common.validators import LMI, get_LMI_default_field
-from unique_toolkit.evals.context_relevancy.schema import StructuredOutputConfig
-from unique_toolkit.tools.config import get_configuration_dict
+from unique_toolkit.agentic.evaluation.context_relevancy.schema import (
+    StructuredOutputConfig,
+)
+from unique_toolkit.agentic.tools.config import get_configuration_dict
 
 
 class ChunkRelevancySortConfig(BaseModel):

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/unique_toolkit/_common/chunk_relevancy_sorter/schemas.py

@@ -2,8 +2,8 @@ from typing import Optional
 
 from pydantic import BaseModel
 
+from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricResult
 from unique_toolkit.content.schemas import ContentChunk
-from unique_toolkit.evals.schemas import EvaluationMetricResult
 
 
 class ChunkRelevancy(BaseModel):

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/unique_toolkit/_common/chunk_relevancy_sorter/service.py

@@ -16,21 +16,23 @@ from unique_toolkit._common.chunk_relevancy_sorter.schemas import (
     ChunkRelevancySorterResult,
 )
 from unique_toolkit._common.validate_required_values import validate_required_values
-from unique_toolkit.app.performance.async_tasks import run_async_tasks_parallel
-from unique_toolkit.app.schemas import BaseEvent, ChatEvent
-from unique_toolkit.content.schemas import ContentChunk
-from unique_toolkit.evals.config import EvaluationMetricConfig
-from unique_toolkit.evals.context_relevancy.schema import (
+from unique_toolkit.agentic.evaluation.config import EvaluationMetricConfig
+from unique_toolkit.agentic.evaluation.context_relevancy.schema import (
     EvaluationSchemaStructuredOutput,
     StructuredOutputConfig,
 )
-from unique_toolkit.evals.context_relevancy.service import ContextRelevancyEvaluator
-from unique_toolkit.evals.exception import EvaluatorException
-from unique_toolkit.evals.schemas import (
+from unique_toolkit.agentic.evaluation.context_relevancy.service import (
+    ContextRelevancyEvaluator,
+)
+from unique_toolkit.agentic.evaluation.exception import EvaluatorException
+from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationMetricInput,
     EvaluationMetricName,
     EvaluationMetricResult,
 )
+from unique_toolkit.app.performance.async_tasks import run_async_tasks_parallel
+from unique_toolkit.app.schemas import BaseEvent, ChatEvent
+from unique_toolkit.content.schemas import ContentChunk
 from unique_toolkit.language_model.infos import LanguageModelInfo
 
 

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/unique_toolkit/_common/chunk_relevancy_sorter/tests/test_service.py

@@ -17,10 +17,15 @@ from unique_toolkit._common.default_language_model import (
     DEFAULT_GPT_35_TURBO,
     DEFAULT_GPT_4o,
 )
+from unique_toolkit.agentic.evaluation.context_relevancy.schema import (
+    StructuredOutputConfig,
+)
+from unique_toolkit.agentic.evaluation.schemas import (
+    EvaluationMetricName,
+    EvaluationMetricResult,
+)
 from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.content.schemas import ContentChunk
-from unique_toolkit.evals.context_relevancy.schema import StructuredOutputConfig
-from unique_toolkit.evals.schemas import EvaluationMetricName, EvaluationMetricResult
 from unique_toolkit.language_model.infos import LanguageModelInfo
 
 

{unique_toolkit-0.8.56 → unique_toolkit-0.9.0}/unique_toolkit/_common/feature_flags/schema.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel, Field
 
-from unique_toolkit.tools.config import get_configuration_dict
+from unique_toolkit.agentic.tools.config import get_configuration_dict
 
 
 class FeatureExtendedSourceSerialization(BaseModel):

unique_toolkit-0.9.0/unique_toolkit/_common/utils/write_configuration.py

@@ -0,0 +1,51 @@
+import json
+from pathlib import Path
+
+from humps import kebabize, pascalize
+from pydantic import BaseModel
+
+
+def write_module_configuration_schema(
+    service_folderpath: Path,
+    write_folderpath: Path,
+    config: BaseModel,
+    sub_name: str = "",
+):
+    filename_prefix = pascalize(service_folderpath.name)
+
+    filepath = (
+        write_folderpath
+        / f"{filename_prefix}{f'-{sub_name}' if sub_name else ''}Schema.json"
+    )
+
+    with open(filepath, "w") as f:
+        json.dump(config.model_json_schema(by_alias=True), f, indent=4)
+
+
+def write_service_configuration(
+    service_folderpath: Path,
+    write_folderpath: Path,
+    config: BaseModel,
+    sub_name: str = "",
+):
+    filename_prefix = kebabize(service_folderpath.name)
+
+    filepath = (
+        write_folderpath
+        / f"{filename_prefix}{f'-{sub_name}' if sub_name else ''}-configuration-schema.json"
+    )
+
+    with open(filepath, "w") as f:
+        json.dump(config.model_json_schema(by_alias=True), f, indent=4)
+    filepath = (
+        write_folderpath
+        / f"{filename_prefix}{f'-{sub_name}' if sub_name else ''}-default-configuration.json"
+    )
+
+    # We exclude language_model_info as it is infered from language_model_name
+    with open(filepath, "w") as f:
+        f.write(
+            config.model_dump_json(
+                by_alias=True, indent=4, exclude=set(["language_model_info"])
+            )
+        )

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/debug_info_manager/debug_info_manager.py

@@ -1,4 +1,4 @@
-from unique_toolkit.tools.schemas import ToolCallResponse
+from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 
 
 class DebugInfoManager:

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/context_relevancy/schema.py

@@ -2,7 +2,7 @@ from pydantic import BaseModel, Field, create_model
 from pydantic.json_schema import SkipJsonSchema
 
 from unique_toolkit._common.utils.structured_output.schema import StructuredOutputModel
-from unique_toolkit.tools.config import get_configuration_dict
+from unique_toolkit.agentic.tools.config import get_configuration_dict
 
 
 class StructuredOutputConfig(BaseModel):

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/context_relevancy/service.py

@@ -8,22 +8,22 @@ from unique_toolkit._common.default_language_model import DEFAULT_GPT_35_TURBO
 from unique_toolkit._common.validate_required_values import (
     validate_required_values,
 )
-from unique_toolkit.app.schemas import BaseEvent, ChatEvent
-from unique_toolkit.evals.config import EvaluationMetricConfig
-from unique_toolkit.evals.context_relevancy.schema import (
+from unique_toolkit.agentic.evaluation.config import EvaluationMetricConfig
+from unique_toolkit.agentic.evaluation.context_relevancy.schema import (
     EvaluationSchemaStructuredOutput,
 )
-from unique_toolkit.evals.exception import EvaluatorException
-from unique_toolkit.evals.output_parser import (
+from unique_toolkit.agentic.evaluation.exception import EvaluatorException
+from unique_toolkit.agentic.evaluation.output_parser import (
     parse_eval_metric_result,
     parse_eval_metric_result_structured_output,
 )
-from unique_toolkit.evals.schemas import (
+from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationMetricInput,
     EvaluationMetricInputFieldName,
     EvaluationMetricName,
     EvaluationMetricResult,
 )
+from unique_toolkit.app.schemas import BaseEvent, ChatEvent
 from unique_toolkit.language_model.infos import (
     LanguageModelInfo,
     ModelCapabilities,

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/evaluation_manager.py

@@ -2,20 +2,23 @@ import asyncio
 from abc import ABC
 from logging import Logger
 
+from unique_toolkit.agentic.evaluation.schemas import (
+    EvaluationAssessmentMessage,
+    EvaluationMetricName,
+    EvaluationMetricResult,
+)
+from unique_toolkit.agentic.tools.utils.execution.execution import (
+    Result,
+    SafeTaskExecutor,
+)
 from unique_toolkit.chat.schemas import (
     ChatMessageAssessmentStatus,
     ChatMessageAssessmentType,
 )
 from unique_toolkit.chat.service import ChatService
-from unique_toolkit.evals.schemas import (
-    EvaluationAssessmentMessage,
-    EvaluationMetricName,
-    EvaluationMetricResult,
-)
 from unique_toolkit.language_model.schemas import (
     LanguageModelStreamResponse,
 )
-from unique_toolkit.tools.utils.execution.execution import Result, SafeTaskExecutor
 
 
 class Evaluation(ABC):

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/hallucination/constants.py

@@ -4,14 +4,14 @@ from pydantic import Field
 
 from unique_toolkit._common.default_language_model import DEFAULT_GPT_4o
 from unique_toolkit._common.validators import LMI
-from unique_toolkit.evals.config import EvaluationMetricConfig
-from unique_toolkit.evals.hallucination.prompts import (
+from unique_toolkit.agentic.evaluation.config import EvaluationMetricConfig
+from unique_toolkit.agentic.evaluation.hallucination.prompts import (
     HALLUCINATION_METRIC_SYSTEM_MSG,
     HALLUCINATION_METRIC_SYSTEM_MSG_DEFAULT,
     HALLUCINATION_METRIC_USER_MSG,
     HALLUCINATION_METRIC_USER_MSG_DEFAULT,
 )
-from unique_toolkit.evals.schemas import (
+from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationMetricInputFieldName,
     EvaluationMetricName,
 )

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/hallucination/hallucination_evaluation.py

@@ -1,26 +1,26 @@
-from unique_toolkit.app.schemas import ChatEvent
-from unique_toolkit.chat.schemas import (
-    ChatMessageAssessmentLabel,
-    ChatMessageAssessmentStatus,
-    ChatMessageAssessmentType,
-)
-from unique_toolkit.evals.evaluation_manager import Evaluation
-from unique_toolkit.evals.hallucination.constants import (
+from unique_toolkit.agentic.evaluation.evaluation_manager import Evaluation
+from unique_toolkit.agentic.evaluation.hallucination.constants import (
     HallucinationConfig,
 )
-from unique_toolkit.evals.hallucination.utils import check_hallucination
-from unique_toolkit.evals.schemas import (
+from unique_toolkit.agentic.evaluation.hallucination.utils import check_hallucination
+from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationAssessmentMessage,
     EvaluationMetricInput,
     EvaluationMetricName,
     EvaluationMetricResult,
 )
+from unique_toolkit.agentic.reference_manager.reference_manager import (
+    ReferenceManager,
+)
+from unique_toolkit.app.schemas import ChatEvent
+from unique_toolkit.chat.schemas import (
+    ChatMessageAssessmentLabel,
+    ChatMessageAssessmentStatus,
+    ChatMessageAssessmentType,
+)
 from unique_toolkit.language_model.schemas import (
     LanguageModelStreamResponse,
 )
-from unique_toolkit.reference_manager.reference_manager import (
-    ReferenceManager,
-)
 
 
 class HallucinationEvaluation(Evaluation):

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/hallucination/service.py

@@ -1,8 +1,11 @@
 import logging
 
+from unique_toolkit.agentic.evaluation.config import EvaluationMetricConfig
+from unique_toolkit.agentic.evaluation.schemas import (
+    EvaluationMetricInput,
+    EvaluationMetricResult,
+)
 from unique_toolkit.app.schemas import ChatEvent
-from unique_toolkit.evals.config import EvaluationMetricConfig
-from unique_toolkit.evals.schemas import EvaluationMetricInput, EvaluationMetricResult
 
 from .constants import hallucination_metric_default_config
 from .utils import check_hallucination

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/hallucination/utils.py

@@ -1,15 +1,15 @@
 import logging
 from string import Template
 
-from unique_toolkit.content.schemas import ContentChunk
-from unique_toolkit.evals.config import EvaluationMetricConfig
-from unique_toolkit.evals.exception import EvaluatorException
-from unique_toolkit.evals.output_parser import parse_eval_metric_result
-from unique_toolkit.evals.schemas import (
+from unique_toolkit.agentic.evaluation.config import EvaluationMetricConfig
+from unique_toolkit.agentic.evaluation.exception import EvaluatorException
+from unique_toolkit.agentic.evaluation.output_parser import parse_eval_metric_result
+from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationMetricInput,
     EvaluationMetricName,
     EvaluationMetricResult,
 )
+from unique_toolkit.content.schemas import ContentChunk
 from unique_toolkit.language_model.schemas import (
     LanguageModelMessages,
     LanguageModelStreamResponse,

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/output_parser.py

@@ -1,8 +1,8 @@
-from unique_toolkit.evals.context_relevancy.schema import (
+from unique_toolkit.agentic.evaluation.context_relevancy.schema import (
     EvaluationSchemaStructuredOutput,
 )
-from unique_toolkit.evals.exception import EvaluatorException
-from unique_toolkit.evals.schemas import (
+from unique_toolkit.agentic.evaluation.exception import EvaluatorException
+from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationMetricName,
     EvaluationMetricResult,
 )

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/schemas.py

@@ -3,13 +3,13 @@ from typing import Optional
 
 from pydantic import BaseModel, ConfigDict, Field
 
+from unique_toolkit.agentic.evaluation.exception import EvaluatorException
 from unique_toolkit.chat import ChatMessage
 from unique_toolkit.chat.schemas import (
     ChatMessageAssessmentLabel,
     ChatMessageAssessmentStatus,
     ChatMessageAssessmentType,
 )
-from unique_toolkit.evals.exception import EvaluatorException
 
 
 class EvaluationMetricName(StrEnum):

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/tests/test_context_relevancy_service.py

@@ -2,24 +2,24 @@ from unittest.mock import MagicMock, patch
 
 import pytest
 
-from unique_toolkit.app.schemas import ChatEvent
-from unique_toolkit.chat.service import LanguageModelName
-from unique_toolkit.evals.config import EvaluationMetricConfig
-from unique_toolkit.evals.context_relevancy.prompts import (
+from unique_toolkit.agentic.evaluation.config import EvaluationMetricConfig
+from unique_toolkit.agentic.evaluation.context_relevancy.prompts import (
     CONTEXT_RELEVANCY_METRIC_SYSTEM_MSG,
 )
-from unique_toolkit.evals.context_relevancy.schema import (
+from unique_toolkit.agentic.evaluation.context_relevancy.schema import (
     EvaluationSchemaStructuredOutput,
 )
-from unique_toolkit.evals.context_relevancy.service import (
+from unique_toolkit.agentic.evaluation.context_relevancy.service import (
     ContextRelevancyEvaluator,
 )
-from unique_toolkit.evals.exception import EvaluatorException
-from unique_toolkit.evals.schemas import (
+from unique_toolkit.agentic.evaluation.exception import EvaluatorException
+from unique_toolkit.agentic.evaluation.schemas import (
     EvaluationMetricInput,
     EvaluationMetricName,
     EvaluationMetricResult,
 )
+from unique_toolkit.app.schemas import ChatEvent
+from unique_toolkit.chat.service import LanguageModelName
 from unique_toolkit.language_model.infos import (
     LanguageModelInfo,
 )

{unique_toolkit-0.8.56/unique_toolkit/evals → unique_toolkit-0.9.0/unique_toolkit/agentic/evaluation}/tests/test_output_parser.py

@@ -1,15 +1,18 @@
 import pytest
 
-from unique_toolkit.evals.context_relevancy.schema import (
+from unique_toolkit.agentic.evaluation.context_relevancy.schema import (
     EvaluationSchemaStructuredOutput,
     Fact,
 )
-from unique_toolkit.evals.exception import EvaluatorException
-from unique_toolkit.evals.output_parser import (
+from unique_toolkit.agentic.evaluation.exception import EvaluatorException
+from unique_toolkit.agentic.evaluation.output_parser import (
     parse_eval_metric_result,
     parse_eval_metric_result_structured_output,
 )
-from unique_toolkit.evals.schemas import EvaluationMetricName, EvaluationMetricResult
+from unique_toolkit.agentic.evaluation.schemas import (
+    EvaluationMetricName,
+    EvaluationMetricResult,
+)
 
 
 def test_parse_eval_metric_result_success():

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/history_manager/history_manager.py

@@ -4,9 +4,12 @@ from typing import Annotated, Awaitable, Callable
 from pydantic import BaseModel, Field
 
 from unique_toolkit._common.validators import LMI
+from unique_toolkit.agentic.history_manager.loop_token_reducer import LoopTokenReducer
+from unique_toolkit.agentic.history_manager.utils import transform_chunks_to_string
+from unique_toolkit.agentic.reference_manager.reference_manager import ReferenceManager
+from unique_toolkit.agentic.tools.config import get_configuration_dict
+from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 from unique_toolkit.app.schemas import ChatEvent
-from unique_toolkit.history_manager.loop_token_reducer import LoopTokenReducer
-from unique_toolkit.history_manager.utils import transform_chunks_to_string
 from unique_toolkit.language_model.infos import LanguageModelInfo, LanguageModelName
 from unique_toolkit.language_model.schemas import (
     LanguageModelAssistantMessage,
@@ -15,9 +18,6 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelMessages,
     LanguageModelToolMessage,
 )
-from unique_toolkit.reference_manager.reference_manager import ReferenceManager
-from unique_toolkit.tools.config import get_configuration_dict
-from unique_toolkit.tools.schemas import ToolCallResponse
 
 DeactivatedNone = Annotated[
     None,

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/history_manager/loop_token_reducer.py

@@ -9,14 +9,15 @@ from unique_toolkit._common.token.token_counting import (
     num_token_for_language_model_messages,
 )
 from unique_toolkit._common.validators import LMI
+from unique_toolkit.agentic.history_manager.history_construction_with_contents import (
+    FileContentSerialization,
+    get_full_history_with_contents,
+)
+from unique_toolkit.agentic.reference_manager.reference_manager import ReferenceManager
 from unique_toolkit.app.schemas import ChatEvent
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content.schemas import ContentChunk
 from unique_toolkit.content.service import ContentService
-from unique_toolkit.history_manager.history_construction_with_contents import (
-    FileContentSerialization,
-    get_full_history_with_contents,
-)
 from unique_toolkit.language_model.schemas import (
     LanguageModelAssistantMessage,
     LanguageModelMessage,
@@ -26,7 +27,6 @@ from unique_toolkit.language_model.schemas import (
     LanguageModelToolMessage,
     LanguageModelUserMessage,
 )
-from unique_toolkit.reference_manager.reference_manager import ReferenceManager
 
 MAX_INPUT_TOKENS_SAFETY_PERCENTAGE = (
     0.1  # 10% safety margin for input tokens we need 10% less does not work.

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/history_manager/utils.py

@@ -2,16 +2,16 @@ import json
 import logging
 from copy import deepcopy
 
+from unique_toolkit.agentic.tools.schemas import Source
+from unique_toolkit.agentic.tools.utils.source_handling.schema import (
+    SourceFormatConfig,
+)
 from unique_toolkit.content.schemas import ContentChunk, ContentMetadata
 from unique_toolkit.language_model.schemas import (
     LanguageModelAssistantMessage,
     LanguageModelMessage,
     LanguageModelToolMessage,
 )
-from unique_toolkit.tools.schemas import Source
-from unique_toolkit.tools.utils.source_handling.schema import (
-    SourceFormatConfig,
-)
 
 logger = logging.getLogger(__name__)
 

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/postprocessor/postprocessor_manager.py

@@ -2,11 +2,11 @@ import asyncio
 from abc import ABC
 from logging import Logger
 
+from unique_toolkit.agentic.tools.utils.execution.execution import SafeTaskExecutor
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.language_model.schemas import (
     LanguageModelStreamResponse,
 )
-from unique_toolkit.tools.utils.execution.execution import SafeTaskExecutor
 
 
 class Postprocessor(ABC):

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/reference_manager/reference_manager.py

@@ -1,5 +1,5 @@
+from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 from unique_toolkit.content.schemas import ContentChunk, ContentReference
-from unique_toolkit.tools.schemas import ToolCallResponse
 
 
 class tool_chunks:

{unique_toolkit-0.8.56/unique_toolkit/short_term_memory → unique_toolkit-0.9.0/unique_toolkit/agentic/short_term_memory_manager}/persistent_short_term_memory_manager.py

@@ -5,9 +5,9 @@ from typing import Generic, Type, TypeVar
 
 from pydantic import BaseModel
 
+from unique_toolkit.agentic.tools.utils.execution.execution import SafeTaskExecutor
 from unique_toolkit.short_term_memory.schemas import ShortTermMemory
 from unique_toolkit.short_term_memory.service import ShortTermMemoryService
-from unique_toolkit.tools.utils.execution.execution import SafeTaskExecutor
 
 TSchema = TypeVar("TSchema", bound=BaseModel)
 

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/thinking_manager/thinking_manager.py

@@ -2,13 +2,13 @@ from logging import Logger
 
 from pydantic import BaseModel, Field
 
+from unique_toolkit.agentic.tools.tool_progress_reporter import (
+    ToolProgressReporter,
+)
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.language_model.schemas import (
     LanguageModelStreamResponse,
 )
-from unique_toolkit.tools.tool_progress_reporter import (
-    ToolProgressReporter,
-)
 
 
 class ThinkingManagerConfig(BaseModel):

unique_toolkit-0.9.0/unique_toolkit/agentic/tools/a2a/__init__.py

@@ -0,0 +1,4 @@
+from unique_toolkit.agentic.tools.a2a.config import SubAgentToolConfig
+from unique_toolkit.agentic.tools.a2a.service import SubAgentTool
+
+__all__ = ["SubAgentToolConfig", "SubAgentTool"]

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/tools/a2a/config.py

@@ -1,5 +1,5 @@
-from unique_toolkit.tools.config import get_configuration_dict
-from unique_toolkit.tools.schemas import BaseToolConfig
+from unique_toolkit.agentic.tools.config import get_configuration_dict
+from unique_toolkit.agentic.tools.schemas import BaseToolConfig
 
 DEFAULT_PARAM_DESCRIPTION_SUB_AGENT_USER_MESSAGE = """
 This is the message that will be sent to the sub-agent.

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/tools/a2a/manager.py

@@ -1,11 +1,11 @@
 from logging import Logger
 
+from unique_toolkit.agentic.tools.a2a.config import SubAgentToolConfig
+from unique_toolkit.agentic.tools.a2a.service import SubAgentTool, ToolProgressReporter
+from unique_toolkit.agentic.tools.config import ToolBuildConfig
+from unique_toolkit.agentic.tools.schemas import BaseToolConfig
+from unique_toolkit.agentic.tools.tool import Tool
 from unique_toolkit.app.schemas import ChatEvent
-from unique_toolkit.tools.a2a.config import SubAgentToolConfig
-from unique_toolkit.tools.a2a.service import SubAgentTool, ToolProgressReporter
-from unique_toolkit.tools.config import ToolBuildConfig
-from unique_toolkit.tools.schemas import BaseToolConfig
-from unique_toolkit.tools.tool import Tool
 
 
 class A2AManager:

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/tools/a2a/memory.py

@@ -1,8 +1,8 @@
 from unique_toolkit import ShortTermMemoryService
-from unique_toolkit.short_term_memory.persistent_short_term_memory_manager import (
+from unique_toolkit.agentic.short_term_memory_manager.persistent_short_term_memory_manager import (
     PersistentShortMemoryManager,
 )
-from unique_toolkit.tools.a2a.schema import SubAgentShortTermMemorySchema
+from unique_toolkit.agentic.tools.a2a.schema import SubAgentShortTermMemorySchema
 
 
 def _get_short_term_memory_name(assistant_id: str) -> str:

{unique_toolkit-0.8.56/unique_toolkit → unique_toolkit-0.9.0/unique_toolkit/agentic}/tools/a2a/schema.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from unique_toolkit.tools.schemas import ToolCallResponse
+from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 
 
 class SubAgentToolInput(BaseModel):