Update agent.py

agent.py

@@ -1,13 +1,9 @@
+from datetime import date
 from langchain.agents import AgentType, initialize_agent, load_tools, tool
 from langchain.chat_models import ChatOpenAI
 
 OPENWEATHERMAP_API_KEY = os.environ["OPENWEATHERMAP_API_KEY"]
 
-config = {
-    "model": "gpt-4-0613",
-    "temperature": 0,
-}
-
 @tool
 def date_tool(text: str) -> str:
     """Returns today's date. Use this for any questions related to knowing today's date. 
@@ -15,11 +11,11 @@ def date_tool(text: str) -> str:
        Any date mathematics should occur outside this function."""
     return str(date.today())
 
-def invoke_agent(openai_api_key, prompt):
+def invoke_agent(model, temperature, openai_api_key, prompt):
     llm = ChatOpenAI(
-        model_name = config["model"],
+        model_name = model,
         openai_api_key = openai_api_key, 
-        temperature = config["temperature"])
+        temperature = temperature)
     
     tools = load_tools(["openweathermap-api"])