alexkueck commited on
Commit
304335a
1 Parent(s): d97945c

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -1
utils.py CHANGED
@@ -385,9 +385,11 @@ def rag_chain(llm, prompt, retriever):
385
 
386
  neu_prompt=prompt
387
  if (len(filtered_docs)<2): #frage neu formulieren
 
388
  neu_prompt = transform_query_direct(prompt)
389
  relevant_docs = retriever.get_relevant_documents(neu_prompt)
390
- filtered_docs = grade_documents_direct(relevant_docs)
 
391
 
392
  if (len(filtered_docs)>0):
393
  llm_chain = LLMChain(llm = llm, prompt = RAG_CHAIN_PROMPT)
 
385
 
386
  neu_prompt=prompt
387
  if (len(filtered_docs)<2): #frage neu formulieren
388
+ relevant_docs=[]
389
  neu_prompt = transform_query_direct(prompt)
390
  relevant_docs = retriever.get_relevant_documents(neu_prompt)
391
+ if (len(relevant_docs)>0):
392
+ filtered_docs = grade_documents_direct(relevant_docs)
393
 
394
  if (len(filtered_docs)>0):
395
  llm_chain = LLMChain(llm = llm, prompt = RAG_CHAIN_PROMPT)