Update utils.py
Browse files
utils.py
CHANGED
@@ -315,7 +315,7 @@ def grade_documents_direct(prompt, documents):
|
|
315 |
parser_tool = PydanticToolsParser(tools=[grade])
|
316 |
|
317 |
# Prompt
|
318 |
-
|
319 |
template="""Du bist ein Bewerter, der die Relevanz von einem erhaltenen Dokument zu einer Nutzeranfrage bewerten soll. \n
|
320 |
Hier ist das erhaltene Dokument: \n\n {context} \n\n
|
321 |
Hier ist die Nutzeranfrage: {question} \n
|
@@ -325,12 +325,12 @@ def grade_documents_direct(prompt, documents):
|
|
325 |
)
|
326 |
|
327 |
# Chain
|
328 |
-
chain =
|
329 |
|
330 |
# Score
|
331 |
filtered_docs = []
|
332 |
for d in documents:
|
333 |
-
score = chain.invoke({"question":
|
334 |
grade = score[0].binary_score
|
335 |
if grade == "ja":
|
336 |
print("---Bewertung: Dokument ist relevant---")
|
|
|
315 |
parser_tool = PydanticToolsParser(tools=[grade])
|
316 |
|
317 |
# Prompt
|
318 |
+
prompt_gesamt = PromptTemplate(
|
319 |
template="""Du bist ein Bewerter, der die Relevanz von einem erhaltenen Dokument zu einer Nutzeranfrage bewerten soll. \n
|
320 |
Hier ist das erhaltene Dokument: \n\n {context} \n\n
|
321 |
Hier ist die Nutzeranfrage: {question} \n
|
|
|
325 |
)
|
326 |
|
327 |
# Chain
|
328 |
+
chain = prompt_gesamt | llm_with_tool | parser_tool
|
329 |
|
330 |
# Score
|
331 |
filtered_docs = []
|
332 |
for d in documents:
|
333 |
+
score = chain.invoke({"question": prompt, "context": d.page_content})
|
334 |
grade = score[0].binary_score
|
335 |
if grade == "ja":
|
336 |
print("---Bewertung: Dokument ist relevant---")
|