unique_toolkit 2026.30.0.dev22__py3-none-any.whl → 2026.30.0.dev23__py3-none-any.whl

unique_toolkit/agentic/debug_info_manager/debug_info_manager.py

@@ -48,6 +48,9 @@ class Analytics(TypedDict):
     language_model: AnalyticsLanguageModel
     loop_iteration_count: int
     mcp_tool_names_used: list[AnalyticsToolName]
+    # Total size of artifacts/files created this turn, in MiB. None when Code
+    # Interpreter did not run; 0.0 when it ran but produced nothing.
+    output_size: float | None
     references_count: int
     skills_used: list[AnalyticsSkill]
     subagent_names_used: list[AnalyticsToolName]
@@ -129,8 +132,8 @@ class DebugInfoManager:
         ]
         # Artifacts are recorded at the postprocessor seam (Code Interpreter's
         # DisplayCodeInterpreterFilesPostProcessor) into debug_info["artifacts"].
-        # Absent when no Code Interpreter ran this turn → both fields stay None,
-        # preserving the always-present, null-when-unknown contract.
+        # Absent when no Code Interpreter ran this turn → artifact fields stay
+        # None, preserving the always-present, null-when-unknown contract.
         analytics = Analytics(
             tools_used=analytics_tools,
             tool_call_count=len(analytics_tools),
@@ -152,6 +155,7 @@ class DebugInfoManager:
             total_time_to_answer_ms=total_time_to_answer_ms,
             artifacts_created_count=artifacts["count"] if artifacts else None,
             artifacts_created_filetype=artifacts["filetypes"] if artifacts else None,
+            output_size=artifacts["output_size"] if artifacts else None,
             # None when the user-memory postprocessor is not activated for this
             # turn; True/False when it ran and did/didn't update the profile.
             context_memory_updated=context_memory_updated,

unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py

@@ -46,16 +46,22 @@ from unique_toolkit.short_term_memory.service import ShortTermMemoryService
 logger = logging.getLogger(__name__)
 
 
+_BYTES_PER_MB = 1024 * 1024
+
+
 class ArtifactsDebugInfo(TypedDict):
     """This turn's successfully-created artifacts, for analytics.
 
-    Feeds analytics.artifacts_created_count / artifacts_created_filetype. The
-    postprocessor returns this to the orchestrator, which owns the
-    ``DebugInfoManager`` and records it.
+    Feeds analytics.artifacts_created_count / artifacts_created_filetype /
+    output_size. The postprocessor returns this to the orchestrator, which
+    owns the ``DebugInfoManager`` and records it.
     """
 
     count: int
     filetypes: list[str]
+    # Total size of this turn's successfully-created artifacts, in MiB
+    # (bytes / 1024**2). 0.0 when the interpreter ran but produced nothing.
+    output_size: float
 
 
 class _ChatLoggerAdapter(logging.LoggerAdapter[logging.Logger]):
@@ -380,6 +386,9 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 chat_id=chat_id,
             )
 
+        # Cleared at the start of each run(); holds this-turn download sizes.
+        self._file_size_map: dict[str, int] = {}
+
     def _build_retry(self) -> AsyncRetrying:
         """Build a tenacity retry policy from the current config.
 
@@ -407,6 +416,10 @@ class DisplayCodeInterpreterFilesPostProcessor(
             [cf.filename for cf in container_files],
         )
 
+        # Per-turn byte sizes of successfully uploaded files. Not persisted to
+        # short-term memory — only feeds this turn's analytics.output_size.
+        self._file_size_map = {}
+
         phase_t0 = time.monotonic()
         try:
             self._content_map: dict[str, str | None] = await self._load_previous_files()  # pyright: ignore[reportAttributeAccessIssue]
@@ -501,17 +514,19 @@ class DisplayCodeInterpreterFilesPostProcessor(
     ) -> ArtifactsDebugInfo | None:
         """Compute this turn's successfully-created artifacts.
 
-        Feeds analytics.artifacts_created_count / artifacts_created_filetype.
+        Feeds analytics.artifacts_created_count / artifacts_created_filetype /
+        output_size.
 
         Returns ``None`` when the Code Interpreter did not execute this turn
         (``code_interpreter_calls`` empty). The postprocessor is registered
         whenever the tool is *enabled*, so ``run()`` also fires on turns where the
         model never invoked it — returning ``None`` there keeps analytics.artifacts_*
