Mirror/ollama4j
1
0
Fork 1
mirror of https://github.com/amithkoujalgi/ollama4j.git synced 2025-10-14 09:28:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update thinking tool model in integration tests

Browse Source 
Replaced the 'deepseek-r1:1.5b' model with 'qwen3:0.6b' for thinking tool tests in OllamaAPIIntegrationTest. Also made minor formatting improvements to comments and string concatenations for better readability.
This commit is contained in:
amithkoujalgi 2025-09-20 00:02:52 +05:30
parent 5da9bc8626
commit 751b11881f
No known key found for this signature in database
GPG Key ID: E29A37746AF94B70
1 changed files with 36 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/test/java/io/github/ollama4j/integrationtests/OllamaAPIIntegrationTest.java
View File

@ -5,7 +5,7 @@
* Licensed under the MIT License (the "License");
* you may not use this file except in compliance with the License.
*
*/
*/
package io.github.ollama4j.integrationtests;
import static org.junit.jupiter.api.Assertions.*;
@ -54,11 +54,13 @@ class OllamaAPIIntegrationTest {
private static final String EMBEDDING_MODEL = "all-minilm";
private static final String VISION_MODEL = "moondream:1.8b";
private static final String THINKING_TOOL_MODEL = "deepseek-r1:1.5b";
private static final String THINKING_TOOL_MODEL_2 = "qwen3:0.6b";
private static final String GENERAL_PURPOSE_MODEL = "gemma3:270m";
private static final String TOOLS_MODEL = "mistral:7b";
@BeforeAll
static void setUp() {
// defaults
int requestTimeoutSeconds = 60;
int numberOfRetriesForModelPull = 5;
@ -685,9 +687,9 @@ class OllamaAPIIntegrationTest {
URISyntaxException,
InterruptedException,
ToolInvocationException {
api.pullModel(THINKING_TOOL_MODEL);
api.pullModel(THINKING_TOOL_MODEL_2);
OllamaChatRequestBuilder builder =
OllamaChatRequestBuilder.getInstance(THINKING_TOOL_MODEL);
OllamaChatRequestBuilder.getInstance(THINKING_TOOL_MODEL_2);
OllamaChatRequest requestModel =
builder.withMessage(
OllamaChatMessageRole.USER,