Bas95 commited on
Commit
2adcbc3
·
verified ·
1 Parent(s): 77c968b

Update model: gemini-2.0-flash

Browse files
backend/agents/delta_agent.py CHANGED
@@ -14,7 +14,7 @@ from google.genai import types
14
  from ..models import Cabin, FlightOption, TripSearchRequest
15
  from ..tools.browser import BrowserSession
16
 
17
- _MODEL = "gemini-1.5-flash"
18
  _MAX_ITER = 30
19
  _ECONOMY_CLASSES = ["L", "U", "T", "X", "V"]
20
  _FIRST_CLASSES = ["Z", "D", "I"]
 
14
  from ..models import Cabin, FlightOption, TripSearchRequest
15
  from ..tools.browser import BrowserSession
16
 
17
+ _MODEL = "gemini-2.0-flash"
18
  _MAX_ITER = 30
19
  _ECONOMY_CLASSES = ["L", "U", "T", "X", "V"]
20
  _FIRST_CLASSES = ["Z", "D", "I"]
backend/agents/ihg_agent.py CHANGED
@@ -17,7 +17,7 @@ from google.genai import types
17
  from ..models import HotelOption
18
  from ..tools.browser import BrowserSession
19
 
20
- _MODEL = "gemini-1.5-flash"
21
  _MAX_ITER = 30
22
 
23
 
 
17
  from ..models import HotelOption
18
  from ..tools.browser import BrowserSession
19
 
20
+ _MODEL = "gemini-2.0-flash"
21
  _MAX_ITER = 30
22
 
23
 
backend/agents/resy_agent.py CHANGED
@@ -15,7 +15,7 @@ from google.genai import types
15
  from ..models import RestaurantOption
16
  from ..tools.browser import BrowserSession
17
 
18
- _MODEL = "gemini-1.5-flash"
19
  _MAX_ITER = 20
20
 
21
 
 
15
  from ..models import RestaurantOption
16
  from ..tools.browser import BrowserSession
17
 
18
+ _MODEL = "gemini-2.0-flash"
19
  _MAX_ITER = 20
20
 
21