Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import logging, os, sys, threading, time
 
 from agent_langchain import agent_langchain
-#from agent_llamaindex import agent_llamaindex
+from agent_llamaindex import agent_llamaindex
 from openai import OpenAI
 from trace import trace_wandb
 
@@ -49,11 +49,11 @@ def invoke(openai_api_key, prompt, agent_option):
                 )
         
                 result = completion["output"]
-            #elif (agent_option == AGENT_LLAMAINDEX):
-            #    result = agent_llamaindex(
-            #        config,
-            #        prompt
-            #    )
+            elif (agent_option == AGENT_LLAMAINDEX):
+                result = agent_llamaindex(
+                    config,
+                    prompt
+                )
             else:
                 client = OpenAI()