ai

Documentation

Index

• type AIConfiguration
• type AIProvider
• type IAI
  • func NewClient(provider string) IAI
• type NoOpAIClient
  • func (c *NoOpAIClient) Configure(token string, model string, 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, nocache bool) (string, error)
• type OpenAIClient
  • func (c *OpenAIClient) Configure(token string, model string, 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, nocache bool) (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"`
}

type IAI

type IAI interface {
	Configure(token string, model string, language string) error
	GetCompletion(ctx context.Context, prompt string) (string, error)
	Parse(ctx context.Context, prompt []string, nocache bool) (string, error)
	GetName() string
}

func NewClient

func NewClient(provider string) IAI

type NoOpAIClient

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

func (*NoOpAIClient) Configure

func (c *NoOpAIClient) Configure(token string, model string, 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, nocache bool) (string, error)

type OpenAIClient

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

func (*OpenAIClient) Configure

func (c *OpenAIClient) Configure(token string, model string, 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, nocache bool) (string, error)