must include references when using retreiver

#3
by RCaz - opened
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -154,15 +154,15 @@ def predict(message, history, request: gr.Request):
154
  }
155
  }
156
  )
157
-
158
  source_context = "\nSources:\n" + "\n".join([
159
- f"{doc.metadata.get('source_url')} ({doc.metadata.get('date')})\n---"
160
  for i, doc in enumerate(relevant_docs)])
161
 
162
  print(gpt_response.content )
163
  print(source_context)
164
 
165
- return gpt_response.content + source_context
166
 
167
 
168
  # setup tracking
 
154
  }
155
  }
156
  )
157
+
158
  source_context = "\nSources:\n" + "\n".join([
159
+ f"{doc.metadata["source"].split("/")[-1]} ({doc.metadata.get('date')})\n---"
160
  for i, doc in enumerate(relevant_docs)])
161
 
162
  print(gpt_response.content )
163
  print(source_context)
164
 
165
+ return gpt_response.content + "\n\n" + source_context
166
 
167
 
168
  # setup tracking