1import type { Image } from './image.js';
2import type { Candidate } from './candidate.js';
3 
4export class ModelOutput {
5  public metadata: string[];
6  public candidates: Candidate[];
7  public chosen: number;
8 
9  constructor(params: { metadata: string[]; candidates: Candidate[]; chosen?: number }) {
10    this.metadata = params.metadata;
11    this.candidates = params.candidates;
12    this.chosen = params.chosen ?? 0;
13  }
14 
15  toString(): string {
16    return this.text;
17  }
18 
19  get text(): string {
20    return this.candidates[this.chosen]?.text ?? '';
21  }
22 
23  get thoughts(): string | null {
24    return this.candidates[this.chosen]?.thoughts ?? null;
25  }
26 
27  get images(): Image[] {
28    return this.candidates[this.chosen]?.images ?? [];
29  }
30 
31  get rcid(): string {
32    return this.candidates[this.chosen]?.rcid ?? '';
33  }
34}
35 
36