From 2b2b910a378dc79c9e92e60c15e0a50afac7fb5b Mon Sep 17 00:00:00 2001
From: duke147 <85614628@qq.com>
Date: Wed, 5 Jun 2024 19:02:34 +0800
Subject: [PATCH] support ernie

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

diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index 43598785..510777fe 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -154,5 +154,15 @@
     "deepseek": {
         "deepseek-chat": 32768,
         "deepseek-coder": 16384
+    },
+    "ernie": {
+        "ernie-bot-turbo": 4096,
+        "ernie-bot": 4096,
+        "ernie-bot-2": 4096,
+        "ernie-bot-2-base": 4096,
+        "ernie-bot-2-base-zh": 4096,
+        "ernie-bot-2-base-en": 4096,
+        "ernie-bot-2-base-en-zh": 4096,
+        "ernie-bot-2-base-zh-en": 4096,
     }
 }