Update app.py

app.py

@@ -369,17 +369,25 @@ def stream_notdiamond_response(response, model):
     """流式处理 notdiamond API 响应。"""
     buffer = ""
     previous_content = ""
+    content_pattern = re.compile(r'data: (.*?)\n\n', re.DOTALL)
     
     for chunk in response.iter_content(1024):
         if chunk:
             buffer += chunk.decode('utf-8')
-            current_content = buffer.split('\n\n')[-1]  # 获取最新的内容块
+            matches = content_pattern.findall(buffer)
             
-            if current_content:
-                new_content = current_content[len(previous_content):]  # 提取新内容
-                previous_content = current_content  # 更新已处理的内容
-                
-                yield create_openai_chunk(new_content, model)
+            for match in matches:
+                try:
+                    data = json.loads(match)
+                    current_content = data['choices'][0]['delta'].get('content', '')
+                    new_content = current_content[len(previous_content):]  # 提取新内容
+                    previous_content = current_content  # 更新已处理的内容
+                    
+                    if new_content:
+                        yield create_openai_chunk(new_content, model)
+                except json.JSONDecodeError:
+                    logger.error(f"Failed to decode JSON: {match}")
+                    continue
     
     yield create_openai_chunk('', model, 'stop')