ragbits-agents 1.4.0.dev202601310254__py3-none-any.whl → 1.4.0.dev202602030301__py3-none-any.whl

ragbits/agents/_main.py

@@ -1036,12 +1036,11 @@ class Agent(
                 }
 
         # Execute POST_TOOL hooks with chaining
-        post_tool_result = await self.hook_manager.execute_post_tool(
+        post_tool_output = await self.hook_manager.execute_post_tool(
             tool_call=tool_call,
-            output=tool_output,
+            tool_return=tool_return,
             error=tool_error,
         )
-        tool_output = post_tool_result.output
 
         # Raise error after hooks have been executed
         if tool_error:
@@ -1051,8 +1050,8 @@ class Agent(
             id=tool_call.id,
             name=tool_call.name,
             arguments=tool_call.arguments,
-            result=tool_return.value,
-            metadata=tool_return.metadata,
+            result=post_tool_output.tool_return.value if post_tool_output.tool_return else None,
+            metadata=post_tool_output.tool_return.metadata if post_tool_output.tool_return else None,
         )
 
     @requires_dependencies(["a2a.types"], "a2a")

ragbits/agents/hooks/manager.py

@@ -8,11 +8,12 @@ organization, and execution of hooks during lifecycle events.
 import hashlib
 import json
 from collections import defaultdict
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
 
 from ragbits.agents.confirmation import ConfirmationRequest
 from ragbits.agents.hooks.base import Hook
 from ragbits.agents.hooks.types import EventType, PostToolInput, PostToolOutput, PreToolInput, PreToolOutput
+from ragbits.agents.tool import ToolReturn
 from ragbits.core.llms.base import ToolCall
 
 if TYPE_CHECKING:
@@ -156,7 +157,7 @@ class HookManager:
     async def execute_post_tool(
         self,
         tool_call: ToolCall,
-        output: Any,  # noqa: ANN401
+        tool_return: ToolReturn | None,
         error: Exception | None,
     ) -> PostToolOutput:
         """
@@ -166,7 +167,7 @@ class HookManager:
 
         Args:
             tool_call: The tool call that was executed
-            output: The tool output
+            tool_return: Object representing the output of the tool (with value passed to the LLM and metadata)
             error: Any error that occurred
 
         Returns:
@@ -175,20 +176,20 @@ class HookManager:
         hooks = self.get_hooks(EventType.POST_TOOL, tool_call.name)
 
         # Start with original output
-        current_output = output
+        current_output = tool_return
 
         for hook in hooks:
             # Create input with current state (chained from previous hook)
             hook_input = PostToolInput(
                 tool_call=tool_call,
-                output=current_output,
+                tool_return=current_output,
                 error=error,
             )
 
             result: PostToolOutput = await hook.execute(hook_input)
 
             # Chain output for next hook
-            current_output = result.output
+            current_output = result.tool_return
 
         # Return final chained result
-        return PostToolOutput(output=current_output)
+        return PostToolOutput(tool_return=current_output)

ragbits/agents/hooks/types.py

@@ -12,6 +12,7 @@ from typing import Any, Literal, TypeAlias
 from pydantic import BaseModel, Field, model_validator
 
 from ragbits.agents.confirmation import ConfirmationRequest
+from ragbits.agents.tool import ToolReturn
 from ragbits.core.llms.base import ToolCall
 
 
@@ -73,13 +74,13 @@ class PostToolInput(HookEventIO):
     Attributes:
         event_type: Always EventType.POST_TOOL (unchangeable)
         tool_call: The original tool call
-        output: The result returned by the tool (None if error occurred)
+        tool_return: The result returned by the tool (None if error occurred)
         error: Any error that occurred during execution (None if successful)
     """
 
     event_type: Literal[EventType.POST_TOOL] = Field(default=EventType.POST_TOOL, frozen=True)
     tool_call: ToolCall
-    output: Any = None
+    tool_return: ToolReturn | None = None
     error: Exception | None = None
 
 
@@ -120,20 +121,20 @@ class PostToolOutput(HookEventIO):
 
     Attributes:
         event_type: Always EventType.POST_TOOL (unchangeable)
-        output: Tool output to use (original or modified) - always present
+        tool_return: Tool output to use (original or modified) - None if the tool execution failed
 
     Example:
         ```python
         # Pass through unchanged
-        return PostToolOutput(output=input.output)
+        return PostToolOutput(tool_return=input.tool_return)
 
         # Modify output
-        return PostToolOutput(output={"filtered": data})
+        return PostToolOutput(tool_return=ToolReturn(value={"filtered": data}))
         ```
     """
 
     event_type: Literal[EventType.POST_TOOL] = Field(default=EventType.POST_TOOL, frozen=True)  # type: ignore[assignment]
-    output: Any
+    tool_return: ToolReturn | None
 
 
 # Type aliases for hook callbacks

{ragbits_agents-1.4.0.dev202601310254.dist-info → ragbits_agents-1.4.0.dev202602030301.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ragbits-agents
-Version: 1.4.0.dev202601310254
+Version: 1.4.0.dev202602030301
 Summary: Building blocks for rapid development of GenAI applications
 Project-URL: Homepage, https://github.com/deepsense-ai/ragbits
 Project-URL: Bug Reports, https://github.com/deepsense-ai/ragbits/issues
@@ -22,7 +22,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: >=3.10
-Requires-Dist: ragbits-core==1.4.0.dev202601310254
+Requires-Dist: ragbits-core==1.4.0.dev202602030301
 Provides-Extra: a2a
 Requires-Dist: a2a-sdk<1.0.0,>=0.2.9; extra == 'a2a'
 Requires-Dist: fastapi<1.0.0,>=0.115.0; extra == 'a2a'

{ragbits_agents-1.4.0.dev202601310254.dist-info → ragbits_agents-1.4.0.dev202602030301.dist-info}/RECORD

@@ -1,5 +1,5 @@
 ragbits/agents/__init__.py,sha256=rKqDbbppy70HmDN6AtC3eXzyVWTTagkRpLvPjWulGuY,1040
-ragbits/agents/_main.py,sha256=bxMBogDl1dWxfGZk3cSztogj8gK8VoYo38jfoMzsBXU,52013
+ragbits/agents/_main.py,sha256=TF0qvhLQ3LS2AmjQ0xjoXl8W47vnFjV1-aypnCEsYPk,52090
 ragbits/agents/cli.py,sha256=xUS7k8IAn0479n165i4YVFKo4Jx0M2iCpaMILZi9xt8,17649
 ragbits/agents/confirmation.py,sha256=cwdd1feSSobxa7gxBvEZcL9e3tcLdc8CyfvvQwaHF1Y,619
 ragbits/agents/exceptions.py,sha256=TiompKlP1QRD4TZ7hpp52dyZqg0rjoq1t5QUTxFZud8,3552
@@ -11,8 +11,8 @@ ragbits/agents/a2a/server.py,sha256=4a-cq87OeMoVOjEM_PbwTSHhPTpJv8r_xOVaZAzFIiI,
 ragbits/agents/hooks/__init__.py,sha256=xvWTFLkbM6aRLEz-Ypw9MMDOKVIJqhZzN5XE0jEvy3o,1984
 ragbits/agents/hooks/base.py,sha256=Rl_SUgnI-e8Rgrg7TAlxDsdC2nqtRAOIGpIsPV3Y_Sk,3137
 ragbits/agents/hooks/confirmation.py,sha256=ykSRR2sF8lbxG4pJN5amtT6qstnDoTbrYWZaafoSa9Y,1648
-ragbits/agents/hooks/manager.py,sha256=ssQjICFmmstQhfVY3Vzhr5mj8K3npGVZ7Q5HL-dj6RE,6910
-ragbits/agents/hooks/types.py,sha256=XIJZIVtjgrBGFh59T8ljkfj3qF9fzY0tqC_Iv59zLak,4407
+ragbits/agents/hooks/manager.py,sha256=Exj-Q9Hajf61n_08pDVtI3ouTZ8EphNnSIxM5xx-EOI,7047
+ragbits/agents/hooks/types.py,sha256=zQkLhrctjU8adELYb78ikX3Q0kqPp_LNrAvfJqHI7GU,4550
 ragbits/agents/mcp/__init__.py,sha256=7icbSe4FCBMgPADNNoeQBKiLY8J6gIlqFS77gFGr6Ks,405
 ragbits/agents/mcp/server.py,sha256=slob6ns-sbLmG-bQAbwz-23MWsaAENeKhE2eNdoI3Vs,16397
 ragbits/agents/mcp/utils.py,sha256=o_UXS-olS1uRdfZHTRJdG697Xmq3q77twfcHzmyRTY0,1394
@@ -25,6 +25,6 @@ ragbits/agents/tools/memory.py,sha256=IrRGpZvdylNbn_Z7FFFwun9Rx3OSQimjVynSt3WgUo
 ragbits/agents/tools/openai.py,sha256=SpoB6my_T6LwfHjrP7ivQL6H4KvwInVampXq-6nGAHE,4955
 ragbits/agents/tools/todo.py,sha256=R86_Hu0HIl5Ujp8B2ctOo1iSLAHmfrzyu6jnyCLs4uc,18576
 ragbits/agents/tools/types.py,sha256=6yNG7IjG476muiCIcXKRgDJSemCeO2ZgycpXLyfu8jc,441
-ragbits_agents-1.4.0.dev202601310254.dist-info/METADATA,sha256=O2-1QB13Z7km26KnXDhIGOpy4vjU5AaRAYyPYwf4Uwg,2273
-ragbits_agents-1.4.0.dev202601310254.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-ragbits_agents-1.4.0.dev202601310254.dist-info/RECORD,,
+ragbits_agents-1.4.0.dev202602030301.dist-info/METADATA,sha256=6yp4rccNkmzSd4B8SM8Rq8TBFoSilNNEAgXqra9EwuE,2273
+ragbits_agents-1.4.0.dev202602030301.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+ragbits_agents-1.4.0.dev202602030301.dist-info/RECORD,,