Update app.py
Browse files
app.py
CHANGED
|
@@ -42,14 +42,14 @@ def respond(
|
|
| 42 |
print(datetime.now())
|
| 43 |
print(system_message)
|
| 44 |
|
| 45 |
-
|
| 46 |
-
retriever = vector_db.as_retriever(search_type="mmr")
|
| 47 |
documents = retriever.invoke(message)
|
| 48 |
|
| 49 |
spacer = " \n"
|
| 50 |
context = ""
|
| 51 |
|
| 52 |
-
print(message)
|
| 53 |
print(len(documents))
|
| 54 |
|
| 55 |
for doc in documents:
|
|
@@ -63,8 +63,8 @@ def respond(
|
|
| 63 |
context += "# Case text: " + doc.page_content + spacer
|
| 64 |
#context += "Case text: " + case_text[:8000] + spacer
|
| 65 |
|
| 66 |
-
print("# Case number: " + doc.metadata["case_nb"] + spacer)
|
| 67 |
-
print("# Case url: " + doc.metadata["case_url"] + spacer)
|
| 68 |
|
| 69 |
|
| 70 |
message = f"""
|
|
@@ -80,6 +80,8 @@ Otherwise, if relevant cases were found, start by summarizing these cases in the
|
|
| 80 |
# Instructions:
|
| 81 |
Always answer the user using the language used in his question: {message}
|
| 82 |
"""
|
|
|
|
|
|
|
| 83 |
|
| 84 |
# for val in history:
|
| 85 |
# if val[0]:
|
|
|
|
| 42 |
print(datetime.now())
|
| 43 |
print(system_message)
|
| 44 |
|
| 45 |
+
retriever = vector_db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": score, "k": 10})
|
| 46 |
+
# retriever = vector_db.as_retriever(search_type="mmr")
|
| 47 |
documents = retriever.invoke(message)
|
| 48 |
|
| 49 |
spacer = " \n"
|
| 50 |
context = ""
|
| 51 |
|
| 52 |
+
#print(message)
|
| 53 |
print(len(documents))
|
| 54 |
|
| 55 |
for doc in documents:
|
|
|
|
| 63 |
context += "# Case text: " + doc.page_content + spacer
|
| 64 |
#context += "Case text: " + case_text[:8000] + spacer
|
| 65 |
|
| 66 |
+
#print("# Case number: " + doc.metadata["case_nb"] + spacer)
|
| 67 |
+
#print("# Case url: " + doc.metadata["case_url"] + spacer)
|
| 68 |
|
| 69 |
|
| 70 |
message = f"""
|
|
|
|
| 80 |
# Instructions:
|
| 81 |
Always answer the user using the language used in his question: {message}
|
| 82 |
"""
|
| 83 |
+
|
| 84 |
+
print(message)
|
| 85 |
|
| 86 |
# for val in history:
|
| 87 |
# if val[0]:
|