🐛 Bug: Fix the bug where the o1 model cannot obtain text format.
Browse files- request.py +1 -1
request.py
CHANGED
@@ -77,7 +77,7 @@ async def get_image_message(base64_image, engine = None):
|
|
77 |
raise ValueError("Unknown engine")
|
78 |
|
79 |
async def get_text_message(role, message, engine = None):
|
80 |
-
if "gpt" == engine or "claude" == engine or "openrouter" == engine or "vertex-claude" == engine:
|
81 |
return {"type": "text", "text": message}
|
82 |
if "gemini" == engine or "vertex-gemini" == engine:
|
83 |
return {"text": message}
|
|
|
77 |
raise ValueError("Unknown engine")
|
78 |
|
79 |
async def get_text_message(role, message, engine = None):
|
80 |
+
if "gpt" == engine or "claude" == engine or "openrouter" == engine or "vertex-claude" == engine or "o1" == engine:
|
81 |
return {"type": "text", "text": message}
|
82 |
if "gemini" == engine or "vertex-gemini" == engine:
|
83 |
return {"text": message}
|