harshalmore31 commited on
Commit
6fba44c
·
1 Parent(s): 5ebc627

Fix model names to valid GPT-4o-mini and Gemini versions for improved compatibility

Browse files
Files changed (1) hide show
  1. mai_dx/main.py +3 -3
mai_dx/main.py CHANGED
@@ -255,7 +255,7 @@ class MaiDxOrchestrator:
255
 
256
  def __init__(
257
  self,
258
- model_name: str = "gpt-4.1", # Fixed: Use valid GPT-4 Turbo model name
259
  max_iterations: int = 10,
260
  initial_budget: int = 10000,
261
  mode: str = "no_budget", # "instant", "question_only", "budgeted", "no_budget", "ensemble"
@@ -2604,13 +2604,13 @@ def run_mai_dxo_demo(
2604
  orchestrator = MaiDxOrchestrator.create_variant(
2605
  variant,
2606
  budget=3000,
2607
- model_name="gpt-4.1", # Fixed: Use valid model name
2608
  max_iterations=5,
2609
  )
2610
  else:
2611
  orchestrator = MaiDxOrchestrator.create_variant(
2612
  variant,
2613
- model_name="gpt-4.1", # Fixed: Use valid model name
2614
  max_iterations=5,
2615
  )
2616
 
 
255
 
256
  def __init__(
257
  self,
258
+ model_name: str = "gpt-4o-mini", # Fixed: Use valid GPT-4 Turbo model name
259
  max_iterations: int = 10,
260
  initial_budget: int = 10000,
261
  mode: str = "no_budget", # "instant", "question_only", "budgeted", "no_budget", "ensemble"
 
2604
  orchestrator = MaiDxOrchestrator.create_variant(
2605
  variant,
2606
  budget=3000,
2607
+ model_name="gemini/gemini-2.5-flash", # Fixed: Use valid model name
2608
  max_iterations=5,
2609
  )
2610
  else:
2611
  orchestrator = MaiDxOrchestrator.create_variant(
2612
  variant,
2613
+ model_name="gemini/gemini-2.5-flash", # Fixed: Use valid model name
2614
  max_iterations=5,
2615
  )
2616