config_goal_test.go RawHistoryFolder

1package agent
2
3import (
4	"encoding/json"
5	"testing"
6)
7
8func TestGoalsConfigTokenBudgetIsOptional(t *testing.T) {
9	var unlimited Config
10	if err := json.Unmarshal([]byte(`{}`), &unlimited); err != nil {
11		t.Fatal(err)
12	}
13	if unlimited.Goals.MaxTokenBudget != nil {
14		t.Fatalf("missing budget = %v, want nil", *unlimited.Goals.MaxTokenBudget)
15	}
16
17	var limited Config
18	if err := json.Unmarshal([]byte(`{"goals":{"max_token_budget":1000000}}`), &limited); err != nil {
19		t.Fatal(err)
20	}
21	if limited.Goals.MaxTokenBudget == nil || *limited.Goals.MaxTokenBudget != 1_000_000 {
22		t.Fatalf("configured budget = %#v", limited.Goals.MaxTokenBudget)
23	}
24}