Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 from datetime import date
 from langchain.agents import AgentType, initialize_agent, tool
 from langchain.chat_models import ChatOpenAI
+from langchain_community.tools.tavily_search import TavilySearchResults
 
 config = {
     "max_tokens": 1000,
@@ -41,7 +42,11 @@ def invoke(openai_api_key, prompt, agent_option):
                              openai_api_key = openai_api_key, 
                              temperature = config["temperature"])
 
-            agent = initialize_agent([time], 
+            search = TavilySearchResults()
+
+            tools = [search]
+            
+            agent = initialize_agent(tools + [time], 
                                      llm, 
                                      agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
                                      handle_parsing_errors = True,
@@ -69,8 +74,8 @@ 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?"],
-                                ["sk-", "What is the weather in SF?"]],
+                    examples = [["sk-", "What is today's date?", AGENT_ON],
+                                ["sk-", "What is the weather in SF?", AGENT_ON]],
                                 cache_examples = False)
 
 demo.launch()