diff --git a/haystack_experimental/components/__init__.py b/haystack_experimental/components/__init__.py index 3d12e900..c1764a6e 100644 --- a/haystack_experimental/components/__init__.py +++ b/haystack_experimental/components/__init__.py @@ -1,27 +1,3 @@ # SPDX-FileCopyrightText: 2022-present deepset GmbH # # SPDX-License-Identifier: Apache-2.0 - - -from .extractors import LLMMetadataExtractor -from .generators.anthropic.chat.chat_generator import AnthropicChatGenerator -from .generators.chat import HuggingFaceAPIChatGenerator, OpenAIChatGenerator -from .generators.ollama.chat.chat_generator import OllamaChatGenerator -from .retrievers.auto_merging_retriever import AutoMergingRetriever -from .retrievers.chat_message_retriever import ChatMessageRetriever -from .splitters import HierarchicalDocumentSplitter -from .tools import ToolInvoker -from .writers import ChatMessageWriter - -_all_ = [ - "AutoMergingRetriever", - "ChatMessageWriter", - "ChatMessageRetriever", - "HuggingFaceAPIChatGenerator", - "OllamaChatGenerator", - "OpenAIChatGenerator", - "AnthropicChatGenerator", - "LLMMetadataExtractor", - "HierarchicalDocumentSplitter", - "ToolInvoker", -] diff --git a/test/components/extractors/test_llm_metadata_extractor.py b/test/components/extractors/test_llm_metadata_extractor.py index ad8238b8..27108523 100644 --- a/test/components/extractors/test_llm_metadata_extractor.py +++ b/test/components/extractors/test_llm_metadata_extractor.py @@ -7,7 +7,7 @@ from haystack.components.builders import PromptBuilder from haystack.components.writers import DocumentWriter from haystack.document_stores.in_memory import InMemoryDocumentStore -from haystack_experimental.components import LLMMetadataExtractor +from haystack_experimental.components.extractors import LLMMetadataExtractor from haystack_experimental.components.extractors import LLMProvider