Update api/config.py

api/config.py

@@ -4,6 +4,9 @@ from dotenv import load_dotenv
 load_dotenv()
 
 BASE_URL = "https://www.blackbox.ai"
+APP_SECRET = os.getenv("APP_SECRET")
+
+# Headers for HTTP requests
 headers = {
     'accept': '*/*',
     'accept-language': 'zh-CN,zh;q=0.9',
@@ -18,11 +21,24 @@ 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',
 }
-APP_SECRET = os.getenv("APP_SECRET")
 
+# Default model configuration
 default_model = 'blackboxai'
-image_models = ['Image Generation', 'repomap']
 userSelectedModel = ['gpt-4o', 'gemini-pro', 'claude-sonnet-3.5', 'blackboxai-pro']
+image_models = ['Image Generation', 'repomap']
+
+# Allowed models including all specified agent and trending models
+ALLOWED_MODELS = [
+    {"id": "gpt-4o", "name": "gpt-4o"},
+    {"id": "gemini-1.5-pro-latest", "name": "gemini-pro"},
+    {"id": "gemini-1.5-pro", "name": "gemini-pro"},
+    {"id": "gemini-pro", "name": "gemini-pro"},
+    {"id": "claude-3-5-sonnet-20240620", "name": "claude-sonnet-3.5"},
+    {"id": "claude-3-5-sonnet", "name": "claude-sonnet-3.5"},
+    {"id": "blackboxai-pro", "name": "BLACKBOXAI-PRO"},
+    {"id": "repomap", "name": "repomap"},
+    # Add any additional agent models here
+]
 
 # Agent and trending agent modes
 agentMode = {
@@ -46,7 +62,6 @@ 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"},