Upload agent.py

agent.py

@@ -1,5 +1,4 @@
-from llama_index.llms.openai import OpenAI # Keep if you might switch back to OpenAI
-from llama_index.llms.gemini import Gemini # Import the Gemini LLM
+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
@@ -70,15 +69,10 @@ class CuongBasicAgent:
         Below is the current conversation consisting of interleaving human and assistant messages.
         """
         react_system_prompt = PromptTemplate(system_prompt)
-        
-        # --- CHANGE HERE: Use Gemini directly ---
-        # Make sure you have your GOOGLE_API_KEY set as an environment variable
-        llm = Gemini(model='gemini-2.0-flash', api_key=os.getenv("GOOGLE_API_KEY")) 
-        # 'gemini-pro' is a commonly used model for general tasks. 
-        # You might also use 'gemini-1.5-pro' for larger contexts.
-        # Check the official Google Gemini API documentation for available model names.
-        # --- END CHANGE ---
+        #llm = DeepInfraLLM(
+        #    model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",api_key=os.getenv("DEEPINFRA_API_KEY"))
 
+        llm = OpenAI(model='gpt-4.1', api_key=os.getenv("OPENAI_API_KEY"))
         agent = ReActAgent.from_tools(
             llm=llm,
             tools=WikipediaToolSpec().to_tool_list() + TavilyToolSpec(api_key=os.getenv('TAVILY_API_KEY')).to_tool_list(),