Merge pull request #88 from seeseemelk/feature/token-streamer

Add ability to stream tokens in chat
2025-01-26 17:37:22 +05:30
parent a06a4025fa 761fbc3398
commit dda807d818
5 changed files with 51 additions and 36 deletions
--- a/src/test/java/io/github/ollama4j/integrationtests/TestRealAPIs.java
+++ b/src/test/java/io/github/ollama4j/integrationtests/TestRealAPIs.java
@@ -321,7 +321,7 @@ class TestRealAPIs {
            assertEquals(1, function.getArguments().size());
            Object noOfDigits = function.getArguments().get("noOfDigits");
            assertNotNull(noOfDigits);
-            assertEquals("5",noOfDigits);
+            assertEquals("5", noOfDigits.toString());
            assertTrue(chatResult.getChatHistory().size()>2);
            List<OllamaChatToolCalls> finalToolCalls = chatResult.getResponseModel().getMessage().getToolCalls();
            assertNull(finalToolCalls);