Update app.py

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 import os
 
 from datetime import date
-from langchain.agents import AgentType, initialize_agent, tool
+from langchain import hub
+from langchain.agents import AgentExecutor, AgentType, create_openai_functions_agent, initialize_agent, tool
 from langchain.chat_models import ChatOpenAI
 from langchain_community.tools.tavily_search import TavilySearchResults
 
@@ -52,15 +53,23 @@ def invoke(openai_api_key, prompt, agent_option):
 
             tools = [search]
             
-            agent = initialize_agent(tools + [time], 
-                                     llm, 
-                                     agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
-                                     handle_parsing_errors = True,
-                                     verbose = True)
-        
-            completion = agent(prompt)
+            #agent = initialize_agent(tools + [time], 
+            #                         llm, 
+            #                         agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+            #                         handle_parsing_errors = True,
+            #                         verbose = True)
+
+            p = hub.pull("hwchase17/openai-functions-agent")
+            
+            agent = create_openai_functions_agent(llm, tools, p)
+
+            agent_executor = AgentExecutor(agent = agent, tools = tools, verbose = True)
+
+            output = agent_executor.invoke({"input": prompt})
+            
+            #completion = agent(prompt)
     
-            output = completion["output"]
+            #output = completion["output"]
     except Exception as e:
         err_msg = e