Documentation
¶
Overview ¶
Package vertex provides a Google Vertex AI provider implementation.
Index ¶
- type Options
- type Provider
- func (p *Provider) EmbeddingModel(modelID string) model.EmbeddingModel
- func (p *Provider) ID() string
- func (p *Provider) ImageModel(modelID string) model.ImageModel
- func (p *Provider) LanguageModel(modelID string) model.LanguageModel
- func (p *Provider) Model(modelID string) stream.Model
- func (p *Provider) Models() []string
- func (p *Provider) RerankingModel(modelID string) model.RerankingModel
- func (p *Provider) SpeechModel(modelID string) model.SpeechModel
- func (p *Provider) TranscriptionModel(modelID string) model.TranscriptionModel
- type VertexEmbeddingModel
- func (m *VertexEmbeddingModel) Dimensions() int
- func (m *VertexEmbeddingModel) Embed(ctx context.Context, opts model.EmbedCallOptions) (*model.EmbedResult, error)
- func (m *VertexEmbeddingModel) ID() string
- func (m *VertexEmbeddingModel) MaxEmbeddingsPerCall() int
- func (m *VertexEmbeddingModel) Provider() string
- type VertexImageModel
- type VertexLanguageModel
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Options ¶
type Options struct {
ProjectID string
Location string // e.g., "us-central1"
AccessToken string // OAuth2 access token
Headers map[string]string
// BaseURL overrides the computed Vertex endpoint. When empty, the
// provider uses https://<location>-aiplatform.googleapis.com/v1/projects/<project>/locations/<location>.
// Primarily useful for tests pointing at httptest.Server.
BaseURL string
}
Options contains configuration for the Vertex AI provider.
type Provider ¶
type Provider struct {
// contains filtered or unexported fields
}
Provider implements the Vertex AI provider.
func (*Provider) EmbeddingModel ¶
func (p *Provider) EmbeddingModel(modelID string) model.EmbeddingModel
func (*Provider) ImageModel ¶
func (p *Provider) ImageModel(modelID string) model.ImageModel
func (*Provider) LanguageModel ¶
func (p *Provider) LanguageModel(modelID string) model.LanguageModel
func (*Provider) RerankingModel ¶
func (p *Provider) RerankingModel(modelID string) model.RerankingModel
func (*Provider) SpeechModel ¶
func (p *Provider) SpeechModel(modelID string) model.SpeechModel
func (*Provider) TranscriptionModel ¶
func (p *Provider) TranscriptionModel(modelID string) model.TranscriptionModel
type VertexEmbeddingModel ¶
type VertexEmbeddingModel struct {
// contains filtered or unexported fields
}
VertexEmbeddingModel implements the EmbeddingModel interface.
func (*VertexEmbeddingModel) Dimensions ¶
func (m *VertexEmbeddingModel) Dimensions() int
func (*VertexEmbeddingModel) Embed ¶
func (m *VertexEmbeddingModel) Embed(ctx context.Context, opts model.EmbedCallOptions) (*model.EmbedResult, error)
func (*VertexEmbeddingModel) ID ¶
func (m *VertexEmbeddingModel) ID() string
func (*VertexEmbeddingModel) MaxEmbeddingsPerCall ¶
func (m *VertexEmbeddingModel) MaxEmbeddingsPerCall() int
func (*VertexEmbeddingModel) Provider ¶
func (m *VertexEmbeddingModel) Provider() string
type VertexImageModel ¶
type VertexImageModel struct {
// contains filtered or unexported fields
}
VertexImageModel implements the ImageModel interface.
func (*VertexImageModel) Generate ¶
func (m *VertexImageModel) Generate(ctx context.Context, opts model.ImageCallOptions) (*model.ImageResult, error)
func (*VertexImageModel) ID ¶
func (m *VertexImageModel) ID() string
func (*VertexImageModel) MaxImagesPerCall ¶
func (m *VertexImageModel) MaxImagesPerCall() int
func (*VertexImageModel) Provider ¶
func (m *VertexImageModel) Provider() string
type VertexLanguageModel ¶
type VertexLanguageModel struct {
// contains filtered or unexported fields
}
VertexLanguageModel implements the LanguageModel interface.
func (*VertexLanguageModel) ID ¶
func (m *VertexLanguageModel) ID() string
func (*VertexLanguageModel) Provider ¶
func (m *VertexLanguageModel) Provider() string
func (*VertexLanguageModel) Stream ¶
func (m *VertexLanguageModel) Stream(ctx context.Context, options *stream.CallOptions) (<-chan stream.Event, error)
Click to show internal directories.
Click to hide internal directories.