chore: Fix e2e tests for other packages

Author: anubhav756

packages/toolbox-langchain/tests/test_e2e.py

@@ -75,13 +75,14 @@ async def test_aload_toolset_specific(
 
     async def test_aload_toolset_all(self, toolbox):
         toolset = await toolbox.aload_toolset()
-        assert len(toolset) == 5
+        assert len(toolset) == 6
         tool_names = [
             "get-n-rows",
             "get-row-by-id",
             "get-row-by-id-auth",
             "get-row-by-email-auth",
             "get-row-by-content-auth",
+            "search-rows",
         ]
         for tool in toolset:
             name = tool._ToolboxTool__core_tool.__name__
@@ -220,13 +221,14 @@ def test_load_toolset_specific(
 
     def test_aload_toolset_all(self, toolbox):
         toolset = toolbox.load_toolset()
-        assert len(toolset) == 5
+        assert len(toolset) == 6
         tool_names = [
             "get-n-rows",
             "get-row-by-id",
             "get-row-by-id-auth",
             "get-row-by-email-auth",
             "get-row-by-content-auth",
+            "search-rows",
         ]
         for tool in toolset:
             name = tool._ToolboxTool__core_tool.__name__

packages/toolbox-llamaindex/tests/test_e2e.py

@@ -75,13 +75,14 @@ async def test_aload_toolset_specific(
 
     async def test_aload_toolset_all(self, toolbox):
         toolset = await toolbox.aload_toolset()
-        assert len(toolset) == 5
+        assert len(toolset) == 6
         tool_names = [
             "get-n-rows",
             "get-row-by-id",
             "get-row-by-id-auth",
             "get-row-by-email-auth",
             "get-row-by-content-auth",
+            "search-rows",
         ]
         for tool in toolset:
             name = tool._ToolboxTool__core_tool.__name__
@@ -220,13 +221,14 @@ def test_load_toolset_specific(
 
     def test_aload_toolset_all(self, toolbox):
         toolset = toolbox.load_toolset()
-        assert len(toolset) == 5
+        assert len(toolset) == 6
         tool_names = [
             "get-n-rows",
             "get-row-by-id",
             "get-row-by-id-auth",
             "get-row-by-email-auth",
             "get-row-by-content-auth",
+            "search-rows",
         ]
         for tool in toolset:
             name = tool._ToolboxTool__core_tool.__name__