unique-toolkit 2026.22.0.dev2__tar.gz → 2026.22.0.dev4__tar.gz

{unique_toolkit-2026.22.0.dev2 → unique_toolkit-2026.22.0.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.22.0.dev2
+Version: 2026.22.0.dev4
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>

{unique_toolkit-2026.22.0.dev2 → unique_toolkit-2026.22.0.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.22.0.dev2"
+version = "2026.22.0.dev4"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-2026.22.0.dev2 → unique_toolkit-2026.22.0.dev4}/unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -13,6 +13,7 @@ from unique_toolkit.agentic.history_manager.history_construction_with_contents i
     get_full_history_with_contents_and_tool_calls_async,
     get_full_history_with_contents_async,
 )
+from unique_toolkit.agentic.history_manager.settings import env_settings
 from unique_toolkit.agentic.history_manager.utils import (
     get_selected_uploaded_content_ids,
     serialize_tool_content_json,
@@ -33,10 +34,8 @@ from unique_toolkit.language_model.schemas import (
 )
 from unique_toolkit.services.factory import UniqueServiceFactory
 
-MAX_INPUT_TOKENS_SAFETY_PERCENTAGE = (
-    0.1  # 10% safety margin for input tokens we need 10% less does not work.
-)
-SAFETY_MARGIN_FOR_AGGRESSIVE_REDUCTION = 0.9
+MAX_INPUT_TOKENS_SAFETY_PERCENTAGE = env_settings.input_correction_reduction_factor
+SAFETY_MARGIN_FOR_AGGRESSIVE_REDUCTION = env_settings.input_aggressive_reduction_factor
 
 
 class SourceReductionResult(BaseModel):

unique_toolkit-2026.22.0.dev4/unique_toolkit/agentic/history_manager/settings.py

@@ -0,0 +1,53 @@
+import os
+import sys
+from pathlib import Path
+
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class Base(BaseSettings):
+    input_correction_reduction_factor: float = Field(
+        default=0.1,
+        ge=0.0,
+        lt=1.0,
+        description="The factor by which the input tokens will be reduced.",
+    )
+    input_aggressive_reduction_factor: float = Field(
+        default=0.9,
+        ge=0.0,
+        lt=1.0,
+        description="The safety margin for the input correction.",
+    )
+
+
+def get_model_config(test: bool = False) -> SettingsConfigDict:
+    if test:
+        env_file = Path(os.getcwd()) / "tests/test.env"
+    else:
+        env_file = Path(os.getcwd()) / ".env"
+
+    return SettingsConfigDict(
+        extra="ignore",
+        env_prefix="TOOLKIT_HISTORY_MANAGER_",
+        case_sensitive=False,
+        env_file=env_file,
+        env_file_encoding="utf-8",
+    )
+
+
+class Settings(Base):
+    model_config = get_model_config()
+
+
+class TestSettings(Base):
+    model_config = get_model_config(test=True)
+
+
+def get_settings() -> Base:
+    if "pytest" in sys.modules:
+        return TestSettings()
+    return Settings()
+
+
+env_settings: Base = get_settings()