From 950de5e670d919b7ee7076821de09772b7d7fb11 Mon Sep 17 00:00:00 2001
From: CREDO23 <bakerathierry@gmail.com>
Date: Wed, 22 Jul 2026 20:06:18 +0200
Subject: [PATCH] fix: forward ToolRuntime to capability tools

---
 surfsense_backend/app/capabilities/core/access/agent.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/surfsense_backend/app/capabilities/core/access/agent.py b/surfsense_backend/app/capabilities/core/access/agent.py
index 35c35774f..3d3864a53 100644
--- a/surfsense_backend/app/capabilities/core/access/agent.py
+++ b/surfsense_backend/app/capabilities/core/access/agent.py
@@ -157,6 +157,9 @@ def _capability_tool(capability: Capability, workspace_id: int) -> BaseTool:
             }
         )
 
+    # Un-stringify for StructuredTool's signature-based runtime injection.
+    _run.__annotations__["runtime"] = ToolRuntime
+
     return StructuredTool.from_function(
         coroutine=_run,
         name=name.replace(".", "_"),