Switch image model to "moondream" in integration test

This commit is contained in:
Amith Koujalgi 2025-03-11 13:12:55 +05:30
parent 7e5ca53fda
commit ac3f505aa6
No known key found for this signature in database
GPG Key ID: 3F065E7150B71F9D

View File

@ -520,7 +520,7 @@ public class OllamaAPIIntegrationTest {
@Test @Test
@Order(19) @Order(19)
void testChatWithImageFromFileWithHistoryRecognition() throws OllamaBaseException, IOException, URISyntaxException, InterruptedException { void testChatWithImageFromFileWithHistoryRecognition() throws OllamaBaseException, IOException, URISyntaxException, InterruptedException {
String imageModel = "llava"; String imageModel = "moondream";
api.pullModel(imageModel); api.pullModel(imageModel);
OllamaChatRequestBuilder builder = OllamaChatRequestBuilder builder =
OllamaChatRequestBuilder.getInstance(imageModel); OllamaChatRequestBuilder.getInstance(imageModel);
@ -531,7 +531,6 @@ public class OllamaAPIIntegrationTest {
OllamaChatResult chatResult = api.chat(requestModel); OllamaChatResult chatResult = api.chat(requestModel);
assertNotNull(chatResult); assertNotNull(chatResult);
assertNotNull(chatResult.getResponseModel()); assertNotNull(chatResult.getResponseModel());
builder.reset(); builder.reset();
requestModel = requestModel =