bstraehle commited on
Commit
056f437
·
1 Parent(s): ab0af2e

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +2 -1
rag.py CHANGED
@@ -113,7 +113,8 @@ def run_rag_chain(config, openai_api_key, rag_option, prompt):
113
  db = retrieve_documents_mongodb()
114
 
115
  rag_chain = RetrievalQA.from_chain_type(llm,
116
- chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT},
 
117
  retriever = db.as_retriever(search_kwargs = {"k": config["k"]}),
118
  return_source_documents = True)
119
 
 
113
  db = retrieve_documents_mongodb()
114
 
115
  rag_chain = RetrievalQA.from_chain_type(llm,
116
+ chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT,
117
+ "verbose": True},
118
  retriever = db.as_retriever(search_kwargs = {"k": config["k"]}),
119
  return_source_documents = True)
120