J266501 commited on
Commit
1214e80
·
verified ·
1 Parent(s): 63c35e4
Files changed (1) hide show
  1. api_service.py +1 -1
api_service.py CHANGED
@@ -132,7 +132,7 @@ async def ask_rag(request: QueryRequest):
132
  })
133
 
134
  sources = [doc.page_content for doc in retrieved_docs]
135
- return QueryResponse(answer=answer)
136
 
137
  except Exception as e:
138
  print(f"Error: {e}")
 
132
  })
133
 
134
  sources = [doc.page_content for doc in retrieved_docs]
135
+ return QueryResponse(answer=answer, source_documents=sources)
136
 
137
  except Exception as e:
138
  print(f"Error: {e}")