dinhquangson commited on
Commit
9f7a757
1 Parent(s): 0abb3c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -205,7 +205,7 @@ def search(prompt: str):
205
  querying.connect("dense_text_embedder.embedding", "retriever.query_embedding")
206
  querying.connect("retriever", "document_joiner")
207
  querying.connect("document_joiner", "ranker")
208
- querying.connect("ranker", "prompt_builder")
209
  querying.connect("prompt_builder", "llm")
210
  querying.debug=True
211
 
@@ -213,7 +213,7 @@ def search(prompt: str):
213
  {"dense_text_embedder": {"text": prompt},
214
  "sparse_text_embedder": {"text": prompt},
215
  "ranker": {"query": prompt},
216
- "prompt_builder": {"query": prompt},
217
  "llm": {"query": prompt},
218
  }
219
  )
 
205
  querying.connect("dense_text_embedder.embedding", "retriever.query_embedding")
206
  querying.connect("retriever", "document_joiner")
207
  querying.connect("document_joiner", "ranker")
208
+ querying.connect("ranker.documents", "prompt_builder.documents")
209
  querying.connect("prompt_builder", "llm")
210
  querying.debug=True
211
 
 
213
  {"dense_text_embedder": {"text": prompt},
214
  "sparse_text_embedder": {"text": prompt},
215
  "ranker": {"query": prompt},
216
+ "prompt_builder": {"documents": documents, "query": prompt},
217
  "llm": {"query": prompt},
218
  }
219
  )