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 |
+
}
|