Update app.py
Browse files
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 |
|