Update tests

deepset-ai · Dec 20, 2024 · ae1efed · ae1efed
1 parent 877a3e6
commit ae1efed
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/test/components/tools/test_tool_openapi.py b/test/components/tools/test_tool_openapi.py
@@ -7,7 +7,7 @@
 import os
 import pytest
 
-from haystack_experimental.tools.openapi import OpenAPIKwargs, create_tools_from_openapi_spec
+from haystack_experimental.tools.openapi import OpenAPIKwargs, _create_tools_from_openapi_spec
 from haystack_experimental.dataclasses.tool import Tool
 
 
@@ -34,7 +34,7 @@ def test_from_openapi_spec_serperdev(self):
         assert serper_api_key is not None
 
         # Direct kwargs usage
-        tools = create_tools_from_openapi_spec(
+        tools = _create_tools_from_openapi_spec(
             spec="https://bit.ly/serperdev_openapi",
             credentials=serper_api_key
         )
@@ -47,7 +47,7 @@ def test_from_openapi_spec_serperdev(self):
         config = OpenAPIKwargs(
             credentials=serper_api_key
         )
-        tools = create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
+        tools = _create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
         assert len(tools) >= 1
         tool = tools[0]
         assert tool.name == "search"
@@ -57,7 +57,7 @@ def test_from_openapi_spec_serperdev(self):
         config = OpenAPIKwargs(**{
             "credentials": serper_api_key
         })
-        tools = create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
+        tools = _create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
         assert len(tools) >= 1
         tool = tools[0]
         assert tool.name == "search"
@@ -83,7 +83,7 @@ def test_from_openapi_spec_serperdev_with_allowed_operations(self):
         assert serper_api_key is not None
 
         # Direct kwargs usage
-        tools = create_tools_from_openapi_spec(
+        tools = _create_tools_from_openapi_spec(
             spec="https://bit.ly/serperdev_openapi",
             credentials=serper_api_key,
             allowed_operations=["search"]
@@ -98,7 +98,7 @@ def test_from_openapi_spec_serperdev_with_allowed_operations(self):
             credentials=serper_api_key,
             allowed_operations=["search"]
         )
-        tools = create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
+        tools = _create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
         assert len(tools) >= 1
         tool = tools[0]
         assert tool.name == "search"
@@ -109,7 +109,7 @@ def test_from_openapi_spec_serperdev_with_allowed_operations(self):
             "credentials": serper_api_key,
             "allowed_operations": ["search"]
         })
-        tools = create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
+        tools = _create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
         assert len(tools) >= 1
         tool = tools[0]
         assert tool.name == "search"
@@ -120,7 +120,7 @@ def test_from_openapi_spec_serperdev_with_allowed_operations(self):
             credentials=serper_api_key,
             allowed_operations=["super-search"]
         )
-        tools = create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
+        tools = _create_tools_from_openapi_spec(spec="https://bit.ly/serperdev_openapi", **config)
         assert len(tools) == 0
 
     @pytest.mark.skipif(