config

Documentation

Index

• Variables
• type ClassSettings
  • func NewClassSettings(cfg moduletools.ClassConfig) ClassSettings

Variables

var (
	DefaultGoogleApiEndpoint          = "us-central1-aiplatform.googleapis.com"
	DefaultGoogleModel                = "chat-bison"
	DefaultGoogleRegion               = "us-central1"
	DefaultGoogleTemperature          = 1.0
	DefaultTokenLimit                 = 1024
	DefaultTokenLimitGemini1_0        = 2048
	DefaultTokenLimitGemini1_0_Vision = 4096
	DefaultTokenLimitGemini1_5        = 8192
	DefaultGoogleTopP                 = 0.95
	DefaultGoogleTopK                 = 40
	DefaulGenerativeAIApiEndpoint     = "generativelanguage.googleapis.com"
	DefaulGenerativeAIModelID         = "chat-bison-001"
)

Types

type ClassSettings

type ClassSettings interface {
	Validate(class *models.Class) error
	// Module settings
	ApiEndpoint() string
	ProjectID() string
	EndpointID() string
	Region() string

	ModelID() string
	Model() string
	// parameters
	// 0.0 - 1.0
	Temperature() float64
	// 1 - 1024 / 2048 Gemini 1.0 / 8192 Gemini 1.5
	TokenLimit() int
	// 1 -
	TopK() int
	// 0.0 - 1.0
	TopP() float64
}

func NewClassSettings

func NewClassSettings(cfg moduletools.ClassConfig) ClassSettings