Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ from haystack.components.generators import OpenAIGenerator
|
|
| 13 |
from haystack.components.builders import PromptBuilder
|
| 14 |
from haystack.components.converters import TextFileToDocument
|
| 15 |
from haystack.components.preprocessors import DocumentSplitter
|
| 16 |
-
from haystack.utils import Secret
|
| 17 |
|
| 18 |
# --- Logging ---
|
| 19 |
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
|
@@ -52,7 +52,7 @@ if not os.path.exists(file_path):
|
|
| 52 |
f.write(r.text)
|
| 53 |
|
| 54 |
# --- Haystack Pipeline for Indexing ---
|
| 55 |
-
document_store = PineconeDocumentStore(api_key=Secret.from_env_var("PINECONE_API_KEY"), index=index_name)
|
| 56 |
|
| 57 |
indexing_pipeline = Pipeline()
|
| 58 |
indexing_pipeline.add_component("converter", TextFileToDocument())
|
|
@@ -85,9 +85,9 @@ query_pipeline = Pipeline()
|
|
| 85 |
query_pipeline.add_component("embedder", OpenAITextEmbedder())
|
| 86 |
query_pipeline.add_component("retriever", PineconeEmbeddingRetriever(document_store=document_store))
|
| 87 |
query_pipeline.add_component("prompt_builder", PromptBuilder(template=template))
|
| 88 |
-
query_pipeline.add_component("llm", OpenAIGenerator(api_key=Secret.from_env_var("OPENAI_API_KEY")))
|
| 89 |
|
| 90 |
-
query_pipeline.connect("embedder.
|
| 91 |
query_pipeline.connect("retriever.documents", "prompt_builder.documents")
|
| 92 |
query_pipeline.connect("prompt_builder", "llm")
|
| 93 |
|
|
|
|
| 13 |
from haystack.components.builders import PromptBuilder
|
| 14 |
from haystack.components.converters import TextFileToDocument
|
| 15 |
from haystack.components.preprocessors import DocumentSplitter
|
| 16 |
+
from haystack.utils import Secret
|
| 17 |
|
| 18 |
# --- Logging ---
|
| 19 |
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
|
|
|
| 52 |
f.write(r.text)
|
| 53 |
|
| 54 |
# --- Haystack Pipeline for Indexing ---
|
| 55 |
+
document_store = PineconeDocumentStore(api_key=Secret.from_env_var("PINECONE_API_KEY"), index=index_name)
|
| 56 |
|
| 57 |
indexing_pipeline = Pipeline()
|
| 58 |
indexing_pipeline.add_component("converter", TextFileToDocument())
|
|
|
|
| 85 |
query_pipeline.add_component("embedder", OpenAITextEmbedder())
|
| 86 |
query_pipeline.add_component("retriever", PineconeEmbeddingRetriever(document_store=document_store))
|
| 87 |
query_pipeline.add_component("prompt_builder", PromptBuilder(template=template))
|
| 88 |
+
query_pipeline.add_component("llm", OpenAIGenerator(api_key=Secret.from_env_var("OPENAI_API_KEY")))
|
| 89 |
|
| 90 |
+
query_pipeline.connect("embedder.embedding", "retriever.query_embedding") # Corrected connection
|
| 91 |
query_pipeline.connect("retriever.documents", "prompt_builder.documents")
|
| 92 |
query_pipeline.connect("prompt_builder", "llm")
|
| 93 |
|