moonshot

type ChatCompletionsFinishReason string

type ChatCompletionsMessage struct {
	Role    ChatCompletionsMessageRole `json:"role"`
	Content string                     `json:"content"`
	Partial bool                       `json:"partial,omitempty"`
	Name    string                     `json:"name,omitempty"`
	// returns only in use tool response
	ToolCalls []*ChatCompletionsResponseToolCalls `json:"tool_calls,omitempty"`
	// use tool request need it
	ToolCallID string `json:"tool_call_id,omitempty"`
}

type ChatCompletionsMessageRole string

type ChatCompletionsModelFamily string

type ChatCompletionsModelID string

type ChatCompletionsParametersType string

type ChatCompletionsRequest struct {
	Messages         []*ChatCompletionsMessage             `json:"messages"`
	Model            ChatCompletionsModelID                `json:"model"`
	MaxTokens        int                                   `json:"max_tokens"`
	Temperature      float64                               `json:"temperature"`
	TopP             float64                               `json:"top_p"`
	N                int                                   `json:"n"`
	PresencePenalty  float64                               `json:"presence_penalty"`
	FrequencyPenalty float64                               `json:"frequency_penalty"`
	ResponseFormat   *ChatCompletionsRequestResponseFormat `json:"response_format"`
	Stop             []string                              `json:"stop"`
	Stream           bool                                  `json:"stream"`
	// When you use a tool, you need to define it
	Tools []*ChatCompletionsTool `json:"tools,omitempty"`
}

type ChatCompletionsRequestResponseFormat struct {
	Type ChatCompletionsResponseFormatType `json:"type"`
}

type ChatCompletionsResponse struct {
	ID      string                            `json:"id"`
	Object  string                            `json:"object"`
	Created int                               `json:"created"`
	Model   string                            `json:"model"`
	Choices []*ChatCompletionsResponseChoices `json:"choices"`
	// returns only in non-stream mode
	Usage *ChatCompletionsResponseUsage `json:"usage,omitempty"`
}

type ChatCompletionsResponseChoices struct {
	Index int `json:"index"`

	// return with no stream
	Message *ChatCompletionsMessage `json:"message,omitempty"`
	// return With stream
	Delta *ChatCompletionsMessage `json:"delta,omitempty"`

	FinishReason ChatCompletionsFinishReason `json:"finish_reason"`

	// returns only in stream mode
	Usage *ChatCompletionsResponseUsage `json:"usage,omitempty"`
}

type ChatCompletionsResponseFormatType string

type ChatCompletionsResponseToolCalls struct {
	Index    int64                                     `json:"index"`
	ID       string                                    `json:"id"`
	Type     string                                    `json:"type"`
	Function *ChatCompletionsResponseToolCallsFunction `json:"function"`
}

type ChatCompletionsResponseToolCallsFunction struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type ChatCompletionsResponseUsage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

type ChatCompletionsStreamResponse struct {
	// contains filtered or unexported fields
}

type ChatCompletionsStreamResponseReceive struct {
	ChatCompletionsResponse
	// contains filtered or unexported fields
}

type ChatCompletionsTool struct {
	Type     ChatCompletionsToolType      `json:"type"`
	Function *ChatCompletionsToolFunction `json:"function"`
}

type ChatCompletionsToolBuiltinFunctionWebSearchArguments struct {
	SearchResult struct {
		SearchId string `json:"search_id"`
	} `json:"search_result"`
	Usage struct {
		TotalTokens int `json:"total_tokens"`
	} `json:"usage"`
}

type ChatCompletionsToolFunction struct {
	Name        string                                 `json:"name"`
	Description string                                 `json:"description,omitempty"`
	Parameters  *ChatCompletionsToolFunctionParameters `json:"parameters,omitempty"`
}

type ChatCompletionsToolFunctionParameters struct {
	Type       ChatCompletionsParametersType                     `json:"type"`
	Properties map[string]*ChatCompletionsToolFunctionProperties `json:"properties"`
	Required   []string                                          `json:"required,omitempty"`
}

type ChatCompletionsToolFunctionProperties struct {
	Type        string   `json:"type"`
	Description string   `json:"description,omitempty"`
	Enum        []string `json:"enum,omitempty"`
}

type ChatCompletionsToolType string

type Client struct {
	// contains filtered or unexported fields
}

type CommonAPIResponse struct {
	Error *CommonAPIResponseError `json:"error,omitempty"`
}

type CommonAPIResponseError struct {
	Message string `json:"message,omitempty"`
	Type    string `json:"type,omitempty"`
}

