unique_toolkit 2026.44.0.dev5__py3-none-any.whl → 2026.44.0.dev6__py3-none-any.whl

unique_toolkit/agentic/history_manager/context_overflow.py

@@ -1,10 +1,4 @@
-"""Keep agent-loop prompts inside the provider's context window.
-
-The local reducer estimates tokens with an OpenAI tokenizer, which can undercount
-other providers by 2x or more. ``ContextOverflowGuard`` verifies near-limit
-payloads with the provider's own counter before the LLM call and recovers when the
-provider still rejects a prompt as too long.
-"""
+"""Keep agent-loop prompts inside the provider's context window, verified by the provider's own token counter."""
 
 from __future__ import annotations
 
@@ -225,7 +219,8 @@ class ContextOverflowGuard:
                 raise
             self._logger.warning(
                 "LLM call still exceeded the context window. Retrying with the "
-                "latest tool call results withheld."
+                "latest tool call results withheld.",
+                exc_info=retry_exc,
             )
             await recompose()
             return await run_iteration()

unique_toolkit/agentic/history_manager/history_manager.py

@@ -237,7 +237,8 @@ class HistoryManager:
 
         The withheld results' chunks are dropped from the reference manager. They
         are the newest chunks, so source numbers of earlier results stay valid.
-        Returns the number of withheld results.
+        Returns the number of newly withheld results; results that already hold
+        *notice* are not counted.
         """
         batch_start: int | None = None
         for index in range(len(self._loop_history) - 1, -1, -1):
@@ -256,12 +257,16 @@ class HistoryManager:
             message = self._loop_history[index]
             if not isinstance(message, LanguageModelToolMessage):
                 continue
+            if message.content == notice:
+                continue
             self._loop_history[index] = LanguageModelToolMessage(
                 content=notice,
                 tool_call_id=message.tool_call_id,
                 name=message.name,
             )
             withheld_tool_call_ids.append(message.tool_call_id)
+        if not withheld_tool_call_ids:
+            return 0
 
         withheld_chunks = {
             id(chunk)

unique_toolkit/agentic/history_manager/loop_token_reducer.py

@@ -140,11 +140,7 @@ class LoopTokenReducer:
         return self._last_token_count
 
     def tighten_effective_token_limit(self, factor: float) -> None:
-        """Shrink the input-token budget for the rest of the run.
-
-        Used after the provider rejected a payload this reducer counted as within
-        budget, i.e. the encoder underestimates the model's real token count.
-        """
+        """Shrink the input-token budget for the rest of the run after the provider counted more tokens than the local encoder."""
         previous_limit = self._effective_token_limit
         self._effective_token_limit = int(previous_limit * factor)
         self._logger.warning(

unique_toolkit/language_model/provider_token_count.py

@@ -1,9 +1,4 @@
-"""Build request bodies for LiteLLM's Anthropic-compatible ``count_tokens`` endpoint.
-
-Agent-loop messages are in OpenAI chat format. The endpoint expects the Anthropic
-Messages format, so system messages, tool calls, tool results and images are
-converted before counting.
-"""
+"""Convert OpenAI chat payloads to request bodies for LiteLLM's Anthropic-compatible ``count_tokens`` endpoint."""
 
 from __future__ import annotations
 

{unique_toolkit-2026.44.0.dev5.dist-info → unique_toolkit-2026.44.0.dev6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.44.0.dev5
+Version: 2026.44.0.dev6
 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.44.0.dev5.dist-info → unique_toolkit-2026.44.0.dev6.dist-info}/RECORD

@@ -113,10 +113,10 @@ unique_toolkit/agentic/evaluation/tests/test_prompt_loaders.py,sha256=zBREdlKf5t
 unique_toolkit/agentic/evaluation/utils.py,sha256=HmyPaDV8wdW-_gOjjW-wDaMKgdrsP5-SHP7OqTmGI_A,264
 unique_toolkit/agentic/feature_flags/__init__.py,sha256=fBR9FaLrc9e8sdYDe1LwP9aNJt5aLLtAq3nIphVWkIk,191
 unique_toolkit/agentic/feature_flags/feature_flags.py,sha256=LlFfcgGV4lRQ_mgOGsbZzjKeTXxjCCOl7agCxvJUhvQ,7133
