From f724c07f1b8cb1b2360d8c6675688b15cbcd85ca Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Fri, 12 Apr 2024 20:58:40 +0200
Subject: Add labels to provider list

---
 g4f/Provider/needs_auth/Groq.py       | 1 +
 g4f/Provider/needs_auth/OpenRouter.py | 1 +
 g4f/Provider/needs_auth/Openai.py     | 1 +
 g4f/Provider/needs_auth/OpenaiChat.py | 1 +
 g4f/Provider/needs_auth/Theb.py       | 1 +
 g4f/Provider/needs_auth/ThebApi.py    | 1 +
 6 files changed, 6 insertions(+)

(limited to 'g4f/Provider/needs_auth')

diff --git a/g4f/Provider/needs_auth/Groq.py b/g4f/Provider/needs_auth/Groq.py
index 87e87e60..922b2dd2 100644
--- a/g4f/Provider/needs_auth/Groq.py
+++ b/g4f/Provider/needs_auth/Groq.py
@@ -4,6 +4,7 @@ from .Openai import Openai
 from ...typing import AsyncResult, Messages
 
 class Groq(Openai):
+    lebel = "Groq"
     url = "https://console.groq.com/playground"
     working = True
     default_model = "mixtral-8x7b-32768"
diff --git a/g4f/Provider/needs_auth/OpenRouter.py b/g4f/Provider/needs_auth/OpenRouter.py
index e5f87076..773d9203 100644
--- a/g4f/Provider/needs_auth/OpenRouter.py
+++ b/g4f/Provider/needs_auth/OpenRouter.py
@@ -6,6 +6,7 @@ from .Openai import Openai
 from ...typing import AsyncResult, Messages
 
 class OpenRouter(Openai):
+    label = "OpenRouter"
     url = "https://openrouter.ai"
     working = True
     default_model = "openrouter/auto"
diff --git a/g4f/Provider/needs_auth/Openai.py b/g4f/Provider/needs_auth/Openai.py
index ea09e950..81ba5981 100644
--- a/g4f/Provider/needs_auth/Openai.py
+++ b/g4f/Provider/needs_auth/Openai.py
@@ -9,6 +9,7 @@ from ...requests import StreamSession, raise_for_status
 from ...errors import MissingAuthError, ResponseError
 
 class Openai(AsyncGeneratorProvider, ProviderModelMixin):
+    label = "OpenAI API"
     url = "https://openai.com"
     working = True
     needs_auth = True
diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py
index 7491725f..3145161a 100644
--- a/g4f/Provider/needs_auth/OpenaiChat.py
+++ b/g4f/Provider/needs_auth/OpenaiChat.py
@@ -35,6 +35,7 @@ from ... import debug
 class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin):
     """A class for creating and managing conversations with OpenAI chat service"""
 
+    lebel = "OpenAI ChatGPT"
     url = "https://chat.openai.com"
     working = True
     supports_gpt_35_turbo = True
diff --git a/g4f/Provider/needs_auth/Theb.py b/g4f/Provider/needs_auth/Theb.py
index 2930e35d..af690063 100644
--- a/g4f/Provider/needs_auth/Theb.py
+++ b/g4f/Provider/needs_auth/Theb.py
@@ -32,6 +32,7 @@ models = {
 }
 
 class Theb(AbstractProvider):
+    label = "TheB.AI"
     url = "https://beta.theb.ai"
     working = True
     supports_gpt_35_turbo = True
diff --git a/g4f/Provider/needs_auth/ThebApi.py b/g4f/Provider/needs_auth/ThebApi.py
index 48879bcb..22fc62ed 100644
--- a/g4f/Provider/needs_auth/ThebApi.py
+++ b/g4f/Provider/needs_auth/ThebApi.py
@@ -28,6 +28,7 @@ models = {
 }
 
 class ThebApi(Openai):
+    label = "TheB.AI API"
     url = "https://theb.ai"
     working = True
     needs_auth = True
-- 
cgit v1.2.3