baseten

package

v0.1.2 Latest Latest Go to latest Published: May 6, 2026 License: Apache-2.0 Imports: 13 Imported by: 0

Details

Package baseten provides a Baseten provider implementation.

This section is empty.

This section is empty.

This section is empty.

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

BasetenLanguageModel implements the LanguageModel interface.

func (m *BasetenLanguageModel) ID() string

func (m *BasetenLanguageModel) Provider() string

func (m *BasetenLanguageModel) Stream(ctx context.Context, options *stream.CallOptions) (<-chan stream.Event, error)

type Options struct {
	APIKey  string
	Headers map[string]string
}

Options contains configuration for the Baseten provider.

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

Provider implements the Baseten provider.

func New(opts Options) *Provider

New creates a new Baseten 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