bstraehle commited on
Commit
5e724ee
1 Parent(s): 0ddb69a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -50,7 +50,9 @@ def invoke(openai_api_key, prompt, rag_option):
50
 
51
  result = completion["result"]
52
  elif (rag_option == RAG_LLAMAINDEX):
53
- result = rag_retrieval(config, prompt)
 
 
54
  else:
55
  completion, chain, callback = llm_chain(config, prompt)
56
 
 
50
 
51
  result = completion["result"]
52
  elif (rag_option == RAG_LLAMAINDEX):
53
+ #result = rag_retrieval(config, prompt)
54
+ rag = LlamaIndexRAG()
55
+ result = rag.rag_retrieval(config, prompt)
56
  else:
57
  completion, chain, callback = llm_chain(config, prompt)
58