Niansuh commited on
Commit
cb023d9
·
verified ·
1 Parent(s): b88e75c

Update api/utils.py

Browse files
Files changed (1) hide show
  1. api/utils.py +2 -1
api/utils.py CHANGED
@@ -52,6 +52,7 @@ def message_to_dict(message, model_prefix: Optional[str] = None):
52
  content = message.content
53
  if model_prefix:
54
  content = f"{model_prefix} {content}"
 
55
  if isinstance(message.content, list) and len(message.content) == 2:
56
  return {
57
  "role": message.role,
@@ -93,7 +94,7 @@ async def process_streaming_response(request: ChatRequest):
93
  headers_api_chat = get_headers_api_chat(referer_url)
94
 
95
  if request.model == 'o1-preview':
96
- delay_seconds = random.randint(20, 60)
97
  logger.info(f"Introducing a delay of {delay_seconds} seconds for model 'o1-preview' (Chat ID: {chat_id})")
98
  await asyncio.sleep(delay_seconds)
99
 
 
52
  content = message.content
53
  if model_prefix:
54
  content = f"{model_prefix} {content}"
55
+ # Allow sending base64 image data for all models
56
  if isinstance(message.content, list) and len(message.content) == 2:
57
  return {
58
  "role": message.role,
 
94
  headers_api_chat = get_headers_api_chat(referer_url)
95
 
96
  if request.model == 'o1-preview':
97
+ delay_seconds = random.randint(1, 60)
98
  logger.info(f"Introducing a delay of {delay_seconds} seconds for model 'o1-preview' (Chat ID: {chat_id})")
99
  await asyncio.sleep(delay_seconds)
100