ai

Documentation

Index

• type AIConfiguration
• type AIProvider
  • func (p *AIProvider) GetBaseURL() string
  • func (p *AIProvider) GetModel() string
  • func (p *AIProvider) GetPassword() string
• type IAI
  • func NewClient(provider string) IAI
• type IAIConfig
• type NoOpAIClient
  • func (c *NoOpAIClient) Configure(config IAIConfig, language string) error
  • func (c *NoOpAIClient) GetCompletion(ctx context.Context, prompt string) (string, error)
  • func (a *NoOpAIClient) GetName() string
  • func (a *NoOpAIClient) Parse(ctx context.Context, prompt []string, cache cache.ICache) (string, error)
• type OpenAIClient
  • func (c *OpenAIClient) Configure(config IAIConfig, language string) error
  • func (c *OpenAIClient) GetCompletion(ctx context.Context, prompt string) (string, error)
  • func (a *OpenAIClient) GetName() string
  • func (a *OpenAIClient) Parse(ctx context.Context, prompt []string, cache cache.ICache) (string, error)

Types

type AIConfiguration

type AIConfiguration struct {
	Providers []AIProvider `mapstructure:"providers"`
}

type AIProvider

type AIProvider struct {
	Name     string `mapstructure:"name"`
	Model    string `mapstructure:"model"`
	Password string `mapstructure:"password"`
	BaseURL  string `mapstructure:"base_url"`
}

func (*AIProvider) GetBaseURL

func (p *AIProvider) GetBaseURL() string

func (*AIProvider) GetModel

func (p *AIProvider) GetModel() string

func (*AIProvider) GetPassword

func (p *AIProvider) GetPassword() string

type IAI

type IAI interface {
	Configure(config IAIConfig, language string) error
	GetCompletion(ctx context.Context, prompt string) (string, error)
	Parse(ctx context.Context, prompt []string, cache cache.ICache) (string, error)
	GetName() string
}

func NewClient

func NewClient(provider string) IAI

type IAIConfig

type IAIConfig interface {
	GetPassword() string
	GetModel() string
	GetBaseURL() string
}

type NoOpAIClient

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

func (*NoOpAIClient) Configure

func (c *NoOpAIClient) Configure(config IAIConfig, language string) error

func (*NoOpAIClient) GetCompletion

func (c *NoOpAIClient) GetCompletion(ctx context.Context, prompt string) (string, error)

func (*NoOpAIClient) GetName

func (a *NoOpAIClient) GetName() string

func (*NoOpAIClient) Parse

func (a *NoOpAIClient) Parse(ctx context.Context, prompt []string, cache cache.ICache) (string, error)

type OpenAIClient

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

func (*OpenAIClient) Configure

func (c *OpenAIClient) Configure(config IAIConfig, language string) error

func (*OpenAIClient) GetCompletion

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

func (*OpenAIClient) GetName

func (a *OpenAIClient) GetName() string

func (*OpenAIClient) Parse

func (a *OpenAIClient) Parse(ctx context.Context, prompt []string, cache cache.ICache) (string, error)