bstraehle commited on
Commit
6b51abc
1 Parent(s): 3c79cf0

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +1 -4
rag_llamaindex.py CHANGED
@@ -121,11 +121,8 @@ class LlamaIndexRAG(BaseRAG):
121
  service_context = self.get_service_context(config)
122
 
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
 
 
121
  service_context = self.get_service_context(config)
122
 
123
  query_engine = index.as_query_engine(
 
124
  service_context = service_context,
125
+ similarity_top_k = config["k"],
 
 
126
  text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"])
127
  )
128