util

Documentation

Index

• func GetPrompt(userPrompt string) (string, error)
• func GetSystemPrompt() (string, error)
• func IsSupportedProvider(provider string) bool
• func SetDefaultProvider(provider string) error
• type Config
  • func GetConfig() (*Config, error)
  • func ParseConfig(filePath string) (*Config, error)
• type LLMConfig
• type SysInfo
  • func GetSysInfo() (*SysInfo, error)

Functions

func GetPrompt

func GetPrompt(userPrompt string) (string, error)

func GetSystemPrompt

func GetSystemPrompt() (string, error)

func IsSupportedProvider

func IsSupportedProvider(provider string) bool

func SetDefaultProvider

func SetDefaultProvider(provider string) error

Types

type Config

type Config struct {
	DefaultProvider string               `toml:"default-provider"`
	LLMProvider     map[string]LLMConfig `toml:"llm-provider"`
}

Config represents the top-level configuration structure

func GetConfig

func GetConfig() (*Config, error)

func ParseConfig

func ParseConfig(filePath string) (*Config, error)

ParseConfig reads and parses the TOML configuration file

type LLMConfig

type LLMConfig struct {
	BaseURL string `toml:"base_url"`
	Model   string `toml:"model"`
	ApiKey  string `toml:"api_key"`
}

LLMConfig represents the configuration for an LLM provider

type SysInfo

type SysInfo struct {
	OS     string
	OSName string
	Arch   string
}

func GetSysInfo

func GetSysInfo() (*SysInfo, error)