From 7f30da99308968f7b06a432aafab8d8a1365ffb9 Mon Sep 17 00:00:00 2001
From: Mobin Chowdhury <47663360+MobinX@users.noreply.github.com>
Date: Sat, 18 May 2024 23:32:27 +0600
Subject: [PATCH] Update models_tokens.py

---
 scrapegraphai/helpers/models_tokens.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index 16a42c17..f1780049 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -27,6 +27,8 @@ models_tokens = {
     },
     "gemini": {
         "gemini-pro": 128000,
+        "gemini-1.5-flash-latest":128000,
+        "gemini-1.5-pro-latest":128000,
         "models/embedding-001": 2048
     },