umaiku commited on
Commit
443f706
·
verified ·
1 Parent(s): 5b9e4ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -29,6 +29,10 @@ def respond(
29
  ):
30
  messages = [{"role": "system", "content": system_message}]
31
 
 
 
 
 
32
  for val in history:
33
  if val[0]:
34
  messages.append({"role": "user", "content": val[0]})
@@ -37,8 +41,6 @@ def respond(
37
 
38
  messages.append({"role": "user", "content": message})
39
 
40
- print(retriever.invoke(message))
41
-
42
  response = ""
43
 
44
  for message in client.chat_completion(
 
29
  ):
30
  messages = [{"role": "system", "content": system_message}]
31
 
32
+ document = retriever.invoke(message)[0].page_content
33
+
34
+ message = message + " document: " + document
35
+
36
  for val in history:
37
  if val[0]:
38
  messages.append({"role": "user", "content": val[0]})
 
41
 
42
  messages.append({"role": "user", "content": message})
43
 
 
 
44
  response = ""
45
 
46
  for message in client.chat_completion(