From c7b5e5eee67089e604f5ab11a7f58078e1639ec4 Mon Sep 17 00:00:00 2001
From: nullstreak <139914347+nullstreak@users.noreply.github.com>
Date: Wed, 13 Dec 2023 13:15:27 +0100
Subject: Add Mistral models

---
 g4f/models.py | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

(limited to 'g4f')

diff --git a/g4f/models.py b/g4f/models.py
index 810722e4..188cbf5c 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -107,6 +107,17 @@ llama2_70b = Model(
     base_provider = "huggingface",
     best_provider = RetryProvider([Llama2, DeepInfra, HuggingChat]))
 
+# Mistal
+mixtral_8x7b = Model(
+    name          = "mistralai/Mixtral-8x7B-Instruct-v0.1",
+    base_provider = "huggingface",
+    best_provider = HuggingChat)
+
+mistral_7b = Model(
+    name          = "mistralai/Mistral-7B-Instruct-v0.1",
+    base_provider = "huggingface",
+    best_provider = HuggingChat)
+
 # Bard
 palm = Model(
     name          = 'palm',
@@ -292,6 +303,10 @@ class ModelUtils:
         'llama2-13b': llama2_13b,
         'llama2-70b': llama2_70b,
         
+        # Mistral
+        'mixtral-8x7b': mixtral_8x7b,
+        'mistral-7b': mistral_7b,
+        
         # Bard
         'palm2'       : palm,
         'palm'        : palm,
@@ -331,4 +346,4 @@ class ModelUtils:
         'pi': pi
     }
 
-_all_models = list(ModelUtils.convert.keys())
\ No newline at end of file
+_all_models = list(ModelUtils.convert.keys())
-- 
cgit v1.2.3