tfidf_retriever: fix missing arg
@@ -58,6 +58,10 @@ class TfidfRetrieverAgent(Agent):
parser.add_argument(
'--retriever-num-retrieved', default=5, type=int,
help='How many docs to retrieve.')
+ parser.add_argument(
+ '--retriever-mode', choices=['keys', 'values'], default='values',
+ help='Whether to retrieve the stored key or the stored value.'
+ )
parser.add_argument(
'--remove-title', type='bool', default=False,
help='Whether to remove the title from the retrieved passage')
