smgc commited on
Commit
5598419
1 Parent(s): db9ffff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -29,7 +29,8 @@ if proxy_url:
29
  else:
30
  transport = None
31
 
32
- sio = socketio.Client(http_session=transport, logger=True, engineio_logger=True)
 
33
 
34
  # 连接选项
35
  connect_opts = {
@@ -161,7 +162,7 @@ def messages():
161
  yield create_event("ping", {"type": "ping"})
162
 
163
  def on_connect():
164
- logging.info("Connected to Perplexity AI WebSocket")
165
  emit_data = {
166
  "version": "2.9",
167
  "source": "default",
@@ -194,7 +195,7 @@ def messages():
194
  response_event.set()
195
 
196
  def on_disconnect():
197
- logging.info("Disconnected from Perplexity AI WebSocket")
198
  response_event.set()
199
 
200
  def on_connect_error(data):
@@ -222,11 +223,11 @@ def messages():
222
  })
223
 
224
  except Exception as e:
225
- logging.error(f"Error during WebSocket connection: {str(e)}")
226
  yield create_event("content_block_delta", {
227
  "type": "content_block_delta",
228
  "index": 0,
229
- "delta": {"type": "text_delta", "text": f"Error during WebSocket connection: {str(e)}"},
230
  })
231
  finally:
232
  if sio.connected:
@@ -259,7 +260,7 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
259
  response_text = []
260
 
261
  def on_connect():
262
- logging.info("Connected to Perplexity AI WebSocket")
263
  emit_data = {
264
  "version": "2.9",
265
  "source": "default",
@@ -289,7 +290,7 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
289
  response_event.set()
290
 
291
  def on_disconnect():
292
- logging.info("Disconnected from Perplexity AI WebSocket")
293
  response_event.set()
294
 
295
  def on_connect_error(data):
 
29
  else:
30
  transport = None
31
 
32
+ # 禁用 socketio engineio 的详细日志
33
+ sio = socketio.Client(http_session=transport, logger=False, engineio_logger=False)
34
 
35
  # 连接选项
36
  connect_opts = {
 
162
  yield create_event("ping", {"type": "ping"})
163
 
164
  def on_connect():
165
+ logging.info("WebSocket connected to Perplexity AI")
166
  emit_data = {
167
  "version": "2.9",
168
  "source": "default",
 
195
  response_event.set()
196
 
197
  def on_disconnect():
198
+ logging.info("WebSocket disconnected from Perplexity AI")
199
  response_event.set()
200
 
201
  def on_connect_error(data):
 
223
  })
224
 
225
  except Exception as e:
226
+ logging.error(f"Error during WebSocket communication: {str(e)}")
227
  yield create_event("content_block_delta", {
228
  "type": "content_block_delta",
229
  "index": 0,
230
+ "delta": {"type": "text_delta", "text": f"Error during WebSocket communication: {str(e)}"},
231
  })
232
  finally:
233
  if sio.connected:
 
260
  response_text = []
261
 
262
  def on_connect():
263
+ logging.info("WebSocket connected to Perplexity AI")
264
  emit_data = {
265
  "version": "2.9",
266
  "source": "default",
 
290
  response_event.set()
291
 
292
  def on_disconnect():
293
+ logging.info("WebSocket disconnected from Perplexity AI")
294
  response_event.set()
295
 
296
  def on_connect_error(data):