Create blackboxai.py
Browse files- api/provider/blackboxai.py +108 -0
api/provider/blackboxai.py
ADDED
@@ -0,0 +1,108 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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': '"Chromium";v="130", "Google Chrome";v="130", "Not?A_Brand";v="99"',
|
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/130.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 |
+
{"id": "Niansuh", "name": "Niansuh"},
|
54 |
+
# GizAI models
|
55 |
+
{"id": "chat-gemini-flash", "name": "chat-gemini-flash"},
|
56 |
+
{"id": "chat-gemini-pro", "name": "chat-gemini-pro"},
|
57 |
+
{"id": "chat-gpt4m", "name": "chat-gpt4m"},
|
58 |
+
{"id": "chat-gpt4", "name": "chat-gpt4"},
|
59 |
+
{"id": "claude-sonnet", "name": "claude-sonnet"},
|
60 |
+
{"id": "claude-haiku", "name": "claude-haiku"},
|
61 |
+
{"id": "llama-3-70b", "name": "llama-3-70b"},
|
62 |
+
{"id": "llama-3-8b", "name": "llama-3-8b"},
|
63 |
+
{"id": "mistral-large", "name": "mistral-large"},
|
64 |
+
{"id": "chat-o1-mini", "name": "chat-o1-mini"},
|
65 |
+
{"id": "flux1", "name": "flux1"},
|
66 |
+
{"id": "sdxl", "name": "sdxl"},
|
67 |
+
{"id": "sd", "name": "sd"},
|
68 |
+
{"id": "sd35", "name": "sd35"},
|
69 |
+
]
|
70 |
+
|
71 |
+
MODEL_MAPPING = {
|
72 |
+
# Existing mappings...
|
73 |
+
# GizAI mappings
|
74 |
+
"chat-gemini-flash": "chat-gemini-flash",
|
75 |
+
"chat-gemini-pro": "chat-gemini-pro",
|
76 |
+
"chat-gpt4m": "chat-gpt4m",
|
77 |
+
"chat-gpt4": "chat-gpt4",
|
78 |
+
"claude-sonnet": "claude-sonnet",
|
79 |
+
"claude-haiku": "claude-haiku",
|
80 |
+
"llama-3-70b": "llama-3-70b",
|
81 |
+
"llama-3-8b": "llama-3-8b",
|
82 |
+
"mistral-large": "mistral-large",
|
83 |
+
"chat-o1-mini": "chat-o1-mini",
|
84 |
+
"flux1": "flux1",
|
85 |
+
"sdxl": "sdxl",
|
86 |
+
"sd": "sd",
|
87 |
+
"sd35": "sd35",
|
88 |
+
# Add any additional mappings if necessary
|
89 |
+
}
|
90 |
+
|
91 |
+
# Agent modes (existing)
|
92 |
+
AGENT_MODE = {
|
93 |
+
'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
|
94 |
+
'Niansuh': {'mode': True, 'id': "NiansuhAIk1HgESy", 'name': "Niansuh"},
|
95 |
+
}
|
96 |
+
|
97 |
+
TRENDING_AGENT_MODE = {
|
98 |
+
# Existing entries...
|
99 |
+
}
|
100 |
+
|
101 |
+
# Model prefixes
|
102 |
+
MODEL_PREFIXES = {
|
103 |
+
# Existing entries...
|
104 |
+
# Add any additional prefixes if necessary
|
105 |
+
}
|
106 |
+
|
107 |
+
# Model referers
|
108 |
+
MODEL_REFERERS =
|