Update _llm_caption.py

This commit is contained in:
hlohaus 2025-05-01 11:27:34 +02:00
parent 4a2f793869
commit c856a6c2c1

View file

@ -49,8 +49,8 @@ def llm_caption(
# Call the OpenAI API # Call the OpenAI API
response = client.chat.completions.create(model=model, messages=messages) response = client.chat.completions.create(model=model, messages=messages)
if asyncio.iscoroutine(response): if asyncio.iscoroutine(response):
async def read_content(): async def read_content(response):
response = await response response = await response
return response.choices[0].message.content return response.choices[0].message.content
return read_content() return read_content(response)
return response.choices[0].message.content return response.choices[0].message.content