From f07173de4fe64bdccfdb9adff34ae0abbfe70a4d Mon Sep 17 00:00:00 2001
From: vlad <master.redstone@yandex.ru>
Date: Thu, 14 Mar 2024 01:29:49 +0300
Subject: GigaChat models add

---
 g4f/models.py | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

(limited to 'g4f/models.py')

diff --git a/g4f/models.py b/g4f/models.py
index ae2ef54b..6fcfdd14 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -2,7 +2,7 @@ from __future__  import annotations
 
 from dataclasses import dataclass
 
-from .Provider   import RetryProvider, ProviderType
+from .Provider import RetryProvider, ProviderType
 from .Provider   import (
     Chatgpt4Online,
     PerplexityLabs,
@@ -15,6 +15,7 @@ from .Provider   import (
     ChatgptAi,
     DeepInfra,
     ChatBase,
+    GigaChat,
     Liaobots,
     FreeGpt,
     Llama2,
@@ -95,6 +96,24 @@ gpt_4_turbo = Model(
     best_provider = Bing
 )
 
+gigachat = Model(
+    name          = 'GigaChat:latest',
+    base_provider = 'gigachat',
+    best_provider = GigaChat
+)
+
+gigachat_plus = Model(
+    name          = 'GigaChat-Plus',
+    base_provider = 'gigachat',
+    best_provider = GigaChat
+)
+
+gigachat_pro = Model(
+    name          = 'GigaChat-Pro',
+    base_provider = 'gigachat',
+    best_provider = GigaChat
+)
+
 llama2_7b = Model(
     name          = "meta-llama/Llama-2-7b-chat-hf",
     base_provider = 'meta',
@@ -271,7 +290,12 @@ class ModelUtils:
         'llama2-70b': llama2_70b,
         'codellama-34b-instruct': codellama_34b_instruct,
         'codellama-70b-instruct': codellama_70b_instruct,
-        
+
+        # GigaChat
+        'gigachat'     : gigachat,
+        'gigachat_plus': gigachat_plus,
+        'gigachat_pro' : gigachat_pro,
+
         'mixtral-8x7b': mixtral_8x7b,
         'mistral-7b': mistral_7b,
         'dolphin-mixtral-8x7b': dolphin_mixtral_8x7b,
-- 
cgit v1.2.3