Source file `thinking.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
type budget_tokens = int

let budget n = if n >= 1024 then Ok n else Error (Printf.sprintf "thinking budget must be >= 1024, got %d" n)

let budget_exn n =
  match budget n with
  | Ok v -> v
  | Error msg -> invalid_arg msg

let to_int n = n

type display =
  | Summarized
  | Omitted

type t =
  | Enabled of {
      budget_tokens : budget_tokens;
      display : display option;
    }
  | Adaptive of { display : display option }
  | Disabled