diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py index 7ae45247..63eedd4d 100644 --- a/scrapegraphai/helpers/models_tokens.py +++ b/scrapegraphai/helpers/models_tokens.py @@ -51,6 +51,7 @@ models_tokens = { "google_genai": { "gemini-pro": 128000, "gemini-1.5-flash-latest": 128000, + "gemini-2.0-flash-latest": 128000, "gemini-1.5-pro-latest": 128000, "models/embedding-001": 2048, },