type CommonResponse struct {
	Code    int    `json:"code,omitempty"`
	Error   string `json:"error,omitempty"`
	Message string `json:"message,omitempty"`
	Method  string `json:"method,omitempty"`
	Scode   string `json:"scode,omitempty"`
	Status  bool   `json:"status,omitempty"`
	UA      string `json:"ua,omitempty"`
	URL     string `json:"url,omitempty"`
}

type Config struct {
	Host   string
	APIKey string
	Debug  bool
}

type ContextCache struct {
	Id          string                     `json:"id"`          // 缓存的唯一标识
	Status      ContextCacheStatus         `json:"status"`      // 缓存的状态
	Object      string                     `json:"object"`      // 缓存的类型
	CreatedAt   int64                      `json:"created_at"`  // 缓存的创建时间
	ExpiredAt   int64                      `json:"expired_at"`  // 缓存的过期时间
	Tokens      int                        `json:"tokens"`      // 缓存的 Token 数量
	Model       ChatCompletionsModelFamily `json:"model"`       // 缓存的模型组名称
	Messages    []ChatCompletionsMessage   `json:"messages"`    // 缓存的消息内容
	Tools       []ChatCompletionsTool      `json:"tools"`       // 缓存使用的工具
	Name        string                     `json:"name"`        // 缓存的名称
	Description string                     `json:"description"` // 缓存的描述信息
	Metadata    map[string]string          `json:"metadata"`    // 缓存的元信息
}

type ContextCacheContent struct {
	CacheId  string `json:"cache_id"`
	Tag      string `json:"tag"`
	ResetTTL int64  `json:"reset_ttl"`
	DryRun   bool   `json:"dry_run"`
}

type ContextCacheCreateRequest struct {
	Model       ChatCompletionsModelFamily `json:"model"`                 // 模型组（model family）名称
	Messages    []ChatCompletionsMessage   `json:"messages"`              // 消息内容
	Tools       []ChatCompletionsTool      `json:"tools,omitempty"`       // 使用的工具
	Name        string                     `json:"name,omitempty"`        // 缓存名称
	Description string                     `json:"description,omitempty"` // 缓存描述信息
	Metadata    map[string]string          `json:"metadata,omitempty"`    // 缓存的元信息
	ExpiredAt   int64                      `json:"expired_at"`            // 缓存的过期时间
	TTL         int64                      `json:"ttl,omitempty"`         // 缓存的有效期
}

type ContextCacheCreateResponse ContextCache

type ContextCacheCreateTagRequest struct {
	Tag     string `json:"tag"`      // 缓存的标签
	CacheId string `json:"cache_id"` // 缓存的唯一标识
}

type ContextCacheCreateTagResponse ContextCacheTag

type ContextCacheDeleteRequest struct {
	Id string `json:"id"` // 缓存的唯一标识
}

type ContextCacheDeleteResponse struct {
	Deleted bool   `json:"deleted"` // 缓存是否被删除
	Id      string `json:"id"`      // 被删除的缓存的唯一标识
	Object  string `json:"object"`  // 返回的数据类型
}

type ContextCacheDeleteTagRequest struct {
	Tag string `json:"_"` // 缓存的标签
}

type ContextCacheDeleteTagResponse struct {
	Deleted bool   `json:"deleted"` // 缓存是否被删除
	Object  string `json:"object"`  // 返回的数据类型
	Tag     string `json:"tag"`     // 被删除的缓存的标签
}

type ContextCacheGetRequest struct {
	Id string `json:"_"` // 缓存的唯一标识
}

type ContextCacheGetResponse ContextCache

type ContextCacheGetTagContentRequest struct {
	Tag string `json:"_"` // 缓存的标签
}

type ContextCacheGetTagContentResponse ContextCache

type ContextCacheGetTagRequest struct {
	Tag string `json:"_"` // 缓存的标签
}

type ContextCacheGetTagResponse ContextCacheTag

type ContextCacheListRequest struct {
	Limit    int               `json:"limit,omitempty"`    // 当前请求单页返回的缓存数量
	Order    ContextCacheOrder `json:"order,omitempty"`    // 当前请求时查询缓存的排序规则
	After    string            `json:"after,omitempty"`    // 当前请求时，应该从哪一个缓存开始进行查找
	Before   string            `json:"before,omitempty"`   // 当前请求时，应该查询到哪一个缓存为止
	Metadata map[string]string `json:"metadata,omitempty"` // 用于筛选缓存的 metadata 信息
}

type ContextCacheListResponse struct {
	Object string         `json:"object"` // 返回的数据类型
	Data   []ContextCache `json:"data"`   // 返回的缓存列表
}

