common : support manually triggering the reasoning budget end sequence (#23949)

2026-06-01 11:37:11 +02:00
parent e22b0de60d
commit 5254a7994d
5 changed files with 108 additions and 1 deletions
@@ -40,3 +40,7 @@ struct llama_sampler * common_reasoning_budget_init(
        common_reasoning_budget_state    initial_state = REASONING_BUDGET_IDLE);

 common_reasoning_budget_state common_reasoning_budget_get_state(const struct llama_sampler * smpl);
+
+// Manually transition the reasoning budget sampler into the FORCING state.
+// Returns true if the transition occurred.
+bool common_reasoning_budget_force(struct llama_sampler * smpl);