ai

Nov 27, 2024

Changes in this version

+ const AzureProvider

+ const HuggingFaceProvider

+ const OpenAIProvider

+ const SQLFixType

+ const Text2sqlType

+ var ServicePromptMap = map[string]string

+ type AIConfig struct

+ AuthToken string

+ BaseURL string

+ Model string

+ Name string

+ Temperature float32

+ TopP float32

+ func ConvertToAIConfig(c registry.ExtraConfig) AIConfig

+ type AIProvider interface

+ Configure func(config AIConfig) error

+ Generate func(ctx context.Context, prompt string) (string, error)

+ func NewClient(name string) AIProvider

+ type AzureAIClient struct

+ func (c *AzureAIClient) Configure(cfg AIConfig) error

+ func (c *AzureAIClient) Generate(ctx context.Context, prompt string) (string, error)

+ type HuggingfaceClient struct

+ func (c *HuggingfaceClient) Configure(cfg AIConfig) error

+ func (c *HuggingfaceClient) Generate(ctx context.Context, prompt string) (string, error)

+ type OpenAIClient struct

+ func (c *OpenAIClient) Configure(cfg AIConfig) error

+ func (c *OpenAIClient) Generate(ctx context.Context, prompt string) (string, error)