chore: Fix e2e tests

Author: anubhav756

packages/toolbox-core/tests/test_e2e.py

@@ -300,13 +300,7 @@ async def test_run_tool_with_optional_id_provided(self, toolbox: ToolboxClient):
 
         response = await tool(email="[email protected]", id=1)
         assert isinstance(response, str)
-        assert '"email":"[email protected]"' in response
-        assert "row1" in response
-        assert "row2" not in response
-        assert "row3" not in response
-        assert "row4" not in response
-        assert "row5" not in response
-        assert "row6" not in response
+        assert response == "null"
 
     async def test_run_tool_with_optional_id_null(self, toolbox: ToolboxClient):
         """Invoke a tool providing both required and optional parameters."""
@@ -331,7 +325,7 @@ async def test_run_tool_with_missing_required_param(self, toolbox: ToolboxClient
     async def test_run_tool_with_required_param_null(self, toolbox: ToolboxClient):
         """Invoke a tool without its required parameter."""
         tool = await toolbox.load_tool("search-rows")
-        with pytest.raises(TypeError, match="missing a required argument: 'email'"):
+        with pytest.raises(ValidationError, match="missing a required argument: 'email'"):
             await tool(email=None, id=5, data="row5")
 
     async def test_run_tool_with_all_default_params(self, toolbox: ToolboxClient):