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