-unique_toolkit/agentic/history_manager/context_overflow.py,sha256=1GrGcf44rKhyT4dMJI7W2l4JvXyPFqoLUy0zp2n4yYw,9677
+unique_toolkit/agentic/history_manager/context_overflow.py,sha256=N5g8bHUHcRz7gey3cCX-xwK7DJkp_O-3IthMnGsIvcE,9472
 unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha256=bJH5RiLM__fg-i_ycact--9xtV5A321xe_cktT__bzA,26284
-unique_toolkit/agentic/history_manager/history_manager.py,sha256=gDrbnOIJlnmh3KvKRtoYKWVAymEgboW61-6GJVWuZ4c,18885
-unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=Z9Hr3Lb_2t5qqqRcUEYSoV8tp7UUO-uf4yyro3jCyQE,31458
+unique_toolkit/agentic/history_manager/history_manager.py,sha256=JM57S-Cou_vzDR3eM52gYRjgu-lhVUDoYyVGE7pgIYY,19078
+unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=yLxBPF6gkzpBTL5aOzPgQ61bsZd2kOOxeNrGBEdWTEg,31350
 unique_toolkit/agentic/history_manager/settings.py,sha256=Lrx3a8yXeU0MXrWlV6RZSHsgYWF9Zrs8LyR96OpqQ_k,1216
 unique_toolkit/agentic/history_manager/utils.py,sha256=DiBip8xtah3I9-G-_Ul3OjlYKunrXR-5RJtzREHWV_0,6579
 unique_toolkit/agentic/loop_runner/__init__.py,sha256=bhwOIloPn2MZyCw60FkHy330twCCRovg3bgye9OjZRE,1496
@@ -402,7 +402,7 @@ unique_toolkit/language_model/infos.py,sha256=_fbEwGA9AjBm7TF1FTsihtKzi4SybTwVN7
 unique_toolkit/language_model/invocation_stats.py,sha256=CQbpUz394IsVd8eNTB1gPSMt5f0UXmkN1amNsqtAW6U,5857
 unique_toolkit/language_model/model_costs.py,sha256=ShuDAzMew1fUkJ3tzPgU5Zqjp6ivUZaG97BFh93tEn4,6358
 unique_toolkit/language_model/prompt.py,sha256=JSawaLjQg3VR-E2fK8engFyJnNdk21zaO8pPIodzN4Q,3991
-unique_toolkit/language_model/provider_token_count.py,sha256=nXgyL-figP1Zv88zlHXFLvopZAL3hL8lUvF_lusP98g,5499
+unique_toolkit/language_model/provider_token_count.py,sha256=mS597Mvrc7EsAc9iJSmASIEwvETSPLtIEsWxusPrQVc,5337
 unique_toolkit/language_model/reference.py,sha256=T3wr2jP_uRg5wKVIbKqLbxJlcgxfbZQGq3XdjbQ2kSo,6984
 unique_toolkit/language_model/schemas.py,sha256=o8zGkfoYayO1gL85G-ulV3Akp_BwjgEWGGUb89j1OgU,35440
 unique_toolkit/language_model/service.py,sha256=lgpzWhqxLBOeFacCafQi4e7BqAxJbOlr8GnCYcKeTdM,13274
@@ -428,6 +428,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.44.0.dev5.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.44.0.dev5.dist-info/METADATA,sha256=eIr1xETlMh66pzXJR__P64AzIJa1TPhYmfn8bL_v8Xw,7821
-unique_toolkit-2026.44.0.dev5.dist-info/RECORD,,
+unique_toolkit-2026.44.0.dev6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.44.0.dev6.dist-info/METADATA,sha256=MgerP-ba0WqiuVpuL_IbEho3hmwCGjgnQWMp6ESf8TI,7821
+unique_toolkit-2026.44.0.dev6.dist-info/RECORD,,