Skip to content

Commit bcc0fed

Browse files
committed
Fix large image passing in tests
1 parent aa339c1 commit bcc0fed

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

integration-tests/src/jvmTest/kotlin/ai/koog/integration/tests/SingleLLMPromptExecutorIntegrationTest.kt

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -691,7 +691,21 @@ class SingleLLMPromptExecutorIntegrationTest {
691691
}
692692

693693
attachments {
694-
image(imageFile.absolutePath)
694+
when (scenario) {
695+
ImageTestScenario.LARGE_IMAGE, ImageTestScenario.LARGE_IMAGE_ANTHROPIC -> {
696+
image(
697+
Attachment.Image(
698+
content = AttachmentContent.Binary.Bytes(imageFile.readBytes()),
699+
format = "jpg",
700+
mimeType = "image/jpeg"
701+
)
702+
)
703+
}
704+
705+
else -> {
706+
image(Path(imageFile.absolutePath))
707+
}
708+
}
695709
}
696710
}
697711
}

0 commit comments

Comments
 (0)