bstraehle commited on
Commit
3c79cf0
1 Parent(s): 7eb4985

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +3 -1
rag_llamaindex.py CHANGED
@@ -123,7 +123,9 @@ class LlamaIndexRAG(BaseRAG):
123
  query_engine = index.as_query_engine(
124
  response_mode = "compact",
125
  service_context = service_context,
126
- similarity_top_k = config["k"],
 
 
127
  text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"])
128
  )
129
 
 
123
  query_engine = index.as_query_engine(
124
  response_mode = "compact",
125
  service_context = service_context,
126
+ #similarity_top_k = config["k"],
127
+ max_top_k = config["k"],
128
+ top_k = config["k"],
129
  text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"])
130
  )
131