dsc

type BalanceResponse struct {
	IsAvailable  bool                `json:"is_available"`
	BalanceInfos []map[string]string `json:"balance_infos"`
}

type ChatRequest struct {
	Model           string           `json:"model"`
	Messages        []prompt.Message `json:"messages"`
	Tools           []toolcall.Tool  `json:"tools,omitzero"`
	Stream          bool             `json:"stream"`
	MaxTokens       int              `json:"max_tokens,omitzero"`
	Thinking        Thinking         `json:"thinking"`
	ReasoningEffort string           `json:"reasoning_effort,omitzero"`
}

type ChatResponse struct {
	ID      string   `json:"id"`
	Choices []Choice `json:"choices"`
}

type Choice struct {
	Message      prompt.Message `json:"message"`
	FinishReason string         `json:"finish_reason"`
}

type Client interface {
	Models() (*ModelsResponse, error)
	Balance() (*BalanceResponse, error)
	FIM(ctx context.Context, req FIMRequest) (*FIMResponse, error)
	Chat(ctx context.Context, messages []prompt.Message, tools []toolcall.Tool) (*ChatResponse, error)
}

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

type FIMChoice struct {
	Text         string `json:"text"`
	Index        int    `json:"index"`
	FinishReason string `json:"finish_reason,omitzero"`
}

type FIMRequest struct {
	Model       string  `json:"model"`
	Prompt      string  `json:"prompt"`
	Suffix      string  `json:"suffix,omitzero"`
	MaxTokens   int     `json:"max_tokens,omitzero"`
	Temperature float64 `json:"temperature,omitzero"`
	TopP        float64 `json:"top_p,omitzero"`
	Stream      bool    `json:"stream,omitzero"`
	Echo        bool    `json:"echo,omitzero"`
	Stop        any     `json:"stop,omitzero"`
}

type FIMResponse struct {
	ID      string      `json:"id"`
	Choices []FIMChoice `json:"choices"`
	Usage   FIMUsage    `json:"usage,omitzero"`
}

type FIMStreamChunk struct {
	ID      string      `json:"id"`
	Object  string      `json:"object"`
	Created int64       `json:"created"`
	Model   string      `json:"model"`
	Choices []FIMChoice `json:"choices"`
	Usage   *FIMUsage   `json:"usage,omitzero"`
}

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

type Model struct {
	ID      string `json:"id"`
	Object  string `json:"object"`
	OwnedBy string `json:"owned_by"`
}

type ModelsResponse struct {
	Object string  `json:"object"`
	Data   []Model `json:"data"`
}

type Thinking struct {
	Type string `json:"type,omitzero"`
}