remove flash

Author: NicolasIRAGNE

src/server/ai_file_selector.py

@@ -5,7 +5,6 @@
 import json
 import os
 import re
-import time
 from typing import TYPE_CHECKING
 
 import google.generativeai as genai
@@ -39,7 +38,7 @@ def __init__(self):
             raise ValueError("GEMINI_API_KEY environment variable is required")
 
         genai.configure(api_key=api_key)
-        self.model = genai.GenerativeModel("gemini-1.5-flash")
+        self.model = genai.GenerativeModel("gemini-1.5-pro")
         self.encoding = tiktoken.get_encoding("o200k_base")
 
     def _count_tokens(self, text: str) -> int:
@@ -208,11 +207,7 @@ async def select_files(
         try:
             # Call Gemini API
             logger.info("Calling Gemini API for file selection")
-            gemini_start_time = time.time()
             response = await self.model.generate_content_async(prompt)
-            gemini_end_time = time.time()
-            gemini_duration = gemini_end_time - gemini_start_time
-            logger.info("Gemini API call completed", extra={"duration": gemini_duration})
 
             if not response.text:
                 raise ValueError("Empty response from Gemini API")