Merge pull request #264 from MobinX/patch-1

VinciGit00 · web-flow · commit ae9986a113de · 2024-05-19T08:45:25.000+02:00
Added gemini flash and pro support
diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
@@ -27,6 +27,8 @@
     },
     "gemini": {
         "gemini-pro": 128000,
+        "gemini-1.5-flash-latest":128000,
+        "gemini-1.5-pro-latest":128000,
         "models/embedding-001": 2048
     },
 
