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