alexkueck commited on
Commit
cfb51f6
1 Parent(s): f37a688

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -290,7 +290,7 @@ def grade_documents_direct(prompt, documents):
290
  binary_score: str = Field(description="Relevanz Bewertung 'ja' oder 'nein'")
291
 
292
  # LLM
293
- model = ChatOpenAI(temperature=0.3, model="gpt-3.5-turbo-1106", streaming=True)
294
 
295
  """
296
  xxx = Du bist ein Bewerter, der die Relevanz von einem erhaltenen Dokument zu einer Nutzeranfrage bewerten soll. \n
@@ -331,8 +331,8 @@ def grade_documents_direct(prompt, documents):
331
  # Score
332
  filtered_docs = []
333
  for d in documents:
334
- print(d.page_content)
335
- score = chain.invoke({ "context": d.page_content})
336
  grade = score[0].binary_score
337
  if grade == "ja":
338
  print("---Bewertung: Dokument ist relevant---")
 
290
  binary_score: str = Field(description="Relevanz Bewertung 'ja' oder 'nein'")
291
 
292
  # LLM
293
+ model = ChatOpenAI(temperature=0.3, model="gpt-4-0125-preview", streaming=True)
294
 
295
  """
296
  xxx = Du bist ein Bewerter, der die Relevanz von einem erhaltenen Dokument zu einer Nutzeranfrage bewerten soll. \n
 
331
  # Score
332
  filtered_docs = []
333
  for d in documents:
334
+ #print(d.page_content)
335
+ score = chain.invoke({"question": prompt, "context": d.page_content})
336
  grade = score[0].binary_score
337
  if grade == "ja":
338
  print("---Bewertung: Dokument ist relevant---")