bstraehle commited on
Commit
431e4ec
·
1 Parent(s): 32dedd9

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. 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)