Skip to content

Commit 79aecdf

Browse files
committed
Add tests for InferenceParameters.setContinueFinalMessage
Two tests covering true/false — consistent with every other boolean setter test in InferenceParametersTest. https://claude.ai/code/session_01GYjU7CXHB6QLbxrbZcQWCn
1 parent f1aecc1 commit 79aecdf

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

src/test/java/net/ladenthin/llama/InferenceParametersTest.java

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,18 @@ public void testSetReasoningBudgetTokensDisabled() {
292292
assertEquals("-1", params.parameters.get("reasoning_budget_tokens"));
293293
}
294294

295+
@Test
296+
public void testSetContinueFinalMessageTrue() {
297+
InferenceParameters params = new InferenceParameters("").setContinueFinalMessage(true);
298+
assertEquals("true", params.parameters.get("continue_final_message"));
299+
}
300+
301+
@Test
302+
public void testSetContinueFinalMessageFalse() {
303+
InferenceParameters params = new InferenceParameters("").setContinueFinalMessage(false);
304+
assertEquals("false", params.parameters.get("continue_final_message"));
305+
}
306+
295307
// -------------------------------------------------------------------------
296308
// MiroStat
297309
// -------------------------------------------------------------------------

0 commit comments

Comments
 (0)