Update api/config.py
Browse files- api/config.py +172 -115
api/config.py
CHANGED
|
@@ -1,115 +1,172 @@
|
|
| 1 |
-
import os
|
| 2 |
-
from dotenv import load_dotenv
|
| 3 |
-
|
| 4 |
-
load_dotenv()
|
| 5 |
-
|
| 6 |
-
BASE_URL = "https://www.blackbox.ai"
|
| 7 |
-
headers = {
|
| 8 |
-
'accept': '*/*',
|
| 9 |
-
'accept-language': 'en-US,en;q=0.9',
|
| 10 |
-
'origin': 'https://www.blackbox.ai',
|
| 11 |
-
'priority': 'u=1, i',
|
| 12 |
-
'sec-ch-ua': '"Google Chrome";v="129", "Not=A?Brand";v="8", "Chromium";v="129"',
|
| 13 |
-
'sec-ch-ua-mobile': '?0',
|
| 14 |
-
'sec-ch-ua-platform': '"Windows"',
|
| 15 |
-
'sec-fetch-dest': 'empty',
|
| 16 |
-
'sec-fetch-mode': 'cors',
|
| 17 |
-
'sec-fetch-site': 'same-origin',
|
| 18 |
-
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
|
| 19 |
-
'AppleWebKit/537.36 (KHTML, like Gecko) '
|
| 20 |
-
'Chrome/129.0.0.0 Safari/537.36',
|
| 21 |
-
}
|
| 22 |
-
APP_SECRET = os.getenv("APP_SECRET")
|
| 23 |
-
|
| 24 |
-
ALLOWED_MODELS = [
|
| 25 |
-
{"id": "blackboxai", "name": "blackboxai"},
|
| 26 |
-
{"id": "blackboxai-pro", "name": "blackboxai-pro"},
|
| 27 |
-
{"id": "flux", "name": "flux"},
|
| 28 |
-
{"id": "llama-3.1-8b", "name": "llama-3.1-8b"},
|
| 29 |
-
{"id": "llama-3.1-70b", "name": "llama-3.1-70b"},
|
| 30 |
-
{"id": "llama-3.1-405b", "name": "llama-3.1-405b"},
|
| 31 |
-
{"id": "gpt-4o", "name": "gpt-4o"},
|
| 32 |
-
{"id": "gemini-pro", "name": "gemini-pro"},
|
| 33 |
-
{"id": "gemini-1.5-flash", "name": "gemini-1.5-flash"},
|
| 34 |
-
{"id": "claude-sonnet-3.5", "name": "claude-sonnet-3.5"},
|
| 35 |
-
{"id": "PythonAgent", "name": "PythonAgent"},
|
| 36 |
-
{"id": "JavaAgent", "name": "JavaAgent"},
|
| 37 |
-
{"id": "JavaScriptAgent", "name": "JavaScriptAgent"},
|
| 38 |
-
{"id": "HTMLAgent", "name": "HTMLAgent"},
|
| 39 |
-
{"id": "GoogleCloudAgent", "name": "GoogleCloudAgent"},
|
| 40 |
-
{"id": "AndroidDeveloper", "name": "AndroidDeveloper"},
|
| 41 |
-
{"id": "SwiftDeveloper", "name": "SwiftDeveloper"},
|
| 42 |
-
{"id": "Next.jsAgent", "name": "Next.jsAgent"},
|
| 43 |
-
{"id": "MongoDBAgent", "name": "MongoDBAgent"},
|
| 44 |
-
{"id": "PyTorchAgent", "name": "PyTorchAgent"},
|
| 45 |
-
{"id": "ReactAgent", "name": "ReactAgent"},
|
| 46 |
-
{"id": "XcodeAgent", "name": "XcodeAgent"},
|
| 47 |
-
{"id": "AngularJSAgent", "name": "AngularJSAgent"},
|
| 48 |
-
{"id": "RepoMap", "name": "RepoMap"},
|
| 49 |
-
{"id": "gemini-1.5-pro-latest", "name": "gemini-pro"},
|
| 50 |
-
{"id": "gemini-1.5-pro", "name": "gemini-1.5-pro"},
|
| 51 |
-
{"id": "claude-3-5-sonnet-20240620", "name": "claude-sonnet-3.5"},
|
| 52 |
-
{"id": "claude-3-5-sonnet", "name": "claude-sonnet-3.5"},
|
| 53 |
-
]
|
| 54 |
-
|
| 55 |
-
MODEL_MAPPING = {
|
| 56 |
-
"blackboxai": "blackboxai",
|
| 57 |
-
"blackboxai-pro": "blackboxai-pro",
|
| 58 |
-
"
|
| 59 |
-
"
|
| 60 |
-
"llama-3.1-
|
| 61 |
-
"llama-3.1-
|
| 62 |
-
"
|
| 63 |
-
"
|
| 64 |
-
"gemini-
|
| 65 |
-
"
|
| 66 |
-
"
|
| 67 |
-
"
|
| 68 |
-
"
|
| 69 |
-
"
|
| 70 |
-
"
|
| 71 |
-
"
|
| 72 |
-
"
|
| 73 |
-
"
|
| 74 |
-
"
|
| 75 |
-
"
|
| 76 |
-
"
|
| 77 |
-
"
|
| 78 |
-
"
|
| 79 |
-
"
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
"
|
| 83 |
-
"
|
| 84 |
-
"
|
| 85 |
-
"
|
| 86 |
-
"
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
| 94 |
-
|
| 95 |
-
"
|
| 96 |
-
"
|
| 97 |
-
"
|
| 98 |
-
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
'
|
| 113 |
-
'
|
| 114 |
-
'
|
| 115 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
from dotenv import load_dotenv
|
| 3 |
+
|
| 4 |
+
load_dotenv()
|
| 5 |
+
|
| 6 |
+
BASE_URL = "https://www.blackbox.ai"
|
| 7 |
+
headers = {
|
| 8 |
+
'accept': '*/*',
|
| 9 |
+
'accept-language': 'en-US,en;q=0.9',
|
| 10 |
+
'origin': 'https://www.blackbox.ai',
|
| 11 |
+
'priority': 'u=1, i',
|
| 12 |
+
'sec-ch-ua': '"Google Chrome";v="129", "Not=A?Brand";v="8", "Chromium";v="129"',
|
| 13 |
+
'sec-ch-ua-mobile': '?0',
|
| 14 |
+
'sec-ch-ua-platform': '"Windows"',
|
| 15 |
+
'sec-fetch-dest': 'empty',
|
| 16 |
+
'sec-fetch-mode': 'cors',
|
| 17 |
+
'sec-fetch-site': 'same-origin',
|
| 18 |
+
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
|
| 19 |
+
'AppleWebKit/537.36 (KHTML, like Gecko) '
|
| 20 |
+
'Chrome/129.0.0.0 Safari/537.36',
|
| 21 |
+
}
|
| 22 |
+
APP_SECRET = os.getenv("APP_SECRET")
|
| 23 |
+
|
| 24 |
+
ALLOWED_MODELS = [
|
| 25 |
+
{"id": "blackboxai", "name": "blackboxai"},
|
| 26 |
+
{"id": "blackboxai-pro", "name": "blackboxai-pro"},
|
| 27 |
+
{"id": "flux", "name": "flux"},
|
| 28 |
+
{"id": "llama-3.1-8b", "name": "llama-3.1-8b"},
|
| 29 |
+
{"id": "llama-3.1-70b", "name": "llama-3.1-70b"},
|
| 30 |
+
{"id": "llama-3.1-405b", "name": "llama-3.1-405b"},
|
| 31 |
+
{"id": "gpt-4o", "name": "gpt-4o"},
|
| 32 |
+
{"id": "gemini-pro", "name": "gemini-pro"},
|
| 33 |
+
{"id": "gemini-1.5-flash", "name": "gemini-1.5-flash"},
|
| 34 |
+
{"id": "claude-sonnet-3.5", "name": "claude-sonnet-3.5"},
|
| 35 |
+
{"id": "PythonAgent", "name": "PythonAgent"},
|
| 36 |
+
{"id": "JavaAgent", "name": "JavaAgent"},
|
| 37 |
+
{"id": "JavaScriptAgent", "name": "JavaScriptAgent"},
|
| 38 |
+
{"id": "HTMLAgent", "name": "HTMLAgent"},
|
| 39 |
+
{"id": "GoogleCloudAgent", "name": "GoogleCloudAgent"},
|
| 40 |
+
{"id": "AndroidDeveloper", "name": "AndroidDeveloper"},
|
| 41 |
+
{"id": "SwiftDeveloper", "name": "SwiftDeveloper"},
|
| 42 |
+
{"id": "Next.jsAgent", "name": "Next.jsAgent"},
|
| 43 |
+
{"id": "MongoDBAgent", "name": "MongoDBAgent"},
|
| 44 |
+
{"id": "PyTorchAgent", "name": "PyTorchAgent"},
|
| 45 |
+
{"id": "ReactAgent", "name": "ReactAgent"},
|
| 46 |
+
{"id": "XcodeAgent", "name": "XcodeAgent"},
|
| 47 |
+
{"id": "AngularJSAgent", "name": "AngularJSAgent"},
|
| 48 |
+
{"id": "RepoMap", "name": "RepoMap"},
|
| 49 |
+
{"id": "gemini-1.5-pro-latest", "name": "gemini-pro"},
|
| 50 |
+
{"id": "gemini-1.5-pro", "name": "gemini-1.5-pro"},
|
| 51 |
+
{"id": "claude-3-5-sonnet-20240620", "name": "claude-sonnet-3.5"},
|
| 52 |
+
{"id": "claude-3-5-sonnet", "name": "claude-sonnet-3.5"},
|
| 53 |
+
]
|
| 54 |
+
|
| 55 |
+
MODEL_MAPPING = {
|
| 56 |
+
"blackboxai": "blackboxai",
|
| 57 |
+
"blackboxai-pro": "blackboxai-pro",
|
| 58 |
+
"flux": "flux",
|
| 59 |
+
"ImageGeneration": "flux",
|
| 60 |
+
"llama-3.1-8b": "llama-3.1-8b",
|
| 61 |
+
"llama-3.1-70b": "llama-3.1-70b",
|
| 62 |
+
"llama-3.1-405b": "llama-3.1-405b",
|
| 63 |
+
"gpt-4o": "gpt-4o",
|
| 64 |
+
"gemini-pro": "gemini-pro",
|
| 65 |
+
"gemini-1.5-flash": "gemini-1.5-flash",
|
| 66 |
+
"claude-sonnet-3.5": "claude-sonnet-3.5",
|
| 67 |
+
"PythonAgent": "PythonAgent",
|
| 68 |
+
"JavaAgent": "JavaAgent",
|
| 69 |
+
"JavaScriptAgent": "JavaScriptAgent",
|
| 70 |
+
"HTMLAgent": "HTMLAgent",
|
| 71 |
+
"GoogleCloudAgent": "GoogleCloudAgent",
|
| 72 |
+
"AndroidDeveloper": "AndroidDeveloper",
|
| 73 |
+
"SwiftDeveloper": "SwiftDeveloper",
|
| 74 |
+
"Next.jsAgent": "Next.jsAgent",
|
| 75 |
+
"MongoDBAgent": "MongoDBAgent",
|
| 76 |
+
"PyTorchAgent": "PyTorchAgent",
|
| 77 |
+
"ReactAgent": "ReactAgent",
|
| 78 |
+
"XcodeAgent": "XcodeAgent",
|
| 79 |
+
"AngularJSAgent": "AngularJSAgent",
|
| 80 |
+
"RepoMap": "RepoMap",
|
| 81 |
+
# Additional mappings
|
| 82 |
+
"gemini-flash": "gemini-1.5-flash",
|
| 83 |
+
"claude-3.5-sonnet": "claude-sonnet-3.5",
|
| 84 |
+
"flux": "flux",
|
| 85 |
+
"gemini-1.5-pro-latest": "gemini-pro",
|
| 86 |
+
"gemini-1.5-pro": "gemini-1.5-pro",
|
| 87 |
+
"claude-3-5-sonnet-20240620": "claude-sonnet-3.5",
|
| 88 |
+
"claude-3-5-sonnet": "claude-sonnet-3.5",
|
| 89 |
+
}
|
| 90 |
+
|
| 91 |
+
USER_SELECTED_MODEL = {
|
| 92 |
+
"gpt-4o": "gpt-4o",
|
| 93 |
+
"gemini-pro": "gemini-pro",
|
| 94 |
+
'claude-sonnet-3.5': "claude-sonnet-3.5",
|
| 95 |
+
"PythonAgent": "PythonAgent",
|
| 96 |
+
"JavaAgent": "JavaAgent",
|
| 97 |
+
"JavaScriptAgent": "JavaScriptAgent",
|
| 98 |
+
"HTMLAgent": "HTMLAgent",
|
| 99 |
+
"GoogleCloudAgent": "GoogleCloudAgent",
|
| 100 |
+
"AndroidDeveloper": "AndroidDeveloper",
|
| 101 |
+
"SwiftDeveloper": "SwiftDeveloper",
|
| 102 |
+
"Next.jsAgent": "Next.jsAgent",
|
| 103 |
+
"MongoDBAgent": "MongoDBAgent",
|
| 104 |
+
"PyTorchAgent": "PyTorchAgent",
|
| 105 |
+
"ReactAgent": "ReactAgent",
|
| 106 |
+
"XcodeAgent": "XcodeAgent",
|
| 107 |
+
"AngularJSAgent": "AngularJSAgent",
|
| 108 |
+
"RepoMap": "RepoMap",
|
| 109 |
+
}
|
| 110 |
+
|
| 111 |
+
MODEL_PREFIXES = {
|
| 112 |
+
'gpt-4o': '@GPT-4o',
|
| 113 |
+
'gemini-pro': '@Gemini-PRO',
|
| 114 |
+
'claude-sonnet-3.5': '@Claude-Sonnet-3.5',
|
| 115 |
+
'PythonAgent': '@Python Agent',
|
| 116 |
+
'JavaAgent': '@Java Agent',
|
| 117 |
+
'JavaScriptAgent': '@JavaScript Agent',
|
| 118 |
+
'HTMLAgent': '@HTML Agent',
|
| 119 |
+
'GoogleCloudAgent': '@Google Cloud Agent',
|
| 120 |
+
'AndroidDeveloper': '@Android Developer',
|
| 121 |
+
'SwiftDeveloper': '@Swift Developer',
|
| 122 |
+
'Next.jsAgent': '@Next.js Agent',
|
| 123 |
+
'MongoDBAgent': '@MongoDB Agent',
|
| 124 |
+
'PyTorchAgent': '@PyTorch Agent',
|
| 125 |
+
'ReactAgent': '@React Agent',
|
| 126 |
+
'XcodeAgent': '@Xcode Agent',
|
| 127 |
+
'AngularJSAgent': '@AngularJS Agent',
|
| 128 |
+
'blackboxai-pro': '@BLACKBOXAI-PRO',
|
| 129 |
+
'ImageGeneration': '@Image Generation',
|
| 130 |
+
}
|
| 131 |
+
|
| 132 |
+
MODEL_REFERERS = {
|
| 133 |
+
"blackboxai": "/?model=blackboxai",
|
| 134 |
+
"gpt-4o": "/?model=gpt-4o",
|
| 135 |
+
"gemini-pro": "/?model=gemini-pro",
|
| 136 |
+
"claude-sonnet-3.5": "/?model=claude-sonnet-3.5",
|
| 137 |
+
# Add more referers if needed
|
| 138 |
+
}
|
| 139 |
+
|
| 140 |
+
MODEL_ALIASES = {
|
| 141 |
+
"gemini-flash": "gemini-1.5-flash",
|
| 142 |
+
"claude-3.5-sonnet": "claude-sonnet-3.5",
|
| 143 |
+
"flux": "ImageGeneration",
|
| 144 |
+
}
|
| 145 |
+
|
| 146 |
+
# Agent modes
|
| 147 |
+
AGENT_MODE = {
|
| 148 |
+
'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
|
| 149 |
+
}
|
| 150 |
+
|
| 151 |
+
TRENDING_AGENT_MODE = {
|
| 152 |
+
"blackboxai": {},
|
| 153 |
+
"gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
|
| 154 |
+
"llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
|
| 155 |
+
'llama-3.1-70b': {'mode': True, 'id': "llama-3.1-70b"},
|
| 156 |
+
'llama-3.1-405b': {'mode': True, 'id': "llama-3.1-405b"},
|
| 157 |
+
'blackboxai-pro': {'mode': True, 'id': "BLACKBOXAI-PRO"},
|
| 158 |
+
'PythonAgent': {'mode': True, 'id': "Python Agent"},
|
| 159 |
+
'JavaAgent': {'mode': True, 'id': "Java Agent"},
|
| 160 |
+
'JavaScriptAgent': {'mode': True, 'id': "JavaScript Agent"},
|
| 161 |
+
'HTMLAgent': {'mode': True, 'id': "HTML Agent"},
|
| 162 |
+
'GoogleCloudAgent': {'mode': True, 'id': "Google Cloud Agent"},
|
| 163 |
+
'AndroidDeveloper': {'mode': True, 'id': "Android Developer"},
|
| 164 |
+
'SwiftDeveloper': {'mode': True, 'id': "Swift Developer"},
|
| 165 |
+
'Next.jsAgent': {'mode': True, 'id': "Next.js Agent"},
|
| 166 |
+
'MongoDBAgent': {'mode': True, 'id': "MongoDB Agent"},
|
| 167 |
+
'PyTorchAgent': {'mode': True, 'id': "PyTorch Agent"},
|
| 168 |
+
'ReactAgent': {'mode': True, 'id': "React Agent"},
|
| 169 |
+
'XcodeAgent': {'mode': True, 'id': "Xcode Agent"},
|
| 170 |
+
'AngularJSAgent': {'mode': True, 'id': "AngularJS Agent"},
|
| 171 |
+
'RepoMap': {'mode': True, 'id': "repomap"},
|
| 172 |
+
}
|