diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index ec83e1fb..f8881d75 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -18,6 +18,7 @@ models_tokens = {
         "gpt-4-0613": 8192,
         "gpt-4-32k": 32768,
         "gpt-4-32k-0613": 32768,
+        "gpt-4o": 128000,
     },
     "azure": {
         "gpt-3.5-turbo": 4096,