summaryrefslogtreecommitdiffstats
path: root/g4f/Provider
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/Provider')
-rw-r--r--g4f/Provider/airforce/AirforceChat.py261
1 files changed, 27 insertions, 234 deletions
diff --git a/g4f/Provider/airforce/AirforceChat.py b/g4f/Provider/airforce/AirforceChat.py
index 8affbe5c..63a0460f 100644
--- a/g4f/Provider/airforce/AirforceChat.py
+++ b/g4f/Provider/airforce/AirforceChat.py
@@ -3,6 +3,7 @@ import re
from aiohttp import ClientSession
import json
from typing import List
+import requests
from ...typing import AsyncResult, Messages
from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
@@ -51,258 +52,50 @@ class AirforceChat(AsyncGeneratorProvider, ProviderModelMixin):
supports_system_message = True
supports_message_history = True
- default_model = 'llama-3-70b-chat'
- text_models = [
- # anthropic
- 'claude-3-haiku-20240307',
- 'claude-3-sonnet-20240229',
- 'claude-3-5-sonnet-20240620',
- 'claude-3-5-sonnet-20241022',
- 'claude-3-opus-20240229',
-
- # openai
- 'chatgpt-4o-latest',
- 'gpt-4',
- 'gpt-4-turbo',
- 'gpt-4o-2024-05-13',
- 'gpt-4o-mini-2024-07-18',
- 'gpt-4o-mini',
- 'gpt-4o-2024-08-06',
- 'gpt-3.5-turbo',
- 'gpt-3.5-turbo-0125',
- 'gpt-3.5-turbo-1106',
- 'gpt-4o',
- 'gpt-4-turbo-2024-04-09',
- 'gpt-4-0125-preview',
- 'gpt-4-1106-preview',
-
- # meta-llama
- default_model,
- 'llama-3-70b-chat-turbo',
- 'llama-3-8b-chat',
- 'llama-3-8b-chat-turbo',
- 'llama-3-70b-chat-lite',
- 'llama-3-8b-chat-lite',
- 'llama-2-13b-chat',
- 'llama-3.1-405b-turbo',
- 'llama-3.1-70b-turbo',
- 'llama-3.1-8b-turbo',
- 'LlamaGuard-2-8b',
- 'llamaguard-7b',
- 'Llama-Vision-Free',
- 'Llama-Guard-7b',
- 'Llama-3.2-90B-Vision-Instruct-Turbo',
- 'Meta-Llama-Guard-3-8B',
- 'Llama-3.2-11B-Vision-Instruct-Turbo',
- 'Llama-Guard-3-11B-Vision-Turbo',
- 'Llama-3.2-3B-Instruct-Turbo',
- 'Llama-3.2-1B-Instruct-Turbo',
- 'llama-2-7b-chat-int8',
- 'llama-2-7b-chat-fp16',
- 'Llama 3.1 405B Instruct',
- 'Llama 3.1 70B Instruct',
- 'Llama 3.1 8B Instruct',
-
- # mistral-ai
- 'Mixtral-8x7B-Instruct-v0.1',
- 'Mixtral-8x22B-Instruct-v0.1',
- 'Mistral-7B-Instruct-v0.1',
- 'Mistral-7B-Instruct-v0.2',
- 'Mistral-7B-Instruct-v0.3',
-
- # Gryphe
- 'MythoMax-L2-13b-Lite',
- 'MythoMax-L2-13b',
-
- # openchat
- 'openchat-3.5-0106',
-
- # qwen
- #'Qwen1.5-72B-Chat', # Empty answer
- #'Qwen1.5-110B-Chat', # Empty answer
- 'Qwen2-72B-Instruct',
- 'Qwen2.5-7B-Instruct-Turbo',
- 'Qwen2.5-72B-Instruct-Turbo',
-
- # google
- 'gemma-2b-it',
- 'gemma-2-9b-it',
- 'gemma-2-27b-it',
-
- # gemini
- 'gemini-1.5-flash',
- 'gemini-1.5-pro',
-
- # databricks
- 'dbrx-instruct',
-
- # deepseek-ai
- 'deepseek-coder-6.7b-base',
- 'deepseek-coder-6.7b-instruct',
- 'deepseek-math-7b-instruct',
-
- # NousResearch
- 'deepseek-math-7b-instruct',
- 'Nous-Hermes-2-Mixtral-8x7B-DPO',
- 'hermes-2-pro-mistral-7b',
-
- # teknium
- 'openhermes-2.5-mistral-7b',
-
- # microsoft
- 'WizardLM-2-8x22B',
- 'phi-2',
-
- # upstage
- 'SOLAR-10.7B-Instruct-v1.0',
-
- # pawan
- 'cosmosrp',
-
- # liquid
- 'lfm-40b-moe',
-
- # DiscoResearch
- 'discolm-german-7b-v1',
-
- # tiiuae
- 'falcon-7b-instruct',
-
- # defog
- 'sqlcoder-7b-2',
-
- # tinyllama
- 'tinyllama-1.1b-chat',
-
- # HuggingFaceH4
- 'zephyr-7b-beta',
- ]
+ default_model = 'llama-3.1-70b-chat'
+ response = requests.get('https://api.airforce/models')
+ data = response.json()
+
+ text_models = [model['id'] for model in data['data']]
models = [*text_models]
model_aliases = {
- # anthropic
- "claude-3-haiku": "claude-3-haiku-20240307",
- "claude-3-sonnet": "claude-3-sonnet-20240229",
- "claude-3.5-sonnet": "claude-3-5-sonnet-20240620",
- "claude-3.5-sonnet": "claude-3-5-sonnet-20241022",
- "claude-3-opus": "claude-3-opus-20240229",
-
- # openai
- "gpt-4o": "chatgpt-4o-latest",
- #"gpt-4": "gpt-4",
- #"gpt-4-turbo": "gpt-4-turbo",
- "gpt-4o": "gpt-4o-2024-05-13",
- "gpt-4o-mini": "gpt-4o-mini-2024-07-18",
- #"gpt-4o-mini": "gpt-4o-mini",
- "gpt-4o": "gpt-4o-2024-08-06",
- "gpt-3.5-turbo": "gpt-3.5-turbo",
- "gpt-3.5-turbo": "gpt-3.5-turbo-0125",
- "gpt-3.5-turbo": "gpt-3.5-turbo-1106",
- #"gpt-4o": "gpt-4o",
- "gpt-4-turbo": "gpt-4-turbo-2024-04-09",
- "gpt-4": "gpt-4-0125-preview",
- "gpt-4": "gpt-4-1106-preview",
-
- # meta-llama
- "llama-3-70b": "llama-3-70b-chat",
- "llama-3-8b": "llama-3-8b-chat",
- "llama-3-8b": "llama-3-8b-chat-turbo",
- "llama-3-70b": "llama-3-70b-chat-lite",
- "llama-3-8b": "llama-3-8b-chat-lite",
- "llama-2-13b": "llama-2-13b-chat",
- "llama-3.1-405b": "llama-3.1-405b-turbo",
- "llama-3.1-70b": "llama-3.1-70b-turbo",
- "llama-3.1-8b": "llama-3.1-8b-turbo",
- "llamaguard-2-8b": "LlamaGuard-2-8b",
- "llamaguard-7b": "llamaguard-7b",
- #"llama_vision_free": "Llama-Vision-Free", # Unknown
- "llamaguard-7b": "Llama-Guard-7b",
- "llama-3.2-90b": "Llama-3.2-90B-Vision-Instruct-Turbo",
- "llamaguard-3-8b": "Meta-Llama-Guard-3-8B",
- "llama-3.2-11b": "Llama-3.2-11B-Vision-Instruct-Turbo",
- "llamaguard-3-11b": "Llama-Guard-3-11B-Vision-Turbo",
- "llama-3.2-3b": "Llama-3.2-3B-Instruct-Turbo",
- "llama-3.2-1b": "Llama-3.2-1B-Instruct-Turbo",
- "llama-2-7b": "llama-2-7b-chat-int8",
- "llama-2-7b": "llama-2-7b-chat-fp16",
- "llama-3.1-405b": "Llama 3.1 405B Instruct",
- "llama-3.1-70b": "Llama 3.1 70B Instruct",
- "llama-3.1-8b": "Llama 3.1 8B Instruct",
-
- # mistral-ai
- "mixtral-8x7b": "Mixtral-8x7B-Instruct-v0.1",
- "mixtral-8x22b": "Mixtral-8x22B-Instruct-v0.1",
- "mixtral-8x7b": "Mistral-7B-Instruct-v0.1",
- "mixtral-8x7b": "Mistral-7B-Instruct-v0.2",
- "mixtral-8x7b": "Mistral-7B-Instruct-v0.3",
-
- # Gryphe
- "mythomax-13b": "MythoMax-L2-13b-Lite",
- "mythomax-13b": "MythoMax-L2-13b",
-
# openchat
"openchat-3.5": "openchat-3.5-0106",
-
- # qwen
- #"qwen-1.5-72b": "Qwen1.5-72B-Chat", # Empty answer
- #"qwen-1.5-110b": "Qwen1.5-110B-Chat", # Empty answer
- "qwen-2-72b": "Qwen2-72B-Instruct",
- "qwen-2-5-7b": "Qwen2.5-7B-Instruct-Turbo",
- "qwen-2-5-72b": "Qwen2.5-72B-Instruct-Turbo",
-
- # google
- "gemma-2b": "gemma-2b-it",
- "gemma-2-9b": "gemma-2-9b-it",
- "gemma-2b-27b": "gemma-2-27b-it",
-
- # gemini
- "gemini-flash": "gemini-1.5-flash",
- "gemini-pro": "gemini-1.5-pro",
-
- # databricks
- "dbrx-instruct": "dbrx-instruct",
-
+
# deepseek-ai
- #"deepseek-coder": "deepseek-coder-6.7b-base",
"deepseek-coder": "deepseek-coder-6.7b-instruct",
- #"deepseek-math": "deepseek-math-7b-instruct",
-
+
# NousResearch
- #"deepseek-math": "deepseek-math-7b-instruct",
"hermes-2-dpo": "Nous-Hermes-2-Mixtral-8x7B-DPO",
- "hermes-2": "hermes-2-pro-mistral-7b",
-
+ "hermes-2-pro": "hermes-2-pro-mistral-7b",
+
# teknium
"openhermes-2.5": "openhermes-2.5-mistral-7b",
-
- # microsoft
- "wizardlm-2-8x22b": "WizardLM-2-8x22B",
- #"phi-2": "phi-2",
-
- # upstage
- "solar-10-7b": "SOLAR-10.7B-Instruct-v1.0",
-
- # pawan
- #"cosmosrp": "cosmosrp",
-
+
# liquid
"lfm-40b": "lfm-40b-moe",
-
+
# DiscoResearch
"german-7b": "discolm-german-7b-v1",
-
- # tiiuae
- #"falcon-7b": "falcon-7b-instruct",
-
- # defog
- #"sqlcoder-7b": "sqlcoder-7b-2",
-
- # tinyllama
- #"tinyllama-1b": "tinyllama-1.1b-chat",
-
+
+ # meta-llama
+ "llama-2-7b": "llama-2-7b-chat-int8",
+ "llama-2-7b": "llama-2-7b-chat-fp16",
+ "llama-3.1-70b": "llama-3.1-70b-chat",
+ "llama-3.1-8b": "llama-3.1-8b-chat",
+ "llama-3.1-70b": "llama-3.1-70b-turbo",
+ "llama-3.1-8b": "llama-3.1-8b-turbo",
+
+ # inferless
+ "neural-7b": "neural-chat-7b-v3-1",
+
# HuggingFaceH4
"zephyr-7b": "zephyr-7b-beta",
+
+ # llmplayground.net
+ #"any-uncensored": "any-uncensored",
}
@classmethod
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-23 15:37:42 +0100