Update api/config.py

api/config.py

@@ -3,35 +3,35 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
-# Base URL and endpoint configurations
+# Base URL and Common Headers
 BASE_URL = "https://www.blackbox.ai"
 API_ENDPOINT = f"{BASE_URL}/api/chat"
 
-# Custom headers for requests
-def get_headers():
-    return {
-        'accept': '*/*',
-        'accept-language': 'en-US,en;q=0.9',
-        'cache-control': 'no-cache',
-        'content-type': 'application/json',
-        'origin': BASE_URL,
-        'pragma': 'no-cache',
-        'priority': 'u=1, i',
-        'referer': f'{BASE_URL}/',
-        'sec-ch-ua': '"Chromium";v="130", "Google Chrome";v="130", "Not?A_Brand";v="99"',
-        'sec-ch-ua-mobile': '?0',
-        'sec-ch-ua-platform': '"Windows"',
-        'sec-fetch-dest': 'empty',
-        'sec-fetch-mode': 'cors',
-        'sec-fetch-site': 'same-origin',
-        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
-                      'AppleWebKit/537.36 (KHTML, like Gecko) '
-                      'Chrome/130.0.0.0 Safari/537.36',
-    }
+common_headers = {
+    'accept': '*/*',
+    'accept-language': 'en-US,en;q=0.9',
+    'cache-control': 'no-cache',
+    'content-type': 'application/json',
+    'origin': BASE_URL,
+    'pragma': 'no-cache',
+    'priority': 'u=1, i',
+    'referer': f'{BASE_URL}/',
+    'sec-ch-ua': '"Chromium";v="130", "Google Chrome";v="130", "Not?A_Brand";v="99"',
+    'sec-ch-ua-mobile': '?0',
+    'sec-ch-ua-platform': '"Windows"',
+    'sec-fetch-dest': 'empty',
+    'sec-fetch-mode': 'cors',
+    'sec-fetch-site': 'same-origin',
+    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.0.0 Safari/537.36',
+}
+
+# Header Configurations for Specific API Calls
+def get_headers_api_chat(referer_url):
+    return {**common_headers, 'Referer': referer_url}
 
-# Model configurations
 APP_SECRET = os.getenv("APP_SECRET")
 
+# Allowed Models and Agent Modes
 ALLOWED_MODELS = [
     {"id": "blackboxai", "name": "blackboxai"},
     {"id": "blackboxai-pro", "name": "blackboxai-pro"},
@@ -62,14 +62,6 @@ ALLOWED_MODELS = [
     {"id": "GitlabAgent", "name": "GitlabAgent"},
     {"id": "GitAgent", "name": "GitAgent"},
     {"id": "RepoMap", "name": "RepoMap"},
-    {"id": "gemini-1.5-pro-latest", "name": "gemini-pro"},
-    {"id": "gemini-1.5-pro", "name": "gemini-1.5-pro"},
-    {"id": "claude-3-5-sonnet-20240620", "name": "claude-sonnet-3.5"},
-    {"id": "claude-3-5-sonnet", "name": "claude-sonnet-3.5"},
-    {"id": "Niansuh", "name": "Niansuh"},
-    {"id": "o1-preview", "name": "o1-preview"},
-    
-    # Added New Agents
     {"id": "FlaskAgent", "name": "FlaskAgent"},
     {"id": "FirebaseAgent", "name": "FirebaseAgent"},
     {"id": "FastAPIAgent", "name": "FastAPIAgent"},
@@ -84,10 +76,6 @@ ALLOWED_MODELS = [
     {"id": "builderAgent", "name": "builderAgent"},
 ]
 
-# Mapping of models for streamlined access
-MODEL_MAPPING = {model['id']: model['name'] for model in ALLOWED_MODELS}
-
-# Agent modes
 AGENT_MODE = {
     'flux': {'mode': True, 'id': "ImageGenerationLV45LJp", 'name': "flux"},
     'Niansuh': {'mode': True, 'id': "NiansuhAIk1HgESy", 'name': "Niansuh"},
@@ -120,8 +108,6 @@ TRENDING_AGENT_MODE = {
     'GitlabAgent': {'mode': True, 'id': "GitlabAgent"},
     'GitAgent': {'mode': True, 'id': "GitAgent"},
     'RepoMap': {'mode': True, 'id': "repomap"},
-    
-    # Added New Agents
     'FlaskAgent': {'mode': True, 'id': "FlaskAgent"},
     'FirebaseAgent': {'mode': True, 'id': "FirebaseAgent"},
     'FastAPIAgent': {'mode': True, 'id': "FastAPIAgent"},