优化流式抗截断

Author: su-kaka

src/router/geminicli/anthropic.py

@@ -258,7 +258,7 @@ async def stream_request_wrapper(payload):
             stream_request_wrapper,
             anti_truncation_payload,
             max_attempts,
-            enable_prefill_mode=("claude" not in str(api_request.get("model", "")).lower()),
+            enable_prefill_mode=True,
         )
 
         # 包装以确保是bytes流

src/router/geminicli/gemini.py

@@ -11,8 +11,6 @@
 if str(project_root) not in sys.path:
     sys.path.insert(0, str(project_root))
 
-# 标准库
-import asyncio
 import json
 
 # 第三方库
@@ -263,7 +261,7 @@ async def stream_request_wrapper(payload):
             stream_request_wrapper,
             anti_truncation_payload,
             max_attempts,
-            enable_prefill_mode=("claude" not in str(api_request.get("model", "")).lower()),
+            enable_prefill_mode=True,
         )
 
         # 迭代 process_stream() 生成器，并展开 response 包装

src/router/geminicli/openai.py

@@ -259,7 +259,7 @@ async def stream_request_wrapper(payload):
             stream_request_wrapper,
             anti_truncation_payload,
             max_attempts,
-            enable_prefill_mode=("claude" not in str(api_request.get("model", "")).lower()),
+            enable_prefill_mode=True,
         )
 
         # 转换为 OpenAI 格式