usbtoo commited on
Commit
7db949a
·
verified ·
1 Parent(s): 56d1f3b

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1318 -0
app.py ADDED
@@ -0,0 +1,1318 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import json
3
+ import uuid
4
+ import time
5
+ import base64
6
+ import sys
7
+ import inspect
8
+ import secrets
9
+ from loguru import logger
10
+ from pathlib import Path
11
+
12
+ import requests
13
+ from flask import Flask, request, Response, jsonify, stream_with_context, render_template, redirect, session
14
+ from curl_cffi import requests as curl_requests
15
+ from werkzeug.middleware.proxy_fix import ProxyFix
16
+
17
+ class Logger:
18
+ def __init__(self, level="INFO", colorize=True, format=None):
19
+ logger.remove()
20
+
21
+ if format is None:
22
+ format = (
23
+ "<green>{time:YYYY-MM-DD HH:mm:ss}</green> | "
24
+ "<level>{level: <8}</level> | "
25
+ "<cyan>{extra[filename]}</cyan>:<cyan>{extra[function]}</cyan>:<cyan>{extra[lineno]}</cyan> | "
26
+ "<level>{message}</level>"
27
+ )
28
+
29
+ logger.add(
30
+ sys.stderr,
31
+ level=level,
32
+ format=format,
33
+ colorize=colorize,
34
+ backtrace=True,
35
+ diagnose=True
36
+ )
37
+
38
+ self.logger = logger
39
+
40
+ def _get_caller_info(self):
41
+ frame = inspect.currentframe()
42
+ try:
43
+ caller_frame = frame.f_back.f_back
44
+ full_path = caller_frame.f_code.co_filename
45
+ function = caller_frame.f_code.co_name
46
+ lineno = caller_frame.f_lineno
47
+
48
+ filename = os.path.basename(full_path)
49
+
50
+ return {
51
+ 'filename': filename,
52
+ 'function': function,
53
+ 'lineno': lineno
54
+ }
55
+ finally:
56
+ del frame
57
+
58
+ def info(self, message, source="API"):
59
+ caller_info = self._get_caller_info()
60
+ self.logger.bind(**caller_info).info(f"[{source}] {message}")
61
+
62
+ def error(self, message, source="API"):
63
+ caller_info = self._get_caller_info()
64
+
65
+ if isinstance(message, Exception):
66
+ self.logger.bind(**caller_info).exception(f"[{source}] {str(message)}")
67
+ else:
68
+ self.logger.bind(**caller_info).error(f"[{source}] {message}")
69
+
70
+ def warning(self, message, source="API"):
71
+ caller_info = self._get_caller_info()
72
+ self.logger.bind(**caller_info).warning(f"[{source}] {message}")
73
+
74
+ def debug(self, message, source="API"):
75
+ caller_info = self._get_caller_info()
76
+ self.logger.bind(**caller_info).debug(f"[{source}] {message}")
77
+
78
+ async def request_logger(self, request):
79
+ caller_info = self._get_caller_info()
80
+ self.logger.bind(**caller_info).info(f"请求: {request.method} {request.path}", "Request")
81
+
82
+ logger = Logger(level="INFO")
83
+ DATA_DIR = Path("/data")
84
+
85
+ if not DATA_DIR.exists():
86
+ DATA_DIR.mkdir(parents=True, exist_ok=True)
87
+ CONFIG = {
88
+ "MODELS": {
89
+ 'grok-2': 'grok-latest',
90
+ 'grok-2-imageGen': 'grok-latest',
91
+ 'grok-2-search': 'grok-latest',
92
+ "grok-3": "grok-3",
93
+ "grok-3-search": "grok-3",
94
+ "grok-3-imageGen": "grok-3",
95
+ "grok-3-deepsearch": "grok-3",
96
+ "grok-3-deepersearch": "grok-3",
97
+ "grok-3-reasoning": "grok-3"
98
+ },
99
+ "API": {
100
+ "IS_TEMP_CONVERSATION": os.environ.get("IS_TEMP_CONVERSATION", "true").lower() == "true",
101
+ "IS_CUSTOM_SSO": os.environ.get("IS_CUSTOM_SSO", "false").lower() == "true",
102
+ "BASE_URL": "https://grok.com",
103
+ "API_KEY": os.environ.get("API_KEY", "sk-123456"),
104
+ "SIGNATURE_COOKIE": None,
105
+ "PICGO_KEY": os.environ.get("PICGO_KEY") or None,
106
+ "TUMY_KEY": os.environ.get("TUMY_KEY") or None,
107
+ "RETRY_TIME": 1000,
108
+ "PROXY": os.environ.get("PROXY") or None
109
+ },
110
+ "ADMIN": {
111
+ "MANAGER_SWITCH": os.environ.get("MANAGER_SWITCH") or None,
112
+ "PASSWORD": os.environ.get("ADMINPASSWORD") or None
113
+ },
114
+ "SERVER": {
115
+ "COOKIE": None,
116
+ "CF_CLEARANCE":os.environ.get("CF_CLEARANCE") or None,
117
+ "PORT": int(os.environ.get("PORT", 5200))
118
+ },
119
+ "RETRY": {
120
+ "RETRYSWITCH": False,
121
+ "MAX_ATTEMPTS": 2
122
+ },
123
+ "TOKEN_STATUS_FILE": str(DATA_DIR / "token_status.json"),
124
+ "SHOW_THINKING": os.environ.get("SHOW_THINKING") == "true",
125
+ "IS_THINKING": False,
126
+ "IS_IMG_GEN": False,
127
+ "IS_IMG_GEN2": False,
128
+ "ISSHOW_SEARCH_RESULTS": os.environ.get("ISSHOW_SEARCH_RESULTS", "true").lower() == "true"
129
+ }
130
+
131
+
132
+ DEFAULT_HEADERS = {
133
+ 'Accept': '*/*',
134
+ 'Accept-Language': 'zh-CN,zh;q=0.9',
135
+ 'Accept-Encoding': 'gzip, deflate, br, zstd',
136
+ 'Content-Type': 'text/plain;charset=UTF-8',
137
+ 'Connection': 'keep-alive',
138
+ 'Origin': 'https://grok.com',
139
+ 'Priority': 'u=1, i',
140
+ 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
141
+ 'Sec-Ch-Ua': '"Not(A:Brand";v="99", "Google Chrome";v="133", "Chromium";v="133"',
142
+ 'Sec-Ch-Ua-Mobile': '?0',
143
+ 'Sec-Ch-Ua-Platform': '"macOS"',
144
+ 'Sec-Fetch-Dest': 'empty',
145
+ 'Sec-Fetch-Mode': 'cors',
146
+ 'Sec-Fetch-Site': 'same-origin',
147
+ 'Baggage': 'sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c'
148
+ }
149
+
150
+ class AuthTokenManager:
151
+ def __init__(self):
152
+ self.token_model_map = {}
153
+ self.expired_tokens = set()
154
+ self.token_status_map = {}
155
+
156
+ self.model_config = {
157
+ "grok-2": {
158
+ "RequestFrequency": 30,
159
+ "ExpirationTime": 1 * 60 * 60 * 1000 # 1小时
160
+ },
161
+ "grok-3": {
162
+ "RequestFrequency": 20,
163
+ "ExpirationTime": 2 * 60 * 60 * 1000 # 2小时
164
+ },
165
+ "grok-3-deepsearch": {
166
+ "RequestFrequency": 10,
167
+ "ExpirationTime": 24 * 60 * 60 * 1000 # 24小时
168
+ },
169
+ "grok-3-deepersearch": {
170
+ "RequestFrequency": 3,
171
+ "ExpirationTime": 24 * 60 * 60 * 1000 # 24小时
172
+ },
173
+ "grok-3-reasoning": {
174
+ "RequestFrequency": 10,
175
+ "ExpirationTime": 24 * 60 * 60 * 1000 # 24小时
176
+ }
177
+ }
178
+ self.token_reset_switch = False
179
+ self.token_reset_timer = None
180
+ self.load_token_status() # 加载令牌状态
181
+ def save_token_status(self):
182
+ try:
183
+ with open(CONFIG["TOKEN_STATUS_FILE"], 'w', encoding='utf-8') as f:
184
+ json.dump(self.token_status_map, f, indent=2, ensure_ascii=False)
185
+ logger.info("令牌状态已保存到配置文件", "TokenManager")
186
+ except Exception as error:
187
+ logger.error(f"保存令牌状态失败: {str(error)}", "TokenManager")
188
+
189
+ def load_token_status(self):
190
+ try:
191
+ token_status_file = Path(CONFIG["TOKEN_STATUS_FILE"])
192
+ if token_status_file.exists():
193
+ with open(token_status_file, 'r', encoding='utf-8') as f:
194
+ self.token_status_map = json.load(f)
195
+ logger.info("已从配置文件加载令牌状态", "TokenManager")
196
+ except Exception as error:
197
+ logger.error(f"加载令牌状态失败: {str(error)}", "TokenManager")
198
+ def add_token(self, token,isinitialization=False):
199
+ sso = token.split("sso=")[1].split(";")[0]
200
+ for model in self.model_config.keys():
201
+ if model not in self.token_model_map:
202
+ self.token_model_map[model] = []
203
+ if sso not in self.token_status_map:
204
+ self.token_status_map[sso] = {}
205
+
206
+ existing_token_entry = next((entry for entry in self.token_model_map[model] if entry["token"] == token), None)
207
+
208
+ if not existing_token_entry:
209
+ self.token_model_map[model].append({
210
+ "token": token,
211
+ "RequestCount": 0,
212
+ "AddedTime": int(time.time() * 1000),
213
+ "StartCallTime": None
214
+ })
215
+
216
+ if model not in self.token_status_map[sso]:
217
+ self.token_status_map[sso][model] = {
218
+ "isValid": True,
219
+ "invalidatedTime": None,
220
+ "totalRequestCount": 0
221
+ }
222
+ if not isinitialization:
223
+ self.save_token_status()
224
+
225
+ def set_token(self, token):
226
+ models = list(self.model_config.keys())
227
+ self.token_model_map = {model: [{
228
+ "token": token,
229
+ "RequestCount": 0,
230
+ "AddedTime": int(time.time() * 1000),
231
+ "StartCallTime": None
232
+ }] for model in models}
233
+
234
+ sso = token.split("sso=")[1].split(";")[0]
235
+ self.token_status_map[sso] = {model: {
236
+ "isValid": True,
237
+ "invalidatedTime": None,
238
+ "totalRequestCount": 0
239
+ } for model in models}
240
+
241
+ def delete_token(self, token):
242
+ try:
243
+ sso = token.split("sso=")[1].split(";")[0]
244
+ for model in self.token_model_map:
245
+ self.token_model_map[model] = [entry for entry in self.token_model_map[model] if entry["token"] != token]
246
+
247
+ if sso in self.token_status_map:
248
+ del self.token_status_map[sso]
249
+
250
+ self.save_token_status()
251
+
252
+ logger.info(f"令牌已成功移除: {token}", "TokenManager")
253
+ return True
254
+ except Exception as error:
255
+ logger.error(f"令牌删除失败: {str(error)}")
256
+ return False
257
+ def reduce_token_request_count(self, model_id, count):
258
+ try:
259
+ normalized_model = self.normalize_model_name(model_id)
260
+
261
+ if normalized_model not in self.token_model_map:
262
+ logger.error(f"模型 {normalized_model} 不存在", "TokenManager")
263
+ return False
264
+
265
+ if not self.token_model_map[normalized_model]:
266
+ logger.error(f"模型 {normalized_model} 没有可用的token", "TokenManager")
267
+ return False
268
+
269
+ token_entry = self.token_model_map[normalized_model][0]
270
+
271
+ # 确保RequestCount不会小于0
272
+ new_count = max(0, token_entry["RequestCount"] - count)
273
+ reduction = token_entry["RequestCount"] - new_count
274
+
275
+ token_entry["RequestCount"] = new_count
276
+
277
+ # 更新token状态
278
+ if token_entry["token"]:
279
+ sso = token_entry["token"].split("sso=")[1].split(";")[0]
280
+ if sso in self.token_status_map and normalized_model in self.token_status_map[sso]:
281
+ self.token_status_map[sso][normalized_model]["totalRequestCount"] = max(
282
+ 0,
283
+ self.token_status_map[sso][normalized_model]["totalRequestCount"] - reduction
284
+ )
285
+ return True
286
+
287
+ except Exception as error:
288
+ logger.error(f"重置校对token请求次数时发生错误: {str(error)}", "TokenManager")
289
+ return False
290
+ def get_next_token_for_model(self, model_id, is_return=False):
291
+ normalized_model = self.normalize_model_name(model_id)
292
+
293
+ if normalized_model not in self.token_model_map or not self.token_model_map[normalized_model]:
294
+ return None
295
+
296
+ token_entry = self.token_model_map[normalized_model][0]
297
+ if is_return:
298
+ return token_entry["token"]
299
+
300
+ if token_entry:
301
+ if token_entry["StartCallTime"] is None:
302
+ token_entry["StartCallTime"] = int(time.time() * 1000)
303
+
304
+ if not self.token_reset_switch:
305
+ self.start_token_reset_process()
306
+ self.token_reset_switch = True
307
+
308
+ token_entry["RequestCount"] += 1
309
+
310
+ if token_entry["RequestCount"] > self.model_config[normalized_model]["RequestFrequency"]:
311
+ self.remove_token_from_model(normalized_model, token_entry["token"])
312
+ next_token_entry = self.token_model_map[normalized_model][0] if self.token_model_map[normalized_model] else None
313
+ return next_token_entry["token"] if next_token_entry else None
314
+
315
+ sso = token_entry["token"].split("sso=")[1].split(";")[0]
316
+ if sso in self.token_status_map and normalized_model in self.token_status_map[sso]:
317
+ if token_entry["RequestCount"] == self.model_config[normalized_model]["RequestFrequency"]:
318
+ self.token_status_map[sso][normalized_model]["isValid"] = False
319
+ self.token_status_map[sso][normalized_model]["invalidatedTime"] = int(time.time() * 1000)
320
+ self.token_status_map[sso][normalized_model]["totalRequestCount"] += 1
321
+
322
+ self.save_token_status()
323
+
324
+ return token_entry["token"]
325
+
326
+ return None
327
+
328
+ def remove_token_from_model(self, model_id, token):
329
+ normalized_model = self.normalize_model_name(model_id)
330
+
331
+ if normalized_model not in self.token_model_map:
332
+ logger.error(f"模型 {normalized_model} 不存在", "TokenManager")
333
+ return False
334
+
335
+ model_tokens = self.token_model_map[normalized_model]
336
+ token_index = next((i for i, entry in enumerate(model_tokens) if entry["token"] == token), -1)
337
+
338
+ if token_index != -1:
339
+ removed_token_entry = model_tokens.pop(token_index)
340
+ self.expired_tokens.add((
341
+ removed_token_entry["token"],
342
+ normalized_model,
343
+ int(time.time() * 1000)
344
+ ))
345
+
346
+ if not self.token_reset_switch:
347
+ self.start_token_reset_process()
348
+ self.token_reset_switch = True
349
+
350
+ logger.info(f"模型{model_id}的令牌已失效,已成功移除令牌: {token}", "TokenManager")
351
+ return True
352
+
353
+ logger.error(f"在模型 {normalized_model} 中未找到 token: {token}", "TokenManager")
354
+ return False
355
+
356
+ def get_expired_tokens(self):
357
+ return list(self.expired_tokens)
358
+
359
+ def normalize_model_name(self, model):
360
+ if model.startswith('grok-') and 'deepsearch' not in model and 'reasoning' not in model:
361
+ return '-'.join(model.split('-')[:2])
362
+ return model
363
+
364
+ def get_token_count_for_model(self, model_id):
365
+ normalized_model = self.normalize_model_name(model_id)
366
+ return len(self.token_model_map.get(normalized_model, []))
367
+
368
+ def get_remaining_token_request_capacity(self):
369
+ remaining_capacity_map = {}
370
+
371
+ for model in self.model_config.keys():
372
+ model_tokens = self.token_model_map.get(model, [])
373
+ model_request_frequency = self.model_config[model]["RequestFrequency"]
374
+
375
+ total_used_requests = sum(token_entry.get("RequestCount", 0) for token_entry in model_tokens)
376
+
377
+ remaining_capacity = (len(model_tokens) * model_request_frequency) - total_used_requests
378
+ remaining_capacity_map[model] = max(0, remaining_capacity)
379
+
380
+ return remaining_capacity_map
381
+
382
+ def get_token_array_for_model(self, model_id):
383
+ normalized_model = self.normalize_model_name(model_id)
384
+ return self.token_model_map.get(normalized_model, [])
385
+
386
+ def start_token_reset_process(self):
387
+ def reset_expired_tokens():
388
+ now = int(time.time() * 1000)
389
+
390
+ tokens_to_remove = set()
391
+ for token_info in self.expired_tokens:
392
+ token, model, expired_time = token_info
393
+ expiration_time = self.model_config[model]["ExpirationTime"]
394
+
395
+ if now - expired_time >= expiration_time:
396
+ if not any(entry["token"] == token for entry in self.token_model_map.get(model, [])):
397
+ if model not in self.token_model_map:
398
+ self.token_model_map[model] = []
399
+
400
+ self.token_model_map[model].append({
401
+ "token": token,
402
+ "RequestCount": 0,
403
+ "AddedTime": now,
404
+ "StartCallTime": None
405
+ })
406
+
407
+ sso = token.split("sso=")[1].split(";")[0]
408
+ if sso in self.token_status_map and model in self.token_status_map[sso]:
409
+ self.token_status_map[sso][model]["isValid"] = True
410
+ self.token_status_map[sso][model]["invalidatedTime"] = None
411
+ self.token_status_map[sso][model]["totalRequestCount"] = 0
412
+
413
+ tokens_to_remove.add(token_info)
414
+
415
+ self.expired_tokens -= tokens_to_remove
416
+
417
+ for model in self.model_config.keys():
418
+ if model not in self.token_model_map:
419
+ continue
420
+
421
+ for token_entry in self.token_model_map[model]:
422
+ if not token_entry.get("StartCallTime"):
423
+ continue
424
+
425
+ expiration_time = self.model_config[model]["ExpirationTime"]
426
+ if now - token_entry["StartCallTime"] >= expiration_time:
427
+ sso = token_entry["token"].split("sso=")[1].split(";")[0]
428
+ if sso in self.token_status_map and model in self.token_status_map[sso]:
429
+ self.token_status_map[sso][model]["isValid"] = True
430
+ self.token_status_map[sso][model]["invalidatedTime"] = None
431
+ self.token_status_map[sso][model]["totalRequestCount"] = 0
432
+
433
+ token_entry["RequestCount"] = 0
434
+ token_entry["StartCallTime"] = None
435
+
436
+ import threading
437
+ # 启动一个线程执行定时任务,每小时执行一次
438
+ def run_timer():
439
+ while True:
440
+ reset_expired_tokens()
441
+ time.sleep(3600)
442
+
443
+ timer_thread = threading.Thread(target=run_timer)
444
+ timer_thread.daemon = True
445
+ timer_thread.start()
446
+
447
+ def get_all_tokens(self):
448
+ all_tokens = set()
449
+ for model_tokens in self.token_model_map.values():
450
+ for entry in model_tokens:
451
+ all_tokens.add(entry["token"])
452
+ return list(all_tokens)
453
+ def get_current_token(self, model_id):
454
+ normalized_model = self.normalize_model_name(model_id)
455
+
456
+ if normalized_model not in self.token_model_map or not self.token_model_map[normalized_model]:
457
+ return None
458
+
459
+ token_entry = self.token_model_map[normalized_model][0]
460
+ return token_entry["token"]
461
+
462
+ def get_token_status_map(self):
463
+ return self.token_status_map
464
+
465
+ class Utils:
466
+ @staticmethod
467
+ def organize_search_results(search_results):
468
+ if not search_results or 'results' not in search_results:
469
+ return ''
470
+
471
+ results = search_results['results']
472
+ formatted_results = []
473
+
474
+ for index, result in enumerate(results):
475
+ title = result.get('title', '未知标题')
476
+ url = result.get('url', '#')
477
+ preview = result.get('preview', '无预览内容')
478
+
479
+ formatted_result = f"\r\n<details><summary>资料[{index}]: {title}</summary>\r\n{preview}\r\n\n[Link]({url})\r\n</details>"
480
+ formatted_results.append(formatted_result)
481
+
482
+ return '\n\n'.join(formatted_results)
483
+
484
+ @staticmethod
485
+ def create_auth_headers(model, is_return=False):
486
+ return token_manager.get_next_token_for_model(model, is_return)
487
+
488
+ @staticmethod
489
+ def get_proxy_options():
490
+ proxy = CONFIG["API"]["PROXY"]
491
+ proxy_options = {}
492
+
493
+ if proxy:
494
+ logger.info(f"使用代理: {proxy}", "Server")
495
+
496
+ if proxy.startswith("socks5://"):
497
+ proxy_options["proxy"] = proxy
498
+
499
+ if '@' in proxy:
500
+ auth_part = proxy.split('@')[0].split('://')[1]
501
+ if ':' in auth_part:
502
+ username, password = auth_part.split(':')
503
+ proxy_options["proxy_auth"] = (username, password)
504
+ else:
505
+ proxy_options["proxies"] = {"https": proxy, "http": proxy}
506
+ return proxy_options
507
+
508
+ class GrokApiClient:
509
+ def __init__(self, model_id):
510
+ if model_id not in CONFIG["MODELS"]:
511
+ raise ValueError(f"不支持的模型: {model_id}")
512
+ self.model_id = CONFIG["MODELS"][model_id]
513
+
514
+ def process_message_content(self, content):
515
+ if isinstance(content, str):
516
+ return content
517
+ return None
518
+
519
+ def get_image_type(self, base64_string):
520
+ mime_type = 'image/jpeg'
521
+ if 'data:image' in base64_string:
522
+ import re
523
+ matches = re.search(r'data:([a-zA-Z0-9]+\/[a-zA-Z0-9-.+]+);base64,', base64_string)
524
+ if matches:
525
+ mime_type = matches.group(1)
526
+
527
+ extension = mime_type.split('/')[1]
528
+ file_name = f"image.{extension}"
529
+
530
+ return {
531
+ "mimeType": mime_type,
532
+ "fileName": file_name
533
+ }
534
+ def upload_base64_file(self, message, model):
535
+ try:
536
+ message_base64 = base64.b64encode(message.encode('utf-8')).decode('utf-8')
537
+ upload_data = {
538
+ "fileName": "message.txt",
539
+ "fileMimeType": "text/plain",
540
+ "content": message_base64
541
+ }
542
+
543
+ logger.info("发送文字文件请求", "Server")
544
+ cookie = f"{Utils.create_auth_headers(model, True)};{CONFIG['SERVER']['CF_CLEARANCE']}"
545
+ proxy_options = Utils.get_proxy_options()
546
+ response = curl_requests.post(
547
+ "https://grok.com/rest/app-chat/upload-file",
548
+ headers={
549
+ **DEFAULT_HEADERS,
550
+ "Cookie":cookie
551
+ },
552
+ json=upload_data,
553
+ impersonate="chrome133a",
554
+ **proxy_options
555
+ )
556
+
557
+ if response.status_code != 200:
558
+ logger.error(f"上传文件失败,状态码:{response.status_code}", "Server")
559
+ raise Exception(f"上传文件失败,状态码:{response.status_code}")
560
+
561
+ result = response.json()
562
+ logger.info(f"上传文件成功: {result}", "Server")
563
+ return result.get("fileMetadataId", "")
564
+
565
+ except Exception as error:
566
+ logger.error(str(error), "Server")
567
+ raise Exception(f"上传文件失败,状态码:{response.status_code}")
568
+ def upload_base64_image(self, base64_data, url):
569
+ try:
570
+ if 'data:image' in base64_data:
571
+ image_buffer = base64_data.split(',')[1]
572
+ else:
573
+ image_buffer = base64_data
574
+
575
+ image_info = self.get_image_type(base64_data)
576
+ mime_type = image_info["mimeType"]
577
+ file_name = image_info["fileName"]
578
+
579
+ upload_data = {
580
+ "rpc": "uploadFile",
581
+ "req": {
582
+ "fileName": file_name,
583
+ "fileMimeType": mime_type,
584
+ "content": image_buffer
585
+ }
586
+ }
587
+
588
+ logger.info("发送图片请求", "Server")
589
+
590
+ proxy_options = Utils.get_proxy_options()
591
+ response = curl_requests.post(
592
+ url,
593
+ headers={
594
+ **DEFAULT_HEADERS,
595
+ "Cookie":CONFIG["SERVER"]['COOKIE']
596
+ },
597
+ json=upload_data,
598
+ impersonate="chrome133a",
599
+ **proxy_options
600
+ )
601
+
602
+ if response.status_code != 200:
603
+ logger.error(f"上传图片失败,状态码:{response.status_code}", "Server")
604
+ return ''
605
+
606
+ result = response.json()
607
+ logger.info(f"上传图片成功: {result}", "Server")
608
+ return result.get("fileMetadataId", "")
609
+
610
+ except Exception as error:
611
+ logger.error(str(error), "Server")
612
+ return ''
613
+ # def convert_system_messages(self, messages):
614
+ # try:
615
+ # system_prompt = []
616
+ # i = 0
617
+ # while i < len(messages):
618
+ # if messages[i].get('role') != 'system':
619
+ # break
620
+
621
+ # system_prompt.append(self.process_message_content(messages[i].get('content')))
622
+ # i += 1
623
+
624
+ # messages = messages[i:]
625
+ # system_prompt = '\n'.join(system_prompt)
626
+
627
+ # if not messages:
628
+ # raise ValueError("没有找到用户或者AI消息")
629
+ # return {"system_prompt":system_prompt,"messages":messages}
630
+ # except Exception as error:
631
+ # logger.error(str(error), "Server")
632
+ # raise ValueError(error)
633
+ def prepare_chat_request(self, request):
634
+ if ((request["model"] == 'grok-2-imageGen' or request["model"] == 'grok-3-imageGen') and
635
+ not CONFIG["API"]["PICGO_KEY"] and not CONFIG["API"]["TUMY_KEY"] and
636
+ request.get("stream", False)):
637
+ raise ValueError("该模型流式输出需要配置PICGO或者TUMY图床密钥!")
638
+
639
+ # system_message, todo_messages = self.convert_system_messages(request["messages"]).values()
640
+ todo_messages = request["messages"]
641
+ if request["model"] in ['grok-2-imageGen', 'grok-3-imageGen', 'grok-3-deepsearch']:
642
+ last_message = todo_messages[-1]
643
+ if last_message["role"] != 'user':
644
+ raise ValueError('此模型最后一条消息必须是用户消息!')
645
+ todo_messages = [last_message]
646
+ file_attachments = []
647
+ messages = ''
648
+ last_role = None
649
+ last_content = ''
650
+ message_length = 0
651
+ convert_to_file = False
652
+ last_message_content = ''
653
+ search = request["model"] in ['grok-2-search', 'grok-3-search']
654
+ deepsearchPreset = ''
655
+ if request["model"] == 'grok-3-deepsearch':
656
+ deepsearchPreset = 'default'
657
+ elif request["model"] == 'grok-3-deepersearch':
658
+ deepsearchPreset = 'deeper'
659
+
660
+ # 移除<think>标签及其内容和base64图片
661
+ def remove_think_tags(text):
662
+ import re
663
+ text = re.sub(r'<think>[\s\S]*?<\/think>', '', text).strip()
664
+ text = re.sub(r'!\[image\]\(data:.*?base64,.*?\)', '[图片]', text)
665
+ return text
666
+
667
+ def process_content(content):
668
+ if isinstance(content, list):
669
+ text_content = ''
670
+ for item in content:
671
+ if item["type"] == 'image_url':
672
+ text_content += ("[图片]" if not text_content else '\n[图片]')
673
+ elif item["type"] == 'text':
674
+ text_content += (remove_think_tags(item["text"]) if not text_content else '\n' + remove_think_tags(item["text"]))
675
+ return text_content
676
+ elif isinstance(content, dict) and content is not None:
677
+ if content["type"] == 'image_url':
678
+ return "[图片]"
679
+ elif content["type"] == 'text':
680
+ return remove_think_tags(content["text"])
681
+ return remove_think_tags(self.process_message_content(content))
682
+ for current in todo_messages:
683
+ role = 'assistant' if current["role"] == 'assistant' else 'user'
684
+ is_last_message = current == todo_messages[-1]
685
+
686
+ if is_last_message and "content" in current:
687
+ if isinstance(current["content"], list):
688
+ for item in current["content"]:
689
+ if item["type"] == 'image_url':
690
+ processed_image = self.upload_base64_image(
691
+ item["image_url"]["url"],
692
+ f"{CONFIG['API']['BASE_URL']}/api/rpc"
693
+ )
694
+ if processed_image:
695
+ file_attachments.append(processed_image)
696
+ elif isinstance(current["content"], dict) and current["content"].get("type") == 'image_url':
697
+ processed_image = self.upload_base64_image(
698
+ current["content"]["image_url"]["url"],
699
+ f"{CONFIG['API']['BASE_URL']}/api/rpc"
700
+ )
701
+ if processed_image:
702
+ file_attachments.append(processed_image)
703
+
704
+
705
+ text_content = process_content(current.get("content", ""))
706
+ if is_last_message and convert_to_file:
707
+ last_message_content = f"{role.upper()}: {text_content or '[图片]'}\n"
708
+ continue
709
+ if text_content or (is_last_message and file_attachments):
710
+ if role == last_role and text_content:
711
+ last_content += '\n' + text_content
712
+ messages = messages[:messages.rindex(f"{role.upper()}: ")] + f"{role.upper()}: {last_content}\n"
713
+ else:
714
+ messages += f"{role.upper()}: {text_content or '[图片]'}\n"
715
+ last_content = text_content
716
+ last_role = role
717
+ message_length += len(messages)
718
+ if message_length >= 40000:
719
+ convert_to_file = True
720
+
721
+ if convert_to_file:
722
+ file_id = self.upload_base64_file(messages, request["model"])
723
+ if file_id:
724
+ file_attachments.insert(0, file_id)
725
+ messages = last_message_content.strip()
726
+ if messages.strip() == '':
727
+ if convert_to_file:
728
+ messages = '基于txt文件内容进行回复:'
729
+ else:
730
+ raise ValueError('消息内容为空!')
731
+ return {
732
+ "temporary": CONFIG["API"].get("IS_TEMP_CONVERSATION", False),
733
+ "modelName": self.model_id,
734
+ "message": messages.strip(),
735
+ "fileAttachments": file_attachments[:4],
736
+ "imageAttachments": [],
737
+ "disableSearch": False,
738
+ "enableImageGeneration": True,
739
+ "returnImageBytes": False,
740
+ "returnRawGrokInXaiRequest": False,
741
+ "enableImageStreaming": False,
742
+ "imageGenerationCount": 1,
743
+ "forceConcise": False,
744
+ "toolOverrides": {
745
+ "imageGen": request["model"] in ['grok-2-imageGen', 'grok-3-imageGen'],
746
+ "webSearch": search,
747
+ "xSearch": search,
748
+ "xMediaSearch": search,
749
+ "trendsSearch": search,
750
+ "xPostAnalyze": search
751
+ },
752
+ "enableSideBySide": True,
753
+ "sendFinalMetadata": True,
754
+ "customPersonality": "",
755
+ "deepsearchPreset": deepsearchPreset,
756
+ "isReasoning": request["model"] == 'grok-3-reasoning',
757
+ "disableTextFollowUps": True
758
+ }
759
+
760
+ class MessageProcessor:
761
+ @staticmethod
762
+ def create_chat_response(message, model, is_stream=False):
763
+ base_response = {
764
+ "id": f"chatcmpl-{uuid.uuid4()}",
765
+ "created": int(time.time()),
766
+ "model": model
767
+ }
768
+
769
+ if is_stream:
770
+ return {
771
+ **base_response,
772
+ "object": "chat.completion.chunk",
773
+ "choices": [{
774
+ "index": 0,
775
+ "delta": {
776
+ "content": message
777
+ }
778
+ }]
779
+ }
780
+
781
+ return {
782
+ **base_response,
783
+ "object": "chat.completion",
784
+ "choices": [{
785
+ "index": 0,
786
+ "message": {
787
+ "role": "assistant",
788
+ "content": message
789
+ },
790
+ "finish_reason": "stop"
791
+ }],
792
+ "usage": None
793
+ }
794
+
795
+ def process_model_response(response, model):
796
+ result = {"token": None, "imageUrl": None}
797
+
798
+ if CONFIG["IS_IMG_GEN"]:
799
+ if response.get("cachedImageGenerationResponse") and not CONFIG["IS_IMG_GEN2"]:
800
+ result["imageUrl"] = response["cachedImageGenerationResponse"]["imageUrl"]
801
+ return result
802
+
803
+ if model == 'grok-2':
804
+ result["token"] = response.get("token")
805
+ elif model in ['grok-2-search', 'grok-3-search']:
806
+ if response.get("webSearchResults") and CONFIG["ISSHOW_SEARCH_RESULTS"]:
807
+ result["token"] = f"\r\n<think>{Utils.organize_search_results(response['webSearchResults'])}</think>\r\n"
808
+ else:
809
+ result["token"] = response.get("token")
810
+ elif model == 'grok-3':
811
+ result["token"] = response.get("token")
812
+ elif model in ['grok-3-deepsearch', 'grok-3-deepersearch']:
813
+ if response.get("messageStepId") and not CONFIG["SHOW_THINKING"]:
814
+ return result
815
+ if response.get("messageStepId") and not CONFIG["IS_THINKING"]:
816
+ result["token"] = "<think>" + response.get("token", "")
817
+ CONFIG["IS_THINKING"] = True
818
+ elif not response.get("messageStepId") and CONFIG["IS_THINKING"] and response.get("messageTag") == "final":
819
+ result["token"] = "</think>" + response.get("token", "")
820
+ CONFIG["IS_THINKING"] = False
821
+ elif (response.get("messageStepId") and CONFIG["IS_THINKING"] and response.get("messageTag") == "assistant") or response.get("messageTag") == "final":
822
+ result["token"] = response.get("token","")
823
+ elif (CONFIG["IS_THINKING"] and response.get("token","").get("action","") == "webSearch"):
824
+ result["token"] = response.get("token","").get("action_input","").get("query","")
825
+ elif (CONFIG["IS_THINKING"] and response.get("webSearchResults")):
826
+ result["token"] = Utils.organize_search_results(response['webSearchResults'])
827
+ elif model == 'grok-3-reasoning':
828
+ if response.get("isThinking") and not CONFIG["SHOW_THINKING"]:
829
+ return result
830
+
831
+ if response.get("isThinking") and not CONFIG["IS_THINKING"]:
832
+ result["token"] = "<think>" + response.get("token", "")
833
+ CONFIG["IS_THINKING"] = True
834
+ elif not response.get("isThinking") and CONFIG["IS_THINKING"]:
835
+ result["token"] = "</think>" + response.get("token", "")
836
+ CONFIG["IS_THINKING"] = False
837
+ else:
838
+ result["token"] = response.get("token")
839
+
840
+ return result
841
+
842
+ def handle_image_response(image_url):
843
+ max_retries = 2
844
+ retry_count = 0
845
+ image_base64_response = None
846
+
847
+ while retry_count < max_retries:
848
+ try:
849
+ proxy_options = Utils.get_proxy_options()
850
+ image_base64_response = curl_requests.get(
851
+ f"https://assets.grok.com/{image_url}",
852
+ headers={
853
+ **DEFAULT_HEADERS,
854
+ "Cookie":CONFIG["SERVER"]['COOKIE']
855
+ },
856
+ impersonate="chrome133a",
857
+ **proxy_options
858
+ )
859
+
860
+ if image_base64_response.status_code == 200:
861
+ break
862
+
863
+ retry_count += 1
864
+ if retry_count == max_retries:
865
+ raise Exception(f"上游服务请求失败! status: {image_base64_response.status_code}")
866
+
867
+ time.sleep(CONFIG["API"]["RETRY_TIME"] / 1000 * retry_count)
868
+
869
+ except Exception as error:
870
+ logger.error(str(error), "Server")
871
+ retry_count += 1
872
+ if retry_count == max_retries:
873
+ raise
874
+
875
+ time.sleep(CONFIG["API"]["RETRY_TIME"] / 1000 * retry_count)
876
+
877
+ image_buffer = image_base64_response.content
878
+
879
+ if not CONFIG["API"]["PICGO_KEY"] and not CONFIG["API"]["TUMY_KEY"]:
880
+ base64_image = base64.b64encode(image_buffer).decode('utf-8')
881
+ image_content_type = image_base64_response.headers.get('content-type', 'image/jpeg')
882
+ return f"![image](data:{image_content_type};base64,{base64_image})"
883
+
884
+ logger.info("开始上传图床", "Server")
885
+
886
+ if CONFIG["API"]["PICGO_KEY"]:
887
+ files = {'source': ('image.jpg', image_buffer, 'image/jpeg')}
888
+ headers = {
889
+ "X-API-Key": CONFIG["API"]["PICGO_KEY"]
890
+ }
891
+
892
+ response_url = requests.post(
893
+ "https://www.picgo.net/api/1/upload",
894
+ files=files,
895
+ headers=headers
896
+ )
897
+
898
+ if response_url.status_code != 200:
899
+ return "生图失败,请查看PICGO图床密钥是否设置正确"
900
+ else:
901
+ logger.info("生图成功", "Server")
902
+ result = response_url.json()
903
+ return f"![image]({result['image']['url']})"
904
+
905
+
906
+ elif CONFIG["API"]["TUMY_KEY"]:
907
+ files = {'file': ('image.jpg', image_buffer, 'image/jpeg')}
908
+ headers = {
909
+ "Accept": "application/json",
910
+ 'Authorization': f"Bearer {CONFIG['API']['TUMY_KEY']}"
911
+ }
912
+
913
+ response_url = requests.post(
914
+ "https://tu.my/api/v1/upload",
915
+ files=files,
916
+ headers=headers
917
+ )
918
+
919
+ if response_url.status_code != 200:
920
+ return "生图失败,请查看TUMY图床密钥是否设置正确"
921
+ else:
922
+ try:
923
+ result = response_url.json()
924
+ logger.info("生图成功", "Server")
925
+ return f"![image]({result['data']['links']['url']})"
926
+ except Exception as error:
927
+ logger.error(str(error), "Server")
928
+ return "生图失败,请查看TUMY图床密钥是否设置正确"
929
+
930
+ def handle_non_stream_response(response, model):
931
+ try:
932
+ logger.info("开始处理非流式响应", "Server")
933
+
934
+ stream = response.iter_lines()
935
+ full_response = ""
936
+
937
+ CONFIG["IS_THINKING"] = False
938
+ CONFIG["IS_IMG_GEN"] = False
939
+ CONFIG["IS_IMG_GEN2"] = False
940
+
941
+ for chunk in stream:
942
+ if not chunk:
943
+ continue
944
+ try:
945
+ line_json = json.loads(chunk.decode("utf-8").strip())
946
+ if line_json.get("error"):
947
+ logger.error(json.dumps(line_json, indent=2), "Server")
948
+ return json.dumps({"error": "RateLimitError"}) + "\n\n"
949
+
950
+ response_data = line_json.get("result", {}).get("response")
951
+ if not response_data:
952
+ continue
953
+
954
+ if response_data.get("doImgGen") or response_data.get("imageAttachmentInfo"):
955
+ CONFIG["IS_IMG_GEN"] = True
956
+
957
+ result = process_model_response(response_data, model)
958
+
959
+ if result["token"]:
960
+ full_response += result["token"]
961
+
962
+ if result["imageUrl"]:
963
+ CONFIG["IS_IMG_GEN2"] = True
964
+ return handle_image_response(result["imageUrl"])
965
+
966
+ except json.JSONDecodeError:
967
+ continue
968
+ except Exception as e:
969
+ logger.error(f"处理流式响应行时出错: {str(e)}", "Server")
970
+ continue
971
+
972
+ return full_response
973
+ except Exception as error:
974
+ logger.error(str(error), "Server")
975
+ raise
976
+ def handle_stream_response(response, model):
977
+ def generate():
978
+ logger.info("开始处理流式响应", "Server")
979
+
980
+ stream = response.iter_lines()
981
+ CONFIG["IS_THINKING"] = False
982
+ CONFIG["IS_IMG_GEN"] = False
983
+ CONFIG["IS_IMG_GEN2"] = False
984
+
985
+ for chunk in stream:
986
+ if not chunk:
987
+ continue
988
+ try:
989
+ line_json = json.loads(chunk.decode("utf-8").strip())
990
+ print(line_json)
991
+ if line_json.get("error"):
992
+ logger.error(json.dumps(line_json, indent=2), "Server")
993
+ yield json.dumps({"error": "RateLimitError"}) + "\n\n"
994
+ return
995
+
996
+ response_data = line_json.get("result", {}).get("response")
997
+ if not response_data:
998
+ continue
999
+
1000
+ if response_data.get("doImgGen") or response_data.get("imageAttachmentInfo"):
1001
+ CONFIG["IS_IMG_GEN"] = True
1002
+
1003
+ result = process_model_response(response_data, model)
1004
+
1005
+ if result["token"]:
1006
+ yield f"data: {json.dumps(MessageProcessor.create_chat_response(result['token'], model, True))}\n\n"
1007
+
1008
+ if result["imageUrl"]:
1009
+ CONFIG["IS_IMG_GEN2"] = True
1010
+ image_data = handle_image_response(result["imageUrl"])
1011
+ yield f"data: {json.dumps(MessageProcessor.create_chat_response(image_data, model, True))}\n\n"
1012
+
1013
+ except json.JSONDecodeError:
1014
+ continue
1015
+ except Exception as e:
1016
+ logger.error(f"处理流式响应行时出错: {str(e)}", "Server")
1017
+ continue
1018
+
1019
+ yield "data: [DONE]\n\n"
1020
+ return generate()
1021
+
1022
+ def initialization():
1023
+ sso_array = os.environ.get("SSO", "").split(',')
1024
+ logger.info("开始加载令牌", "Server")
1025
+ token_manager.load_token_status()
1026
+ for sso in sso_array:
1027
+ if sso:
1028
+ token_manager.add_token(f"sso-rw={sso};sso={sso}",True)
1029
+ token_manager.save_token_status()
1030
+
1031
+ logger.info(f"成功加载令牌: {json.dumps(token_manager.get_all_tokens(), indent=2)}", "Server")
1032
+ logger.info(f"令牌加载完成,共加载: {len(token_manager.get_all_tokens())}个令牌", "Server")
1033
+
1034
+ if CONFIG["API"]["PROXY"]:
1035
+ logger.info(f"代理已设置: {CONFIG['API']['PROXY']}", "Server")
1036
+
1037
+ logger.info("初始化完成", "Server")
1038
+
1039
+
1040
+ app = Flask(__name__)
1041
+ app.wsgi_app = ProxyFix(app.wsgi_app)
1042
+ app.secret_key = os.environ.get('FLASK_SECRET_KEY') or secrets.token_hex(16)
1043
+ app.json.sort_keys = False
1044
+
1045
+ @app.route('/manager/login', methods=['GET', 'POST'])
1046
+ def manager_login():
1047
+ if CONFIG["ADMIN"]["MANAGER_SWITCH"]:
1048
+ if request.method == 'POST':
1049
+ password = request.form.get('password')
1050
+ if password == CONFIG["ADMIN"]["PASSWORD"]:
1051
+ session['is_logged_in'] = True
1052
+ return redirect('/manager')
1053
+ return render_template('login.html', error=True)
1054
+ return render_template('login.html', error=False)
1055
+ else:
1056
+ return redirect('/')
1057
+
1058
+ def check_auth():
1059
+ return session.get('is_logged_in', False)
1060
+
1061
+ @app.route('/manager')
1062
+ def manager():
1063
+ if not check_auth():
1064
+ return redirect('/manager/login')
1065
+ return render_template('manager.html')
1066
+
1067
+ @app.route('/manager/api/get')
1068
+ def get_manager_tokens():
1069
+ if not check_auth():
1070
+ return jsonify({"error": "Unauthorized"}), 401
1071
+ return jsonify(token_manager.get_token_status_map())
1072
+
1073
+ @app.route('/manager/api/add', methods=['POST'])
1074
+ def add_manager_token():
1075
+ if not check_auth():
1076
+ return jsonify({"error": "Unauthorized"}), 401
1077
+ try:
1078
+ sso = request.json.get('sso')
1079
+ if not sso:
1080
+ return jsonify({"error": "SSO token is required"}), 400
1081
+ token_manager.add_token(f"sso-rw={sso};sso={sso}")
1082
+ return jsonify({"success": True})
1083
+ except Exception as e:
1084
+ return jsonify({"error": str(e)}), 500
1085
+
1086
+ @app.route('/manager/api/delete', methods=['POST'])
1087
+ def delete_manager_token():
1088
+ if not check_auth():
1089
+ return jsonify({"error": "Unauthorized"}), 401
1090
+ try:
1091
+ sso = request.json.get('sso')
1092
+ if not sso:
1093
+ return jsonify({"error": "SSO token is required"}), 400
1094
+ token_manager.delete_token(f"sso-rw={sso};sso={sso}")
1095
+ return jsonify({"success": True})
1096
+ except Exception as e:
1097
+ return jsonify({"error": str(e)}), 500
1098
+
1099
+ @app.route('/manager/api/cf_clearance', methods=['POST'])
1100
+ def setCf_Manager_clearance():
1101
+ if not check_auth():
1102
+ return jsonify({"error": "Unauthorized"}), 401
1103
+ try:
1104
+ cf_clearance = request.json.get('cf_clearance')
1105
+ if not cf_clearance:
1106
+ return jsonify({"error": "cf_clearance is required"}), 400
1107
+ CONFIG["SERVER"]['CF_CLEARANCE'] = cf_clearance
1108
+ return jsonify({"success": True})
1109
+ except Exception as e:
1110
+ return jsonify({"error": str(e)}), 500
1111
+
1112
+
1113
+ @app.route('/get/tokens', methods=['GET'])
1114
+ def get_tokens():
1115
+ auth_token = request.headers.get('Authorization', '').replace('Bearer ', '')
1116
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1117
+ return jsonify({"error": '自定义的SSO令牌模式无法获取轮询sso令牌状态'}), 403
1118
+ elif auth_token != CONFIG["API"]["API_KEY"]:
1119
+ return jsonify({"error": 'Unauthorized'}), 401
1120
+ return jsonify(token_manager.get_token_status_map())
1121
+
1122
+ @app.route('/add/token', methods=['POST'])
1123
+ def add_token():
1124
+ auth_token = request.headers.get('Authorization', '').replace('Bearer ', '')
1125
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1126
+ return jsonify({"error": '自定义的SSO令牌模式无法添加sso令牌'}), 403
1127
+ elif auth_token != CONFIG["API"]["API_KEY"]:
1128
+ return jsonify({"error": 'Unauthorized'}), 401
1129
+
1130
+ try:
1131
+ sso = request.json.get('sso')
1132
+ token_manager.add_token(f"sso-rw={sso};sso={sso}")
1133
+ return jsonify(token_manager.get_token_status_map().get(sso, {})), 200
1134
+ except Exception as error:
1135
+ logger.error(str(error), "Server")
1136
+ return jsonify({"error": '添加sso令牌失败'}), 500
1137
+
1138
+ @app.route('/set/cf_clearance', methods=['POST'])
1139
+ def setCf_clearance():
1140
+ auth_token = request.headers.get('Authorization', '').replace('Bearer ', '')
1141
+ if auth_token != CONFIG["API"]["API_KEY"]:
1142
+ return jsonify({"error": 'Unauthorized'}), 401
1143
+ try:
1144
+ cf_clearance = request.json.get('cf_clearance')
1145
+ CONFIG["SERVER"]['CF_CLEARANCE'] = cf_clearance
1146
+ return jsonify({"message": '设置cf_clearance成功'}), 200
1147
+ except Exception as error:
1148
+ logger.error(str(error), "Server")
1149
+ return jsonify({"error": '设置cf_clearance失败'}), 500
1150
+
1151
+ @app.route('/delete/token', methods=['POST'])
1152
+ def delete_token():
1153
+ auth_token = request.headers.get('Authorization', '').replace('Bearer ', '')
1154
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1155
+ return jsonify({"error": '自定义的SSO令牌模式无法删除sso令牌'}), 403
1156
+ elif auth_token != CONFIG["API"]["API_KEY"]:
1157
+ return jsonify({"error": 'Unauthorized'}), 401
1158
+
1159
+ try:
1160
+ sso = request.json.get('sso')
1161
+ token_manager.delete_token(f"sso-rw={sso};sso={sso}")
1162
+ return jsonify({"message": '删除sso令牌成功'}), 200
1163
+ except Exception as error:
1164
+ logger.error(str(error), "Server")
1165
+ return jsonify({"error": '删除sso令牌失败'}), 500
1166
+
1167
+ @app.route('/v1/models', methods=['GET'])
1168
+ def get_models():
1169
+ return jsonify({
1170
+ "object": "list",
1171
+ "data": [
1172
+ {
1173
+ "id": model,
1174
+ "object": "model",
1175
+ "created": int(time.time()),
1176
+ "owned_by": "grok"
1177
+ }
1178
+ for model in CONFIG["MODELS"].keys()
1179
+ ]
1180
+ })
1181
+
1182
+ @app.route('/v1/chat/completions', methods=['POST'])
1183
+ def chat_completions():
1184
+ response_status_code = 500
1185
+ try:
1186
+ auth_token = request.headers.get('Authorization',
1187
+ '').replace('Bearer ', '')
1188
+ if auth_token:
1189
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1190
+ result = f"sso={auth_token};sso-rw={auth_token}"
1191
+ token_manager.set_token(result)
1192
+ elif auth_token != CONFIG["API"]["API_KEY"]:
1193
+ return jsonify({"error": 'Unauthorized'}), 401
1194
+ else:
1195
+ return jsonify({"error": 'API_KEY缺失'}), 401
1196
+
1197
+ data = request.json
1198
+ model = data.get("model")
1199
+ stream = data.get("stream", False)
1200
+
1201
+ retry_count = 0
1202
+ grok_client = GrokApiClient(model)
1203
+ request_payload = grok_client.prepare_chat_request(data)
1204
+ logger.info(json.dumps(request_payload,indent=2))
1205
+
1206
+ while retry_count < CONFIG["RETRY"]["MAX_ATTEMPTS"]:
1207
+ retry_count += 1
1208
+ CONFIG["API"]["SIGNATURE_COOKIE"] = Utils.create_auth_headers(model)
1209
+
1210
+ if not CONFIG["API"]["SIGNATURE_COOKIE"]:
1211
+ raise ValueError('该模型无可用令牌')
1212
+
1213
+ logger.info(
1214
+ f"当前令牌: {json.dumps(CONFIG['API']['SIGNATURE_COOKIE'], indent=2)}","Server")
1215
+ logger.info(
1216
+ f"当前可用模型的全部可用数量: {json.dumps(token_manager.get_remaining_token_request_capacity(), indent=2)}","Server")
1217
+
1218
+ if CONFIG['SERVER']['CF_CLEARANCE']:
1219
+ CONFIG["SERVER"]['COOKIE'] = f"{CONFIG['API']['SIGNATURE_COOKIE']};{CONFIG['SERVER']['CF_CLEARANCE']}"
1220
+ else:
1221
+ CONFIG["SERVER"]['COOKIE'] = CONFIG['API']['SIGNATURE_COOKIE']
1222
+ logger.info(json.dumps(request_payload,indent=2),"Server")
1223
+ try:
1224
+ proxy_options = Utils.get_proxy_options()
1225
+ response = curl_requests.post(
1226
+ f"{CONFIG['API']['BASE_URL']}/rest/app-chat/conversations/new",
1227
+ headers={
1228
+ **DEFAULT_HEADERS,
1229
+ "Cookie":CONFIG["SERVER"]['COOKIE']
1230
+ },
1231
+ data=json.dumps(request_payload),
1232
+ impersonate="chrome133a",
1233
+ stream=True,
1234
+ **proxy_options)
1235
+ logger.info(CONFIG["SERVER"]['COOKIE'],"Server")
1236
+ if response.status_code == 200:
1237
+ response_status_code = 200
1238
+ logger.info("请求成功", "Server")
1239
+ logger.info(f"当前{model}剩余可用令牌数: {token_manager.get_token_count_for_model(model)}","Server")
1240
+
1241
+ try:
1242
+ if stream:
1243
+ return Response(stream_with_context(
1244
+ handle_stream_response(response, model)),content_type='text/event-stream')
1245
+ else:
1246
+ content = handle_non_stream_response(response, model)
1247
+ return jsonify(
1248
+ MessageProcessor.create_chat_response(content, model))
1249
+
1250
+ except Exception as error:
1251
+ logger.error(str(error), "Server")
1252
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1253
+ raise ValueError(f"自定义SSO令��当前模型{model}的请求次数已失效")
1254
+ token_manager.remove_token_from_model(model, CONFIG["API"]["SIGNATURE_COOKIE"])
1255
+ if token_manager.get_token_count_for_model(model) == 0:
1256
+ raise ValueError(f"{model} 次数已达上限,请切换其他模型或者重新对话")
1257
+ elif response.status_code == 403:
1258
+ response_status_code = 403
1259
+ token_manager.reduce_token_request_count(model,1)#重置去除当前因为错误未成功请求的次数,确保不会因为错误未成功请求的次数导致次数上限
1260
+ if token_manager.get_token_count_for_model(model) == 0:
1261
+ raise ValueError(f"{model} 次数已达上限,请切换其他模型或者重新对话")
1262
+ print("状态码:", response.status_code)
1263
+ print("响应头:", response.headers)
1264
+ print("响应内容:", response.text)
1265
+ raise ValueError(f"IP暂时被封无法破盾,请稍后重试或者更换ip")
1266
+ elif response.status_code == 429:
1267
+ response_status_code = 429
1268
+ token_manager.reduce_token_request_count(model,1)
1269
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1270
+ raise ValueError(f"自定义SSO令牌当前模型{model}的请求次数已失效")
1271
+
1272
+ token_manager.remove_token_from_model(
1273
+ model, CONFIG["API"]["SIGNATURE_COOKIE"])
1274
+ if token_manager.get_token_count_for_model(model) == 0:
1275
+ raise ValueError(f"{model} 次数已达上限,请切换其他模型或者重新对话")
1276
+
1277
+ else:
1278
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1279
+ raise ValueError(f"自定义SSO令牌当前模型{model}的请求次数已失效")
1280
+
1281
+ logger.error(f"令牌异常错误状态!status: {response.status_code}","Server")
1282
+ token_manager.remove_token_from_model(model, CONFIG["API"]["SIGNATURE_COOKIE"])
1283
+ logger.info(
1284
+ f"当前{model}剩余可用令牌数: {token_manager.get_token_count_for_model(model)}",
1285
+ "Server")
1286
+
1287
+ except Exception as e:
1288
+ logger.error(f"请求处理异常: {str(e)}", "Server")
1289
+ if CONFIG["API"]["IS_CUSTOM_SSO"]:
1290
+ raise
1291
+ continue
1292
+ if response_status_code == 403:
1293
+ raise ValueError('IP暂时被封无法破盾,请稍后重试或者更换ip')
1294
+ elif response_status_code == 500:
1295
+ raise ValueError('当前模型所有令牌暂无可用,请稍后重试')
1296
+
1297
+ except Exception as error:
1298
+ logger.error(str(error), "ChatAPI")
1299
+ return jsonify(
1300
+ {"error": {
1301
+ "message": str(error),
1302
+ "type": "server_error"
1303
+ }}), response_status_code
1304
+
1305
+ @app.route('/', defaults={'path': ''})
1306
+ @app.route('/<path:path>')
1307
+ def catch_all(path):
1308
+ return 'api运行正常', 200
1309
+
1310
+ if __name__ == '__main__':
1311
+ token_manager = AuthTokenManager()
1312
+ initialization()
1313
+
1314
+ app.run(
1315
+ host='0.0.0.0',
1316
+ port=CONFIG["SERVER"]["PORT"],
1317
+ debug=False
1318
+ )