aury-agent 0.0.14__py3-none-any.whl → 0.0.15__py3-none-any.whl

aury/agents/core/types/tool.py

@@ -46,6 +46,14 @@ class ToolContext:
     # Caller's middleware chain (for sub-agent inheritance)
     middleware: Any | None = None  # MiddlewareChain
     
+    # Step number in the current invocation (for usage tracking)
+    step: int | None = None
+    
+    @property
+    def tool_call_id(self) -> str:
+        """Alias for call_id for backward compatibility."""
+        return self.call_id
+    
     async def emit(self, block: Any) -> None:
         """Emit a block event.
         

aury/agents/react/step.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import json
+import time
 from typing import TYPE_CHECKING, Any
 
 from ..core.base import ToolInjectionMode
@@ -263,12 +264,14 @@ async def execute_step(agent: "ReactAgent") -> str | None:
         )
     
     # Call LLM
+    llm_start_time = time.time()
     await agent.bus.publish(
         Events.LLM_START,
         {
             "provider": agent.llm.provider,
             "model": agent.llm.model,
             "step": agent._current_step,
+            "invocation_id": agent._current_invocation.id if agent._current_invocation else None,
             "enable_thinking": enable_thinking,
         },
     )
@@ -556,6 +559,9 @@ async def execute_step(agent: "ReactAgent") -> str | None:
 
         elif event.type == "usage":
             if event.usage:
+                # Calculate duration
+                duration_ms = int((time.time() - llm_start_time) * 1000)
+                
                 # Store usage for middleware
                 agent._last_usage = {
                     "provider": agent.llm.provider,
@@ -565,6 +571,12 @@ async def execute_step(agent: "ReactAgent") -> str | None:
                     "cache_read_tokens": event.usage.cache_read_tokens,
                     "cache_write_tokens": event.usage.cache_write_tokens,
                     "reasoning_tokens": event.usage.reasoning_tokens,
+                    # Additional context for billing/tracking
+                    "invocation_id": agent._current_invocation.id if agent._current_invocation else None,
+                    "step": agent._current_step,
+                    "duration_ms": duration_ms,
+                    # Include ctx.metadata for trace_id, tenant_id, etc.
+                    "meta": dict(agent.ctx.metadata) if agent.ctx.metadata else {},
                 }
                 await agent.bus.publish(
                     Events.USAGE_RECORDED,

aury/agents/react/tools.py

@@ -226,6 +226,7 @@ async def execute_tool(
             abort_signal=agent._abort,
             update_metadata=agent._noop_update_metadata,
             middleware=agent.middleware,
+            step=agent._current_step,
         )
         
         # Emit running status before execution

{aury_agent-0.0.14.dist-info → aury_agent-0.0.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aury-agent
-Version: 0.0.14
+Version: 0.0.15
 Summary: Aury Agent Framework - React Agent and Workflow orchestration
 Author: Aury Team
 License: MIT

{aury_agent-0.0.14.dist-info → aury_agent-0.0.15.dist-info}/RECORD

@@ -82,7 +82,7 @@ aury/agents/core/types/message.py,sha256=W0d8elddcbvW98bejDkHIzKiTigr6O5KxVuVUne
 aury/agents/core/types/recall.py,sha256=BZSDgv8Q0eK7Sshc-jA6ycpFpMHzndZ-lS86MPto9-c,4898
 aury/agents/core/types/session.py,sha256=JX7Zg3ojKaGE4pixIoqGgWHPvgRs19hb9mY2IDA1ICM,9656
 aury/agents/core/types/subagent.py,sha256=vIxDkfekAaHj_V2UdSIaygDKb5RjiLqi8twTnhIUF_o,4651
-aury/agents/core/types/tool.py,sha256=Fg3Fq9yodPx-j530zLMSNwwWkfzWApHjUcFDGtZkwdE,14448
+aury/agents/core/types/tool.py,sha256=ccpCCP9VuTiJzAl4agtMqBYJEPxNYNo58VRBCYImsno,14688
 aury/agents/eval/__init__.py,sha256=uRGZN-RBpNlwOmrn66EwZBNKufDVKgMc3tODuMM6WB4,8850
 aury/agents/hitl/__init__.py,sha256=gcxGdXB9UlttWBXB4dNHwwAN3Vc2ZfmSMpaGM6BmsSE,1062
 aury/agents/hitl/ask_user.py,sha256=RtBHe29-UyiSxQDB-yMgxzO9BWmXyMaDm5xvD6iI5oo,9298
@@ -118,8 +118,8 @@ aury/agents/react/context.py,sha256=gARzqE4ez_xxA5JQJ2XoK-_2hnkV9MohscLnXpvYaL0,
 aury/agents/react/factory.py,sha256=3JOqFrOkp_UJMiBNitliP28s7C9eDP7YwuYIfigDCB4,10840
 aury/agents/react/pause.py,sha256=nUE2wHUDJsLdA3FHcnKVD9PdsGCrjQW1V8ZJUcssS5A,7857
 aury/agents/react/persistence.py,sha256=4BYmx3Dxu1grEpt4Qw3fUG26LgYyAhRnXVQC03OJDtE,6354
-aury/agents/react/step.py,sha256=pnG1xZ3pkK26CSVH8hx85J6L9pbg8jIRXi8P_SRvjjw,30470
-aury/agents/react/tools.py,sha256=rIZk3tKLUA_JiUaGvAJFiWgtVmUSWg-cU2iGVp-1LRc,18866
+aury/agents/react/step.py,sha256=0-Qj205ux1Y9vLOoChv5zGjcz60MvIneAcSjKqGW7YA,31162
+aury/agents/react/tools.py,sha256=dE2FfJa2iImsV0TNiP1QCw69rB0aGOhGYwMqObMUwJs,18904
 aury/agents/sandbox/__init__.py,sha256=adH29NhzpZUIFVCxhRUZxyANKpFauUjsPVd8iktUOaI,635
 aury/agents/sandbox/local.py,sha256=Dl-zJZ5FUz7WONtuze7MdpCDBjm9YZbjLIkGa9lYbwY,8328
 aury/agents/sandbox/remote.py,sha256=6NcYIs_thjyxa5LgIqwoPNsHYpT70DG6xGbjDYV6-X4,6444
@@ -150,7 +150,7 @@ aury/agents/workflow/expression.py,sha256=Hsx2jBigtt5zbJb9uK9pveipqMbBBUUAvlgYjB
 aury/agents/workflow/parser.py,sha256=mfiFZ_TtFq3IxAqPlTGcNkluiZ8ww16y3NYwbmbsrwE,6149
 aury/agents/workflow/state.py,sha256=AxTSo7P9b1DbWjGdQdzFY4_7m6CJGNFo7xkGK28SPZw,4234
 aury/agents/workflow/types.py,sha256=PVvjTQRgTObx5Mq_lXFogyLjGChOi5pUgcJwF5ZzVtg,2354
-aury_agent-0.0.14.dist-info/METADATA,sha256=J9hqoZ94RCaqkdCJCzwIbOkPLFRz-lIBhyyTjlOifPw,2115
-aury_agent-0.0.14.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-aury_agent-0.0.14.dist-info/entry_points.txt,sha256=7EC5SMAKdC9HZLG8RfPhGno4OKOeSjaGsOWi1gOL-K8,56
-aury_agent-0.0.14.dist-info/RECORD,,
+aury_agent-0.0.15.dist-info/METADATA,sha256=jS3XZmaGW9usD3GvRQsmmu03PQ-QxW6M2yQHOHugrDE,2115
+aury_agent-0.0.15.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+aury_agent-0.0.15.dist-info/entry_points.txt,sha256=7EC5SMAKdC9HZLG8RfPhGno4OKOeSjaGsOWi1gOL-K8,56
+aury_agent-0.0.15.dist-info/RECORD,,