siliconflow

type ImageGenerations struct {
	Prompt            string  `json:"prompt"`
	ImageSize         string  `json:"image_size"`
	Seed              int64   `json:"seed,omitempty"`
	NumInferenceSteps int64   `json:"num_inference_steps,omitempty"`
	NegativePrompt    string  `json:"negative_prompt,omitempty"`
	BatchSize         int     `json:"batch_size,omitempty"`
	GuidanceScale     float64 `json:"guidance_scale,omitempty"`
}

type ImageImings struct {
	Inference float64 `json:"inference"`
}

type ImageRes struct {
	Images  []*ImagesData `json:"images"`
	Timings *ImageImings  `json:"timings,omitempty"`
	Seed    int64         `json:"seed,omitempty"`
}

type ImagesData struct {
	Url string `json:"url"`
}

type Meta struct {
	Tokens *Tokens `json:"tokens,omitempty"`
}

type RerankRequest struct {
	Model           string   `json:"model"`
	Query           string   `json:"query"`
	TopN            int      `json:"top_n"`
	ReturnDocuments bool     `json:"return_documents"`
	Documents       []string `json:"documents"`
}

type RerankResponse struct {
	Id      string               `json:"id"`
	Results []types.RerankResult `json:"results"`
	Meta    *Meta                `json:"meta,omitempty"`
}

type Response struct {
	Code    int       `json:"code"`
	Message string    `json:"message"`
	Data    *UserInfo `json:"data,omitempty"`
}

type SiliError struct {
	Code    int    `json:"code,omitempty"`
	Message string `json:"message,omitempty"`
}

type SiliconflowProvider struct {
	openai.OpenAIProvider
}

type SiliconflowProviderFactory struct{}

type Tokens struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
}

type UserInfo struct {
	ID           string `json:"id"`
	Balance      string `json:"balance"`
	TotalBalance string `json:"totalBalance"`
}