From e0ffc838b06c0f024026a275fc7f7b4243ad5cf9 Mon Sep 17 00:00:00 2001
From: VinciGit00 <mvincig11@gmail.com>
Date: Tue, 30 Apr 2024 10:08:54 +0200
Subject: [PATCH] feat: add cluade integration

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

diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
index 9dcfd047..6b9ed637 100644
--- a/scrapegraphai/helpers/models_tokens.py
+++ b/scrapegraphai/helpers/models_tokens.py
@@ -32,11 +32,16 @@
         "mistral-openorca": 32000,
         "stablelm-zephyr": 8192
     },
-    
     "groq": {
         "llama3-8b-8192": 8192,
         "llama3-70b-8192": 8192,
         "mixtral-8x7b-32768": 32768,
         "gemma-7b-it": 8192,
     },
+    "claude": {
+        "claude_instant": 100000,
+        "claude2": 9000,
+        "claude2.1": 200000,
+        "claude3": 200000
+    }
 }