bstraehle commited on
Commit
adc41aa
·
1 Parent(s): dce66ba

Update rag_llamaindex.py

Browse files
Files changed (1) hide show
  1. rag_llamaindex.py +2 -3
rag_llamaindex.py CHANGED
@@ -106,9 +106,8 @@ class LlamaIndexRAG(BaseRAG):
106
 
107
  ###
108
  prompt_template = LangchainPromptTemplate(
109
- template = os.environ["RAG_TEMPLATE"],
110
- template_var_mappings = {"context": "context",
111
- "question": "question"},
112
  )
113
 
114
  query_engine.update_prompts(
 
106
 
107
  ###
108
  prompt_template = LangchainPromptTemplate(
109
+ os.environ["RAG_TEMPLATE"],
110
+ template_vars = {"context", "question"},
 
111
  )
112
 
113
  query_engine.update_prompts(