Update utils.py
Browse files
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 |
-
|
|
|
|
| 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
|