From 03f4a3aa29c42a9a312c4afb6818de3450e7cedf Mon Sep 17 00:00:00 2001 From: Marco Vinciguerra Date: Sat, 20 Jul 2024 13:19:21 +0200 Subject: [PATCH] fix: azure models --- scrapegraphai/helpers/models_tokens.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py index 0ef10277..7becbf8b 100644 --- a/scrapegraphai/helpers/models_tokens.py +++ b/scrapegraphai/helpers/models_tokens.py @@ -18,7 +18,17 @@ models_tokens = { "gpt-4o": 128000, }, "azure": { - "gpt-3.5-turbo": 4096, + "gpt-3.5-turbo-0125": 16385, + "gpt-3.5": 4096, + "gpt-3.5-turbo": 16385, + "gpt-3.5-turbo-1106": 16385, + "gpt-3.5-turbo-instruct": 4096, + "gpt-4-0125-preview": 128000, + "gpt-4-turbo-preview": 128000, + "gpt-4-turbo": 128000, + "gpt-4-turbo-2024-04-09": 128000, + "gpt-4-1106-preview": 128000, + "gpt-4-vision-preview": 128000, "gpt-4": 8192, "gpt-4-0613": 8192, "gpt-4-32k": 32768,