Skip to content

Commit f4c22b5

Browse files
basketikuncodex
andcommitted
fix helper image extraction to use latest user turn
Co-Authored-By: Codex <noreply@openai.com>
1 parent 74065c4 commit f4c22b5

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@
1717
> [!IMPORTANT]
1818
> 本项目基于对 ChatGPT 官网相关能力的逆向研究实现,存在账号受限、临时封禁或永久封禁的风险。请勿使用你自己的重要账号、常用账号或高价值账号进行测试。
1919
20+
> [!CAUTION]
21+
> 旧版本存在已知漏洞,请尽快升级到最新版本。公网部署时请尽量不要放置敏感信息,并自行做好访问控制与隔离。
22+
2023
## 快速开始
2124

2225
已发布镜像支持 `linux/amd64``linux/arm64`,在 x86 服务器和 Apple Silicon / ARM Linux 设备上都会自动拉取匹配架构的版本。

utils/helper.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -183,16 +183,15 @@ def extract_chat_image(body: dict[str, object]) -> list[tuple[bytes, str]]:
183183
messages = body.get("messages")
184184
if not isinstance(messages, list):
185185
return []
186-
all_images = []
187186
for message in reversed(messages):
188187
if not isinstance(message, dict):
189188
continue
190189
if str(message.get("role") or "").strip().lower() != "user":
191190
continue
192191
images = extract_image_from_message_content(message.get("content"))
193192
if images:
194-
all_images.extend(images)
195-
return all_images
193+
return images
194+
return []
196195

197196

198197
def extract_chat_prompt(body: dict[str, object]) -> str:

0 commit comments

Comments
 (0)