i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-01-02 15:52:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): 优化字节流迭代器以支持自动解压 gzip (#39)

Browse Source
This commit is contained in:
hank9999
2025-12-24 01:11:35 +08:00
committed by GitHub
parent d44cfaddf6
commit e38cd6819b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/api/handlers/base/cli_handler_base.py
View File

@@ -476,8 +476,8 @@ class CliMessageHandlerBase(BaseMessageHandler):
stream_response.raise_for_status() stream_response.raise_for_status()
# 使用字节流迭代器（避免 aiter_lines 的性能问题） # 使用字节流迭代器（避免 aiter_lines 的性能问题, aiter_bytes 会自动解压 gzip/deflate）
byte_iterator = stream_response.aiter_raw() byte_iterator = stream_response.aiter_bytes()
# 预读第一个数据块，检测嵌套错误（HTTP 200 但响应体包含错误） # 预读第一个数据块，检测嵌套错误（HTTP 200 但响应体包含错误）
prefetched_chunks = await self._prefetch_and_check_embedded_error( prefetched_chunks = await self._prefetch_and_check_embedded_error(
@@ -531,7 +531,7 @@ class CliMessageHandlerBase(BaseMessageHandler):
# 检查是否需要格式转换 # 检查是否需要格式转换
needs_conversion = self._needs_format_conversion(ctx) needs_conversion = self._needs_format_conversion(ctx)
async for chunk in stream_response.aiter_raw(): async for chunk in stream_response.aiter_bytes():
# 在第一次输出数据前更新状态为 streaming # 在第一次输出数据前更新状态为 streaming
if not streaming_status_updated: if not streaming_status_updated:
self._update_usage_to_streaming_with_ctx(ctx) self._update_usage_to_streaming_with_ctx(ctx)