alexkueck commited on
Commit
04ceb17
1 Parent(s): 873efb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -215,8 +215,8 @@ def llm_chain(prompt):
215
  #prompt mit RAG!!!
216
  def rag_chain(prompt, db):
217
  rag_template = "Nutze die folgenden Kontext Teile am Ende, um die Frage zu beantworten . " + template + "Frage: " + prompt + "Kontext Teile: "
218
- retriever = db.as_retriever(search_kwargs={"k": 3})
219
- retrieved_chunks = retriever(prompt)
220
  neu_prompt = rag_template
221
  for i, chunk in enumerate(retrieved_chunks):
222
  neu_prompt += f"{i+1}. {chunk}\n"
 
215
  #prompt mit RAG!!!
216
  def rag_chain(prompt, db):
217
  rag_template = "Nutze die folgenden Kontext Teile am Ende, um die Frage zu beantworten . " + template + "Frage: " + prompt + "Kontext Teile: "
218
+ retriever = db.as_retriever()
219
+ retrieved_chunks = retriever.search(prompt, k=3)
220
  neu_prompt = rag_template
221
  for i, chunk in enumerate(retrieved_chunks):
222
  neu_prompt += f"{i+1}. {chunk}\n"