Spaces:
Running
Running
Commit
·
ae9bdd2
1
Parent(s):
45de2ca
fix: added haystack-ai-integrations in requirements
Browse files- pipelines.py +4 -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 |
-
#
|
| 11 |
-
|
| 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
|