Update api/utils.py
Browse files- 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(
|
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 |
|