vertex

package
v0.1.2 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: May 6, 2026 License: Apache-2.0 Imports: 16 Imported by: 0

Documentation

Overview

Package vertex provides a Google Vertex AI provider implementation.

Index

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 New

func New(opts Options) *Provider

New creates a new Vertex AI provider.

func (*Provider) EmbeddingModel

func (p *Provider) EmbeddingModel(modelID string) model.EmbeddingModel

func (*Provider) ID

func (p *Provider) ID() string

func (*Provider) ImageModel

func (p *Provider) ImageModel(modelID string) model.ImageModel

func (*Provider) LanguageModel

func (p *Provider) LanguageModel(modelID string) model.LanguageModel

func (*Provider) Model

func (p *Provider) Model(modelID string) stream.Model

func (*Provider) Models

func (p *Provider) Models() []string

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 (*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 (*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)

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL