fix e2e test

deepset-ai · Nov 29, 2023 · 6b3f107 · 6b3f107
1 parent 14581ed
commit 6b3f107
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/e2e/pipelines/test_gradient_rag_pipelines.py b/e2e/pipelines/test_gradient_rag_pipelines.py
@@ -2,13 +2,13 @@
 import json
 import pytest
 
-from haystack.preview import Pipeline, Document
+from haystack import Pipeline, Document
 from haystack.components.embedders.gradient_document_embedder import GradientDocumentEmbedder
 from haystack.components.embedders.gradient_text_embedder import GradientTextEmbedder
-from haystack.preview.document_stores import InMemoryDocumentStore
-from haystack.preview.components.writers import DocumentWriter
-from haystack.preview.components.retrievers import InMemoryEmbeddingRetriever
-from haystack.preview.components.generators.gradient.base import GradientGenerator
+from haystack.document_stores import InMemoryDocumentStore
+from haystack.components.writers import DocumentWriter
+from haystack.components.retrievers import InMemoryEmbeddingRetriever
+from haystack.components.generators.gradient.base import GradientGenerator
 from haystack.components.builders.answer_builder import AnswerBuilder
 from haystack.components.builders.prompt_builder import PromptBuilder
 

diff --git a/haystack/components/generators/__init__.py b/haystack/components/generators/__init__.py
@@ -1,6 +1,7 @@
 from haystack.components.generators.cohere import CohereGenerator
+from haystack.components.generators.gradient.base import GradientGenerator
 from haystack.components.generators.hugging_face_local import HuggingFaceLocalGenerator
 from haystack.components.generators.hugging_face_tgi import HuggingFaceTGIGenerator
 from haystack.components.generators.openai import GPTGenerator
 
-__all__ = ["HuggingFaceLocalGenerator", "HuggingFaceTGIGenerator", "GPTGenerator", "CohereGenerator"]
+__all__ = ["HuggingFaceLocalGenerator", "HuggingFaceTGIGenerator", "GPTGenerator", "CohereGenerator", "GradientGenerator"]
diff --git a/...omponents/generators/gradient/__init__.py → ...omponents/generators/gradient/__init__.py b/...omponents/generators/gradient/__init__.py → ...omponents/generators/gradient/__init__.py
diff --git a/...ew/components/generators/gradient/base.py → ...ck/components/generators/gradient/base.py b/...ew/components/generators/gradient/base.py → ...ck/components/generators/gradient/base.py
@@ -3,7 +3,7 @@
 import logging
 from haystack.lazy_imports import LazyImport
 
-from haystack.preview import component, default_to_dict
+from haystack import component, default_to_dict
 
 with LazyImport(message="Run 'pip install gradientai'") as gradientai_import:
     from gradientai import Gradient