bstraehle commited on
Commit
202d100
·
1 Parent(s): 8ded9c8

Update rag_llamaindex.py

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