Update app.py

app.py

@@ -27,6 +27,7 @@ if proxy_url:
     transport = requests.Session()
     transport.proxies.update(proxies)
 else:
+    proxies = None
     transport = None
 
 sio = socketio.Client(http_session=transport, logger=True, engineio_logger=True)
@@ -79,18 +80,18 @@ def calculate_tokens_via_tiktoken(text, model="gpt-3.5-turbo"):
     """
     使用 tiktoken 库根据 GPT 模型计算 token 数量。
     Claude 模型与 GPT 模型的 token 计算机制类似，因此可以使用 tiktoken。
-    这里我们指定本地的编码文件路径。
     """
-    # 假设您将编码文件放在 /app/cl100k_base.tiktoken
-    encoding_path = '/app/cl100k_base.tiktoken'
-    
-    # 使用 tiktoken 的 Encoding 类手动加载本地编码文件
-    with open(encoding_path, 'rb') as f:
-        encoding = tiktoken.Encoding.load(f)
-    
-    # 对文本进行 tokenization
-    tokens = encoding.encode(text)
-    return len(tokens)
+    # 临时禁用代理
+    original_proxies = requests.Session().proxies  # 保存原始代理配置
+    requests.Session().proxies = {}  # 禁用代理
+
+    try:
+        encoding = tiktoken.encoding_for_model(model)  # 获取模型的编码器
+        tokens = encoding.encode(text)  # 对文本进行 tokenization
+        return len(tokens)
+    finally:
+        # 恢复代理设置
+        requests.Session().proxies = original_proxies
 
 @app.route('/')
 def root():
@@ -340,8 +341,8 @@ def not_found(error):
 @app.errorhandler(500)
 def server_error(error):
     logging.error(f"Server error: {str(error)}")
-        log_request(request.remote_addr, request.path, 500)
-        return "Something broke!", 500
+    log_request(request.remote_addr, request.path, 500)
+    return "Something broke!", 500
 
 def create_event(event, data):
     if isinstance(data, dict):