From 855144876d796ceebb0930fec45ead6cc3834f14 Mon Sep 17 00:00:00 2001
From: sandeepchittilla <62606281+sandeepchittilla@users.noreply.github.com>
Date: Thu, 15 Aug 2024 14:59:48 +0100
Subject: [PATCH] feat: Add new feature to support gpt-4o variant models with
 different pricing

---
 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 8f863a9c..7e91c8ea 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -16,6 +16,8 @@ models_tokens = {
         "gpt-4-32k": 32768,
         "gpt-4-32k-0613": 32768,
         "gpt-4o": 128000,
+        "gpt-4o-2024-08-06": 128000,
+        "gpt-4o-2024-05-13": 128000,
         "gpt-4o-mini":128000,
 
     },