Update (g4f/Provider/)

kqlio67 · Nov 9, 2024 · c1e6276 · c1e6276
1 parent a96f9bd
commit c1e6276
Show file tree

Hide file tree

Showing 7 changed files with 13 additions and 13 deletions.
diff --git a/g4f/Provider/local/Ollama.py b/g4f/Provider/local/Ollama.py
@@ -3,10 +3,10 @@
 import requests
 import os
 
-from ..needs_auth.Openai import Openai
+from ..needs_auth.OpenaiAPI import OpenaiAPI
 from ...typing import AsyncResult, Messages
 
-class Ollama(Openai):
+class Ollama(OpenaiAPI):
     label = "Ollama"
     url = "https://ollama.com"
     needs_auth = False

diff --git a/g4f/Provider/needs_auth/DeepInfra.py b/g4f/Provider/needs_auth/DeepInfra.py
@@ -2,9 +2,9 @@
 
 import requests
 from ...typing import AsyncResult, Messages
-from .Openai import Openai
+from .OpenaiAPI import OpenaiAPI
 
-class DeepInfra(Openai):
+class DeepInfra(OpenaiAPI):
     label = "DeepInfra"
     url = "https://deepinfra.com"
     working = True

diff --git a/g4f/Provider/needs_auth/Groq.py b/g4f/Provider/needs_auth/Groq.py
@@ -1,9 +1,9 @@
 from __future__ import annotations
 
-from .Openai import Openai
+from .OpenaiAPI import OpenaiAPI
 from ...typing import AsyncResult, Messages
 
-class Groq(Openai):
+class Groq(OpenaiAPI):
     label = "Groq"
     url = "https://console.groq.com/playground"
     working = True

diff --git a/g4f/Provider/needs_auth/Openai.py → g4f/Provider/needs_auth/OpenaiAPI.py b/g4f/Provider/needs_auth/Openai.py → g4f/Provider/needs_auth/OpenaiAPI.py
@@ -9,7 +9,7 @@
 from ...errors import MissingAuthError, ResponseError
 from ...image import to_data_uri
 
-class Openai(AsyncGeneratorProvider, ProviderModelMixin):
+class OpenaiAPI(AsyncGeneratorProvider, ProviderModelMixin):
     label = "OpenAI API"
     url = "https://platform.openai.com"
     working = True

diff --git a/g4f/Provider/needs_auth/PerplexityApi.py b/g4f/Provider/needs_auth/PerplexityApi.py
@@ -1,9 +1,9 @@
 from __future__ import annotations
 
-from .Openai import Openai
+from .OpenaiAPI import OpenaiAPI
 from ...typing import AsyncResult, Messages
 
-class PerplexityApi(Openai):
+class PerplexityApi(OpenaiAPI):
     label = "Perplexity API"
     url = "https://www.perplexity.ai"
     working = True

diff --git a/g4f/Provider/needs_auth/ThebApi.py b/g4f/Provider/needs_auth/ThebApi.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from ...typing import CreateResult, Messages
-from .Openai import Openai
+from .OpenaiAPI import OpenaiAPI
 
 models = {
     "theb-ai": "TheB.AI",
@@ -27,7 +27,7 @@
     "qwen-7b-chat": "Qwen 7B"
 }
 
-class ThebApi(Openai):
+class ThebApi(OpenaiAPI):
     label = "TheB.AI API"
     url = "https://theb.ai"
     working = True
@@ -58,4 +58,4 @@ def create_async_generator(
                 "top_p": top_p,
             }
         }
-        return super().create_async_generator(model, messages, api_base=api_base, extra_data=data, **kwargs)
+        return super().create_async_generator(model, messages, api_base=api_base, extra_data=data, **kwargs)
diff --git a/g4f/Provider/needs_auth/__init__.py b/g4f/Provider/needs_auth/__init__.py
@@ -11,7 +11,7 @@
 from .Groq             import Groq
 from .HuggingFace      import HuggingFace
 from .MetaAI           import MetaAI
-from .Openai           import Openai
+from .OpenaiAPI        import OpenaiAPI
 from .OpenaiChat       import OpenaiChat
 from .PerplexityApi    import PerplexityApi
 from .Poe              import Poe