aljaz/opencode
1
0
Fork 0
mirror of https://github.com/aljazceru/opencode.git synced 2026-01-04 00:14:59 +01:00
Code Issues Packages Projects Releases Wiki Activity

rework llm

Browse Source
This commit is contained in:
Kujtim Hoxha
2025-03-27 22:35:48 +01:00
parent 904061c243
commit afd9ad0560
61 changed files with 5882 additions and 2074 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
cmd/root.go
View File

@@ -7,12 +7,12 @@ import (
tea "github.com/charmbracelet/bubbletea"
"github.com/kujtimiihoxha/termai/internal/app"
"github.com/kujtimiihoxha/termai/internal/config"
"github.com/kujtimiihoxha/termai/internal/db"
"github.com/kujtimiihoxha/termai/internal/llm/models"
"github.com/kujtimiihoxha/termai/internal/llm/agent"
"github.com/kujtimiihoxha/termai/internal/tui"
zone "github.com/lrstanley/bubblezone"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
var rootCmd = &cobra.Command{
@@ -25,11 +25,10 @@ var rootCmd = &cobra.Command{
return nil
}
debug, _ := cmd.Flags().GetBool("debug")
viper.Set("debug", debug)
if debug {
viper.Set("log.level", "debug")
err := config.Load(debug)
if err != nil {
return err
}
conn, err := db.Connect()
if err != nil {
return err
@@ -49,6 +48,8 @@ var rootCmd = &cobra.Command{
defer unsub()
go func() {
// Set this up once
agent.GetMcpTools(ctx)
for msg := range ch {
tui.Send(msg)
}
@@ -95,16 +96,6 @@ func setupSubscriptions(app *app.App) (chan tea.Msg, func()) {
wg.Done()
}()
}
{
sub := app.LLM.Subscribe(ctx)
wg.Add(1)
go func() {
for ev := range sub {
ch <- ev
}
wg.Done()
}()
}
{
sub := app.Permissions.Subscribe(ctx)
wg.Add(1)
@@ -129,40 +120,7 @@ func Execute() {
}
}
func loadConfig() {
viper.SetConfigName(".termai")
viper.SetConfigType("yaml")
viper.AddConfigPath("$HOME")
viper.AddConfigPath("$XDG_CONFIG_HOME/termai")
viper.AddConfigPath(".")
viper.SetEnvPrefix("TERMAI")
// SET DEFAULTS
viper.SetDefault("log.level", "info")
viper.SetDefault("data.dir", ".termai")
// LLM
viper.SetDefault("models.big", string(models.DefaultBigModel))
viper.SetDefault("models.small", string(models.DefaultLittleModel))
viper.SetDefault("providers.openai.key", os.Getenv("OPENAI_API_KEY"))
viper.SetDefault("providers.anthropic.key", os.Getenv("ANTHROPIC_API_KEY"))
viper.SetDefault("providers.groq.key", os.Getenv("GROQ_API_KEY"))
viper.SetDefault("providers.common.max_tokens", 4000)
viper.SetDefault("agents.default", "coder")
viper.ReadInConfig()
workdir, err := os.Getwd()
if err != nil {
panic(err)
}
viper.Set("wd", workdir)
}
func init() {
loadConfig()
rootCmd.Flags().BoolP("help", "h", false, "Help")
rootCmd.Flags().BoolP("debug", "d", false, "Help")
}