124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
package lmcli
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"git.mlow.ca/mlow/lmcli/pkg/lmcli/model"
|
|
"git.mlow.ca/mlow/lmcli/pkg/lmcli/provider"
|
|
"git.mlow.ca/mlow/lmcli/pkg/lmcli/provider/anthropic"
|
|
"git.mlow.ca/mlow/lmcli/pkg/lmcli/provider/openai"
|
|
"git.mlow.ca/mlow/lmcli/pkg/lmcli/tools"
|
|
"git.mlow.ca/mlow/lmcli/pkg/util/tty"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Context struct {
|
|
Config *Config
|
|
Store ConversationStore
|
|
|
|
Chroma *tty.ChromaHighlighter
|
|
EnabledTools []model.Tool
|
|
}
|
|
|
|
func NewContext() (*Context, error) {
|
|
configFile := filepath.Join(configDir(), "config.yaml")
|
|
config, err := NewConfig(configFile)
|
|
if err != nil {
|
|
Fatal("%v\n", err)
|
|
}
|
|
|
|
databaseFile := filepath.Join(dataDir(), "conversations.db")
|
|
db, err := gorm.Open(sqlite.Open(databaseFile), &gorm.Config{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Error establishing connection to store: %v", err)
|
|
}
|
|
store, err := NewSQLStore(db)
|
|
if err != nil {
|
|
Fatal("%v\n", err)
|
|
}
|
|
|
|
chroma := tty.NewChromaHighlighter("markdown", *config.Chroma.Formatter, *config.Chroma.Style)
|
|
|
|
var enabledTools []model.Tool
|
|
for _, toolName := range *config.Tools.EnabledTools {
|
|
tool, ok := tools.AvailableTools[toolName]
|
|
if ok {
|
|
enabledTools = append(enabledTools, tool)
|
|
}
|
|
}
|
|
|
|
return &Context{config, store, chroma, enabledTools}, nil
|
|
}
|
|
|
|
func (c *Context) GetModels() (models []string) {
|
|
for _, m := range *c.Config.Anthropic.Models {
|
|
models = append(models, m)
|
|
}
|
|
for _, m := range *c.Config.OpenAI.Models {
|
|
models = append(models, m)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (c *Context) GetCompletionProvider(model string) (provider.ChatCompletionClient, error) {
|
|
for _, m := range *c.Config.Anthropic.Models {
|
|
if m == model {
|
|
anthropic := &anthropic.AnthropicClient{
|
|
APIKey: *c.Config.Anthropic.APIKey,
|
|
}
|
|
return anthropic, nil
|
|
}
|
|
}
|
|
for _, m := range *c.Config.OpenAI.Models {
|
|
if m == model {
|
|
openai := &openai.OpenAIClient{
|
|
APIKey: *c.Config.OpenAI.APIKey,
|
|
}
|
|
return openai, nil
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("unknown model: %s", model)
|
|
}
|
|
|
|
func configDir() string {
|
|
var configDir string
|
|
|
|
xdgConfigHome := os.Getenv("XDG_CONFIG_HOME")
|
|
if xdgConfigHome != "" {
|
|
configDir = filepath.Join(xdgConfigHome, "lmcli")
|
|
} else {
|
|
userHomeDir, _ := os.UserHomeDir()
|
|
configDir = filepath.Join(userHomeDir, ".config/lmcli")
|
|
}
|
|
|
|
os.MkdirAll(configDir, 0755)
|
|
return configDir
|
|
}
|
|
|
|
func dataDir() string {
|
|
var dataDir string
|
|
|
|
xdgDataHome := os.Getenv("XDG_DATA_HOME")
|
|
if xdgDataHome != "" {
|
|
dataDir = filepath.Join(xdgDataHome, "lmcli")
|
|
} else {
|
|
userHomeDir, _ := os.UserHomeDir()
|
|
dataDir = filepath.Join(userHomeDir, ".local/share/lmcli")
|
|
}
|
|
|
|
os.MkdirAll(dataDir, 0755)
|
|
return dataDir
|
|
}
|
|
|
|
func Fatal(format string, args ...any) {
|
|
fmt.Fprintf(os.Stderr, format, args...)
|
|
os.Exit(1)
|
|
}
|
|
|
|
func Warn(format string, args ...any) {
|
|
fmt.Fprintf(os.Stderr, format, args...)
|
|
}
|