Upload 2 files

Dockerfile

@@ -1,5 +1,8 @@
 FROM hpyp/bbapi:latest
 
+# 复制重试中间件文件到容器中
+COPY retry_middleware.py /app/retry_middleware.py
+
 # ENV APP_SECRET=
 
 EXPOSE 8001

retry_middleware.py

@@ -0,0 +1,48 @@
+import time
+from functools import wraps
+
+def retry_on_unsafe_content(max_retries=3, delay=1):
+    """
+    装饰器函数，用于在收到 'content is not safe' 响应时自动重试
+    
+    Args:
+        max_retries (int): 最大重试次数
+        delay (int): 重试之间的延迟时间（秒）
+    """
+    def decorator(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            retries = 0
+            while retries < max_retries:
+                try:
+                    response = await func(*args, **kwargs)
+                    
+                    # 检查响应中是否包含 "content is not safe" 错误
+                    if isinstance(response, dict) and response.get('error') and 'content is not safe' in response['error'].lower():
+                        retries += 1
+                        if retries < max_retries:
+                            print(f"检测到内容安全问题，正在进行第 {retries} 次重试...")
+                            time.sleep(delay)
+                            continue
+                    return response
+                except Exception as e:
+                    if 'content is not safe' in str(e).lower():
+                        retries += 1
+                        if retries < max_retries:
+                            print(f"检测到内容安全问题，正在进行第 {retries} 次重试...")
+                            time.sleep(delay)
+                            continue
+                    raise e
+            
+            # 如果所有重试都失败，返回最后一次的响应
+            return response
+        return wrapper
+    return decorator
+
+# 使用示例:
+"""
+@retry_on_unsafe_content(max_retries=3, delay=1)
+async def your_ai_chat_function(message):
+    # 原有的AI聊天处理逻辑
+    pass
+"""