aury-agent 0.0.9__py3-none-any.whl → 0.0.10__py3-none-any.whl

aury/agents/middleware/message.py

@@ -21,15 +21,10 @@ if TYPE_CHECKING:
 
 
 class MessageBackendMiddleware(BaseMiddleware):
-    """Middleware that persists messages directly via MessageBackend.
+    """Middleware that persists messages via MessageBackend.
     
-    This is the recommended middleware for message persistence.
-    Uses backends.message from the context for direct storage.
-    
-    Features:
-    - Saves both truncated (for LLM context) and raw (for audit) messages
-    - Supports namespace isolation for sub-agents
-    - Works with any MessageBackend implementation
+    Simple middleware that passes messages to the backend.
+    Storage details (raw/truncated handling) are left to the application layer.
     
     Example:
         agent = ReactAgent.create(
@@ -38,21 +33,6 @@ class MessageBackendMiddleware(BaseMiddleware):
         )
     """
     
-    def __init__(
-        self,
-        *,
-        save_raw: bool = False,
-        max_history: int = 100,
-    ):
-        """Initialize MessageBackendMiddleware.
-        
-        Args:
-            save_raw: Also save raw (untruncated) messages for audit
-            max_history: Max messages to keep in history (for truncation)
-        """
-        self.save_raw = save_raw
-        self.max_history = max_history
-    
     async def on_message_save(
         self,
         message: dict[str, Any],
@@ -78,45 +58,17 @@ class MessageBackendMiddleware(BaseMiddleware):
             return message
         
         backend = ctx.backends.message
-        
-        # Extract message fields
-        role = message.get("role", "")
-        content = message.get("content", "")
         invocation_id = ctx.invocation_id or ""
         agent_id = ctx.agent_id
-        tool_call_id = message.get("tool_call_id")
-        
-        # Build message dict for backend
-        msg_dict = {
-            "role": role,
-            "content": content,
-        }
-        if tool_call_id:
-            msg_dict["tool_call_id"] = tool_call_id
         
-        # Include tool_calls if present (for assistant messages)
-        if "tool_calls" in message:
-            msg_dict["tool_calls"] = message["tool_calls"]
-        
-        # Save truncated message (for LLM context)
+        # Save message
         await backend.add(
             session_id=session_id,
-            message=msg_dict,
-            type="truncated",
+            message=message,
             agent_id=agent_id,
             invocation_id=invocation_id,
         )
         
-        # Optionally save raw message (for audit)
-        if self.save_raw:
-            await backend.add(
-                session_id=session_id,
-                message=message,  # Full original message
-                type="raw",
-                agent_id=agent_id,
-                invocation_id=invocation_id,
-            )
-        
         # Pass through to other middlewares
         return message
 

aury/agents/react/factory.py

@@ -40,6 +40,7 @@ def create_react_agent(
     snapshot: "SnapshotBackend | None" = None,
     # ContextProvider system
     context_providers: "list[ContextProvider] | None" = None,
+    message_provider: "ContextProvider | None" = None,
     enable_history: bool = True,
     history_limit: int = 50,
     # Tool customization
@@ -64,8 +65,9 @@ def create_react_agent(
         memory: Memory manager (optional)
         snapshot: Snapshot backend (optional)
         context_providers: Additional custom context providers (optional)
-        enable_history: Enable message history (default True)
-        history_limit: Max conversation turns to keep (default 50)
+        message_provider: Custom message context provider (replaces default MessageContextProvider)
+        enable_history: Enable message history (default True, ignored if message_provider is set)
+        history_limit: Max conversation turns to keep (default 50, ignored if message_provider is set)
         delegate_tool_class: Custom DelegateTool class (optional)
         
     Returns:
@@ -128,7 +130,7 @@ def create_react_agent(
         middleware_chain = MiddlewareChain()
         # Add message persistence middleware first (uses backends.message)
         if enable_history and backends.message is not None:
-            middleware_chain.use(MessageBackendMiddleware(max_history=history_limit))
+            middleware_chain.use(MessageBackendMiddleware())
         # Add user middlewares
         if middlewares:
             for mw in middlewares:
@@ -152,10 +154,13 @@ def create_react_agent(
     # === Build providers ===
     default_providers: list["ContextProvider"] = []
     
-    # MessageContextProvider - for fetching history (uses backends.message)
-    if enable_history:
-        message_provider = MessageContextProvider(max_messages=history_limit * 2)
+    # MessageContextProvider - for fetching history
+    # Use custom message_provider if provided, otherwise use default
+    if message_provider is not None:
         default_providers.append(message_provider)
+    elif enable_history:
+        default_message_provider = MessageContextProvider(max_messages=history_limit * 2)
+        default_providers.append(default_message_provider)
     
     # Combine default + custom context_providers
     all_providers = default_providers + (context_providers or [])

aury/agents/react/step.py

@@ -293,10 +293,10 @@ async def execute_step(agent: "ReactAgent") -> str | None:
         if event.type == "content":
             # Text content
             if event.delta:
-                # === Middleware: on_model_stream ===
-                stream_chunk = {"delta": event.delta, "type": "content"}
+                # === Middleware: on_text_stream ===
+                stream_chunk = {"delta": event.delta}
                 if agent.middleware:
-                    stream_chunk = await agent.middleware.process_stream_chunk(
+                    stream_chunk = await agent.middleware.process_text_stream(
                         stream_chunk
                     )
                     if stream_chunk is None:
@@ -328,10 +328,10 @@ async def execute_step(agent: "ReactAgent") -> str | None:
             # Thinking content - only emit if thinking is enabled
             stream_thinking = agent._get_stream_thinking()
             if event.delta and enable_thinking:
-                # === Middleware: on_model_stream (type=thinking) ===
-                stream_chunk = {"delta": event.delta, "type": "thinking"}
+                # === Middleware: on_thinking_stream ===
+                stream_chunk = {"delta": event.delta}
                 if agent.middleware:
-                    stream_chunk = await agent.middleware.process_stream_chunk(
+                    stream_chunk = await agent.middleware.process_thinking_stream(
                         stream_chunk
                     )
                     if stream_chunk is None:
@@ -548,6 +548,67 @@ async def execute_step(agent: "ReactAgent") -> str | None:
                 data={"message": event.error or "Unknown LLM error"},
             ))
 
+    # === Middleware: on_text_stream_end (flush buffered content) ===
+    if agent.middleware:
+        final_chunks = await agent.middleware.process_text_stream_end()
+        for final_chunk in final_chunks:
+            final_delta = final_chunk.get("delta", "")
+            if final_delta:
+                agent._text_buffer += final_delta
+                # Emit the final text content
+                if agent._current_text_block_id is None:
+                    agent._current_text_block_id = generate_id("blk")
+                await agent.ctx.emit(BlockEvent(
+                    block_id=agent._current_text_block_id,
+                    kind=BlockKind.TEXT,
+                    op=BlockOp.DELTA,
+                    data={"content": final_delta},
+                ))
+                logger.debug(
+                    f"Emitted final text chunk from middleware: {len(final_delta)} chars",
+                    extra={"invocation_id": agent._current_invocation.id},
+                )
+    
+    # Emit text block completed status
+    if agent._current_text_block_id:
+        await agent.ctx.emit(BlockEvent(
+            block_id=agent._current_text_block_id,
+            kind=BlockKind.TEXT,
+            op=BlockOp.PATCH,
+            data={"status": "completed"},
+        ))
+    
+    # === Middleware: on_thinking_stream_end (flush buffered thinking) ===
+    if agent.middleware:
+        thinking_chunks = await agent.middleware.process_thinking_stream_end()
+        for thinking_chunk in thinking_chunks:
+            thinking_delta = thinking_chunk.get("delta", "")
+            if thinking_delta:
+                agent._thinking_buffer += thinking_delta
+                # Emit the final thinking content (if streaming thinking)
+                if agent.config.stream_thinking:
+                    if agent._current_thinking_block_id is None:
+                        agent._current_thinking_block_id = generate_id("blk")
+                    await agent.ctx.emit(BlockEvent(
+                        block_id=agent._current_thinking_block_id,
+                        kind=BlockKind.THINKING,
+                        op=BlockOp.DELTA,
+                        data={"content": thinking_delta},
+                    ))
+                logger.debug(
+                    f"Emitted final thinking chunk from middleware: {len(thinking_delta)} chars",
+                    extra={"invocation_id": agent._current_invocation.id},
+                )
+    
+    # Emit thinking block completed status
+    if agent._current_thinking_block_id:
+        await agent.ctx.emit(BlockEvent(
+            block_id=agent._current_thinking_block_id,
+            kind=BlockKind.THINKING,
+            op=BlockOp.PATCH,
+            data={"status": "completed"},
+        ))
+    
     # If thinking was buffered, emit it now
     if agent._thinking_buffer and not agent.config.stream_thinking:
         await agent.ctx.emit(BlockEvent(

aury/agents/react/tools.py

@@ -288,6 +288,7 @@ async def process_tool_results(agent: "ReactAgent") -> None:
                     role="tool",
                     content=invocation.result,
                     tool_call_id=invocation.tool_call_id,
+                    name=invocation.tool_name,  # Required for Gemini
                 )
             )
         
@@ -339,6 +340,7 @@ async def process_tool_results(agent: "ReactAgent") -> None:
             {
                 "type": "tool_result",
                 "tool_use_id": invocation.tool_call_id,
+                "tool_name": invocation.tool_name,
                 "content": result.output,
                 "is_error": invocation.is_error,
             }
@@ -352,5 +354,6 @@ async def process_tool_results(agent: "ReactAgent") -> None:
                 role="tool",
                 content=tr["content"],
                 tool_call_id=tr["tool_use_id"],
+                name=tr["tool_name"],  # Required for Gemini
             )
         )

aury/agents/tool/decorator.py

@@ -6,7 +6,7 @@ import inspect
 from functools import wraps
 from typing import Any, Callable, get_type_hints, get_origin, get_args, Union
 
-from ..core.types.tool import BaseTool, ToolContext, ToolResult, ToolInfo
+from ..core.types.tool import BaseTool, ToolContext, ToolResult, ToolInfo, ToolConfig
 
 
 def _type_to_schema(t: type) -> dict[str, Any]:
@@ -196,20 +196,12 @@ def tool(
         accepts_ctx = "ctx" in sig.parameters or "context" in sig.parameters
         ctx_param_name = "ctx" if "ctx" in sig.parameters else "context"
         
-        class FunctionTool:
+        class FunctionTool(BaseTool):
             """Tool created from decorated function."""
             
-            @property
-            def name(self) -> str:
-                return tool_name
-            
-            @property
-            def description(self) -> str:
-                return tool_desc
-            
-            @property
-            def parameters(self) -> dict[str, Any]:
-                return parameters_schema
+            _name = tool_name
+            _description = tool_desc
+            _parameters = parameters_schema
             
             async def execute(
                 self,
@@ -237,13 +229,6 @@ def tool(
                 
                 except Exception as e:
                     return ToolResult.error(str(e))
-            
-            def get_info(self) -> ToolInfo:
-                return ToolInfo(
-                    name=tool_name,
-                    description=tool_desc,
-                    parameters=parameters_schema,
-                )
         
         return FunctionTool()
     

{aury_agent-0.0.9.dist-info → aury_agent-0.0.10.dist-info}/METADATA

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

{aury_agent-0.0.9.dist-info → aury_agent-0.0.10.dist-info}/RECORD

@@ -2,7 +2,7 @@ aury/__init__.py,sha256=yTnJ3EOYYknROar-CUqU825NRcsc06fAp_dEIEUwRMI,94
 aury/agents/__init__.py,sha256=kaSH-1392iNti6GfFJra9nl7lu4iz1QHANz-y7IDl8Q,1842
 aury/agents/plugin.py,sha256=V2KCu4MIBoar6BmCevKp7AzGZvHbGi3CizQSE2gCOzs,5438
 aury/agents/a2a/__init__.py,sha256=xwZLlAOhv2nGU6FHaCd8lygaJ5IPdVnvw-cB_w1Dwxg,4979
-aury/agents/backends/__init__.py,sha256=f7AOdMM2Tj9Qn7uJTR5SvMiA3slbrItHW6uDtiyy0R8,5946
+aury/agents/backends/__init__.py,sha256=7vFpBzRUvWDUKv1aadFgIeQ8zcX-9F2J_BDRvT5BsiM,5898
 aury/agents/backends/sandbox.py,sha256=imOItqBu47GUc8sqHKDJFYlRoBlfn4VV-8CG8cL9480,8029
 aury/agents/backends/shell.py,sha256=Bc72UQ5LVu-ZKxb7K3V8wS8UI_WV0pib-F2Ou7_p3Rw,6006
 aury/agents/backends/artifact/__init__.py,sha256=YMKBSAoItGQMTqKBodWg8xaitJOMuOq4MGI9MuzpjkE,210
@@ -18,9 +18,9 @@ aury/agents/backends/invocation/types.py,sha256=2b3DcCmum6Ctkn7JuNZeSvnxRlN_VJ9Z
 aury/agents/backends/memory/__init__.py,sha256=mIaSJOnFNybMPNNOtR2zch_wFs6b11JFMp9LL8tvOLM,162
 aury/agents/backends/memory/memory.py,sha256=HWpME3vl4nLM5J3lv_iHuKjpbbLBLWE23SFFDG8NQJk,5621
 aury/agents/backends/memory/types.py,sha256=qbKPZc-E_SMnGVq1Eh_5Da_3Fp235p6xzV9b044qI2A,3680
-aury/agents/backends/message/__init__.py,sha256=Ea1WhfRsYjfsXvc2yzQVDSuKvLCZ0XXj-os-RgfmT6k,199
-aury/agents/backends/message/memory.py,sha256=yqh8lq1Hh1USaM7C4q-a8RbWRKFsyUSI2KGs1G1ZW74,3715
-aury/agents/backends/message/types.py,sha256=YepmOcfRUdtnaKXk6Oua6yKNeyDyB7fOkIWPt-Qknzc,3610
+aury/agents/backends/message/__init__.py,sha256=NgQZPYZ7NJfZ59OXSx_s7LRvkXGYtkXx9f-F5QatLys,167
+aury/agents/backends/message/memory.py,sha256=9DNMotsjje_jfwsixmgNlSxrigO93m6zgkFRbgrLxbU,2916
+aury/agents/backends/message/types.py,sha256=eYRmjGbnD3NG84Tdz_OmfqnRe_j_ovHT6sgXxQ4mM14,2564
 aury/agents/backends/session/__init__.py,sha256=27WuH6ulc3RBrPN3pTK2iittVdGno5iMlY9knBKnZ-E,167
 aury/agents/backends/session/memory.py,sha256=qMBnNgq5HxA39VHGUtftzJqCtsSpHWuetqF5nFFnmgQ,2837
 aury/agents/backends/session/types.py,sha256=ukfmE__QIbDNeMz0pj6gYEyjHYtGgolpqPGEu7X_u6E,3309
@@ -50,13 +50,13 @@ aury/agents/context_providers/__init__.py,sha256=QlbTh2vAHbxfZcXjcY5mOsLqOiYkIYG
 aury/agents/context_providers/artifact.py,sha256=tjjaseE2iosKLx6BAJ7kR5xmo9VsNY2ukkbj5NLHlC0,9144
 aury/agents/context_providers/base.py,sha256=FhFRWJS_afDFTHl7Yjr6vGIje58IpEOlEolgqpFR-QQ,6246
 aury/agents/context_providers/memory.py,sha256=HKIR5fRNzsZSV1-IjV325cTYIFqmNVPFPujufSI1CrY,1942
-aury/agents/context_providers/message.py,sha256=3YkscDxhTzKkkCtCTOdRJHabXym0crJlu3eP2_A0lYY,4558
+aury/agents/context_providers/message.py,sha256=LDQ1BPozyunh_4FM9WLcnf4CX3piOeD94zUkn9QG84o,4524
 aury/agents/context_providers/skill.py,sha256=w_R1txvu45ruMjuOtckuD9oG_hqeFGqAa_NZ_0T8WNs,1355
 aury/agents/context_providers/subagent.py,sha256=Y-6bYjaq6nbLS9d3siyy5eAt6SzFfk6N3Y1WU209lhI,1364
 aury/agents/context_providers/tool.py,sha256=wQGmcZi_XuB9ukweg5PUVO93GLzley0xftje-iVSfq8,2266
 aury/agents/core/__init__.py,sha256=-euIGAwoe2gU_2xEqJ8YmOMnDOaOlgWXwXxzIQnDG8E,1716
 aury/agents/core/base.py,sha256=nqsQ7fG0HZ08bqJh7KdpBgIxtJzDXZ6qLMf5hZ5vBuc,21279
-aury/agents/core/context.py,sha256=fDJ9S1sPuFbQJ9XihGdiPXDqZB01SxijxhiOose6C00,28933
+aury/agents/core/context.py,sha256=co_v6Mz7fYR0jaTlUHlnSBWaOgr_1_84lVO0FPo_tKo,28932
 aury/agents/core/context_builder.py,sha256=HF0hfnKBVCmlcFDmdyj53org90318ixmZl1LYQBl0fM,9075
 aury/agents/core/factory.py,sha256=8rx_2jD9Sj2tK-8_We09fHT1czjbN-yF1ws0hmUX2w4,6829
 aury/agents/core/isolator.py,sha256=zi_ZXk-a6b7D2EAeWH3FKkO5_7PwK975zRd9enbBf_E,2916
@@ -79,7 +79,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=oRn9jKFe8Tg-12ZFQcGJ0VLaJIWBfk8YQDzXUuXSNEk,9406
 aury/agents/core/types/subagent.py,sha256=vIxDkfekAaHj_V2UdSIaygDKb5RjiLqi8twTnhIUF_o,4651
-aury/agents/core/types/tool.py,sha256=DqlK306lmeabb61BscqHbhdN7mx3whzndh3GYGqSb8Q,6366
+aury/agents/core/types/tool.py,sha256=sP0mA_YufOjlPnEKcQSAbp719GkJulNuvQ76yYmahFU,7213
 aury/agents/eval/__init__.py,sha256=uRGZN-RBpNlwOmrn66EwZBNKufDVKgMc3tODuMM6WB4,8850
 aury/agents/hitl/__init__.py,sha256=6H8cBU-dSR0eSOGWte_Iy7OvTKGDl4ZMb0E56KrfFkw,1020
 aury/agents/hitl/ask_user.py,sha256=3MAJOo0rI0Wl7ZpZdDSQVBkO0GJFYH6bT7hnrcED4qI,9071
@@ -90,7 +90,7 @@ aury/agents/hitl/revert.py,sha256=i3F8Zcibuo3fjhJTK-rQEYSPMKxmRQeua5cjoDzAR14,66
 aury/agents/llm/__init__.py,sha256=teantqedfyQ6IxLD6LayLr2cbgFYbMT2I4TUSfkOrQ4,625
 aury/agents/llm/adapter.py,sha256=icX_RVk738PqxueCFj58WTPVhFXQpgNkjn__u2dZDEM,14916
 aury/agents/llm/openai.py,sha256=v4Qlc5EIJDNcqkGHA2aIEaTTKaPVGMPf1227aKcef58,11192
-aury/agents/llm/provider.py,sha256=zZ5xfbOEfcYSORrg05rylKVtfOzdzvolQ4GTPWEir2A,15376
+aury/agents/llm/provider.py,sha256=oE7sBF64pq8h1-bowSwApmAl3SVlHV_I4aCMePD29Mk,16106
 aury/agents/mcp/__init__.py,sha256=7zsY-5LhQBLpiQI-XWe2uUHdTwJrDHrr0pDd_SXfO3o,4355
 aury/agents/memory/__init__.py,sha256=PMGMQ1VE-j3M0n4dIeb9DYO3qrCyv5AMSfMWTyhPARE,816
 aury/agents/memory/compaction.py,sha256=3inOVsbwSgTghcsSLgvUdX8IG2J6_0Qbr36vAqanV0M,12757
@@ -98,27 +98,25 @@ aury/agents/memory/manager.py,sha256=9UsuZUYmp5Pju9lRa0bsque6gHpLIE5UwdqtQKZ3fO0
 aury/agents/memory/processor.py,sha256=Y4PZIMFdNtn_TQ3fRdhynFLUgcURi8Xu-ZeRq5kDS_s,4816
 aury/agents/memory/store.py,sha256=SX5AYMbifCEFfLqPahhDRZLswZK0g6vFy523F1CTUWg,5822
 aury/agents/memory/types.py,sha256=N8Rhc79OEjdJ3Xe7iG1AwnWK1Ej6b-zI51FIqmyGmLg,4560
-aury/agents/messages/__init__.py,sha256=NiDuJPpzlWkQ-kQT7uxxZrWUVWJ0JS_x3q7QlTmaF3Y,930
+aury/agents/messages/__init__.py,sha256=AAKgyGtnI60APvIo8RBgipF1jyaaOydUlL1TZVwXUH4,715
 aury/agents/messages/config.py,sha256=1ed8Akbna75Z5-eR6qnDPBFo91qwbdkSgQHc3wF5uBQ,1685
-aury/agents/messages/raw_store.py,sha256=QT6tJlJO-cBXKuVqYUsGXavWrA0cAsoJp6zcgMx6anY,6788
 aury/agents/messages/store.py,sha256=1zE24GHnXpPOhDQbehmAGQpNxW9Gv1-sas9dm0FYt4s,3969
 aury/agents/messages/types.py,sha256=riNl-C6f_zakpS673U0KRlAVSvQg5L4Pd6FNs2Isssg,2582
-aury/agents/middleware/__init__.py,sha256=T-A9u_T_x8QwgamDjUJc5GhF0VfMwOGLYezspypo4eg,896
-aury/agents/middleware/base.py,sha256=pm07DHWydXPug47_WkB4z57XhxvPcHR-oKo_U2UAOyo,9501
-aury/agents/middleware/chain.py,sha256=pp49Orth_ILEQkM8lmVT_1PFMnICpycqL4YfHShS82I,15359
-aury/agents/middleware/message.py,sha256=1Zw7c2WWjUFzKQ21mC_Ga0j6UbLnp4VHp4SC5FhsWSo,3581
+aury/agents/middleware/__init__.py,sha256=8iafxYOQtYgzYUorN68gdq2xFbiGLd-D7JR263lIf9U,822
+aury/agents/middleware/base.py,sha256=M4sKdIPOqHOsiiYveQaqgZgK-n4bQW2FSg5pbah_b04,10445
+aury/agents/middleware/chain.py,sha256=7HlTvTZI5bOMIzp3UbIjCwhqXytJTgMndz8zhGnxEg8,17609
+aury/agents/middleware/message.py,sha256=hDGGb0QwL04K2llbUId1938sd4DIJSpq_KnRJ6fcriE,2017
 aury/agents/middleware/message_container.py,sha256=0I_C-mIVVUqSzMzaYKUZGPtJUbV3Plo1msW8BIU-3RE,4073
-aury/agents/middleware/raw_message.py,sha256=ogcvGhqYLu6j35UJC4AGydokC91PHEQY81t-lcNy9y4,5064
 aury/agents/middleware/truncation.py,sha256=SM1ldZ6kC-joQVaoukNwFqKiHQ0symJAnzBJVANkxmg,4786
 aury/agents/middleware/types.py,sha256=T63AgePVSs4b1NT0MZ5PyaQJBLoVqNoAQt9NSGjBwBs,2512
 aury/agents/react/__init__.py,sha256=yixRP3RSXQtVaamtwjaOpmW55sZb-xp4XRjxYcy0HfU,90
 aury/agents/react/agent.py,sha256=DYWxuspMs3srOuxW4Sxis_Hkg4R6YkM0aM-rLRjD9Oo,23300
 aury/agents/react/context.py,sha256=Q7pY5j6b-OvKAmI1UsuawC3BmVxFkKN1oXxeBTVFejQ,11067
-aury/agents/react/factory.py,sha256=Y5S1wxIU2gvQsIb_7xBz8Nzr4IUZZkuF10hIbkT_DHk,10423
+aury/agents/react/factory.py,sha256=7aWb2YqSdlIq161Ke0JZwnjpaA9U7Q5xejAnkrmjMUw,10776
 aury/agents/react/pause.py,sha256=bcrhbuoJQUz7WxZSsKr5qjxQlDdpwai54ns_qio0du4,7306
 aury/agents/react/persistence.py,sha256=bwRk5Dq1qIoIbfOsVe6BhGCuXJM5TUX-zyHt9zEVPzw,5845
-aury/agents/react/step.py,sha256=Btwo9VtPKNpetaLnBWtJ6bdRCYl13n7bmXmoNbTZr38,26302
-aury/agents/react/tools.py,sha256=0mDKeVIl6lmy3nLn0b_t-F0mEmtt7wwl7IbpV-Lfer4,13186
+aury/agents/react/step.py,sha256=oUvmVUlzGX6aa_YcbhHAmA3UHgKJcMvK3IrMFzEX-IM,28994
+aury/agents/react/tools.py,sha256=VxXec1K8C0h85yQZBVOQt0dhsmhSKCVV7iV_LXlasAs,13368
 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
@@ -127,7 +125,7 @@ aury/agents/skill/__init__.py,sha256=CkWSSclFphXCrQkQDbf6ndclP4dsLvWYhSW8Wj5lVcw
 aury/agents/skill/loader.py,sha256=g2mhC0CLzP0s8BlNk21e_IEonHNyekK7EbqUOFkfOHc,5516
 aury/agents/skill/types.py,sha256=yDzMxF9S5D4NNA3k1KnQvhhVBd4plCu2oJXCWTO1jSI,2696
 aury/agents/tool/__init__.py,sha256=_IC06waxr9cC-HXdalwDByOBAZgfQiMwjsXypj-9mVE,731
-aury/agents/tool/decorator.py,sha256=u6x3NNGcXEPf1dp1amaJKQhIdApqm8redYsUcBYqZy4,8484
+aury/agents/tool/decorator.py,sha256=dTd9tj851ydik-cjClsHB97gN34qXsv3IV9677y7u5g,8052
 aury/agents/tool/set.py,sha256=6UgI12E2b-C0Kfwg3KUr4e5kraBK887wbveAs8GudyY,7728
 aury/agents/tool/builtin/__init__.py,sha256=Pp5NlE0i1jiNlfZ6kVH-iSeE553RMysa4icmOgPzezI,513
 aury/agents/tool/builtin/ask_user.py,sha256=RnAJ969YE77AYG_foI7KlVsISvHj5Ykrdz0l2ZYZh-g,4871
@@ -149,7 +147,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.9.dist-info/METADATA,sha256=TZx8DIqJFPAeQt1yDq2lhlJ7X3uOCbnZ6zjIWq06OoQ,2114
-aury_agent-0.0.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-aury_agent-0.0.9.dist-info/entry_points.txt,sha256=7EC5SMAKdC9HZLG8RfPhGno4OKOeSjaGsOWi1gOL-K8,56
-aury_agent-0.0.9.dist-info/RECORD,,
+aury_agent-0.0.10.dist-info/METADATA,sha256=O36iw-FfvggVYFrIhm5tnHP33J3jS9vsrQCZdMCgCJA,2115
+aury_agent-0.0.10.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+aury_agent-0.0.10.dist-info/entry_points.txt,sha256=7EC5SMAKdC9HZLG8RfPhGno4OKOeSjaGsOWi1gOL-K8,56
+aury_agent-0.0.10.dist-info/RECORD,,

aury/agents/messages/raw_store.py

@@ -1,224 +0,0 @@
-"""Raw message store for complete message storage.
-
-RawMessageStore stores complete, untruncated messages for:
-- HITL recovery (restore exact state)
-- Full-context recall (when truncated history is insufficient)
-- Audit/debugging
-
-Messages are stored per invocation and can be cleaned up after invocation completes.
-"""
-from __future__ import annotations
-
-from typing import Any, Protocol, runtime_checkable
-from datetime import datetime
-
-from ..core.types.session import generate_id
-
-
-@runtime_checkable
-class RawMessageStore(Protocol):
-    """Protocol for raw message storage.
-    
-    Stores complete messages per invocation.
-    """
-    
-    async def add(
-        self,
-        invocation_id: str,
-        message: dict[str, Any],
-    ) -> str:
-        """Add a message and return its ID.
-        
-        Args:
-            invocation_id: Invocation this message belongs to
-            message: Complete message dict
-            
-        Returns:
-            Generated message ID
-        """
-        ...
-    
-    async def get(self, msg_id: str) -> dict[str, Any] | None:
-        """Get a single message by ID.
-        
-        Args:
-            msg_id: Message ID
-            
-        Returns:
-            Message dict or None if not found
-        """
-        ...
-    
-    async def get_many(self, msg_ids: list[str]) -> list[dict[str, Any]]:
-        """Get multiple messages by IDs.
-        
-        Args:
-            msg_ids: List of message IDs
-            
-        Returns:
-            List of message dicts (in same order, None for missing)
-        """
-        ...
-    
-    async def get_by_invocation(self, invocation_id: str) -> list[dict[str, Any]]:
-        """Get all messages for an invocation.
-        
-        Args:
-            invocation_id: Invocation ID
-            
-        Returns:
-            List of messages in chronological order
-        """
-        ...
-    
-    async def delete_by_invocation(self, invocation_id: str) -> int:
-        """Delete all messages for an invocation.
-        
-        Args:
-            invocation_id: Invocation ID
-            
-        Returns:
-            Number of messages deleted
-        """
-        ...
-
-
-class StateBackendRawMessageStore:
-    """RawMessageStore implementation backed by StateBackend."""
-    
-    def __init__(self, backend: Any):  # StateBackend
-        """Initialize with StateBackend.
-        
-        Args:
-            backend: StateBackend instance
-        """
-        self._backend = backend
-    
-    async def add(
-        self,
-        invocation_id: str,
-        message: dict[str, Any],
-    ) -> str:
-        """Add a message using StateBackend."""
-        msg_id = generate_id("rmsg")
-        
-        # Store message with metadata
-        await self._backend.set("raw_messages", msg_id, {
-            "id": msg_id,
-            "invocation_id": invocation_id,
-            "message": message,
-            "created_at": datetime.now().isoformat(),
-        })
-        
-        # Add to invocation's message list
-        inv_key = f"raw_msg_ids:{invocation_id}"
-        msg_ids = await self._backend.get("raw_messages", inv_key) or []
-        msg_ids.append(msg_id)
-        await self._backend.set("raw_messages", inv_key, msg_ids)
-        
-        return msg_id
-    
-    async def get(self, msg_id: str) -> dict[str, Any] | None:
-        """Get a single message."""
-        data = await self._backend.get("raw_messages", msg_id)
-        if data:
-            return data.get("message")
-        return None
-    
-    async def get_many(self, msg_ids: list[str]) -> list[dict[str, Any]]:
-        """Get multiple messages."""
-        results = []
-        for msg_id in msg_ids:
-            msg = await self.get(msg_id)
-            if msg:
-                results.append(msg)
-        return results
-    
-    async def get_by_invocation(self, invocation_id: str) -> list[dict[str, Any]]:
-        """Get all messages for an invocation."""
-        inv_key = f"raw_msg_ids:{invocation_id}"
-        msg_ids = await self._backend.get("raw_messages", inv_key) or []
-        return await self.get_many(msg_ids)
-    
-    async def delete_by_invocation(self, invocation_id: str) -> int:
-        """Delete all messages for an invocation."""
-        inv_key = f"raw_msg_ids:{invocation_id}"
-        msg_ids = await self._backend.get("raw_messages", inv_key) or []
-        
-        # Delete each message
-        for msg_id in msg_ids:
-            await self._backend.remove("raw_messages", msg_id)
-        
-        # Delete the index
-        await self._backend.remove("raw_messages", inv_key)
-        
-        return len(msg_ids)
-
-
-class InMemoryRawMessageStore:
-    """In-memory raw message store for testing."""
-    
-    def __init__(self) -> None:
-        self._messages: dict[str, dict[str, Any]] = {}  # msg_id -> message data
-        self._invocation_index: dict[str, list[str]] = {}  # inv_id -> [msg_ids]
-    
-    async def add(
-        self,
-        invocation_id: str,
-        message: dict[str, Any],
-    ) -> str:
-        """Add a message."""
-        msg_id = generate_id("rmsg")
-        
-        self._messages[msg_id] = {
-            "id": msg_id,
-            "invocation_id": invocation_id,
-            "message": message,
-            "created_at": datetime.now().isoformat(),
-        }
-        
-        if invocation_id not in self._invocation_index:
-            self._invocation_index[invocation_id] = []
-        self._invocation_index[invocation_id].append(msg_id)
-        
-        return msg_id
-    
-    async def get(self, msg_id: str) -> dict[str, Any] | None:
-        """Get a single message."""
-        data = self._messages.get(msg_id)
-        if data:
-            return data.get("message")
-        return None
-    
-    async def get_many(self, msg_ids: list[str]) -> list[dict[str, Any]]:
-        """Get multiple messages."""
-        results = []
-        for msg_id in msg_ids:
-            msg = await self.get(msg_id)
-            if msg:
-                results.append(msg)
-        return results
-    
-    async def get_by_invocation(self, invocation_id: str) -> list[dict[str, Any]]:
-        """Get all messages for an invocation."""
-        msg_ids = self._invocation_index.get(invocation_id, [])
-        return await self.get_many(msg_ids)
-    
-    async def delete_by_invocation(self, invocation_id: str) -> int:
-        """Delete all messages for an invocation."""
-        msg_ids = self._invocation_index.pop(invocation_id, [])
-        for msg_id in msg_ids:
-            self._messages.pop(msg_id, None)
-        return len(msg_ids)
-    
-    def clear(self) -> None:
-        """Clear all messages (for testing)."""
-        self._messages.clear()
-        self._invocation_index.clear()
-
-
-__all__ = [
-    "RawMessageStore",
-    "StateBackendRawMessageStore",
-    "InMemoryRawMessageStore",
-]