llm

Apr 21, 2025

Changes in this version

+ const GeminiModel1Flash

+ const GeminiModel1Pro

+ const GeminiModel2Flash

+ const GeminiModel2Pro

+ const OpenAIModelGPT35Turbo

+ const OpenAIModelGPT4

+ const OpenAIModelGPT4O

+ const OpenAIModelGPT4OMini

+ const OpenAIModelGPT4Turbo

+ const OpenAIModelGPT4Vision

+ const VertexModelGemini15Flash

+ const VertexModelGemini15Pro

+ const VertexModelGemini20Flash

+ const VertexModelGemini20FlashImageGen

+ const VertexModelGemini20FlashLite

+ const VertexModelGemini25ProPreview

+ var ErrInvalidRequest = errors.New("invalid request")

+ var ErrServiceUnavailable = errors.New("service unavailable")

+ func CountTokens(text string) int

+ func EstimateMaxTokens(promptTokens, contextWindowSize int) int

+ func Version() string

+ type CompletionRequest struct

+ MaxTokens int

+ SystemPrompt string

+ Temperature float64

+ UserPrompt string

+ type CompletionResponse struct

+ Text string

+ TokensUsed int

+ type MockModel struct

+ Error error

+ Response CompletionResponse

+ func NewMockModel() *MockModel

+ func NewMockModelWithOptions(options ModelOptions) *MockModel

+ func (m *MockModel) Complete(ctx context.Context, request CompletionRequest) (CompletionResponse, error)

+ func (m *MockModel) GetApiKey() string

+ func (m *MockModel) GetMaxTokens() int

+ func (m *MockModel) GetModel() string

+ func (m *MockModel) GetOutputFormat() OutputFormat

+ func (m *MockModel) GetProjectID() string

+ func (m *MockModel) GetProvider() Provider

+ func (m *MockModel) GetRegion() string

+ func (m *MockModel) GetTemperature() float64

+ func (m *MockModel) GetVerbose() bool

+ type ModelInterface interface

+ Complete func(ctx context.Context, request CompletionRequest) (CompletionResponse, error)

+ GetApiKey func() string

+ GetMaxTokens func() int

+ GetModel func() string

+ GetOutputFormat func() OutputFormat

+ GetProjectID func() string

+ GetProvider func() Provider

+ GetRegion func() string

+ GetTemperature func() float64

+ GetVerbose func() bool

+ func ImageModel(provider Provider) (ModelInterface, error)

+ func JSONModel(provider Provider) (ModelInterface, error)

+ func NewModel(options ModelOptions) (ModelInterface, error)

+ func TextModel(provider Provider) (ModelInterface, error)

+ type ModelOptions struct

+ ApiKey string

+ MaxTokens int

+ Model string

+ OutputFormat OutputFormat

+ ProjectID string

+ Provider Provider

+ Region string

+ Temperature float64

+ Verbose bool

+ type OutputFormat string

+ const OutputFormatEnum

+ const OutputFormatImageJPG

+ const OutputFormatImagePNG

+ const OutputFormatJSON

+ const OutputFormatText

+ const OutputFormatXML

+ const OutputFormatYAML

+ type Provider string

+ const ProviderAnthropic

+ const ProviderCustom

+ const ProviderGemini

+ const ProviderMock

+ const ProviderOpenAI

+ const ProviderVertex