openaiprovider

Documentation

Index

• Constants
• func NewOAIProviderConfig(apiKey, model string) oaiProviderConfig
• func NewTextToJSONProvider(config oaiProviderConfig) (baseprovider.TextToJSONProvider, error)
• func NewTextToSpeechProvider(config oaiProviderConfig) (baseprovider.TextToSpeechProvider, error)
• func NewTextToTextProvider(config oaiProviderConfig) (baseprovider.TextToTextProvider, error)
• func NewTextToTextReasoningProvider(config oaiProviderConfig) (baseprovider.TextToTextReasoningProvider, error)
• type TextToEmbeddingProvider
  • func NewTextToEmbeddingProvider(config oaiProviderConfig) (TextToEmbeddingProvider, error)
  • func (p TextToEmbeddingProvider) Close() error
  • func (p TextToEmbeddingProvider) GenerateEmbedding(ctx context.Context, message string) ([]float32, error)
  • func (p TextToEmbeddingProvider) GetModel() string
• type TextToJSONProvider
  • func (p TextToJSONProvider) Close() error
  • func (p TextToJSONProvider) GenerateCompletion(ctx context.Context, messages []chat.Message, ...) error
  • func (p TextToJSONProvider) GetModel() string
• type TextToSpeechProvider
  • func (p TextToSpeechProvider) Close() error
  • func (p TextToSpeechProvider) GenerateSpeech(ctx context.Context, message string) ([]byte, error)
  • func (p TextToSpeechProvider) GetModel() string
• type TextToTextProvider
  • func (p TextToTextProvider) Close() error
  • func (p TextToTextProvider) GenerateCompletion(ctx context.Context, messages []chat.Message, ...) error
  • func (p TextToTextProvider) GetModel() string
• type TextToTextReasoningProvider
  • func (p *TextToTextReasoningProvider) Close() error
  • func (p *TextToTextReasoningProvider) GenerateCompletion(ctx context.Context, messages []chat.Message, ...) error
  • func (p *TextToTextReasoningProvider) GetModel() string

Constants

const (
	OpenAITextToEmbeddingDefaultModel = openai.LargeEmbedding3
	OpenAITextToEmbeddingFastModel    = openai.SmallEmbedding3
)

const (
	OpenAITextToJSONDefaultModel     = openai.GPT4o
	OpenAITextToJSONDefaultModelFast = openai.GPT4o
)

const (
	OpenAITextToTextReasoningDefaultModel     = openai.O1Preview
	OpenAITextToTextReasoningDefaultModelFast = openai.O1Mini
)

const (
	OpenAITextToTextDefaultModel     = openai.GPT4o
	OpenAITextToTextDefaultModelFast = openai.GPT4o
)

const (
	OpenAITextToSpeechDefaultModel = openai.TTSModel1
)

Functions

func NewOAIProviderConfig

func NewOAIProviderConfig(apiKey, model string) oaiProviderConfig

func NewTextToJSONProvider

func NewTextToJSONProvider(
	config oaiProviderConfig,
) (baseprovider.TextToJSONProvider, error)

func NewTextToSpeechProvider

func NewTextToSpeechProvider(
	config oaiProviderConfig,
) (baseprovider.TextToSpeechProvider, error)

func NewTextToTextProvider

func NewTextToTextProvider(
	config oaiProviderConfig,
) (baseprovider.TextToTextProvider, error)

func NewTextToTextReasoningProvider

func NewTextToTextReasoningProvider(
	config oaiProviderConfig,
) (baseprovider.TextToTextReasoningProvider, error)

Types

type TextToEmbeddingProvider

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

func NewTextToEmbeddingProvider

func NewTextToEmbeddingProvider(
	config oaiProviderConfig,
) (TextToEmbeddingProvider, error)

func (TextToEmbeddingProvider) Close

func (p TextToEmbeddingProvider) Close() error

func (TextToEmbeddingProvider) GenerateEmbedding

func (p TextToEmbeddingProvider) GenerateEmbedding(
	ctx context.Context,
	message string,
) ([]float32, error)

func (TextToEmbeddingProvider) GetModel

func (p TextToEmbeddingProvider) GetModel() string

For now, we are always using the default model

type TextToJSONProvider

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

func (TextToJSONProvider) Close

func (p TextToJSONProvider) Close() error

func (TextToJSONProvider) GenerateCompletion

func (p TextToJSONProvider) GenerateCompletion(
	ctx context.Context,
	messages []chat.Message,
	completionCh chan<- completion.Completion,
) error

func (TextToJSONProvider) GetModel

func (p TextToJSONProvider) GetModel() string

type TextToSpeechProvider

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

func (TextToSpeechProvider) Close

func (p TextToSpeechProvider) Close() error

func (TextToSpeechProvider) GenerateSpeech

func (p TextToSpeechProvider) GenerateSpeech(
	ctx context.Context,
	message string,
) ([]byte, error)

func (TextToSpeechProvider) GetModel

func (p TextToSpeechProvider) GetModel() string

For now, we are always using the default model

type TextToTextProvider

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

func (TextToTextProvider) Close

func (p TextToTextProvider) Close() error

func (TextToTextProvider) GenerateCompletion

func (p TextToTextProvider) GenerateCompletion(
	ctx context.Context,
	messages []chat.Message,
	completionCh chan<- completion.Completion,
) error

func (TextToTextProvider) GetModel

func (p TextToTextProvider) GetModel() string

type TextToTextReasoningProvider

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

func (*TextToTextReasoningProvider) Close

func (p *TextToTextReasoningProvider) Close() error

func (*TextToTextReasoningProvider) GenerateCompletion

func (p *TextToTextReasoningProvider) GenerateCompletion(
	ctx context.Context,
	messages []chat.Message,
	completionCh chan<- completion.Completion,
) error

func (*TextToTextReasoningProvider) GetModel

func (p *TextToTextReasoningProvider) GetModel() string