Fix image generation in OpenaiChat

xtekky · Apr 19, 2024 · 37b91e6 · 37b91e6
1 parent 7e543f4
commit 37b91e6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/g4f/Provider/needs_auth/OpenaiChat.py b/g4f/Provider/needs_auth/OpenaiChat.py
@@ -533,6 +533,8 @@ async def iter_messages_line(cls, session: StreamSession, line: bytes, fields: C
  return
  if line["message"]["metadata"]["message_type"] not in ("next", "continue", "variant"):
  return
+ if line["message"]["recipient"] != "all":
+ return
  if fields.conversation_id is None:
  fields.conversation_id = line["conversation_id"]
  fields.message_id = line["message"]["id"]