Fix unexpected None type for images

This commit is contained in:
doctorpangloss 2025-02-15 16:52:08 -08:00
parent d4218f3f19
commit 3a3e31a0a5

View File

@ -67,6 +67,7 @@ class OpenAILanguageModelWrapper(LanguageModel):
prompt = tokens.get("inputs", [])
prompt = "".join(prompt)
images = tokens.get("images", [])
images = [image for image in images if image is not None]
messages: list[ChatCompletionMessageParam] = [
{
"role": "user",