aru-code 0.24.0__tar.gz → 0.24.1__tar.gz

{aru_code-0.24.0/aru_code.egg-info → aru_code-0.24.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aru-code
-Version: 0.24.0
+Version: 0.24.1
 Summary: A Claude Code clone built with Agno agents
 Author-email: Estevao <estevaofon@gmail.com>
 License-Expression: MIT

aru_code-0.24.1/aru/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.24.1"

{aru_code-0.24.0 → aru_code-0.24.1}/aru/cli.py

@@ -199,7 +199,8 @@ async def run_cli(skip_permissions: bool = False, resume_id: str | None = None):
     # Apply tree_depth from config
     session._tree_max_depth = config.tree_depth
 
-    # Wire file-mutation callback and atexit cleanup
+    # Wire session and file-mutation callback
+    ctx.session = session
     ctx.on_file_mutation = session.invalidate_context_cache
     atexit.register(lambda: cleanup_processes(ctx.tracked_processes))
 
@@ -745,6 +746,7 @@ async def run_oneshot(prompt: str, print_only: bool = False, skip_permissions: b
     if config.default_model:
         session.model_ref = config.default_model
 
+    ctx.session = session
     ctx.model_id = session.model_id
     small_ref = config.model_aliases.get("small") if config else None
     if not small_ref:

{aru_code-0.24.0 → aru_code-0.24.1}/aru/runtime.py

@@ -122,6 +122,9 @@ class RuntimeContext:
     # -- Plugins --
     plugin_manager: Any = None  # aru.plugins.manager.PluginManager (lazy to avoid circular)
 
+    # -- Session --
+    session: Any = None  # aru.session.Session (set by CLI, used for sub-agent cost tracking)
+
     # -- Checkpoints --
     checkpoint_manager: Any = None  # aru.checkpoints.CheckpointManager (lazy)
 

{aru_code-0.24.0 → aru_code-0.24.1}/aru/tools/codebase.py

@@ -1289,6 +1289,22 @@ Do not create documentation files unless explicitly asked.
             from aru.permissions import permission_scope
             with permission_scope(agent_perm):
                 result = await sub.arun(task, stream=False)
+            # Track sub-agent token usage in the session so cost is not underestimated.
+            # We accumulate totals directly instead of calling track_tokens() because
+            # track_tokens() also reads cache_patch globals (last_call_metrics) which
+            # may belong to a different concurrent sub-agent — we must not touch last_*.
+            if result and hasattr(result, "metrics") and result.metrics:
+                try:
+                    session = get_ctx().session
+                    if session is not None:
+                        m = result.metrics
+                        session.total_input_tokens += getattr(m, "input_tokens", 0) or 0
+                        session.total_output_tokens += getattr(m, "output_tokens", 0) or 0
+                        session.total_cache_read_tokens += getattr(m, "cache_read_tokens", 0) or 0
+                        session.total_cache_write_tokens += getattr(m, "cache_write_tokens", 0) or 0
+                        session.api_calls += 1
+                except (LookupError, AttributeError):
+                    pass
             if result and result.content:
                 return _truncate_output(f"[{label}] {result.content}")
             return f"[{label}] Task completed but no output was returned."

{aru_code-0.24.0 → aru_code-0.24.1/aru_code.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aru-code
-Version: 0.24.0
+Version: 0.24.1
 Summary: A Claude Code clone built with Agno agents
 Author-email: Estevao <estevaofon@gmail.com>
 License-Expression: MIT

{aru_code-0.24.0 → aru_code-0.24.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aru-code"
-version = "0.24.0"
+version = "0.24.1"
 description = "A Claude Code clone built with Agno agents"
 readme = "README.md"
 license = "MIT"

aru_code-0.24.0/aru/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.24.0"