HarshKalia-24 commited on
Commit
ae9bdd2
·
1 Parent(s): 45de2ca

fix: added haystack-ai-integrations in requirements

Browse files
Files changed (2) hide show
  1. pipelines.py +4 -2
  2. requirements.txt +4 -5
pipelines.py CHANGED
@@ -5,7 +5,8 @@ from haystack.dataclasses import Document
5
  from haystack.document_stores.in_memory import InMemoryDocumentStore
6
  from haystack.components.embedders import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder
7
  from haystack.components.retrievers.in_memory import InMemoryEmbeddingRetriever
8
- from haystack.components.rankers import SentenceTransformersSimilarityRanker
 
9
  from haystack_integrations.components.generators.google_ai import GoogleAIGeminiGenerator
10
  from haystack.components.preprocessors import DocumentSplitter
11
 
@@ -18,7 +19,8 @@ document_store = InMemoryDocumentStore()
18
  doc_embedder = SentenceTransformersDocumentEmbedder(model="BAAI/bge-large-en-v1.5")
19
  text_embedder = SentenceTransformersTextEmbedder(model="BAAI/bge-large-en-v1.5")
20
  retriever = InMemoryEmbeddingRetriever(document_store=document_store, top_k=5)
21
- reranker = SentenceTransformersSimilarityRanker(model="cross-encoder/ms-marco-MiniLM-L-6-v2")
 
22
 
23
  # Initialize generator
24
  generator = GoogleAIGeminiGenerator(
 
5
  from haystack.document_stores.in_memory import InMemoryDocumentStore
6
  from haystack.components.embedders import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder
7
  from haystack.components.retrievers.in_memory import InMemoryEmbeddingRetriever
8
+ # from haystack.components.rankers import SentenceTransformersSimilarityRanker
9
+ from haystack_integrations.components.rankers.sentence_transformers import SentenceTransformersRanker
10
  from haystack_integrations.components.generators.google_ai import GoogleAIGeminiGenerator
11
  from haystack.components.preprocessors import DocumentSplitter
12
 
 
19
  doc_embedder = SentenceTransformersDocumentEmbedder(model="BAAI/bge-large-en-v1.5")
20
  text_embedder = SentenceTransformersTextEmbedder(model="BAAI/bge-large-en-v1.5")
21
  retriever = InMemoryEmbeddingRetriever(document_store=document_store, top_k=5)
22
+ # reranker = SentenceTransformersSimilarityRanker(model="cross-encoder/ms-marco-MiniLM-L-6-v2")
23
+ ranker = SentenceTransformersRanker(model="sentence-transformers/all-MiniLM-L6-v2")
24
 
25
  # Initialize generator
26
  generator = GoogleAIGeminiGenerator(
requirements.txt CHANGED
@@ -5,13 +5,12 @@ python-multipart==0.0.9
5
  pillow==10.3.0
6
  pdfplumber==0.11.0
7
  pytesseract==0.3.10
8
- sentence-transformers==3.0.1
9
 
10
- # Google Generative AI
11
- google-generativeai==0.7.2
12
 
13
- # Haystack v2 core + integrations
14
- haystack-ai==2.1.0
15
 
16
  # gRPC (needed by Google / Haystack)
17
  grpcio
 
5
  pillow==10.3.0
6
  pdfplumber==0.11.0
7
  pytesseract==0.3.10
 
8
 
9
+ # Sentence Transformers (explicit, though Haystack pulls it too)
10
+ sentence-transformers==3.0.1
11
 
12
+ # Haystack v2 core + integrations (sentence-transformers + google-ai)
13
+ haystack-ai[sentence-transformers,google]==2.1.0
14
 
15
  # gRPC (needed by Google / Haystack)
16
  grpcio