type ContextCacheListTagRequest struct {
	Limit  int               `json:"limit,omitempty"`  // 当前请求单页返回的缓存数量
	Order  ContextCacheOrder `json:"order,omitempty"`  // 当前请求时查询缓存的排序规则
	After  string            `json:"after,omitempty"`  // 当前请求时，应该从哪一个缓存开始进行查找
	Before string            `json:"before,omitempty"` // 当前请求时，应该查询到哪一个缓存为止
}

type ContextCacheListTagResponse struct {
	Object string            `json:"object"` // 返回的数据类型
	Data   []ContextCacheTag `json:"data"`   // 返回的缓存标签列表
}

type ContextCacheOrder string

type ContextCacheStatus string

type ContextCacheTag struct {
	Tag       string `json:"tag"`        // 缓存的标签
	CacheId   string `json:"cache_id"`   // 缓存的唯一标识
	Object    string `json:"object"`     // 缓存的类型
	OwnedBy   string `json:"owned_by"`   // 缓存的拥有者
	CreatedAt int    `json:"created_at"` // 缓存的创建时间
}

type ContextCacheUpdateRequest struct {
	Id        string            `json:"_"`                  // 缓存的唯一标识
	Metadata  map[string]string `json:"metadata,omitempty"` // 缓存的元信息
	ExpiredAt int64             `json:"expired_at"`         // 缓存的过期时间
	TTL       int64             `json:"ttl,omitempty"`      // 缓存的有效期
}

type ContextCacheUpdateResponse ContextCache

type FileContentResponse struct {
	Content  string `json:"content"`
	FileType string `json:"file_type"`
	Filename string `json:"filename"`
	Title    string `json:"title"`
	Type     string `json:"type"`
}

type FilesBatchDeleteRequest struct {
	FileIDList []string `json:"file_ids"`
}

type FilesBatchDeleteResponse struct {
	RespList  []*FilesDeleteResponse `json:"resp_list"`
	ErrorList []error                `json:"error_list"`
}

type FilesDeleteResponse struct {
	Deleted bool   `json:"deleted"`
	ID      string `json:"id"`
	Object  string `json:"object"`
}

type FilesInfoResponse struct {
	ID            string `json:"id"`
	Object        string `json:"object"`
	Bytes         int    `json:"bytes"`
	CreatedAt     int    `json:"created_at"`
	Filename      string `json:"filename"`
	Purpose       string `json:"purpose"`
	Status        string `json:"status"`
	StatusDetails string `json:"status_details"`
}

type FilesListRequest struct {
}

type FilesListResponse struct {
	Object string                   `json:"object"`
	Data   []*FilesListResponseData `json:"data"`
}

type FilesListResponseData struct {
	ID           string       `json:"id"`
	Object       string       `json:"object"`
	Bytes        int          `json:"bytes"`
	CreatedAt    int          `json:"created_at"`
	Filename     string       `json:"filename"`
	Purpose      FilesPurpose `json:"purpose"`
	Status       string       `json:"status"`
	StatusDetail string       `json:"status_detail"`
}

type FilesPurpose string

type FilesUploadBytesRequest struct {
	Name    string
	Bytes   []byte
	Purpose FilesPurpose
}

type FilesUploadBytesResponse struct {
	ID            string `json:"id"`
	Object        string `json:"object"`
	Bytes         int    `json:"bytes"`
	CreatedAt     int    `json:"created_at"`
	Filename      string `json:"filename"`
	Purpose       string `json:"purpose"`
	Status        string `json:"status"`
	StatusDetails string `json:"status_details"`
}

type FilesUploadRequest struct {
	Name    string
	Path    string
	Purpose FilesPurpose
}

type FilesUploadResponse struct {
	ID            string `json:"id"`
	Object        string `json:"object"`
	Bytes         int    `json:"bytes"`
	CreatedAt     int    `json:"created_at"`
	Filename      string `json:"filename"`
	Purpose       string `json:"purpose"`
	Status        string `json:"status"`
	StatusDetails string `json:"status_details"`
}

type IChat interface {
	Completions(ctx context.Context, req *ChatCompletionsRequest) (*ChatCompletionsResponse, error)
	CompletionsStream(ctx context.Context, req *ChatCompletionsRequest) (*ChatCompletionsStreamResponse, error)
}

