Update api/config.py
Browse files- api/config.py +117 -115
api/config.py
CHANGED
@@ -1,115 +1,117 @@
|
|
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 |
-
"ImageGeneration": "flux",
|
59 |
-
"llama-3.1-8b": "llama-3.1-8b",
|
60 |
-
"llama-3.1-70b": "llama-3.1-70b",
|
61 |
-
"llama-3.1-405b": "llama-3.1-405b",
|
62 |
-
"gpt-4o": "gpt-4o",
|
63 |
-
"gemini-pro": "gemini-pro",
|
64 |
-
"gemini-1.5-flash": "gemini-1.5-flash",
|
65 |
-
"claude-sonnet-3.5": "claude-sonnet-3.5",
|
66 |
-
"PythonAgent": "PythonAgent",
|
67 |
-
"JavaAgent": "JavaAgent",
|
68 |
-
"JavaScriptAgent": "JavaScriptAgent",
|
69 |
-
"HTMLAgent": "HTMLAgent",
|
70 |
-
"GoogleCloudAgent": "GoogleCloudAgent",
|
71 |
-
"AndroidDeveloper": "AndroidDeveloper",
|
72 |
-
"SwiftDeveloper": "SwiftDeveloper",
|
73 |
-
"Next.jsAgent": "Next.jsAgent",
|
74 |
-
"MongoDBAgent": "MongoDBAgent",
|
75 |
-
"PyTorchAgent": "PyTorchAgent",
|
76 |
-
"ReactAgent": "ReactAgent",
|
77 |
-
"XcodeAgent": "XcodeAgent",
|
78 |
-
"AngularJSAgent": "AngularJSAgent",
|
79 |
-
"RepoMap": "RepoMap",
|
80 |
-
# Additional mappings
|
81 |
-
"gemini-flash": "gemini-1.5-flash",
|
82 |
-
"claude-3.5-sonnet": "claude-sonnet-3.5",
|
83 |
-
"
|
84 |
-
"gemini-1.5-pro": "gemini-
|
85 |
-
"
|
86 |
-
"claude-3-5-sonnet": "claude-sonnet-3.5",
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
}
|
93 |
-
|
94 |
-
|
95 |
-
|
96 |
-
"
|
97 |
-
"
|
98 |
-
|
99 |
-
'llama-3.1-
|
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 |
+
"ImageGeneration": "flux",
|
59 |
+
"llama-3.1-8b": "llama-3.1-8b",
|
60 |
+
"llama-3.1-70b": "llama-3.1-70b",
|
61 |
+
"llama-3.1-405b": "llama-3.1-405b",
|
62 |
+
"gpt-4o": "gpt-4o",
|
63 |
+
"gemini-pro": "gemini-pro",
|
64 |
+
"gemini-1.5-flash": "gemini-1.5-flash",
|
65 |
+
"claude-sonnet-3.5": "claude-sonnet-3.5",
|
66 |
+
"PythonAgent": "PythonAgent",
|
67 |
+
"JavaAgent": "JavaAgent",
|
68 |
+
"JavaScriptAgent": "JavaScriptAgent",
|
69 |
+
"HTMLAgent": "HTMLAgent",
|
70 |
+
"GoogleCloudAgent": "GoogleCloudAgent",
|
71 |
+
"AndroidDeveloper": "AndroidDeveloper",
|
72 |
+
"SwiftDeveloper": "SwiftDeveloper",
|
73 |
+
"Next.jsAgent": "Next.jsAgent",
|
74 |
+
"MongoDBAgent": "MongoDBAgent",
|
75 |
+
"PyTorchAgent": "PyTorchAgent",
|
76 |
+
"ReactAgent": "ReactAgent",
|
77 |
+
"XcodeAgent": "XcodeAgent",
|
78 |
+
"AngularJSAgent": "AngularJSAgent",
|
79 |
+
"RepoMap": "RepoMap",
|
80 |
+
# Additional mappings
|
81 |
+
"gemini-flash": "gemini-1.5-flash",
|
82 |
+
"claude-3.5-sonnet": "claude-sonnet-3.5",
|
83 |
+
"flux": "ImageGeneration",
|
84 |
+
"gemini-1.5-pro-latest": "gemini-pro",
|
85 |
+
"gemini-1.5-pro": "gemini-1.5-pro",
|
86 |
+
"claude-3-5-sonnet-20240620": "claude-sonnet-3.5",
|
87 |
+
"claude-3-5-sonnet": "claude-sonnet-3.5",
|
88 |
+
}
|
89 |
+
|
90 |
+
# Agent modes
|
91 |
+
AGENT_MODE = {
|
92 |
+
'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
|
93 |
+
}
|
94 |
+
|
95 |
+
TRENDING_AGENT_MODE = {
|
96 |
+
"blackboxai": {},
|
97 |
+
"gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
|
98 |
+
"llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
|
99 |
+
'llama-3.1-70b': {'mode': True, 'id': "llama-3.1-70b"},
|
100 |
+
'llama-3.1-405b': {'mode': True, 'id': "llama-3.1-405b"},
|
101 |
+
'blackboxai-pro': {'mode': True, 'id': "BLACKBOXAI-PRO"},
|
102 |
+
'PythonAgent': {'mode': True, 'id': "Python Agent"},
|
103 |
+
'JavaAgent': {'mode': True, 'id': "Java Agent"},
|
104 |
+
'JavaScriptAgent': {'mode': True, 'id': "JavaScript Agent"},
|
105 |
+
'HTMLAgent': {'mode': True, 'id': "HTML Agent"},
|
106 |
+
'GoogleCloudAgent': {'mode': True, 'id': "Google Cloud Agent"},
|
107 |
+
'AndroidDeveloper': {'mode': True, 'id': "Android Developer"},
|
108 |
+
'SwiftDeveloper': {'mode': True, 'id': "Swift Developer"},
|
109 |
+
'Next.jsAgent': {'mode': True, 'id': "Next.js Agent"},
|
110 |
+
'MongoDBAgent': {'mode': True, 'id': "MongoDB Agent"},
|
111 |
+
'PyTorchAgent': {'mode': True, 'id': "PyTorch Agent"},
|
112 |
+
'ReactAgent': {'mode': True, 'id': "React Agent"},
|
113 |
+
'XcodeAgent': {'mode': True, 'id': "Xcode Agent"},
|
114 |
+
'AngularJSAgent': {'mode': True, 'id': "AngularJS Agent"},
|
115 |
+
'RepoMap': {'mode': True, 'id': "repomap"},
|
116 |
+
}
|
117 |
+
|