Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,19 +4,15 @@ import logging
|
|
| 4 |
import gradio as gr
|
| 5 |
import requests
|
| 6 |
from pinecone import Pinecone, ServerlessSpec
|
| 7 |
-
from haystack.components.embedders import
|
| 8 |
from haystack.components.writers import DocumentWriter
|
| 9 |
from haystack.document_stores.pinecone import PineconeDocumentStore
|
| 10 |
-
from haystack.components.retrievers.in_memory import InMemoryEmbeddingRetriever
|
| 11 |
from haystack.components.retrievers.pinecone import PineconeEmbeddingRetriever
|
| 12 |
from haystack import Pipeline
|
| 13 |
-
from haystack.components.routers import DocumentJoiner
|
| 14 |
from haystack.components.generators import OpenAIGenerator
|
| 15 |
from haystack.components.builders import PromptBuilder
|
| 16 |
from haystack.components.converters import TextFileToDocument
|
| 17 |
from haystack.components.preprocessors import DocumentSplitter
|
| 18 |
-
from haystack.components.routers import FileTypeRouter
|
| 19 |
-
from haystack.document_stores.in_memory import InMemoryDocumentStore
|
| 20 |
|
| 21 |
# --- Logging ---
|
| 22 |
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
|
@@ -60,7 +56,7 @@ document_store = PineconeDocumentStore(api_key=api_key, index=index_name)
|
|
| 60 |
indexing_pipeline = Pipeline()
|
| 61 |
indexing_pipeline.add_component("converter", TextFileToDocument())
|
| 62 |
indexing_pipeline.add_component("splitter", DocumentSplitter(split_by="word", split_length=100))
|
| 63 |
-
indexing_pipeline.add_component("embedder",
|
| 64 |
indexing_pipeline.add_component("writer", DocumentWriter(document_store))
|
| 65 |
|
| 66 |
indexing_pipeline.connect("converter.documents", "splitter.documents")
|
|
@@ -85,7 +81,7 @@ Context:
|
|
| 85 |
Question: {{ query }}
|
| 86 |
"""
|
| 87 |
query_pipeline = Pipeline()
|
| 88 |
-
query_pipeline.add_component("embedder",
|
| 89 |
query_pipeline.add_component("retriever", PineconeEmbeddingRetriever(document_store=document_store))
|
| 90 |
query_pipeline.add_component("prompt_builder", PromptBuilder(template=template))
|
| 91 |
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 # Corrected imports
|
| 8 |
from haystack.components.writers import DocumentWriter
|
| 9 |
from haystack.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
|
| 13 |
from haystack.components.builders import PromptBuilder
|
| 14 |
from haystack.components.converters import TextFileToDocument
|
| 15 |
from haystack.components.preprocessors import DocumentSplitter
|
|
|
|
|
|
|
| 16 |
|
| 17 |
# --- Logging ---
|
| 18 |
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
|
|
|
|
| 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()) # Corrected component name
|
| 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()) # Corrected component name
|
| 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))
|