umaiku commited on
Commit
7e5bae2
·
verified ·
1 Parent(s): b70a745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -30,12 +30,20 @@ def respond(
30
  messages = [{"role": "system", "content": system_message}]
31
 
32
  retriever = vector_db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": score})
33
- document = retriever.invoke(message)
34
 
 
35
  if document == []:
36
  message = message + "\nNo cases were found about this subject"
37
  else:
38
- message = message + "\nUse the following jurisprudence case to answer " + document[0].page_content + "\n Give the following url " + document[0].metadata["case_url"]
 
 
 
 
 
 
 
39
 
40
  print(message)
41
 
 
30
  messages = [{"role": "system", "content": system_message}]
31
 
32
  retriever = vector_db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": score})
33
+ documents = retriever.invoke(message)
34
 
35
+ """
36
  if document == []:
37
  message = message + "\nNo cases were found about this subject"
38
  else:
39
+ message = message + "\nUse the following jurisprudence case to answer " + documents[0].page_content + "\n Give the following url " + documents[0].metadata["case_url"]
40
+ """
41
+
42
+ message = f"""
43
+ The user is asking for information about the following: {message}.
44
+ Answer him in his own language using the information from the following link and mention this link: {" \n ".join(doc.metadata["case_url"] for doc in documents)}
45
+
46
+ """
47
 
48
  print(message)
49