api_models.go

package moonshot

import (
	"context"
)

type IModels interface {
	List(ctx context.Context) (*ListModelsResponse, error)
}

type models struct {
	client *Client
}

// Models returns a new models controller
func (c *Client) Models() IModels {
	return &models{
		client: c,
	}
}

type ListModelsRequest struct {
}

type ListModelsResponse struct {
	CommonResponse
	Object string                   `json:"object"`
	Data   []*ListModelResponseData `json:"data"`
}

type ListModelResponseData struct {
	Created    int                                 `json:"created"`
	ID         string                              `json:"id"`
	Object     string                              `json:"object"`
	OwnedBy    string                              `json:"owned_by"`
	Permission []*ListModelsResponseDataPermission `json:"permission"`
	Root       string                              `json:"root"`
	Parent     string                              `json:"parent"`
}

type ListModelsResponseDataPermission struct {
	Created            int    `json:"created"`
	ID                 string `json:"id"`
	Object             string `json:"object"`
	AllowCreateEngine  bool   `json:"allow_create_engine"`
	AllowSampling      bool   `json:"allow_sampling"`
	AllowLogprobs      bool   `json:"allow_logprobs"`
	AllowSearchIndices bool   `json:"allow_search_indices"`
	AllowView          bool   `json:"allow_view"`
	AllowFineTuning    bool   `json:"allow_fine_tuning"`
	Organization       string `json:"organization"`
	Group              string `json:"group"`
	IsBlocking         bool   `json:"is_blocking"`
}

// List lists all models
func (m *models) List(ctx context.Context) (*ListModelsResponse, error) {
	const path = "/v1/models"
	resp, err := m.client.HTTPClient().SetPath(path).Get(ctx)
	if err != nil {
		return nil, err
	}
	if !resp.StatusOK() {
		return nil, ResponseToError(resp)
	}
	listModelResp := new(ListModelsResponse)
	err = resp.Unmarshal(listModelResp)
	if err != nil {
		return nil, err
	}
	return listModelResp, nil
}