tonyliu404 commited on
Commit
eb68218
·
verified ·
1 Parent(s): aea6906

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -99,7 +99,7 @@ def vector_search(user_query, collection):
99
  def mongo_retriever(query):
100
  print("mongo retriever query: ", query)
101
  documents = vector_search(query, mongo_collection)
102
- print("DOCuMENTS RETRIEVED: ", documents)
103
  return documents
104
 
105
 
@@ -171,9 +171,8 @@ rag_chain = (
171
  )
172
 
173
  def get_response(query):
174
- temp = rag_chain.invoke(query)
175
- print("RAG RESPONSE TEMP: ", temp)
176
- return temp
177
 
178
 
179
  ##############################################
 
99
  def mongo_retriever(query):
100
  print("mongo retriever query: ", query)
101
  documents = vector_search(query, mongo_collection)
102
+ print("Documents Retrieved: ", documents)
103
  return documents
104
 
105
 
 
171
  )
172
 
173
  def get_response(query):
174
+ print("get_response query: ", query)
175
+ return rag_chain.invoke(query)
 
176
 
177
 
178
  ##############################################