alexkueck commited on
Commit
d97945c
1 Parent(s): 953a275

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +4 -1
utils.py CHANGED
@@ -377,8 +377,11 @@ def llm_chain2(llm, prompt):
377
  #langchain nutzen, um prompt an llm zu leiten, aber vorher in der VektorDB suchen, um passende splits zum Prompt hinzuzufügen
378
  def rag_chain(llm, prompt, retriever):
379
  #Langgraph nutzen für ein wenig mehr Intelligenz beim Dokumente suchen
 
 
380
  relevant_docs = retriever.get_relevant_documents(prompt)
381
- filtered_docs = grade_documents_direct(prompt, relevant_docs)
 
382
 
383
  neu_prompt=prompt
384
  if (len(filtered_docs)<2): #frage neu formulieren
 
377
  #langchain nutzen, um prompt an llm zu leiten, aber vorher in der VektorDB suchen, um passende splits zum Prompt hinzuzufügen
378
  def rag_chain(llm, prompt, retriever):
379
  #Langgraph nutzen für ein wenig mehr Intelligenz beim Dokumente suchen
380
+ relevant_docs=[]
381
+ filtered_docs=[]
382
  relevant_docs = retriever.get_relevant_documents(prompt)
383
+ if (len(relevant_docs)>0):
384
+ filtered_docs = grade_documents_direct(prompt, relevant_docs)
385
 
386
  neu_prompt=prompt
387
  if (len(filtered_docs)<2): #frage neu formulieren