Update app.py

app.py

@@ -215,7 +215,7 @@ def format_messages_for_deepsider(messages: List[ChatMessage]) -> str:
             next_code = False
             continue
 
-        if msg.role == 'assistant' and '验证码提示' in msg.content and 'clId' in msg.content:
+        if msg.role == 'assistant' and ('验证码提示' in msg.content or '验证码已发送至您的邮箱' in msg.content) and 'clId' in msg.content:
             next_code = True
             continue
 
@@ -343,7 +343,7 @@ async def stream_openai_response(response, request_id: str, model: str, token: s
                                 ]
                             }
 
-                            if '验证码提示' in content:
+                            if '验证码提示' in content or '验证码已发送至您的邮箱' in content:
                                 codeFlag = True
                             yield f"data: {json.dumps(chunk)}\n\n"
 
@@ -464,7 +464,7 @@ async def create_chat_completion(
 
     if len(chat_request.messages) > 1:
         msg = chat_request.messages[-2]
-        if msg.role == 'assistant' and '验证码提示' in msg.content and 'clId' in msg.content:
+        if msg.role == 'assistant' and  ('验证码提示' in msg.content or '验证码已发送至您的邮箱' in msg.content) and 'clId' in msg.content:
             isCode = True
             clId = re.search(r'\[clId:(.*)]', msg.content).group(1)