Update api/config.py

api/config.py

@@ -8,23 +8,27 @@ headers = {
     'accept': '*/*',
     'accept-language': 'en-US,en;q=0.9',
     'origin': 'https://www.blackbox.ai',
+    'cache-control': 'no-cache',
+    'pragma': 'no-cache',
     'priority': 'u=1, i',
-    'sec-ch-ua': '"Google Chrome";v="129", "Not=A?Brand";v="8", "Chromium";v="129"',
+    'sec-ch-ua': '"Chromium";v="129", "Not=A?Brand";v="8"',
     'sec-ch-ua-mobile': '?0',
-    'sec-ch-ua-platform': '"Windows"',
+    'sec-ch-ua-platform': '"Linux"',
     'sec-fetch-dest': 'empty',
     'sec-fetch-mode': 'cors',
     'sec-fetch-site': 'same-origin',
-    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
+    'user-agent': 'Mozilla/5.0 (X11; Linux x86_64) '
                   'AppleWebKit/537.36 (KHTML, like Gecko) '
-                  'Chrome/129.0.0.0 Safari/537.36',
+                  'Chrome/129.0.0.0 Safari/537.36'
 }
+
 APP_SECRET = os.getenv("APP_SECRET")
 
+# Define all allowed models, including agent models
 ALLOWED_MODELS = [
     {"id": "blackboxai", "name": "blackboxai"},
     {"id": "blackboxai-pro", "name": "blackboxai-pro"},
-    {"id": "flux", "name": "flux"},
+    {"id": "ImageGeneration", "name": "ImageGeneration"},
     {"id": "llama-3.1-8b", "name": "llama-3.1-8b"},
     {"id": "llama-3.1-70b", "name": "llama-3.1-70b"},
     {"id": "llama-3.1-405b", "name": "llama-3.1-405b"},
@@ -52,10 +56,11 @@ ALLOWED_MODELS = [
     {"id": "claude-3-5-sonnet", "name": "claude-sonnet-3.5"},
 ]
 
+# Mapping of requested model names to actual API model identifiers
 MODEL_MAPPING = {
     "blackboxai": "blackboxai",
     "blackboxai-pro": "blackboxai-pro",
-    "ImageGeneration": "flux",
+    "ImageGeneration": "ImageGeneration",
     "llama-3.1-8b": "llama-3.1-8b",
     "llama-3.1-70b": "llama-3.1-70b",
     "llama-3.1-405b": "llama-3.1-405b",
@@ -87,9 +92,9 @@ MODEL_MAPPING = {
     "claude-3-5-sonnet": "claude-sonnet-3.5",
 }
 
-# Agent modes
+# Agent modes configuration
 AGENT_MODE = {
-    'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
+    'ImageGeneration': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "Image Generation"},
 }
 
 TRENDING_AGENT_MODE = {
@@ -115,3 +120,50 @@ TRENDING_AGENT_MODE = {
     'RepoMap': {'mode': True, 'id': "repomap"},
 }
 
+# User selected models mapping
+USER_SELECTED_MODEL = {
+    "gpt-4o": "gpt-4o",
+    "gemini-pro": "gemini-pro",
+    'claude-sonnet-3.5': "claude-sonnet-3.5",
+}
+
+# Model prefixes for prompt formatting
+MODEL_PREFIXES = {
+    'gpt-4o': '@GPT-4o',
+    'gemini-pro': '@Gemini-PRO',
+    'claude-sonnet-3.5': '@Claude-Sonnet-3.5',
+    'PythonAgent': '@Python Agent',
+    'JavaAgent': '@Java Agent',
+    'JavaScriptAgent': '@JavaScript Agent',
+    'HTMLAgent': '@HTML Agent',
+    'GoogleCloudAgent': '@Google Cloud Agent',
+    'AndroidDeveloper': '@Android Developer',
+    'SwiftDeveloper': '@Swift Developer',
+    'Next.jsAgent': '@Next.js Agent',
+    'MongoDBAgent': '@MongoDB Agent',
+    'PyTorchAgent': '@PyTorch Agent',
+    'ReactAgent': '@React Agent',
+    'XcodeAgent': '@Xcode Agent',
+    'AngularJSAgent': '@AngularJS Agent',
+    'blackboxai-pro': '@BLACKBOXAI-PRO',
+    'ImageGeneration': '@Image Generation',
+}
+
+# Model referers for API requests
+MODEL_REFERERS = {
+    "blackboxai": "/?model=blackboxai",
+    "gpt-4o": "/?model=gpt-4o",
+    "gemini-pro": "/?model=gemini-pro",
+    "claude-sonnet-3.5": "/?model=claude-sonnet-3.5"
+}
+
+# Model aliases for flexibility in model naming
+MODEL_ALIASES = {
+    "gemini-flash": "gemini-1.5-flash",
+    "claude-3.5-sonnet": "claude-sonnet-3.5",
+    "flux": "ImageGeneration",
+    "gemini-1.5-pro-latest": "gemini-pro",
+    "gemini-1.5-pro": "gemini-1.5-pro",
+    "claude-3-5-sonnet-20240620": "claude-sonnet-3.5",
+    "claude-3-5-sonnet": "claude-sonnet-3.5",
+}