feat(tools/tom_consult): declare resources for parallel execution (#2663)

Author: VascoSch92

openhands-tools/openhands/tools/tom_consult/definition.py

@@ -11,7 +11,13 @@
 
 from openhands.sdk.io import LocalFileStore
 from openhands.sdk.llm import ImageContent, TextContent
-from openhands.sdk.tool import Action, Observation, ToolDefinition, register_tool
+from openhands.sdk.tool import (
+    Action,
+    DeclaredResources,
+    Observation,
+    ToolDefinition,
+    register_tool,
+)
 
 
 if TYPE_CHECKING:
@@ -143,6 +149,14 @@ def to_llm_content(self) -> Sequence[TextContent | ImageContent]:
 class TomConsultTool(ToolDefinition[ConsultTomAction, ConsultTomObservation]):
     """Tool for consulting Tom agent."""
 
+    def declared_resources(self, action: Action) -> DeclaredResources:  # noqa: ARG002
+        """Declare resources for parallel execution.
+
+        Consulting Tom is a read-only LLM call with no shared mutable
+        state, so it is always safe to run in parallel.
+        """
+        return DeclaredResources(keys=(), declared=True)
+
     @classmethod
     @override
     def create(

tests/tools/tom_consult/__init__.py

@@ -0,0 +1,38 @@
+"""Tests for TomConsultTool declared_resources."""
+
+import pytest
+
+from openhands.sdk.tool import DeclaredResources
+from openhands.tools.tom_consult.definition import (
+    ConsultTomAction,
+    ConsultTomObservation,
+    TomConsultTool,
+)
+
+
+@pytest.mark.parametrize(
+    "action",
+    [
+        ConsultTomAction(reason="unclear intent", use_user_message=True),
+        ConsultTomAction(
+            reason="need guidance",
+            use_user_message=False,
+            custom_query="What does the user prefer?",
+        ),
+    ],
+    ids=["use-user-message", "custom-query"],
+)
+def test_consult_tom_declared_resources(action):
+    """TomConsultTool always declares safe with no resource keys."""
+    tool = TomConsultTool(
+        action_type=ConsultTomAction,
+        observation_type=ConsultTomObservation,
+        description="test",
+        executor=None,
+    )
+
+    resources = tool.declared_resources(action)
+
+    assert isinstance(resources, DeclaredResources)
+    assert resources.declared is True
+    assert resources.keys == ()