Upload 2 files

agent.py

@@ -1,4 +1,5 @@
-from llama_index.llms.openai import OpenAI
+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.tools.wikipedia.base import WikipediaToolSpec
 from llama_index.core.llms import ChatMessage
 from llama_index.core.agent import ReActAgent
@@ -69,10 +70,15 @@ class CuongBasicAgent:
         Below is the current conversation consisting of interleaving human and assistant messages.
         """
         react_system_prompt = PromptTemplate(system_prompt)
-        #llm = DeepInfraLLM(
-        #    model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",api_key=os.getenv("DEEPINFRA_API_KEY"))
+        
+        # --- CHANGE HERE: Use Gemini directly ---
+        # Make sure you have your GOOGLE_API_KEY set as an environment variable
+        llm = Gemini(model='gemini-pro', 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 = OpenAI(model='Gemini-2.5', api_key=os.getenv("GEMINI_API_KEY"))
         agent = ReActAgent.from_tools(
             llm=llm,
             tools=WikipediaToolSpec().to_tool_list() + TavilyToolSpec(api_key=os.getenv('TAVILY_API_KEY')).to_tool_list(),

requirements.txt

@@ -1,7 +1,9 @@
-gradio
-requests
-pandas
-llama-index-llms-openai
-llama-index-tools-wikipedia
-llama-index-llms-deepinfra
-llama-index-tools-tavily_research
+gradio>=4.0.0
+requests>=2.31.0
+pandas>=2.0.0
+llama-index-core>=0.10.0 # Fundamental LlamaIndex package
+llama-index-llms-openai>=0.1.0 # For OpenAI API (or Gemini via OpenAI compatibility)
+llama-index-llms-gemini>=0.1.0 # Crucial if you are directly using llama_index.llms.gemini.Gemini
+llama-index-llms-deepinfra>=0.1.0 # If you plan to use DeepInfra models
+llama-index-tools-wikipedia>=0.1.0
+llama-index-tools-tavily-research>=0.1.0