google

type Candidate struct {
	Content      Content `json:"content,omitempty"`
	FinishReason string  `json:"finishReason,omitempty"`
	TokenCount   uint    `json:"tokenCount,omitempty"`
	Index        uint    `json:"index,omitempty"`
}

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

type CompletionMessageResponse struct {
	Candidates []Candidate `json:"candidates,omitempty"`
}

type Config struct {
	BaseURL        string
	HTTPClient     *http.Client
	ThinkingBudget int
}

type Content struct {
	Parts []Part `json:"parts,omitempty"`
	Role  string `json:"role,omitempty"`
}

type ErrorAccumulator interface {
	Write(p []byte) error
	Bytes() []byte
}

type GenerationConfig struct {
	StopSequences    []string        `json:"stopSequences,omitempty"`
	ResponseMimeType string          `json:"responseMimeType,omitempty"`
	CandidateCount   uint            `json:"candidateCount,omitempty"`
	MaxOutputTokens  uint            `json:"maxOutputTokens,omitempty"`
	Temperature      float64         `json:"temperature,omitempty"`
	TopP             float64         `json:"topP,omitempty"`
	TopK             int64           `json:"topK,omitempty"`
	ThinkingConfig   *ThinkingConfig `json:"thinkingConfig,omitempty"`
}

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

type JSONMarshaller struct{}

type JSONUnmarshaler struct{}

type Marshaller interface {
	Marshal(value any) ([]byte, error)
}

type MessageCompletionRequest struct {
	Contents         []Content        `json:"contents,omitempty"`
	GenerationConfig GenerationConfig `json:"generationConfig,omitempty"`
}

type Part struct {
	Text string `json:"text,omitempty"`
}

type RequestBuilder interface {
	Build(ctx context.Context, method, url string, body any, header http.Header) (*http.Request, error)
}

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

type ThinkingConfig struct {
	ThinkingBudget int `json:"thinkingBudget,omitempty"`
}

type Unmarshaler interface {
	Unmarshal(data []byte, v any) error
}