yym68686 commited on
Commit
e5b8220
·
1 Parent(s): cfee5f1

🐛 Bug: Fix the bug where the o1 model cannot obtain text format.

Browse files
Files changed (1) hide show
  1. 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}