Update agent.py

agent.py

@@ -1,4 +1,20 @@
 from langchain.agents import AgentType, initialize_agent, load_tools, tool
 from langchain.chat_models import ChatOpenAI
 
-def invoke(openai_api_key, prompt, agent_option):
+def invoke_agent(openai_api_key, prompt):
+    llm = ChatOpenAI(
+        model_name = config["model"],
+        openai_api_key = openai_api_key, 
+        temperature = config["temperature"])
+    
+    tools = load_tools(["openweathermap-api"])
+            
+    agent = initialize_agent(
+        tools +      # built-in tools
+        [date_tool], # custom tools
+        llm,
+        agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+        handle_parsing_errors = True,
+        verbose = True)
+
+    return agent(prompt)