-        as null ("did not run") rather than 0/[] ("ran, created nothing"). When it
-        did run, count is scoped to this response's ``container_files`` (not
-        ``self._content_map``, which also holds prior-message files from short-term
-        memory), successful uploads only, with a deduped, sorted set of file
-        extensions.
+        / output_size as null ("did not run") rather than 0/[]/0.0 ("ran, created
+        nothing"). When it did run, metrics are scoped to this response's
+        ``container_files`` (not ``self._content_map``, which also holds
+        prior-message files from short-term memory), successful uploads only, with
+        a deduped, sorted set of file extensions and total size in MiB from
+        ``self._file_size_map`` (populated at download time).
         """
         # Interpreter did not run this turn → leave artifacts null, not 0/[].
         if not loop_response.code_interpreter_calls:
@@ -522,9 +537,13 @@ class DisplayCodeInterpreterFilesPostProcessor(
             for cf in loop_response.container_files
             if self._content_map.get(cf.filename) is not None
         }
+        total_bytes = sum(
+            self._file_size_map.get(name, 0) for name in created_filenames
+        )
         return ArtifactsDebugInfo(
             count=len(created_filenames),
             filetypes=sorted({_artifact_filetype(name) for name in created_filenames}),
+            output_size=total_bytes / _BYTES_PER_MB,
         )
 
     @override
@@ -763,6 +782,9 @@ class DisplayCodeInterpreterFilesPostProcessor(
                 upload_ms,
                 pipeline_ms,
             )
+            # Record size only on successful upload so failed downloads don't
+            # inflate analytics.output_size (mirrors content_map success gating).
+            self._file_size_map[container_file.filename] = len(file_bytes)
             return _ContentInfo(filename=container_file.filename, content_id=content.id)
 
     async def _download_file_bytes_with_progress(

{unique_toolkit-2026.30.0.dev22.dist-info → unique_toolkit-2026.30.0.dev23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.30.0.dev22
+Version: 2026.30.0.dev23
 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.30.0.dev22.dist-info → unique_toolkit-2026.30.0.dev23.dist-info}/RECORD

@@ -83,7 +83,7 @@ unique_toolkit/_common/utils/write_configuration.py,sha256=fzvr4C-XBL3OSM3Od9Tbq
 unique_toolkit/_common/validate_required_values.py,sha256=cA8GPwkV1EPsG6ffIkmAn3Dx4LRRMC1EAURmT7PyuCg,551
 unique_toolkit/_common/validators.py,sha256=ElnkMsyEY24TfzfTVHvireyT39EnZgW5N40T0P4b6gE,2638
 unique_toolkit/agentic/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-unique_toolkit/agentic/debug_info_manager/debug_info_manager.py,sha256=MoLHxJSK5yYb_MIzQEx0DICPfzHYZi04ofLbVEvcNso,8666
+unique_toolkit/agentic/debug_info_manager/debug_info_manager.py,sha256=qbQp-e9BO-vGLb6IDbYAM8q8xt2iXMRoC5PX-SjlsRk,8920
 unique_toolkit/agentic/debug_info_manager/test/test_debug_info_manager.py,sha256=bDjrHLqfwtd7ewdIdyTFHabG-15t3iXvs34ywoYXcik,11888
 unique_toolkit/agentic/evaluation/config.py,sha256=u5-iuT-4mfA2_9UZjZ-TN3YutVf35aR4EsF9l61Odnk,1696
 unique_toolkit/agentic/evaluation/context_relevancy/prompts/__init__.py,sha256=IKGvHW2viBylTUjEJNKvIXO4YrYwdJXBHSxA9rKtiRI,355
@@ -218,7 +218,7 @@ unique_toolkit/agentic/tools/openai_builtin/code_interpreter/config.py,sha256=C4
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/__init__.py,sha256=SzEiUqxIw8D4_51SvTt-8oTOYTGirYkmH4J6SgojzME,634
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/artifacts.py,sha256=wAgDrcXZbCdDbNWNpbMigBUXySzbKQXSfg3KsZfAWfs,3115
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/code_display.py,sha256=Q4crOkYG-C_uTSc7rW14cP1H4IWxsKpfuftafDHbjEo,2897
-unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py,sha256=YZ3AWE97RaGREO3bO6r9c5CYvKeVVvOlIGwwuRvyYuU,61797
+unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/generated_files.py,sha256=OIbaRhjorsNe6LhVJYVvLxxMKwphc6ot13vi58axnnM,62861
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/schemas.py,sha256=W7egCmYrPrzic0aXsu9wY5fvQ8y_cl4Tzq_cjVs0mZs,542
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/tool/__init__.py,sha256=r6yXmGTLJzXNq5laHOlre9veu--VbCQVlDRVm2QUGzI,326
 unique_toolkit/agentic/tools/openai_builtin/code_interpreter/tool/config.py,sha256=sbLV-fYt_veBAdonqAHY9sm-eHxpjIZgc1KVbm03pZY,19042
@@ -418,6 +418,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.30.0.dev22.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.30.0.dev22.dist-info/METADATA,sha256=g3QbTNdj0uJ646TdRxg3L2XegUywt3EVUp5vYpGHrCs,7460
-unique_toolkit-2026.30.0.dev22.dist-info/RECORD,,
+unique_toolkit-2026.30.0.dev23.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.30.0.dev23.dist-info/METADATA,sha256=utPGYgyaIlzKn9GzqkzYKL5hcp5de76B_N0AoVQSFLc,7460
+unique_toolkit-2026.30.0.dev23.dist-info/RECORD,,