Fix: import and lint errors

Author: VirajAgarwal1

examples/query/rag_pipeline.py

@@ -49,16 +49,14 @@
 # interacting with LLMs using a custom prompt.
 prompt_messages = [
     ChatMessage.from_system("You are a helpful assistant that answers questions based on the provided documents."),
-    ChatMessage.from_user(
-        """Given these documents, answer the question.
+    ChatMessage.from_user("""Given these documents, answer the question.
 Documents:
 {% for doc in documents %}
     {{ doc.content }}
 {% endfor %}
 
 Question: {{question}}
-Answer:"""
-    ),
+Answer:"""),
 ]
 rag_pipeline = Pipeline()
 rag_pipeline.add_component(

examples/search/rag_pipeline.py

@@ -37,16 +37,14 @@
 # interacting with LLMs using a custom prompt.
 prompt_messages = [
     ChatMessage.from_system("You are a helpful assistant that answers questions based on the provided documents."),
-    ChatMessage.from_user(
-        """Given these documents, answer the question.
+    ChatMessage.from_user("""Given these documents, answer the question.
 Documents:
 {% for doc in documents %}
     {{ doc.content }}
 {% endfor %}
 
 Question: {{question}}
-Answer:"""
-    ),
+Answer:"""),
 ]
 rag_pipeline = Pipeline()
 rag_pipeline.add_component(

pyproject.toml

@@ -59,6 +59,7 @@ extra-dependencies = [
     "pytest",
     "pandas",
     "sentence-transformers",
+    "sentence-transformers-haystack",
 ]
 
 [tool.hatch.envs.default.scripts]

tests/search_document_store/test_document_store.py

@@ -18,17 +18,16 @@
 from couchbase.auth import PasswordAuthenticator
 from couchbase_haystack import CouchbaseClusterOptions
 from couchbase_haystack import CouchbasePasswordAuthenticator
-from couchbase.management.logic.search_index_logic import SearchIndex
+from couchbase.management.search import SearchIndex
 from couchbase.exceptions import SearchIndexNotFoundException
 from datetime import timedelta
 from sentence_transformers import SentenceTransformer
-from couchbase.management.logic.collections_logic import ScopeSpec, CollectionSpec
+from couchbase.management.collections import ScopeSpec, CollectionSpec
 from couchbase.result import SearchResult
 
 from ..common.common import IS_GLOBAL_LEVEL_INDEX
 from ..common import common
 
-
 model = SentenceTransformer('all-MiniLM-L6-v2')
 
 

tests/search_document_store/test_embedding_retrieval.py

@@ -12,7 +12,7 @@
 from couchbase.cluster import Cluster, ClusterOptions
 from couchbase.options import ClusterOptions, KnownConfigProfiles
 from couchbase.auth import PasswordAuthenticator
-from couchbase.management.logic.search_index_logic import SearchIndex
+from couchbase.management.search import SearchIndex
 from couchbase.exceptions import SearchIndexNotFoundException
 from couchbase import search
 from couchbase_haystack.document_stores.search_filters import NumericRangeQuery

tests/search_document_store/test_retriever.py

@@ -12,8 +12,7 @@
 from haystack import GeneratedAnswer, Pipeline
 from haystack.components.builders.answer_builder import AnswerBuilder
 from haystack.components.builders.prompt_builder import PromptBuilder
-from haystack.components.embedders import SentenceTransformersTextEmbedder
-from haystack.components.generators import HuggingFaceAPIGenerator
+from haystack_integrations.components.embedders.sentence_transformers import SentenceTransformersTextEmbedder
 import couchbase.search as search
 from couchbase.search import SearchQuery
 from tests.common.common import IS_GLOBAL_LEVEL_INDEX