google

package

v1.2.5 Latest Latest Go to latest Published: Jan 7, 2026 License: MIT Imports: 6 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Config struct {
	types.BaseConfig
	ProjectID string `json:"project_id,omitempty"`
	Location  string `json:"location,omitempty"`
}

Config holds Google AI-specific configuration

func (c *Config) Validate() error

Validate validates Google-specific configuration

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

Provider implements the Google AI provider

func NewProvider() *Provider

NewProvider creates a new Google AI provider

func (p *Provider) Close() error

Close cleans up resources

func (p *Provider) Complete(ctx context.Context, req *types.CompletionRequest) (*types.CompletionResponse, error)

Complete performs a completion request

func (p *Provider) EstimateTokens(ctx context.Context, messages []*types.Message, model string) (int, error)

EstimateTokens estimates token count for messages

func (p *Provider) GetModels(ctx context.Context) ([]*types.Model, error)

GetModels returns available Google AI models

func (p *Provider) GetName() string

GetName returns the provider name

func (p *Provider) Initialize(config types.Config) error

Initialize sets up the provider with configuration

func (p *Provider) Stream(ctx context.Context, req *types.CompletionRequest, callback types.StreamCallback) error

Stream performs a streaming completion request

func (p *Provider) ValidateModel(model string) error

ValidateModel checks if a model is supported