Update rag_llamaindex.py
Browse files- 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 |
-
|
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)
|