smgc commited on
Commit
40d6495
1 Parent(s): 15dd0c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -164,12 +164,12 @@ def messages():
164
  def generate():
165
  nonlocal total_output_tokens
166
 
167
- yield 'retry: 1000\n\n' # 设置重试间隔
168
-
169
  def send_event(event_type, data):
170
- event = create_event(event_type, data)
 
 
171
  logger.debug(f"Sending event: {event_type}")
172
- return event
173
 
174
  # 发送初始事件
175
  yield send_event("message_start", {
 
164
  def generate():
165
  nonlocal total_output_tokens
166
 
 
 
167
  def send_event(event_type, data):
168
+ if isinstance(data, dict):
169
+ data = json.dumps(data, ensure_ascii=False)
170
+ event_data = f"event: {event_type}\ndata: {data}\n\n"
171
  logger.debug(f"Sending event: {event_type}")
172
+ return event_data
173
 
174
  # 发送初始事件
175
  yield send_event("message_start", {