From b587e391714adc20e553b90f5eaa05018eca7da0 Mon Sep 17 00:00:00 2001 From: dvonthenen Date: Thu, 30 Nov 2023 08:26:26 -0800 Subject: [PATCH] Missing SummaryInfo on Prerecorded --- pkg/api/prerecorded/v1/interfaces/types.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/pkg/api/prerecorded/v1/interfaces/types.go b/pkg/api/prerecorded/v1/interfaces/types.go index fd13c8d1..ea68661a 100644 --- a/pkg/api/prerecorded/v1/interfaces/types.go +++ b/pkg/api/prerecorded/v1/interfaces/types.go @@ -18,7 +18,12 @@ type Metadata struct { Duration float64 `json:"duration,omitempty"` Channels int `json:"channels,omitempty"` Models []string `json:"models,omitempty"` - ModelInfo map[string]struct { + SummaryInfo struct { + InputTokens int `json:"input_tokens,omitempty"` + OutputTokens int `json:"output_tokens,omitempty"` + ModelUUID string `json:"model_uuid,omitempty"` + } `json:"summary_info,omitempty"` + ModelInfo map[string]struct { Name string `json:"name,omitempty"` Version string `json:"version,omitempty"` Arch string `json:"arch,omitempty"`