Update rag_llamaindex.py
Browse files- rag_llamaindex.py +1 -10
rag_llamaindex.py
CHANGED
@@ -100,18 +100,9 @@ class LlamaIndexRAG(BaseRAG):
|
|
100 |
)
|
101 |
|
102 |
query_engine = index.as_query_engine(
|
|
|
103 |
service_context = self.get_service_context(config),
|
104 |
similarity_top_k = config["k"]
|
105 |
)
|
106 |
|
107 |
-
###
|
108 |
-
prompt_template = PromptTemplate(
|
109 |
-
os.environ["RAG_TEMPLATE"]
|
110 |
-
)
|
111 |
-
|
112 |
-
query_engine.update_prompts(
|
113 |
-
{"response_synthesizer:text_qa_template": prompt_template}
|
114 |
-
)
|
115 |
-
###
|
116 |
-
|
117 |
return query_engine.query(prompt)
|
|
|
100 |
)
|
101 |
|
102 |
query_engine = index.as_query_engine(
|
103 |
+
text_qa_template = PromptTemplate(os.environ["RAG_TEMPLATE"]),
|
104 |
service_context = self.get_service_context(config),
|
105 |
similarity_top_k = config["k"]
|
106 |
)
|
107 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
108 |
return query_engine.query(prompt)
|