bgregorutti commited on
Commit
1b91870
·
1 Parent(s): 56f7ede

Update sorbobotapp/conversation_retrieval_chain.py

Browse files

Just for testing: remove the "one document" condition

sorbobotapp/conversation_retrieval_chain.py CHANGED
@@ -17,8 +17,8 @@ class CustomConversationalRetrievalChain(ConversationalRetrievalChain):
17
  def _handle_docs(self, docs):
18
  if len(docs) == 0:
19
  return False, "No documents found. Can you rephrase ?"
20
- elif len(docs) == 1:
21
- return False, "Only one document found. Can you rephrase ?"
22
  elif len(docs) > 10:
23
  return False, "Too many documents found. Can you specify your request ?"
24
  return True, ""
 
17
  def _handle_docs(self, docs):
18
  if len(docs) == 0:
19
  return False, "No documents found. Can you rephrase ?"
20
+ #elif len(docs) == 1:
21
+ # return False, "Only one document found. Can you rephrase ?"
22
  elif len(docs) > 10:
23
  return False, "Too many documents found. Can you specify your request ?"
24
  return True, ""