Update app.py

app.py

@@ -36,25 +36,30 @@ def invoke(openai_api_key, prompt, agent_option):
     output = ""
     
     try:
-        llm = ChatOpenAI(model_name = config["model_name"],
-                         openai_api_key = openai_api_key, 
-                         temperature = config["temperature"])
-
         if (agent_option == AGENT_OFF):
-            agent = initialize_agent([], # no tools
-                                     llm,
-                                     agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
-                                     handle_parsing_errors = True,
-                                     verbose = True)
+            client = OpenAI(api_key = openai_api_key)
+    
+            completion = client.chat.completions.create(
+                messages = [{"role": "user", "content": prompt}],
+                model = config["model"],
+                temperature = config["temperature"],)
+    
+            output = completion.choices[0].message.content
         else:
+            llm = ChatOpenAI(
+                model_name = config["model_name"],
+                openai_api_key = openai_api_key, 
+                temperature = config["temperature"])
+    
             tools = load_tools(["openweathermap-api"])
             
-            agent = initialize_agent(tools + # built-in tools
-                                     [time], # custom tools
-                                     llm,
-                                     agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
-                                     handle_parsing_errors = True,
-                                     verbose = True)
+            agent = initialize_agent(
+                tools + # built-in tools
+                [time], # custom tools
+                llm,
+                agent = AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+                handle_parsing_errors = True,
+                verbose = True)
 
         completion = agent(prompt)