smgc commited on
Commit
4b8e78c
1 Parent(s): 69aca69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -6
app.py CHANGED
@@ -4,9 +4,9 @@ import uuid
4
  from datetime import datetime
5
  from flask import Flask, request, Response, jsonify
6
  import socketio
 
7
 
8
  app = Flask(__name__)
9
- sio = socketio.Client()
10
 
11
  # 从环境变量中获取API密钥
12
  API_KEY = os.environ.get('PPLX_KEY')
@@ -14,16 +14,24 @@ API_KEY = os.environ.get('PPLX_KEY')
14
  # 代理设置
15
  proxy_url = os.environ.get('PROXY_URL')
16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  # 连接选项
18
  connect_opts = {
19
  'transports': ['websocket'],
20
  }
21
 
22
- # 如果设置了代理,添加到连接选项中
23
- if proxy_url:
24
- connect_opts['http'] = proxy_url
25
- connect_opts['https'] = proxy_url
26
-
27
  # 其他选项
28
  sio_opts = {
29
  'auth': {
 
4
  from datetime import datetime
5
  from flask import Flask, request, Response, jsonify
6
  import socketio
7
+ import requests
8
 
9
  app = Flask(__name__)
 
10
 
11
  # 从环境变量中获取API密钥
12
  API_KEY = os.environ.get('PPLX_KEY')
 
14
  # 代理设置
15
  proxy_url = os.environ.get('PROXY_URL')
16
 
17
+ # 设置代理
18
+ if proxy_url:
19
+ proxies = {
20
+ 'http': proxy_url,
21
+ 'https': proxy_url
22
+ }
23
+ transport = requests.Session()
24
+ transport.proxies.update(proxies)
25
+ else:
26
+ transport = None
27
+
28
+ sio = socketio.Client(http_session=transport)
29
+
30
  # 连接选项
31
  connect_opts = {
32
  'transports': ['websocket'],
33
  }
34
 
 
 
 
 
 
35
  # 其他选项
36
  sio_opts = {
37
  'auth': {