Update app.py

app.py

@@ -2,15 +2,8 @@ import gradio as gr
 import os
 
 from datetime import date
-from langchain import hub
-from langchain.agents import AgentExecutor, AgentType, create_openai_functions_agent, initialize_agent, tool
+from langchain.agents import AgentType, initialize_agent, tool
 from langchain.chat_models import ChatOpenAI
-from langchain_community.tools.tavily_search import TavilySearchResults
-
-from dotenv import load_dotenv, find_dotenv
-_ = load_dotenv(find_dotenv())
-
-TAVILY_API_KEY = os.environ["TAVILY_API_KEY"]
 
 config = {
     "max_tokens": 1000,
@@ -49,27 +42,15 @@ def invoke(openai_api_key, prompt, agent_option):
                              openai_api_key = openai_api_key, 
                              temperature = config["temperature"])
 
-            search = TavilySearchResults()
-
-            tools = [search]
-            
-            #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)
+            agent = initialize_agent([time], 
+                                     llm, 
+                                     agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+                                     handle_parsing_errors = True,
+                                     verbose = True)
 
-            output = agent_executor.invoke({"input": prompt})
-            
-            #completion = agent(prompt)
+            completion = agent(prompt)
     
-            #output = completion["output"]
+            output = completion["output"]
     except Exception as e:
         err_msg = e
 
@@ -89,8 +70,7 @@ demo = gr.Interface(fn = invoke,
                     outputs = [gr.Textbox(label = "Completion", lines = 1)],
                     title = "Generative AI - LLM & Agent",
                     description = description,
-                    examples = [["sk-", "What is today's date?", AGENT_ON],
-                                ["sk-", "What is the weather in SF?", AGENT_ON]],
+                    examples = [["sk-", "What is today's date?", AGENT_ON]],
                                 cache_examples = False)
 
 demo.launch()