mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-16 05:20:51 +00:00
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
|
from haystack.preview import Document
|
||
|
|
from haystack.preview.components.retrievers import InMemoryBM25Retriever
|
||
|
|
from haystack.preview.document_stores import InMemoryDocumentStore
|
||
|
|
from haystack.preview.pipeline import Pipeline
|
||
|
|
|
||
|
|
# Create components and a query pipeline
|
||
|
|
document_store = InMemoryDocumentStore()
|
||
|
|
retriever = InMemoryBM25Retriever(document_store=document_store)
|
||
|
|
|
||
|
|
pipeline = Pipeline()
|
||
|
|
pipeline.add_component(instance=retriever, name="retriever")
|
||
|
|
|
||
|
|
# Add Documents
|
||
|
|
documents = [
|
||
|
|
Document(content="There are over 7,000 languages spoken around the world today."),
|
||
|
|
Document(
|
||
|
|
content="Elephants have been observed to behave in a way that indicates a high level of self-awareness, such as recognizing themselves in mirrors."
|
||
|
|
),
|
||
|
|
Document(
|
||
|
|
content="In certain parts of the world, like the Maldives, Puerto Rico, and San Diego, you can witness the phenomenon of bioluminescent waves."
|
||
|
|
),
|
||
|
|
]
|
||
|
|
document_store.write_documents(documents)
|
||
|
|
|
||
|
|
# Run the pipeline
|
||
|
|
result = pipeline.run(data={"retriever": {"query": "How many languages are there?"}})
|
||
|
|
|
||
|
|
print(result["retriever"]["documents"][0])
|