From abea4ddbcca40c1c1da51507363867fb3664228f Mon Sep 17 00:00:00 2001
From: kqlio67 <kqlio67@users.noreply.github.com>
Date: Thu, 12 Sep 2024 21:35:31 +0300
Subject: Bug fixes and improvements to HuggingChat and HuggingFace providers

---
 g4f/Provider/HuggingChat.py |  1 +
 g4f/Provider/HuggingFace.py | 23 +++++------------------
 2 files changed, 6 insertions(+), 18 deletions(-)

(limited to 'g4f')

diff --git a/g4f/Provider/HuggingChat.py b/g4f/Provider/HuggingChat.py
index fad44957..7edb2f9f 100644
--- a/g4f/Provider/HuggingChat.py
+++ b/g4f/Provider/HuggingChat.py
@@ -12,6 +12,7 @@ class HuggingChat(AbstractProvider, ProviderModelMixin):
     working = True
     supports_stream = True
     default_model = "meta-llama/Meta-Llama-3.1-70B-Instruct"
+    
     models = [
         default_model,
         'CohereForAI/c4ai-command-r-plus-08-2024',
diff --git a/g4f/Provider/HuggingFace.py b/g4f/Provider/HuggingFace.py
index 4fe02739..586e5f5f 100644
--- a/g4f/Provider/HuggingFace.py
+++ b/g4f/Provider/HuggingFace.py
@@ -9,29 +9,16 @@ from .helper import get_connector
 from ..errors import RateLimitError, ModelNotFoundError
 from ..requests.raise_for_status import raise_for_status
 
+from .HuggingChat import HuggingChat
+
 class HuggingFace(AsyncGeneratorProvider, ProviderModelMixin):
     url = "https://huggingface.co/chat"
     working = True
     needs_auth = True
     supports_message_history = True
-    default_model = "meta-llama/Meta-Llama-3.1-70B-Instruct"
-    models = [
-        default_model,
-        'CohereForAI/c4ai-command-r-plus-08-2024',
-        'mistralai/Mixtral-8x7B-Instruct-v0.1',
-        'NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO',
-        'mistralai/Mistral-7B-Instruct-v0.3',
-        'microsoft/Phi-3-mini-4k-instruct',
-    ]
-    
-    model_aliases = {
-        "llama-3.1-70b": "meta-llama/Meta-Llama-3.1-70B-Instruct",
-        "command-r-plus": "CohereForAI/c4ai-command-r-plus-08-2024",
-        "mixtral-8x7b": "mistralai/Mixtral-8x7B-Instruct-v0.1",
-        "mixtral-8x7b": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
-        "mistral-7b": "mistralai/Mistral-7B-Instruct-v0.3",
-        "phi-3-mini-4k": "microsoft/Phi-3-mini-4k-instruct",
-    }
+    default_model = HuggingChat.default_model
+    models = HuggingChat.models
+    model_aliases = HuggingChat.model_aliases
 
     @classmethod
     def get_model(cls, model: str) -> str:
-- 
cgit v1.2.3