type IChatCompletionsBuilder interface {
	AddUserContent(content string) IChatCompletionsBuilder
	AddSystemContent(content string) IChatCompletionsBuilder
	AddAssistantContent(content string, partialMode ...bool) IChatCompletionsBuilder
	AddToolContent(content, name, toolCallID string) IChatCompletionsBuilder
	AddPrompt(prompt string) IChatCompletionsBuilder
	AddMessage(message *ChatCompletionsMessage) IChatCompletionsBuilder
	AddMessageFromChoices(choices []*ChatCompletionsResponseChoices) IChatCompletionsBuilder

	SetModel(model ChatCompletionsModelID) IChatCompletionsBuilder
	SetTemperature(temperature float64) IChatCompletionsBuilder
	SetStream(enable bool) IChatCompletionsBuilder
	SetMaxTokens(num int) IChatCompletionsBuilder
	SetTopP(num float64) IChatCompletionsBuilder
	SetN(num int) IChatCompletionsBuilder
	SetPresencePenalty(num float64) IChatCompletionsBuilder
	SetFrequencyPenalty(num float64) IChatCompletionsBuilder
	SetStop(stop []string) IChatCompletionsBuilder
	SetTool(tool *ChatCompletionsTool) IChatCompletionsBuilder
	SetTools(tools []*ChatCompletionsTool) IChatCompletionsBuilder
	SetContextCacheContent(content *ContextCacheContent) IChatCompletionsBuilder
	SetResponseFormat(format ChatCompletionsResponseFormatType) IChatCompletionsBuilder

	ToRequest() *ChatCompletionsRequest
}

type IContextCache interface {
	Create(ctx context.Context, req *ContextCacheCreateRequest) (*ContextCacheCreateResponse, error)
	List(ctx context.Context, req *ContextCacheListRequest) (*ContextCacheListResponse, error)
	Delete(ctx context.Context, req *ContextCacheDeleteRequest) (*ContextCacheDeleteResponse, error)
	Update(ctx context.Context, req *ContextCacheUpdateRequest) (*ContextCacheUpdateResponse, error)
	Get(ctx context.Context, req *ContextCacheGetRequest) (*ContextCacheGetResponse, error)

	CreateTag(ctx context.Context, req *ContextCacheCreateTagRequest) (*ContextCacheCreateTagResponse, error)
	ListTag(ctx context.Context, req *ContextCacheListTagRequest) (*ContextCacheListTagResponse, error)
	DeleteTag(ctx context.Context, req *ContextCacheDeleteTagRequest) (*ContextCacheDeleteTagResponse, error)
	GetTag(ctx context.Context, req *ContextCacheGetTagRequest) (*ContextCacheGetTagResponse, error)
	GetTagContent(ctx context.Context, req *ContextCacheGetTagContentRequest) (*ContextCacheGetTagContentResponse, error)
}

type IFiles interface {
	Upload(ctx context.Context, req *FilesUploadRequest) (resp *FilesUploadResponse, err error)
	UploadBytes(ctx context.Context, req *FilesUploadBytesRequest) (resp *FilesUploadBytesResponse, err error)
	List(ctx context.Context) (res *FilesListResponse, err error)
	Delete(ctx context.Context, fileID string) (resp *FilesDeleteResponse, err error)
	BatchDelete(ctx context.Context, req *FilesBatchDeleteRequest) (resp *FilesBatchDeleteResponse, err error)
	Info(ctx context.Context, fileID string) (resp *FilesInfoResponse, err error)
	Content(ctx context.Context, fileID string) (resp *FileContentResponse, err error)
}

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

type ITokenizers interface {
	EstimateTokenCount(ctx context.Context, req *TokenizersEstimateTokenCountRequest) (resp *TokenizersEstimateTokenCountResponse, err error)
}

type IUsers interface {
	Balance(ctx context.Context) (*UsersBalanceResponse, error)
}

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 ListModelsRequest struct {
}

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

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"`
}

type Option func(*Config)

type TokenizersEstimateTokenCountRequest struct {
	Model    ChatCompletionsModelID    `json:"model"`
	Messages []*ChatCompletionsMessage `json:"messages"`
}

type TokenizersEstimateTokenCountResponse struct {
	CommonResponse
	Data *TokenizersEstimateTokenCountResponseData `json:"data"`
}

type TokenizersEstimateTokenCountResponseData struct {
	TotalTokens int `json:"total_tokens"`
}

type UsersBalanceResponse struct {
	Code   int                       `json:"code"`
	Data   *UsersBalanceResponseData `json:"data"`
	Scode  string                    `json:"scode"`
	Status bool                      `json:"status"`
}

type UsersBalanceResponseData struct {
	// AvailableBalance including cash balance and voucher balance. When it is less than or equal to 0, the user cannot call the completions API
	AvailableBalance float64 `json:"available_balance"`
	// VoucherBalance will not be negative
	VoucherBalance float64 `json:"voucher_balance"`
	// CashBalance may be negative, which means that the user owes the cost. When it is negative, the AvailableBalance can be the amount of VoucherBalance
	CashBalance float64 `json:"cash_balance"`
}