diff --git a/controller/channel-test.go b/controller/channel-test.go index 5e44aa75..dae6eb17 100644 --- a/controller/channel-test.go +++ b/controller/channel-test.go @@ -58,7 +58,7 @@ func testChannel(channel *model.Channel, request ChatRequest) error { if err != nil { return err } - if response.Usage.CompletionTokens == 0 { + if response.Id == "" { return errors.New(fmt.Sprintf("type %s, code %v, message %s", response.Error.Type, response.Error.Code, response.Error.Message)) } return nil diff --git a/controller/relay.go b/controller/relay.go index a9a5a364..c2c834e1 100644 --- a/controller/relay.go +++ b/controller/relay.go @@ -70,6 +70,7 @@ type OpenAIErrorWithStatusCode struct { } type TextResponse struct { + Id string `json:"id"` Usage `json:"usage"` Error OpenAIError `json:"error"` }