Merge pull request #269 from hud-evals/invalidate-routing

Invalidate routing on add_tool

Author: lorenss-m

docs/docs.json

@@ -33,7 +33,7 @@
         "icon": "code",
         "versions": [
           {
-            "version": "0.5.5",
+            "version": "0.5.6",
             "groups": [
               {
                 "group": "Get Started",

hud/agents/claude.py

@@ -303,7 +303,7 @@ def to_api_tool(tool: types.Tool) -> BetaToolUnionParam | None:
                         display_width_px=computer_settings.ANTHROPIC_COMPUTER_WIDTH,
                         display_height_px=computer_settings.ANTHROPIC_COMPUTER_HEIGHT,
                     )
-                elif tool.name == "computer":
+                elif tool.name == "computer" or tool.name.endswith("_computer"):
                     logger.warning(
                         "Renamed tool %s to 'computer', dropping original 'computer' tool",
                         selected_computer_tool.name,

hud/agents/gemini_cua.py

@@ -123,6 +123,9 @@ def _to_gemini_tool(self, tool: types.Tool) -> genai_types.Tool | None:
                 )
             )
 
+        if tool.name == "computer" or tool.name.endswith("_computer"):
+            return None
+
         # For non-computer tools, use the parent implementation
         return super()._to_gemini_tool(tool)
 

hud/agents/operator.py

@@ -129,6 +129,8 @@ def _to_openai_tool(
                 display_height=self._operator_display_height,
                 environment=self._operator_environment,
             )
+        if tool.name == "computer" or tool.name.endswith("_computer"):
+            return None
         return super()._to_openai_tool(tool)
 
     def _extract_tool_call(self, item: Any) -> MCPToolCall | None:

hud/environment/environment.py

@@ -182,6 +182,10 @@ def as_tools(self) -> list[mcp_types.Tool]:
 
         return tools
 
+    def add_tool(self, obj: Any, **kwargs: Any) -> None:
+        super().add_tool(obj, **kwargs)
+        self._routing_built = False
+
     async def call_tool(self, call: Any, /, **kwargs: Any) -> Any:
         """Call a tool, auto-detecting format and returning matching result format.
 
@@ -446,6 +450,10 @@ async def _execute_tool(self, name: str, arguments: dict[str, Any]) -> MCPToolRe
             logger.debug("Mock mode: returning mock result for tool %s", name)
             return self._get_mock_result(name, arguments)
 
+        # Rebuild routing if invalidated (e.g., after add_tool)
+        if not self._routing_built:
+            await self._build_routing()
+
         if self._router.is_local(name):
             # Call tool manager directly to avoid FastMCP context requirement
             result = await self._tool_manager.call_tool(name, arguments)

hud/utils/tests/test_version.py

@@ -5,4 +5,4 @@ def test_import():
     """Test that the package can be imported."""
     import hud
 
-    assert hud.__version__ == "0.5.5"
+    assert hud.__version__ == "0.5.6"

hud/version.py

@@ -4,4 +4,4 @@
 
 from __future__ import annotations
 
-__version__ = "0.5.5"
+__version__ = "0.5.6"

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hud-python"
-version = "0.5.5"
+version = "0.5.6"
 description = "SDK for the HUD platform."
 readme = "README.md"
 requires-python = ">=3.11, <3.13"