Niansuh commited on
Commit
ffe19dc
·
verified ·
1 Parent(s): ec71f25

Update api/config.py

Browse files
Files changed (1) hide show
  1. 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
- "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
- "gemini-1.5-pro-latest": "gemini-pro",
84
- "gemini-1.5-pro": "gemini-1.5-pro",
85
- "claude-3-5-sonnet-20240620": "claude-sonnet-3.5",
86
- "claude-3-5-sonnet": "claude-sonnet-3.5",
87
- }
88
-
89
- # Agent modes
90
- AGENT_MODE = {
91
- 'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
92
- }
93
-
94
- TRENDING_AGENT_MODE = {
95
- "blackboxai": {},
96
- "gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
97
- "llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
98
- 'llama-3.1-70b': {'mode': True, 'id': "llama-3.1-70b"},
99
- 'llama-3.1-405b': {'mode': True, 'id': "llama-3.1-405b"},
100
- 'blackboxai-pro': {'mode': True, 'id': "BLACKBOXAI-PRO"},
101
- 'PythonAgent': {'mode': True, 'id': "Python Agent"},
102
- 'JavaAgent': {'mode': True, 'id': "Java Agent"},
103
- 'JavaScriptAgent': {'mode': True, 'id': "JavaScript Agent"},
104
- 'HTMLAgent': {'mode': True, 'id': "HTML Agent"},
105
- 'GoogleCloudAgent': {'mode': True, 'id': "Google Cloud Agent"},
106
- 'AndroidDeveloper': {'mode': True, 'id': "Android Developer"},
107
- 'SwiftDeveloper': {'mode': True, 'id': "Swift Developer"},
108
- 'Next.jsAgent': {'mode': True, 'id': "Next.js Agent"},
109
- 'MongoDBAgent': {'mode': True, 'id': "MongoDB Agent"},
110
- 'PyTorchAgent': {'mode': True, 'id': "PyTorch Agent"},
111
- 'ReactAgent': {'mode': True, 'id': "React Agent"},
112
- 'XcodeAgent': {'mode': True, 'id': "Xcode Agent"},
113
- 'AngularJSAgent': {'mode': True, 'id': "AngularJS Agent"},
114
- 'RepoMap': {'mode': True, 'id': "repomap"},
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
+ }