Skip to content

Commit 175ab2b

Browse files
Merge pull request #295 from deepset-ai/sentence-transformers-update
update notebooks for sentence transformers new package
2 parents 50cf66a + 0ec8228 commit 175ab2b

20 files changed

Lines changed: 2550 additions & 685 deletions

notebooks/agent-breakpoints.ipynb

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,7 @@
2727
"%%bash\n",
2828
"\n",
2929
"pip install \"haystack-ai>=2.16.1\"\n",
30-
"pip install \"transformers[torch,sentencepiece]\"\n",
31-
"pip install \"sentence-transformers>=3.0.0\""
30+
"pip install \"transformers[torch,sentencepiece]\""
3231
]
3332
},
3433
{

notebooks/astradb_haystack_integration.ipynb

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
},
5757
"outputs": [],
5858
"source": [
59-
"!pip install astra-haystack sentence-transformers"
59+
"!pip install astra-haystack sentence-transformers-haystack"
6060
]
6161
},
6262
{
@@ -157,7 +157,7 @@
157157
"\n",
158158
"from haystack import Document, Pipeline\n",
159159
"\n",
160-
"from haystack.components.embedders import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder\n",
160+
"from haystack_integrations.components.embedders.sentence_transformers import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder\n",
161161
"from haystack.components.writers import DocumentWriter\n",
162162
"from haystack.document_stores.types import DuplicatePolicy\n",
163163
"\n",
@@ -291,7 +291,6 @@
291291
"rag_pipeline.connect(\"retriever\", \"prompt_builder.documents\")\n",
292292
"rag_pipeline.connect(\"prompt_builder\", \"llm\")\n",
293293
"rag_pipeline.connect(\"llm.replies\", \"answer_builder.replies\")\n",
294-
"rag_pipeline.connect(\"llm.meta\", \"answer_builder.meta\")\n",
295294
"rag_pipeline.connect(\"retriever\", \"answer_builder.documents\")\n",
296295
"\n",
297296
"\n",

notebooks/async_pipeline.ipynb

Lines changed: 193 additions & 51 deletions
Large diffs are not rendered by default.

notebooks/evaluating_ai_with_haystack.ipynb

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
},
3434
"outputs": [],
3535
"source": [
36-
"!pip install haystack-ai \"sentence-transformers>=3.0.0\" pypdf \"flow-judge[hf]\""
36+
"!pip install haystack-ai sentence-transformers-haystack pypdf \"flow-judge[hf]\""
3737
]
3838
},
3939
{
@@ -255,7 +255,7 @@
255255
"from haystack import Pipeline\n",
256256
"from haystack.document_stores.in_memory import InMemoryDocumentStore\n",
257257
"from haystack.components.converters import PyPDFToDocument\n",
258-
"from haystack.components.embedders import SentenceTransformersDocumentEmbedder\n",
258+
"from haystack_integrations.components.embedders.sentence_transformers import SentenceTransformersDocumentEmbedder\n",
259259
"from haystack.components.preprocessors import DocumentCleaner, DocumentSplitter\n",
260260
"from haystack.components.writers import DocumentWriter\n",
261261
"from haystack.document_stores.types import DuplicatePolicy\n",
@@ -269,7 +269,7 @@
269269
"pipeline.add_component(\"cleaner\", DocumentCleaner())\n",
270270
"pipeline.add_component(\"splitter\", DocumentSplitter(split_length=250, split_by=\"word\")) # default splitting by word\n",
271271
"pipeline.add_component(\"writer\", DocumentWriter(document_store=document_store, policy=DuplicatePolicy.SKIP))\n",
272-
"pipeline.add_component(\"embedder\", SentenceTransformersDocumentEmbedder(embedding_model))\n",
272+
"pipeline.add_component(\"embedder\", SentenceTransformersDocumentEmbedder(model=embedding_model))\n",
273273
"pipeline.connect(\"converter\", \"cleaner\")\n",
274274
"pipeline.connect(\"cleaner\", \"splitter\")\n",
275275
"pipeline.connect(\"splitter\", \"embedder\")\n",
@@ -363,7 +363,7 @@
363363
"source": [
364364
"from haystack import Pipeline\n",
365365
"from haystack.components.builders import ChatPromptBuilder\n",
366-
"from haystack.components.embedders import SentenceTransformersTextEmbedder\n",
366+
"from haystack_integrations.components.embedders.sentence_transformers import SentenceTransformersTextEmbedder\n",
367367
"from haystack.components.generators.chat import OpenAIChatGenerator\n",
368368
"from haystack.components.retrievers import InMemoryEmbeddingRetriever\n",
369369
"from haystack.dataclasses import ChatMessage\n",

notebooks/extracting_metadata_filters_from_a_user_query.ipynb

Lines changed: 64 additions & 238 deletions
Large diffs are not rendered by default.

notebooks/hybrid_rag_pipeline_with_breakpoints.ipynb

Lines changed: 85 additions & 54 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)