Shafaq25 commited on
Commit
3137489
·
verified ·
1 Parent(s): bb772b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 # Import the Secret class
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) # Use Secret object
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"))) # Use Secret object
89
 
90
- query_pipeline.connect("embedder.query", "retriever.query")
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