Shafaq25 commited on
Commit
a14ce03
·
verified ·
1 Parent(s): b328b6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 # 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
@@ -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()) # Corrected component name
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()) # 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))
 
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))