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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -7
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 OpenAIAPIDocumentEmbedder, OpenAIAPIQueryEmbedder
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", OpenAIAPIDocumentEmbedder())
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", OpenAIAPIQueryEmbedder())
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))