Spaces:

bstraehle
/

rag

Running

bstraehle commited on Jan 6

Commit

f6fce88

•

1 Parent(s): d53a430

Update rag_llamaindex.py

Files changed (1) hide show

rag_llamaindex.py CHANGED Viewed

@@ -100,7 +100,7 @@ class LlamaIndexRAG(BaseRAG):
         )
         query_engine = index.as_query_engine(
-            text_qa_template = PromptTemplate(os.environ["PROMPT_TEMPLATE"]),
             service_context = self.get_service_context(config),
             similarity_top_k = config["k"]
         )

         )
         query_engine = index.as_query_engine(
+            text_qa_template = PromptTemplate(os.environ["LLAMAINDEX_TEMPLATE"]),
             service_context = self.get_service_context(config),
             similarity_top_k = config["k"]
         )