Niansuh commited on
Commit
fd78367
·
verified ·
1 Parent(s): e33313f

Update api/config.py

Browse files
Files changed (1) hide show
  1. api/config.py +8 -8
api/config.py CHANGED
@@ -18,6 +18,7 @@ common_headers = {
18
  'sec-ch-ua-full-version': '"131.0.6778.69"',
19
  'sec-ch-ua-full-version-list': '"Google Chrome";v="131.0.6778.69", "Chromium";v="131.0.6778.69", "Not_A Brand";v="24.0.0.0"',
20
  'sec-ch-ua-mobile': '?0',
 
21
  'sec-ch-ua-platform': '"Windows"',
22
  'sec-ch-ua-platform-version': '"19.0.0"',
23
  'sec-fetch-dest': 'empty',
@@ -27,15 +28,14 @@ common_headers = {
27
  }
28
 
29
  # Header Configurations for Specific API Calls
30
- def get_headers_api_chat(referer_url):
31
- return {**common_headers, 'Content-Type': 'application/json', 'Referer': referer_url}
32
 
33
- def get_headers_chat(chat_url, next_action, next_router_state_tree):
34
  return {
35
  **common_headers,
36
  'Accept': 'text/x-component',
37
  'Content-Type': 'text/plain;charset=UTF-8',
38
- 'Referer': chat_url,
39
  'next-action': next_action,
40
  'next-router-state-tree': next_router_state_tree,
41
  'next-url': '/',
@@ -81,7 +81,7 @@ ALLOWED_MODELS = [
81
  {"id": "o1-preview", "name": "o1-preview"},
82
  {"id": "claude-3-5-sonnet-20241022", "name": "claude-3-5-sonnet-20241022"},
83
  {"id": "claude-3-5-sonnet-x", "name": "claude-3-5-sonnet-x"},
84
-
85
  # Added New Agents
86
  {"id": "FlaskAgent", "name": "FlaskAgent"},
87
  {"id": "FirebaseAgent", "name": "FirebaseAgent"},
@@ -140,7 +140,7 @@ MODEL_MAPPING = {
140
  "o1-preview": "o1-preview",
141
  "claude-3-5-sonnet-20241022": "claude-3-5-sonnet-20241022",
142
  "claude-3-5-sonnet-x": "claude-3-5-sonnet-x",
143
-
144
  # Added New Agents
145
  "FlaskAgent": "FlaskAgent",
146
  "FirebaseAgent": "FirebaseAgent",
@@ -191,7 +191,7 @@ TRENDING_AGENT_MODE = {
191
  'GitlabAgent': {'mode': True, 'id': "GitlabAgent"},
192
  'GitAgent': {'mode': True, 'id': "GitAgent"},
193
  'RepoMap': {'mode': True, 'id': "repomap"},
194
-
195
  # Added New Agents
196
  'FlaskAgent': {'mode': True, 'id': "FlaskAgent"},
197
  'FirebaseAgent': {'mode': True, 'id': "FirebaseAgent"},
@@ -233,7 +233,7 @@ MODEL_PREFIXES = {
233
  'blackboxai-pro': '@BLACKBOXAI-PRO',
234
  'flux': '@Image Generation',
235
  # Add any additional prefixes if necessary
236
-
237
  # Added New Agents
238
  'FlaskAgent': '@Flask Agent',
239
  'FirebaseAgent': '@Firebase Agent',
 
18
  'sec-ch-ua-full-version': '"131.0.6778.69"',
19
  'sec-ch-ua-full-version-list': '"Google Chrome";v="131.0.6778.69", "Chromium";v="131.0.6778.69", "Not_A Brand";v="24.0.0.0"',
20
  'sec-ch-ua-mobile': '?0',
21
+ 'sec-ch-ua-model:': '',
22
  'sec-ch-ua-platform': '"Windows"',
23
  'sec-ch-ua-platform-version': '"19.0.0"',
24
  'sec-fetch-dest': 'empty',
 
28
  }
29
 
30
  # Header Configurations for Specific API Calls
31
+ def get_headers_api_chat():
32
+ return {**common_headers, 'Content-Type': 'application/json'}
33
 
34
+ def get_headers_chat(next_action, next_router_state_tree):
35
  return {
36
  **common_headers,
37
  'Accept': 'text/x-component',
38
  'Content-Type': 'text/plain;charset=UTF-8',
 
39
  'next-action': next_action,
40
  'next-router-state-tree': next_router_state_tree,
41
  'next-url': '/',
 
81
  {"id": "o1-preview", "name": "o1-preview"},
82
  {"id": "claude-3-5-sonnet-20241022", "name": "claude-3-5-sonnet-20241022"},
83
  {"id": "claude-3-5-sonnet-x", "name": "claude-3-5-sonnet-x"},
84
+
85
  # Added New Agents
86
  {"id": "FlaskAgent", "name": "FlaskAgent"},
87
  {"id": "FirebaseAgent", "name": "FirebaseAgent"},
 
140
  "o1-preview": "o1-preview",
141
  "claude-3-5-sonnet-20241022": "claude-3-5-sonnet-20241022",
142
  "claude-3-5-sonnet-x": "claude-3-5-sonnet-x",
143
+
144
  # Added New Agents
145
  "FlaskAgent": "FlaskAgent",
146
  "FirebaseAgent": "FirebaseAgent",
 
191
  'GitlabAgent': {'mode': True, 'id': "GitlabAgent"},
192
  'GitAgent': {'mode': True, 'id': "GitAgent"},
193
  'RepoMap': {'mode': True, 'id': "repomap"},
194
+
195
  # Added New Agents
196
  'FlaskAgent': {'mode': True, 'id': "FlaskAgent"},
197
  'FirebaseAgent': {'mode': True, 'id': "FirebaseAgent"},
 
233
  'blackboxai-pro': '@BLACKBOXAI-PRO',
234
  'flux': '@Image Generation',
235
  # Add any additional prefixes if necessary
236
+
237
  # Added New Agents
238
  'FlaskAgent': '@Flask Agent',
239
  'FirebaseAgent': '@Firebase Agent',