alexkueck commited on
Commit
5418168
1 Parent(s): a3aa151

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -315,7 +315,7 @@ def grade_documents_direct(prompt, documents):
315
  parser_tool = PydanticToolsParser(tools=[grade])
316
 
317
  # Prompt
318
- prompt = 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,12 +325,12 @@ def grade_documents_direct(prompt, documents):
325
  )
326
 
327
  # Chain
328
- chain = prompt | llm_with_tool | parser_tool
329
 
330
  # Score
331
  filtered_docs = []
332
  for d in documents:
333
- score = chain.invoke({"question": question, "context": d.page_content})
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---")