🐛 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}
|