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

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

@ -54,11 +54,13 @@ class OllamaAPIIntegrationTest {
private static final String EMBEDDING_MODEL = "all-minilm"; private static final String EMBEDDING_MODEL = "all-minilm";
private static final String VISION_MODEL = "moondream:1.8b"; 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 = "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 GENERAL_PURPOSE_MODEL = "gemma3:270m";
private static final String TOOLS_MODEL = "mistral:7b"; private static final String TOOLS_MODEL = "mistral:7b";
@BeforeAll @BeforeAll
static void setUp() { static void setUp() {
// defaults
int requestTimeoutSeconds = 60; int requestTimeoutSeconds = 60;
int numberOfRetriesForModelPull = 5; int numberOfRetriesForModelPull = 5;
@ -685,9 +687,9 @@ class OllamaAPIIntegrationTest {
URISyntaxException, URISyntaxException,
InterruptedException, InterruptedException,
ToolInvocationException { ToolInvocationException {
api.pullModel(THINKING_TOOL_MODEL); api.pullModel(THINKING_TOOL_MODEL_2);
OllamaChatRequestBuilder builder = OllamaChatRequestBuilder builder =
OllamaChatRequestBuilder.getInstance(THINKING_TOOL_MODEL); OllamaChatRequestBuilder.getInstance(THINKING_TOOL_MODEL_2);
OllamaChatRequest requestModel = OllamaChatRequest requestModel =
builder.withMessage( builder.withMessage(
OllamaChatMessageRole.USER, OllamaChatMessageRole.USER,