nomyo-ai/nomyo-router
2
2
Fork 0
Code Issues 1 Pull requests Projects Releases 5 Packages 1 Wiki Activity Actions

fix: conditional to_thread for the image_transform to relieve threadpool pressure

Browse source
This commit is contained in:
Alpha Nerd 2026-04-07 13:28:34 +02:00
parent 81013ec3b1
commit 2c87472483
Signed by: alpha-nerd
SSH key fingerprint: SHA256:QkkAgVoYi9TQ0UKPkiKSfnerZy2h4qhi3SVPXJmBN+M
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
router.py
View file

@ -1059,6 +1059,7 @@ async def _make_chat_request(model: str, messages: list, tools=None, stream: boo
if ":latest" in model: if ":latest" in model:
model = model.split(":latest")[0] model = model.split(":latest")[0]
if messages: if messages:
if any("images" in m for m in messages):
messages = await asyncio.to_thread(transform_images_to_data_urls, messages) messages = await asyncio.to_thread(transform_images_to_data_urls, messages)
messages = transform_tool_calls_to_openai(messages) messages = transform_tool_calls_to_openai(messages)
messages = _strip_assistant_prefill(messages) messages = _strip_assistant_prefill(messages)
@ -1965,6 +1966,7 @@ async def chat_proxy(request: Request):
model = model.split(":latest") model = model.split(":latest")
model = model[0] model = model[0]
if messages: if messages:
if any("images" in m for m in messages):
messages = await asyncio.to_thread(transform_images_to_data_urls, messages) messages = await asyncio.to_thread(transform_images_to_data_urls, messages)
messages = transform_tool_calls_to_openai(messages) messages = transform_tool_calls_to_openai(messages)
messages = _strip_assistant_prefill(messages) messages = _strip_assistant_prefill(messages)