feat(g4f/models.py): add new provider GizAI and update best providers

This commit is contained in:
kqlio67 2024-10-26 18:54:47 +03:00
parent 7fba6f59a7
commit cc0ed0481e

View File

@ -23,7 +23,6 @@ from .Provider import (
DDG, DDG,
DeepInfra, DeepInfra,
DeepInfraChat, DeepInfraChat,
DeepInfraImage,
Editee, Editee,
Free2GPT, Free2GPT,
FreeChatgpt, FreeChatgpt,
@ -31,6 +30,7 @@ from .Provider import (
FreeNetfly, FreeNetfly,
Gemini, Gemini,
GeminiPro, GeminiPro,
GizAI,
GigaChat, GigaChat,
GPROChat, GPROChat,
HuggingChat, HuggingChat,
@ -87,6 +87,8 @@ class Model:
"""Returns a list of all model names.""" """Returns a list of all model names."""
return _all_models return _all_models
### Default ###
default = Model( default = Model(
name = "", name = "",
base_provider = "", base_provider = "",
@ -113,6 +115,8 @@ default = Model(
]) ])
) )
############ ############
### Text ### ### Text ###
############ ############
@ -136,13 +140,13 @@ gpt_35_turbo = Model(
gpt_4o = Model( gpt_4o = Model(
name = 'gpt-4o', name = 'gpt-4o',
base_provider = 'OpenAI', base_provider = 'OpenAI',
best_provider = IterListProvider([NexraChatGPT4o, Blackbox, ChatGptEs, AmigoChat, DarkAI, Editee, Liaobots, Airforce, OpenaiChat]) best_provider = IterListProvider([NexraChatGPT4o, Blackbox, ChatGptEs, AmigoChat, DarkAI, Editee, GizAI, Airforce, Liaobots, OpenaiChat])
) )
gpt_4o_mini = Model( gpt_4o_mini = Model(
name = 'gpt-4o-mini', name = 'gpt-4o-mini',
base_provider = 'OpenAI', base_provider = 'OpenAI',
best_provider = IterListProvider([DDG, ChatGptEs, FreeNetfly, Pizzagpt, MagickPen, AmigoChat, RubiksAI, Liaobots, Airforce, ChatgptFree, Koala, OpenaiChat, ChatGpt]) best_provider = IterListProvider([DDG, ChatGptEs, FreeNetfly, Pizzagpt, MagickPen, AmigoChat, RubiksAI, Liaobots, Airforce, GizAI, ChatgptFree, Koala, OpenaiChat, ChatGpt])
) )
gpt_4_turbo = Model( gpt_4_turbo = Model(
@ -167,7 +171,7 @@ o1 = Model(
o1_mini = Model( o1_mini = Model(
name = 'o1-mini', name = 'o1-mini',
base_provider = 'OpenAI', base_provider = 'OpenAI',
best_provider = AmigoChat best_provider = IterListProvider([AmigoChat, GizAI])
) )
@ -216,13 +220,13 @@ llama_3_70b = Model(
llama_3_1_8b = Model( llama_3_1_8b = Model(
name = "llama-3.1-8b", name = "llama-3.1-8b",
base_provider = "Meta Llama", base_provider = "Meta Llama",
best_provider = IterListProvider([Blackbox, DeepInfraChat, ChatHub, Cloudflare, Airforce, PerplexityLabs]) best_provider = IterListProvider([Blackbox, DeepInfraChat, ChatHub, Cloudflare, Airforce, GizAI, PerplexityLabs])
) )
llama_3_1_70b = Model( llama_3_1_70b = Model(
name = "llama-3.1-70b", name = "llama-3.1-70b",
base_provider = "Meta Llama", base_provider = "Meta Llama",
best_provider = IterListProvider([DDG, HuggingChat, Blackbox, FreeGpt, TeachAnything, Free2GPT, DeepInfraChat, DarkAI, Airforce, AiMathGPT, RubiksAI, HuggingFace, PerplexityLabs]) best_provider = IterListProvider([DDG, HuggingChat, Blackbox, FreeGpt, TeachAnything, Free2GPT, DeepInfraChat, DarkAI, Airforce, AiMathGPT, RubiksAI, GizAI, HuggingFace, PerplexityLabs])
) )
llama_3_1_405b = Model( llama_3_1_405b = Model(
@ -299,7 +303,7 @@ mistral_nemo = Model(
mistral_large = Model( mistral_large = Model(
name = "mistral-large", name = "mistral-large",
base_provider = "Mistral", base_provider = "Mistral",
best_provider = Editee best_provider = IterListProvider([Editee, GizAI])
) )
@ -347,13 +351,13 @@ phi_3_5_mini = Model(
gemini_pro = Model( gemini_pro = Model(
name = 'gemini-pro', name = 'gemini-pro',
base_provider = 'Google DeepMind', base_provider = 'Google DeepMind',
best_provider = IterListProvider([GeminiPro, Blackbox, AIChatFree, GPROChat, NexraGeminiPro, AmigoChat, Editee, Liaobots, Airforce]) best_provider = IterListProvider([GeminiPro, Blackbox, AIChatFree, GPROChat, NexraGeminiPro, AmigoChat, Editee, GizAI, Airforce, Liaobots])
) )
gemini_flash = Model( gemini_flash = Model(
name = 'gemini-flash', name = 'gemini-flash',
base_provider = 'Google DeepMind', base_provider = 'Google DeepMind',
best_provider = IterListProvider([Blackbox, Liaobots, Airforce]) best_provider = IterListProvider([Blackbox, GizAI, Airforce, Liaobots])
) )
gemini = Model( gemini = Model(
@ -424,14 +428,14 @@ claude_3_sonnet = Model(
claude_3_haiku = Model( claude_3_haiku = Model(
name = 'claude-3-haiku', name = 'claude-3-haiku',
base_provider = 'Anthropic', base_provider = 'Anthropic',
best_provider = IterListProvider([DDG, Airforce, Liaobots]) best_provider = IterListProvider([DDG, Airforce, GizAI, Liaobots])
) )
# claude 3.5 # claude 3.5
claude_3_5_sonnet = Model( claude_3_5_sonnet = Model(
name = 'claude-3.5-sonnet', name = 'claude-3.5-sonnet',
base_provider = 'Anthropic', base_provider = 'Anthropic',
best_provider = IterListProvider([Blackbox, Editee, AmigoChat, Airforce, Liaobots]) best_provider = IterListProvider([Blackbox, Editee, AmigoChat, Airforce, GizAI, Liaobots])
) )
@ -753,14 +757,14 @@ sdxl_lora = Model(
sdxl = Model( sdxl = Model(
name = 'sdxl', name = 'sdxl',
base_provider = 'Stability AI', base_provider = 'Stability AI',
best_provider = IterListProvider([ReplicateHome, DeepInfraImage]) best_provider = IterListProvider([ReplicateHome])
) )
sd_1_5 = Model( sd_1_5 = Model(
name = 'sd-1.5', name = 'sd-1.5',
base_provider = 'Stability AI', base_provider = 'Stability AI',
best_provider = NexraSD15 best_provider = IterListProvider([NexraSD15, GizAI])
) )
@ -771,6 +775,13 @@ sd_3 = Model(
) )
sd_3_5 = Model(
name = 'sd-3.5',
base_provider = 'Stability AI',
best_provider = GizAI
)
### Playground ### ### Playground ###
playground_v2_5 = Model( playground_v2_5 = Model(
name = 'playground-v2.5', name = 'playground-v2.5',
@ -791,7 +802,7 @@ flux = Model(
flux_pro = Model( flux_pro = Model(
name = 'flux-pro', name = 'flux-pro',
base_provider = 'Flux AI', base_provider = 'Flux AI',
best_provider = IterListProvider([AmigoChat, NexraFluxPro]) best_provider = IterListProvider([NexraFluxPro, AmigoChat])
) )
@ -840,7 +851,7 @@ flux_4o = Model(
flux_schnell = Model( flux_schnell = Model(
name = 'flux-schnell', name = 'flux-schnell',
base_provider = 'Flux AI', base_provider = 'Flux AI',
best_provider = ReplicateHome best_provider = IterListProvider([ReplicateHome, GizAI])
) )
@ -1123,6 +1134,7 @@ class ModelUtils:
'sdxl-turbo': sdxl_turbo, 'sdxl-turbo': sdxl_turbo,
'sd-1.5': sd_1_5, 'sd-1.5': sd_1_5,
'sd-3': sd_3, 'sd-3': sd_3,
'sd-3.5': sd_3_5,
### Playground ### ### Playground ###