bstraehle commited on
Commit
31b16dc
·
1 Parent(s): cd34af5

Update rag_llamaindex.py

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