umaiku commited on
Commit
ae1f860
·
verified ·
1 Parent(s): f0a1be9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -42,14 +42,14 @@ def respond(
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})
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]: