Update api/config.py

api/config.py

@@ -1,15 +1,9 @@
-# api/config.py
-
 import os
 from dotenv import load_dotenv
 
-# Load environment variables from .env file
 load_dotenv()
 
-# Base URL for external API
 BASE_URL = "https://www.blackbox.ai"
-
-# Common headers for HTTP requests
 headers = {
     'accept': '*/*',
     'accept-language': 'zh-CN,zh;q=0.9',
@@ -24,30 +18,21 @@ headers = {
     'sec-fetch-site': 'same-origin',
     'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36',
 }
-
-# Application secret key for authentication
 APP_SECRET = os.getenv("APP_SECRET")
 
-# Default model used by the application
 default_model = 'blackboxai'
-
-# Image-related models
 image_models = ['Image Generation', 'repomap']
-
-# User-selected models
 userSelectedModel = ['gpt-4o', 'gemini-pro', 'claude-sonnet-3.5', 'blackboxai-pro']
 
-# Agent modes with their configurations
+# Agent and trending agent modes
 agentMode = {
     'Image Generation': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "Image Generation"},
 }
-
-# Trending agent modes with their configurations
 trendingAgentMode = {
     "gemini-1.5-flash": {'mode': True, 'id': 'Gemini'},
     "llama-3.1-8b": {'mode': True, 'id': "llama-3.1-8b"},
-    'llama-3.1-70b': {'mode': True, 'id': "llama-3.1-70b"},
-    'llama-3.1-405b': {'mode': True, 'id': "llama-3.1-405"},
+    "llama-3.1-70b": {'mode': True, 'id': "llama-3.1-70b"},
+    "llama-3.1-405b": {'mode': True, 'id': "llama-3.1-405b"},
     'Python Agent': {'mode': True, 'id': "Python Agent"},
     'Java Agent': {'mode': True, 'id': "Java Agent"},
     'JavaScript Agent': {'mode': True, 'id': "JavaScript Agent"},
@@ -61,6 +46,7 @@ trendingAgentMode = {
     'React Agent': {'mode': True, 'id': "React Agent"},
     'Xcode Agent': {'mode': True, 'id': "Xcode Agent"},
     'AngularJS Agent': {'mode': True, 'id': "AngularJS Agent"},
+    'blackboxai-pro': {'mode': True, 'id': "BLACKBOXAI-PRO"},
     'repomap': {'mode': True, 'id': "repomap"},
     'Heroku Agent': {'mode': True, 'id': "Heroku Agent"},
     'Godot Agent': {'mode': True, 'id': "Godot Agent"},
@@ -81,95 +67,15 @@ trendingAgentMode = {
     'builder Agent': {'mode': True, 'id': "builder Agent"},
 }
 
-# Prefixes for trending agent modes (excluding specific models)
-model_prefixes = {
-    mode: f"@{value['id']}"
-    for mode, value in trendingAgentMode.items()
-    if mode not in ["gemini-1.5-flash", "llama-3.1-8b", "llama-3.1-70b", "llama-3.1-405b", "repomap"]
-}
+# Model prefixes for trending agent modes
+model_prefixes = {mode: f"@{value['id']}" for mode, value in trendingAgentMode.items() if mode not in ["gemini-1.5-flash", "llama-3.1-8b", "llama-3.1-70b", "llama-3.1-405b", "repomap"]}
 
-# Comprehensive list of all models
-models = [
-    default_model,
-    *userSelectedModel,
-    *list(agentMode.keys()),
-    *list(trendingAgentMode.keys())
-]
+# Consolidated models list
+models = [default_model, *userSelectedModel, *list(agentMode.keys()), *list(trendingAgentMode.keys())]
 
-# Aliases for certain models to handle alternative naming
+# Aliases for model names
 model_aliases = {
     "gemini-flash": "gemini-1.5-flash",
     "claude-3.5-sonnet": "claude-sonnet-3.5",
     "flux": "Image Generation",
 }
-
-# Allowed models for API requests
-ALLOWED_MODELS = [
-    {"id": default_model, "name": "BlackboxAI"},
-    {"id": "gpt-4o", "name": "GPT-4o"},
-    {"id": "gemini-pro", "name": "Gemini Pro"},
-    {"id": "claude-sonnet-3.5", "name": "Claude Sonnet 3.5"},
-    {"id": "blackboxai-pro", "name": "BlackboxAI-Pro"},
-    {"id": "Image Generation", "name": "Image Generation"},
-    {"id": "repomap", "name": "Repomap"},
-]
-
-# Add agentMode models to ALLOWED_MODELS
-for model_key, model_info in agentMode.items():
-    ALLOWED_MODELS.append({"id": model_info['id'], "name": model_info['name']})
-
-# Add trendingAgentMode models to ALLOWED_MODELS
-for model_key, model_info in trendingAgentMode.items():
-    ALLOWED_MODELS.append({"id": model_info['id'], "name": model_key})
-
-# Mapping from user-provided model identifiers to internal model identifiers
-MODEL_MAPPING = {
-    "blackboxai": "blackboxai",
-    "gpt-4o": "gpt-4o",
-    "gemini-pro": "gemini-pro",
-    "claude-sonnet-3.5": "claude-sonnet-3.5",
-    "blackboxai-pro": "BLACKBOXAI-PRO",
-    "Image Generation": "ImageGenerationLV45LJp",
-    "repomap": "repomap",
-    "gemini-1.5-flash": "Gemini",
-    "llama-3.1-8b": "llama-3.1-8b",
-    "llama-3.1-70b": "llama-3.1-70b",
-    "llama-3.1-405b": "llama-3.1-405",
-    "Python Agent": "Python Agent",
-    "Java Agent": "Java Agent",
-    "JavaScript Agent": "JavaScript Agent",
-    "HTML Agent": "HTML Agent",
-    "Google Cloud Agent": "Google Cloud Agent",
-    "Android Developer": "Android Developer",
-    "Swift Developer": "Swift Developer",
-    "Next.js Agent": "Next.js Agent",
-    "MongoDB Agent": "MongoDB Agent",
-    "PyTorch Agent": "PyTorch Agent",
-    "React Agent": "React Agent",
-    "Xcode Agent": "Xcode Agent",
-    "AngularJS Agent": "AngularJS Agent",
-    "Heroku Agent": "Heroku Agent",
-    "Godot Agent": "Godot Agent",
-    "Go Agent": "Go Agent",
-    "Gitlab Agent": "Gitlab Agent",
-    "Git Agent": "Git Agent",
-    "Flask Agent": "Flask Agent",
-    "Firebase Agent": "Firebase Agent",
-    "FastAPI Agent": "FastAPI Agent",
-    "Erlang Agent": "Erlang Agent",
-    "Electron Agent": "Electron Agent",
-    "Docker Agent": "Docker Agent",
-    "DigitalOcean Agent": "DigitalOcean Agent",
-    "Bitbucket Agent": "Bitbucket Agent",
-    "Azure Agent": "Azure Agent",
-    "Flutter Agent": "Flutter Agent",
-    "Youtube Agent": "Youtube Agent",
-    "builder Agent": "builder Agent",
-}
-
-# Reverse mapping for model aliases
-REVERSE_MODEL_MAPPING = {alias: original for alias, original in model_aliases.items()}
-
-# Function to resolve model aliases
-def resolve_model(model_name: str) -> str:
-    return REVERSE_MODEL_MAPPING.get(model_name, model_name)