From fa20daf6e540e8a40bdea4160d28e757f21d6f3a Mon Sep 17 00:00:00 2001 From: anjeongkyun Date: Sun, 21 Apr 2024 23:37:18 +0900 Subject: [PATCH] Adds test case of testRequestWithOptions --- .../jackson/TestChatRequestSerialization.java | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/src/test/java/io/github/amithkoujalgi/ollama4j/unittests/jackson/TestChatRequestSerialization.java b/src/test/java/io/github/amithkoujalgi/ollama4j/unittests/jackson/TestChatRequestSerialization.java index c5a7060..f6237cc 100644 --- a/src/test/java/io/github/amithkoujalgi/ollama4j/unittests/jackson/TestChatRequestSerialization.java +++ b/src/test/java/io/github/amithkoujalgi/ollama4j/unittests/jackson/TestChatRequestSerialization.java @@ -51,12 +51,27 @@ public class TestChatRequestSerialization extends AbstractRequestSerializationTe public void testRequestWithOptions() { OptionsBuilder b = new OptionsBuilder(); OllamaChatRequestModel req = builder.withMessage(OllamaChatMessageRole.USER, "Some prompt") - .withOptions(b.setMirostat(1).build()).build(); + .withOptions(b.setMirostat(1).build()) + .withOptions(b.setTemperature(1L).build()) + .withOptions(b.setMirostatEta(1L).build()) + .withOptions(b.setMirostatTau(1L).build()) + .withOptions(b.setNumGpu(1).build()) + .withOptions(b.setSeed(1).build()) + .withOptions(b.setTopK(1).build()) + .withOptions(b.setTopP(1).build()) + .build(); String jsonRequest = serializeRequest(req); - OllamaChatRequestModel deserializeRequest = deserializeRequest(jsonRequest,OllamaChatRequestModel.class); + OllamaChatRequestModel deserializeRequest = deserializeRequest(jsonRequest, OllamaChatRequestModel.class); assertEqualsAfterUnmarshalling(deserializeRequest, req); assertEquals(1, deserializeRequest.getOptions().get("mirostat")); + assertEquals(1.0, deserializeRequest.getOptions().get("temperature")); + assertEquals(1.0, deserializeRequest.getOptions().get("mirostat_eta")); + assertEquals(1.0, deserializeRequest.getOptions().get("mirostat_tau")); + assertEquals(1, deserializeRequest.getOptions().get("num_gpu")); + assertEquals(1, deserializeRequest.getOptions().get("seed")); + assertEquals(1, deserializeRequest.getOptions().get("top_k")); + assertEquals(1.0, deserializeRequest.getOptions().get("top_p")); } @Test