google

type Candidate struct {
	Content      Content `json:"content"`
	FinishReason string  `json:"finishReason"`
}

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

type ClientRequest struct {
	Contents []*Content `json:"contents"`
}

type Config struct {
	APIKey string `json:"google_api_key"`
	Base   string `json:"google_api_base"`
}

type Content struct {
	Parts []map[string]interface{} `json:"parts"`
	Role  string                   `json:"role"`
}

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

type Error struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Status  string `json:"status"`
}

type FNewModelMode func(string) IModelMode

type IModelMode interface {
	Endpoint() string
	convert.IConverter
}

type ModelConfig struct {
	ResponseMimeType string  `json:"response_format"`
	MaxOutputTokens  int     `json:"max_tokens_to_sample"`
	Temperature      float64 `json:"temperature"`
	TopP             float64 `json:"top_p"`
	TopK             int     `json:"top_k"`
}

type ModelFactory struct {
}

type Response struct {
	Candidates    []Candidate   `json:"candidates"`
	UsageMetadata UsageMetadata `json:"usageMetadata"`
	Error         Error         `json:"error"`
}

type UsageMetadata struct {
	PromptTokenCount     int `json:"promptTokenCount"`
	CandidatesTokenCount int `json:"candidatesTokenCount"`
	TotalTokenCount      int `json:"totalTokenCount"`
}