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---")
|