Adds test of testWithTemplate

This commit is contained in:
anjeongkyun 2024-04-21 23:43:49 +09:00
parent fa20daf6e5
commit 9900ae92fb

View File

@ -86,4 +86,12 @@ public class TestChatRequestSerialization extends AbstractRequestSerializationTe
String requestFormatProperty = jsonObject.getString("format"); String requestFormatProperty = jsonObject.getString("format");
assertEquals("json", requestFormatProperty); assertEquals("json", requestFormatProperty);
} }
@Test
public void testWithTemplate() {
OllamaChatRequestModel req = builder.withTemplate("System Template")
.build();
String jsonRequest = serializeRequest(req);
assertEqualsAfterUnmarshalling(deserializeRequest(jsonRequest, OllamaChatRequestModel.class), req);
}
} }