feat: 恢复ratelimit

2026-08-09 20:50:58 +00:00 · 2024-02-01 18:38:20 +08:00
parent 0dec10ddf2
commit f340a44abf
8 changed files with 58 additions and 41 deletions
@@ -12,6 +12,7 @@ from .longtext import longtext
 from .respback import respback
 from .wrapper import wrapper
 from .preproc import preproc
+from .ratelimit import ratelimit


 stage_order = [
@@ -19,7 +20,9 @@ stage_order = [
    "BanSessionCheckStage",
    "PreContentFilterStage",
    "PreProcessor",
+    "RequireRateLimitOccupancy",
    "MessageProcessor",
+    "ReleaseRateLimitOccupancy",
    "PostContentFilterStage",
    "ResponseWrapper",
    "LongTextProcessStage",