Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,9 +4,9 @@ import logging
|
|
| 4 |
import gradio as gr
|
| 5 |
import requests
|
| 6 |
from pinecone import Pinecone, ServerlessSpec
|
| 7 |
-
from haystack.components.embedders import OpenAIDocumentEmbedder, OpenAITextEmbedder
|
| 8 |
from haystack.components.writers import DocumentWriter
|
| 9 |
-
from
|
| 10 |
from haystack.components.retrievers.pinecone import PineconeEmbeddingRetriever
|
| 11 |
from haystack import Pipeline
|
| 12 |
from haystack.components.generators import OpenAIGenerator
|
|
@@ -56,7 +56,7 @@ document_store = PineconeDocumentStore(api_key=api_key, index=index_name)
|
|
| 56 |
indexing_pipeline = Pipeline()
|
| 57 |
indexing_pipeline.add_component("converter", TextFileToDocument())
|
| 58 |
indexing_pipeline.add_component("splitter", DocumentSplitter(split_by="word", split_length=100))
|
| 59 |
-
indexing_pipeline.add_component("embedder", OpenAIDocumentEmbedder())
|
| 60 |
indexing_pipeline.add_component("writer", DocumentWriter(document_store))
|
| 61 |
|
| 62 |
indexing_pipeline.connect("converter.documents", "splitter.documents")
|
|
@@ -81,7 +81,7 @@ Context:
|
|
| 81 |
Question: {{ query }}
|
| 82 |
"""
|
| 83 |
query_pipeline = Pipeline()
|
| 84 |
-
query_pipeline.add_component("embedder", OpenAITextEmbedder())
|
| 85 |
query_pipeline.add_component("retriever", PineconeEmbeddingRetriever(document_store=document_store))
|
| 86 |
query_pipeline.add_component("prompt_builder", PromptBuilder(template=template))
|
| 87 |
query_pipeline.add_component("llm", OpenAIGenerator(api_key=openai_api_key))
|
|
|
|
| 4 |
import gradio as gr
|
| 5 |
import requests
|
| 6 |
from pinecone import Pinecone, ServerlessSpec
|
| 7 |
+
from haystack.components.embedders import OpenAIDocumentEmbedder, OpenAITextEmbedder
|
| 8 |
from haystack.components.writers import DocumentWriter
|
| 9 |
+
from haystack_integrations.document_stores.pinecone import PineconeDocumentStore
|
| 10 |
from haystack.components.retrievers.pinecone import PineconeEmbeddingRetriever
|
| 11 |
from haystack import Pipeline
|
| 12 |
from haystack.components.generators import OpenAIGenerator
|
|
|
|
| 56 |
indexing_pipeline = Pipeline()
|
| 57 |
indexing_pipeline.add_component("converter", TextFileToDocument())
|
| 58 |
indexing_pipeline.add_component("splitter", DocumentSplitter(split_by="word", split_length=100))
|
| 59 |
+
indexing_pipeline.add_component("embedder", OpenAIDocumentEmbedder())
|
| 60 |
indexing_pipeline.add_component("writer", DocumentWriter(document_store))
|
| 61 |
|
| 62 |
indexing_pipeline.connect("converter.documents", "splitter.documents")
|
|
|
|
| 81 |
Question: {{ query }}
|
| 82 |
"""
|
| 83 |
query_pipeline = Pipeline()
|
| 84 |
+
query_pipeline.add_component("embedder", OpenAITextEmbedder())
|
| 85 |
query_pipeline.add_component("retriever", PineconeEmbeddingRetriever(document_store=document_store))
|
| 86 |
query_pipeline.add_component("prompt_builder", PromptBuilder(template=template))
|
| 87 |
query_pipeline.add_component("llm", OpenAIGenerator(api_key=openai_api_key))
|