Expose LLMProvider

deepset-ai · Jun 6, 2024 · c8a80c1 · c8a80c1
1 parent b6c0d84
commit c8a80c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/haystack_experimental/components/tools/openapi/__init__.py b/haystack_experimental/components/tools/openapi/__init__.py
@@ -2,6 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+from haystack_experimental.components.tools.openapi._openapi import LLMProvider
 from haystack_experimental.components.tools.openapi.openapi_tool import OpenAPITool
 
-__all__ = ["OpenAPITool"]
+__all__ = ["LLMProvider", "OpenAPITool"]
diff --git a/haystack_experimental/components/tools/openapi/openapi_tool.py b/haystack_experimental/components/tools/openapi/openapi_tool.py
@@ -10,10 +10,10 @@
 from haystack.components.generators.chat import OpenAIChatGenerator
 from haystack.dataclasses import ChatMessage, ChatRole
 from haystack.lazy_imports import LazyImport
+from haystack_experimental.components.tools.openapi import LLMProvider
 
 from haystack_experimental.components.tools.openapi._openapi import (
     ClientConfiguration,
-    LLMProvider,
     OpenAPIServiceClient,
 )