unique-toolkit 2026.38.0.dev9__tar.gz → 2026.38.0.dev11__tar.gz

{unique_toolkit-2026.38.0.dev9 → unique_toolkit-2026.38.0.dev11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.38.0.dev9
+Version: 2026.38.0.dev11
 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>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.38.0.dev1,<2026.38.0rc0
+Requires-Dist: unique-sdk>=2026.38.0.dev2,<2026.38.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.38.0.dev9 → unique_toolkit-2026.38.0.dev11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.38.0.dev9"
+version = "2026.38.0.dev11"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -28,7 +28,7 @@ dependencies = [
     "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=2026.38.0.dev1,<2026.38.0rc0",
+    "unique-sdk>=2026.38.0.dev2,<2026.38.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.38.0.dev9 → unique_toolkit-2026.38.0.dev11}/unique_toolkit/agentic_table/schemas.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from enum import StrEnum
 from typing import Annotated, Any, Generic, Literal, TypeVar, override
 
@@ -501,6 +502,15 @@ class MagicTableSheet(BaseModel):
     created_by: str
     company_id: str
     created_at: str
+    user_aborted_at: datetime | None = Field(
+        default=None,
+        description=(
+            "When the user stopped an in-progress refresh. Set by "
+            "stopMagicTableRefresh and returned on GET sheet/library so the "
+            "agent can poll and exit. None when the run was not aborted or "
+            "when an older backend omits the field."
+        ),
+    )
     magic_table_cells: list[MagicTableCell] = Field(
         default_factory=list, description="The cells in the sheet"
     )

{unique_toolkit-2026.38.0.dev9 → unique_toolkit-2026.38.0.dev11}/unique_toolkit/language_model/model_costs.py

@@ -6,7 +6,7 @@ import time
 from pathlib import Path
 
 import yaml
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from unique_toolkit.language_model.schemas import LanguageModelTokenUsage
 
@@ -37,6 +37,50 @@ class ModelCost(BaseModel):
     cached_input: float | None = Field(default=None, alias="cachedInput")
     cache_write: float | None = Field(default=None, alias="cacheWrite")
     cache_write_1h: float | None = Field(default=None, alias="cacheWrite1h")
+    # Long-context tier: above the threshold the whole request (not just the
+    # overflow) is billed at these multipliers. All-or-none, see validator below.
+    long_context_threshold: int | None = Field(
+        default=None, alias="longContextThreshold"
+    )
+    long_context_input_multiplier: float | None = Field(
+        default=None, alias="longContextInputMultiplier"
+    )
+    long_context_completion_multiplier: float | None = Field(
+        default=None, alias="longContextCompletionMultiplier"
+    )
+
+    @model_validator(mode="after")
+    def _validate_long_context_all_or_none(self) -> "ModelCost":
+        fields = (
+            self.long_context_threshold,
+            self.long_context_input_multiplier,
+            self.long_context_completion_multiplier,
+        )
+        defined = [field for field in fields if field is not None]
+        if defined and len(defined) != len(fields):
+            raise ValueError(
+                "longContextThreshold, longContextInputMultiplier and "
+                "longContextCompletionMultiplier must be set together"
+            )
+        return self
+
+    def multipliers_for(self, prompt_tokens: int) -> tuple[float, float]:
+        """Return ``(input_multiplier, completion_multiplier)`` for a request.
+
+        ``prompt_tokens`` is the provider-reported input size of the single
+        request (for OpenAI this already includes cached tokens).
+        """
+        if (
+            self.long_context_threshold is None
+            or self.long_context_input_multiplier is None
+            or self.long_context_completion_multiplier is None
+            or prompt_tokens <= self.long_context_threshold
+        ):
+            return 1.0, 1.0
+        return (
+            self.long_context_input_multiplier,
+            self.long_context_completion_multiplier,
+        )
 
 
 class ModelCostCatalog(BaseModel):
@@ -119,7 +163,10 @@ def calculate_invocation_cost_usd(
     if model_cost is None:
         return None
 
+    input_multiplier, completion_multiplier = model_cost.multipliers_for(
+        token_usage.prompt_tokens
+    )
     return (
-        token_usage.prompt_tokens * model_cost.input
-        + token_usage.completion_tokens * model_cost.completion
+        token_usage.prompt_tokens * model_cost.input * input_multiplier
+        + token_usage.completion_tokens * model_cost.completion * completion_multiplier
     ) / _TOKENS_PER_MILLION