Update app.py

app.py

@@ -81,17 +81,20 @@ def calculate_tokens_via_tiktoken(text, model="gpt-3.5-turbo"):
     使用 tiktoken 库根据 GPT 模型计算 token 数量。
     Claude 模型与 GPT 模型的 token 计算机制类似，因此可以使用 tiktoken。
     """
-    # 临时禁用代理
-    original_proxies = requests.Session().proxies  # 保存原始代理配置
-    requests.Session().proxies = {}  # 禁用代理
+    # 临时禁用代理环境变量
+    original_http_proxy = os.environ.pop('http_proxy', None)
+    original_https_proxy = os.environ.pop('https_proxy', None)
 
     try:
         encoding = tiktoken.encoding_for_model(model)  # 获取模型的编码器
         tokens = encoding.encode(text)  # 对文本进行 tokenization
         return len(tokens)
     finally:
-        # 恢复代理设置
-        requests.Session().proxies = original_proxies
+        # 恢复代理
+        if original_http_proxy:
+            os.environ['http_proxy'] = original_http_proxy
+        if original_https_proxy:
+            os.environ['https_proxy'] = original_https_proxy
 
 @app.route('/')
 def root():