Gaurav vashistha commited on
Commit
50e246b
Β·
1 Parent(s): 1900bac

Prioritize gemini-1.5-flash-001 to avoid 404s

Browse files
Files changed (1) hide show
  1. agents/visual_analyst.py +9 -8
agents/visual_analyst.py CHANGED
@@ -17,19 +17,20 @@ class VisualAnalyst:
17
  # --- SMART MODEL SELECTION ---
18
  print("πŸ” Checking available Gemini models for this key...")
19
  try:
 
20
  my_models = [m.name for m in genai.list_models() if 'generateContent' in m.supported_generation_methods]
21
  print(f"πŸ“‹ Available Models: {my_models}")
22
 
23
- # Priority list (Newest -> Oldest)
24
  preferred_order = [
25
- 'models/gemini-1.5-flash',
26
- 'models/gemini-1.5-flash-001',
27
- 'models/gemini-1.5-pro',
28
  'models/gemini-1.5-pro-001',
29
- 'models/gemini-pro-vision' # Legacy fallback
 
30
  ]
31
 
32
- selected_model = "models/gemini-1.5-flash" # Default
33
 
34
  for model_name in preferred_order:
35
  if model_name in my_models:
@@ -40,8 +41,8 @@ class VisualAnalyst:
40
  self.model = genai.GenerativeModel(selected_model)
41
 
42
  except Exception as e:
43
- print(f"⚠️ Model list failed ({e}), defaulting to gemini-1.5-flash")
44
- self.model = genai.GenerativeModel('gemini-1.5-flash')
45
 
46
  async def analyze_image(self, image_path: str):
47
  # Adaptation: Read file path to bytes, as main.py passes a path
 
17
  # --- SMART MODEL SELECTION ---
18
  print("πŸ” Checking available Gemini models for this key...")
19
  try:
20
+ # Filter for models that support content generation
21
  my_models = [m.name for m in genai.list_models() if 'generateContent' in m.supported_generation_methods]
22
  print(f"πŸ“‹ Available Models: {my_models}")
23
 
24
+ # UPDATED PRIORITY: Specific versions FIRST
25
  preferred_order = [
26
+ 'models/gemini-1.5-flash-001', # <--- Specific version (Safest)
27
+ 'models/gemini-1.5-flash', # Alias (Fallback)
 
28
  'models/gemini-1.5-pro-001',
29
+ 'models/gemini-1.5-pro',
30
+ 'models/gemini-pro-vision'
31
  ]
32
 
33
+ selected_model = "models/gemini-1.5-flash-001" # Default safety
34
 
35
  for model_name in preferred_order:
36
  if model_name in my_models:
 
41
  self.model = genai.GenerativeModel(selected_model)
42
 
43
  except Exception as e:
44
+ print(f"⚠️ Model list failed ({e}), defaulting to gemini-1.5-flash-001")
45
+ self.model = genai.GenerativeModel('gemini-1.5-flash-001')
46
 
47
  async def analyze_image(self, image_path: str):
48
  # Adaptation: Read file path to bytes, as main.py passes a path