🐛 Bug: Fix image input bug in o1-mini model

💻 Code: Change request header to curl/7.68.0
yym68686 · Dec 7, 2024 · c211d80 · c211d80
1 parent 89ab406
commit c211d80
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -713,7 +713,7 @@ async def ensure_config(request: Request, call_next):
 
         default_config = {
             "headers": {
-                "User-Agent": "OpenAI/Python 1.55.3",
+                "User-Agent": "curl/7.68.0",
                 "Accept": "*/*",
             },
             "http2": True,

diff --git a/request.py b/request.py
@@ -658,7 +658,7 @@ async def get_gpt_payload(request, engine, provider):
                 if item.type == "text":
                     text_message = await get_text_message(msg.role, item.text, engine)
                     content.append(text_message)
-                elif item.type == "image_url" and provider.get("image", True):
+                elif item.type == "image_url" and provider.get("image", True) and "o1-mini" not in model:
                     image_message = await get_image_message(item.image_url.url, engine)
                     content.append(image_message)
         else: