Update agent.py

agent.py

@@ -1,4 +1,4 @@
-from llama_index.llms.groq import Groq
+from llama_index.llms.openai import OpenAI
 from llama_index.tools.wikipedia.base import WikipediaToolSpec
 from llama_index.core.llms import ChatMessage
 from llama_index.core.agent import ReActAgent
@@ -69,10 +69,10 @@ class JoroBasicAgent:
         Below is the current conversation consisting of interleaving human and assistant messages.
         """
         react_system_prompt = PromptTemplate(system_prompt)
-        #llm = Groq(model="qwen-qwq-32b")
-        llm = DeepInfraLLM(
-            model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",api_key=os.getenv("DEEPINFRA_API_KEY"))
+        #llm = DeepInfraLLM(
+        #    model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",api_key=os.getenv("DEEPINFRA_API_KEY"))
 
+        llm = OpenAI(model='gpt-4o')
         agent = ReActAgent.from_tools(
             llm=llm,
             tools=WikipediaToolSpec().to_tool_list() + TavilyToolSpec(api_key=os.getenv('TAVILY_API_KEY')).to_tool_list(),