errors.go

// This file was auto-generated by Fern from our API Definition.

package api

import (
	json "encoding/json"
	core "github.com/cohere-ai/cohere-go/v2/core"
)

type BadRequestError struct {
	*core.APIError
	Body interface{}
}

func (b *BadRequestError) UnmarshalJSON(data []byte) error {
	var body interface{}
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	b.StatusCode = 400
	b.Body = body
	return nil
}

func (b *BadRequestError) MarshalJSON() ([]byte, error) {
	return json.Marshal(b.Body)
}

func (b *BadRequestError) Unwrap() error {
	return b.APIError
}

// This error is returned when a request is cancelled by the user.
type ClientClosedRequestError struct {
	*core.APIError
	Body *ClientClosedRequestErrorBody
}

func (c *ClientClosedRequestError) UnmarshalJSON(data []byte) error {
	var body *ClientClosedRequestErrorBody
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	c.StatusCode = 499
	c.Body = body
	return nil
}

func (c *ClientClosedRequestError) MarshalJSON() ([]byte, error) {
	return json.Marshal(c.Body)
}

func (c *ClientClosedRequestError) Unwrap() error {
	return c.APIError
}

type ForbiddenError struct {
	*core.APIError
	Body interface{}
}

func (f *ForbiddenError) UnmarshalJSON(data []byte) error {
	var body interface{}
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	f.StatusCode = 403
	f.Body = body
	return nil
}

func (f *ForbiddenError) MarshalJSON() ([]byte, error) {
	return json.Marshal(f.Body)
}

func (f *ForbiddenError) Unwrap() error {
	return f.APIError
}

// This error is returned when a request to the server times out. This could be due to:
//   - An internal services taking too long to respond
type GatewayTimeoutError struct {
	*core.APIError
	Body *GatewayTimeoutErrorBody
}

func (g *GatewayTimeoutError) UnmarshalJSON(data []byte) error {
	var body *GatewayTimeoutErrorBody
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	g.StatusCode = 504
	g.Body = body
	return nil
}

func (g *GatewayTimeoutError) MarshalJSON() ([]byte, error) {
	return json.Marshal(g.Body)
}

func (g *GatewayTimeoutError) Unwrap() error {
	return g.APIError
}

type InternalServerError struct {
	*core.APIError
	Body interface{}
}

func (i *InternalServerError) UnmarshalJSON(data []byte) error {
	var body interface{}
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	i.StatusCode = 500
	i.Body = body
	return nil
}

func (i *InternalServerError) MarshalJSON() ([]byte, error) {
	return json.Marshal(i.Body)
}

func (i *InternalServerError) Unwrap() error {
	return i.APIError
}

type NotFoundError struct {
	*core.APIError
	Body interface{}
}

func (n *NotFoundError) UnmarshalJSON(data []byte) error {
	var body interface{}
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	n.StatusCode = 404
	n.Body = body
	return nil
}

func (n *NotFoundError) MarshalJSON() ([]byte, error) {
	return json.Marshal(n.Body)
}

func (n *NotFoundError) Unwrap() error {
	return n.APIError
}

// This error is returned when the requested feature is not implemented.
type NotImplementedError struct {
	*core.APIError
	Body *NotImplementedErrorBody
}

func (n *NotImplementedError) UnmarshalJSON(data []byte) error {
	var body *NotImplementedErrorBody
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	n.StatusCode = 501
	n.Body = body
	return nil
}

func (n *NotImplementedError) MarshalJSON() ([]byte, error) {
	return json.Marshal(n.Body)
}

func (n *NotImplementedError) Unwrap() error {
	return n.APIError
}

type ServiceUnavailableError struct {
	*core.APIError
	Body interface{}
}

func (s *ServiceUnavailableError) UnmarshalJSON(data []byte) error {
	var body interface{}
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	s.StatusCode = 503
	s.Body = body
	return nil
}

func (s *ServiceUnavailableError) MarshalJSON() ([]byte, error) {
	return json.Marshal(s.Body)
}

func (s *ServiceUnavailableError) Unwrap() error {
	return s.APIError
}

// Too many requests
type TooManyRequestsError struct {
	*core.APIError
	Body *TooManyRequestsErrorBody
}

func (t *TooManyRequestsError) UnmarshalJSON(data []byte) error {
	var body *TooManyRequestsErrorBody
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	t.StatusCode = 429
	t.Body = body
	return nil
}

func (t *TooManyRequestsError) MarshalJSON() ([]byte, error) {
	return json.Marshal(t.Body)
}

func (t *TooManyRequestsError) Unwrap() error {
	return t.APIError
}

type UnauthorizedError struct {
	*core.APIError
	Body interface{}
}

func (u *UnauthorizedError) UnmarshalJSON(data []byte) error {
	var body interface{}
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	u.StatusCode = 401
	u.Body = body
	return nil
}

func (u *UnauthorizedError) MarshalJSON() ([]byte, error) {
	return json.Marshal(u.Body)
}

func (u *UnauthorizedError) Unwrap() error {
	return u.APIError
}

// This error is returned when the request is not well formed. This could be because:
//   - JSON is invalid
//   - The request is missing required fields
//   - The request contains an invalid combination of fields
type UnprocessableEntityError struct {
	*core.APIError
	Body *UnprocessableEntityErrorBody
}

func (u *UnprocessableEntityError) UnmarshalJSON(data []byte) error {
	var body *UnprocessableEntityErrorBody
	if err := json.Unmarshal(data, &body); err != nil {
		return err
	}
	u.StatusCode = 422
	u.Body = body
	return nil
}

func (u *UnprocessableEntityError) MarshalJSON() ([]byte, error) {
	return json.Marshal(u.Body)
}

func (u *UnprocessableEntityError) Unwrap() error {
	return u.APIError
}