all gemini

README.md

@@ -67,7 +67,7 @@ tag: agent-demo-track
 ### 1. Prerequisites
 
 - Python 3.8+
-- Mistral API key
+- Gemini API key
 
 ### 2. Installation
 
@@ -85,7 +85,7 @@ pip install -r requirements.txt
 Create a `.env` file in the project root:
 
 ```bash
-Mistral_API_KEY=your_Mistral_api_key_here
+GEMINI_API_KEY=your_Gemini_api_key_here
 ```
 
 ### 4. Launch the Application
@@ -170,7 +170,7 @@ After the initial analysis is complete:
 
 - **Frontend**: Gradio with custom CSS and real-time progress
 - **AI Agents**: 
-  - Primary SmolAgents powered by Mistral Medium Latest
+  - Primary SmolAgents powered by Gemini
   - Specialized follow-up agent for interactive analysis ⭐
 - **Search**: BM25 keyword matching with TF-IDF preprocessing
 - **Translation**: LLM-powered bilingual query translation

agent.py

@@ -25,9 +25,8 @@ from smolagents import (
 def create_web_agent(step_callback):
     search_tool = DuckDuckGoSearchTool()
     model = LiteLLMModel(
-                #model_id="gemini/gemini-2.5-flash-preview-05-20",
-                model_id="mistral/mistral-medium-latest",
-                api_key=os.getenv("MISTRAL_API_KEY"),
+                model_id="gemini/gemini-2.5-flash-preview-05-20",
+                api_key=os.getenv("GEMINI_API_KEY"),
             )
     web_agent = CodeAgent(
             tools=[

app.py

@@ -32,9 +32,8 @@ def initialize_models():
         # Initialize LLM for translation
         try:
             model = LiteLLMModel(
-                #model_id="gemini/gemini-2.5-flash-preview-05-20",
-                model_id="mistral/mistral-medium-latest",
-                api_key=os.getenv("MISTRAL_API_KEY")
+                model_id="gemini/gemini-2.5-flash-preview-05-20",
+                api_key=os.getenv("GEMINI_API_KEY")
             )
             llm_translator = CodeAgent(tools=[], model=model, max_steps=1)
             print("✅ LLM translator initialized")

followup_agent.py

@@ -22,9 +22,8 @@ def create_followup_agent():
     """Create a specialized agent for follow-up analysis"""
     search_tool = DuckDuckGoSearchTool()
     model = LiteLLMModel(
-        #model_id="gemini/gemini-2.5-flash-preview-05-20",
-        model_id="mistral/mistral-medium-latest",
-        api_key=os.getenv("MISTRAL_API_KEY"),
+        model_id="gemini/gemini-2.5-flash-preview-05-20",
+        api_key=os.getenv("GEMINI_API_KEY"),
     )
     
     followup_agent = CodeAgent(

tools/retrieval_tools.py

@@ -33,9 +33,8 @@ def _initialize_retrieval_system():
     if _llm_translator is None:
         try:
             model = LiteLLMModel(
-                #model_id="gemini/gemini-2.5-flash-preview-05-20",
-                model_id="mistral/mistral-medium-latest",
-                api_key=os.getenv("MISTRAL_API_KEY")
+                model_id="gemini/gemini-2.5-flash-preview-05-20",
+                api_key=os.getenv("GEMINI_API_KEY")
             )
             _llm_translator = CodeAgent(tools=[], model=model, max_steps=1)
             print("✅ LLM translator initialized")