diff --git a/.sqlfluff b/.sqlfluff deleted file mode 100644 index 8917e1d9..00000000 --- a/.sqlfluff +++ /dev/null @@ -1,2 +0,0 @@ -[sqlfluff:rules] -exclude_rules = AM04 diff --git a/LICENSE b/LICENSE deleted file mode 100644 index e69de29b..00000000 diff --git a/cmd/root.go b/cmd/root.go index e9144486..3879a3cf 100644 --- a/cmd/root.go +++ b/cmd/root.go @@ -1,6 +1,3 @@ -/* -Copyright © 2025 NAME HERE -*/ package cmd import ( @@ -11,6 +8,7 @@ import ( tea "github.com/charmbracelet/bubbletea" "github.com/kujtimiihoxha/termai/internal/app" "github.com/kujtimiihoxha/termai/internal/db" + "github.com/kujtimiihoxha/termai/internal/llm/models" "github.com/kujtimiihoxha/termai/internal/tui" "github.com/spf13/cobra" "github.com/spf13/viper" @@ -64,7 +62,7 @@ func setupSubscriptions(app *app.App) (chan tea.Msg, func()) { wg := sync.WaitGroup{} ctx, cancel := context.WithCancel(app.Context) - if viper.GetBool("debug") { + { sub := app.Logger.Subscribe(ctx) wg.Add(1) go func() { @@ -84,6 +82,26 @@ func setupSubscriptions(app *app.App) (chan tea.Msg, func()) { wg.Done() }() } + { + sub := app.Messages.Subscribe(ctx) + wg.Add(1) + go func() { + for ev := range sub { + ch <- ev + } + wg.Done() + }() + } + { + sub := app.LLM.Subscribe(ctx) + wg.Add(1) + go func() { + for ev := range sub { + ch <- ev + } + wg.Done() + }() + } return ch, func() { cancel() wg.Wait() @@ -111,15 +129,26 @@ func loadConfig() { viper.SetDefault("log.level", "info") viper.SetDefault("data.dir", ".termai") + // LLM + viper.SetDefault("models.big", string(models.DefaultBigModel)) + viper.SetDefault("models.little", 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.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() - // Here you will define your flags and configuration settings. - // Cobra supports persistent flags, which, if defined here, - // will be global for your application. rootCmd.Flags().BoolP("help", "h", false, "Help") rootCmd.Flags().BoolP("debug", "d", false, "Help") diff --git a/go.mod b/go.mod index 2bc72387..c536fc72 100644 --- a/go.mod +++ b/go.mod @@ -9,9 +9,13 @@ require ( github.com/charmbracelet/glamour v0.9.1 github.com/charmbracelet/huh v0.6.0 github.com/charmbracelet/lipgloss v1.1.0 + github.com/cloudwego/eino v0.3.17 + github.com/cloudwego/eino-ext/components/model/claude v0.0.0-20250320062631-616205c32186 + github.com/cloudwego/eino-ext/components/model/openai v0.0.0-20250320062631-616205c32186 github.com/go-logfmt/logfmt v0.6.0 github.com/golang-migrate/migrate/v4 v4.18.2 github.com/google/uuid v1.6.0 + github.com/kujtimiihoxha/vimtea v0.0.3-0.20250317175717-9d8ba9c69840 github.com/mattn/go-runewidth v0.0.16 github.com/mattn/go-sqlite3 v1.14.24 github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 @@ -19,52 +23,97 @@ require ( github.com/muesli/termenv v0.16.0 github.com/spf13/cobra v1.9.1 github.com/spf13/viper v1.20.0 - golang.org/x/exp v0.0.0-20230315142452-642cacee5cc0 + golang.org/x/exp v0.0.0-20230713183714-613f0c0eb8a1 ) require ( github.com/alecthomas/chroma/v2 v2.15.0 // indirect + github.com/anthropics/anthropic-sdk-go v0.2.0-alpha.8 // indirect github.com/atotto/clipboard v0.1.4 // indirect + github.com/aws/aws-sdk-go-v2 v1.33.0 // indirect + github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.3 // indirect + github.com/aws/aws-sdk-go-v2/config v1.29.1 // indirect + github.com/aws/aws-sdk-go-v2/credentials v1.17.54 // indirect + github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.16.24 // indirect + github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.28 // indirect + github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.28 // indirect + github.com/aws/aws-sdk-go-v2/internal/ini v1.8.1 // indirect + github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.12.1 // indirect + github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.12.9 // indirect + github.com/aws/aws-sdk-go-v2/service/sso v1.24.11 // indirect + github.com/aws/aws-sdk-go-v2/service/ssooidc v1.28.10 // indirect + github.com/aws/aws-sdk-go-v2/service/sts v1.33.9 // indirect + github.com/aws/smithy-go v1.22.1 // indirect github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect github.com/aymerick/douceur v0.2.0 // indirect + github.com/bytedance/sonic v1.12.2 // indirect + github.com/bytedance/sonic/loader v0.2.0 // indirect github.com/charmbracelet/colorprofile v0.2.3-0.20250311203215-f60798e515dc // indirect github.com/charmbracelet/x/ansi v0.8.0 // indirect github.com/charmbracelet/x/cellbuf v0.0.13-0.20250311204145-2c3ea96c31dd // indirect github.com/charmbracelet/x/exp/strings v0.0.0-20240722160745-212f7b056ed0 // indirect github.com/charmbracelet/x/term v0.2.1 // indirect + github.com/cloudwego/base64x v0.1.4 // indirect + github.com/cloudwego/eino-ext/libs/acl/openai v0.0.0-20250305023926-469de0301955 // indirect + github.com/cloudwego/iasm v0.2.0 // indirect github.com/dlclark/regexp2 v1.11.4 // indirect github.com/dustin/go-humanize v1.0.1 // indirect github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect github.com/fsnotify/fsnotify v1.8.0 // indirect + github.com/getkin/kin-openapi v0.118.0 // indirect + github.com/go-openapi/jsonpointer v0.19.5 // indirect + github.com/go-openapi/swag v0.19.5 // indirect github.com/go-viper/mapstructure/v2 v2.2.1 // indirect + github.com/goph/emperror v0.17.2 // indirect github.com/gorilla/css v1.0.1 // indirect github.com/hashicorp/errwrap v1.1.0 // indirect github.com/hashicorp/go-multierror v1.1.1 // indirect github.com/inconshreveable/mousetrap v1.1.0 // indirect - github.com/kujtimiihoxha/vimtea v0.0.3-0.20250317175717-9d8ba9c69840 // indirect + github.com/invopop/yaml v0.1.0 // indirect + github.com/josharian/intern v1.0.0 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/cpuid/v2 v2.0.9 // indirect github.com/lucasb-eyer/go-colorful v1.2.0 // indirect + github.com/mailru/easyjson v0.7.7 // indirect github.com/mattn/go-isatty v0.0.20 // indirect github.com/mattn/go-localereader v0.0.1 // indirect github.com/microcosm-cc/bluemonday v1.0.27 // indirect github.com/mitchellh/hashstructure/v2 v2.0.2 // indirect + github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 // indirect github.com/muesli/cancelreader v0.2.2 // indirect + github.com/nikolalohinski/gonja v1.5.3 // indirect github.com/pelletier/go-toml/v2 v2.2.3 // indirect + github.com/perimeterx/marshmallow v1.1.4 // indirect + github.com/pkg/errors v0.9.1 // indirect github.com/rivo/uniseg v0.4.7 // indirect github.com/sagikazarmark/locafero v0.7.0 // indirect github.com/sahilm/fuzzy v0.1.1 // indirect + github.com/sashabaranov/go-openai v1.32.5 // indirect + github.com/sirupsen/logrus v1.9.3 // indirect + github.com/slongfield/pyfmt v0.0.0-20220222012616-ea85ff4c361f // indirect github.com/sourcegraph/conc v0.3.0 // indirect github.com/spf13/afero v1.12.0 // indirect github.com/spf13/cast v1.7.1 // indirect github.com/spf13/pflag v1.0.6 // indirect github.com/subosito/gotenv v1.6.0 // indirect + github.com/tidwall/gjson v1.18.0 // indirect + github.com/tidwall/match v1.1.1 // indirect + github.com/tidwall/pretty v1.2.1 // indirect + github.com/tidwall/sjson v1.2.5 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect + github.com/yargevad/filepathx v1.0.0 // indirect github.com/yuin/goldmark v1.7.8 // indirect github.com/yuin/goldmark-emoji v1.0.5 // indirect go.uber.org/atomic v1.9.0 // indirect go.uber.org/multierr v1.9.0 // indirect + golang.org/x/arch v0.11.0 // indirect golang.org/x/net v0.33.0 // indirect golang.org/x/sync v0.12.0 // indirect golang.org/x/sys v0.30.0 // indirect golang.org/x/text v0.23.0 // indirect + gopkg.in/yaml.v2 v2.4.0 // indirect gopkg.in/yaml.v3 v3.0.1 // indirect ) diff --git a/go.sum b/go.sum index b94a4c55..6e4f85db 100644 --- a/go.sum +++ b/go.sum @@ -1,21 +1,64 @@ github.com/MakeNowJust/heredoc v1.0.0 h1:cXCdzVdstXyiTqTvfqk9SDHpKNjxuom+DOlyEeQ4pzQ= github.com/MakeNowJust/heredoc v1.0.0/go.mod h1:mG5amYoWBHf8vpLOuehzbGGw0EHxpZZ6lCpQ4fNJ8LE= +github.com/airbrake/gobrake v3.6.1+incompatible/go.mod h1:wM4gu3Cn0W0K7GUuVWnlXZU11AGBXMILnrdOU8Kn00o= github.com/alecthomas/assert/v2 v2.11.0 h1:2Q9r3ki8+JYXvGsDyBXwH3LcJ+WK5D0gc5E8vS6K3D0= github.com/alecthomas/assert/v2 v2.11.0/go.mod h1:Bze95FyfUr7x34QZrjL+XP+0qgp/zg8yS+TtBj1WA3k= github.com/alecthomas/chroma/v2 v2.15.0 h1:LxXTQHFoYrstG2nnV9y2X5O94sOBzf0CIUpSTbpxvMc= github.com/alecthomas/chroma/v2 v2.15.0/go.mod h1:gUhVLrPDXPtp/f+L1jo9xepo9gL4eLwRuGAunSZMkio= github.com/alecthomas/repr v0.4.0 h1:GhI2A8MACjfegCPVq9f1FLvIBS+DrQ2KQBFZP1iFzXc= github.com/alecthomas/repr v0.4.0/go.mod h1:Fr0507jx4eOXV7AlPV6AVZLYrLIuIeSOWtW57eE/O/4= +github.com/anthropics/anthropic-sdk-go v0.2.0-alpha.8 h1:ss/c/eeyILgoK2sMsTJdcdLdhY3wZSt//+nanM41B9w= +github.com/anthropics/anthropic-sdk-go v0.2.0-alpha.8/go.mod h1:GJxtdOs9K4neo8Gg65CjJ7jNautmldGli5/OFNabOoo= github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4= github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI= +github.com/aws/aws-sdk-go-v2 v1.33.0 h1:Evgm4DI9imD81V0WwD+TN4DCwjUMdc94TrduMLbgZJs= +github.com/aws/aws-sdk-go-v2 v1.33.0/go.mod h1:P5WJBrYqqbWVaOxgH0X/FYYD47/nooaPOZPlQdmiN2U= +github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.3 h1:tW1/Rkad38LA15X4UQtjXZXNKsCgkshC3EbmcUmghTg= +github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.3/go.mod h1:UbnqO+zjqk3uIt9yCACHJ9IVNhyhOCnYk8yA19SAWrM= +github.com/aws/aws-sdk-go-v2/config v1.29.1 h1:JZhGawAyZ/EuJeBtbQYnaoftczcb2drR2Iq36Wgz4sQ= +github.com/aws/aws-sdk-go-v2/config v1.29.1/go.mod h1:7bR2YD5euaxBhzt2y/oDkt3uNRb6tjFp98GlTFueRwk= +github.com/aws/aws-sdk-go-v2/credentials v1.17.54 h1:4UmqeOqJPvdvASZWrKlhzpRahAulBfyTJQUaYy4+hEI= +github.com/aws/aws-sdk-go-v2/credentials v1.17.54/go.mod h1:RTdfo0P0hbbTxIhmQrOsC/PquBZGabEPnCaxxKRPSnI= +github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.16.24 h1:5grmdTdMsovn9kPZPI23Hhvp0ZyNm5cRO+IZFIYiAfw= +github.com/aws/aws-sdk-go-v2/feature/ec2/imds v1.16.24/go.mod h1:zqi7TVKTswH3Ozq28PkmBmgzG1tona7mo9G2IJg4Cis= +github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.28 h1:igORFSiH3bfq4lxKFkTSYDhJEUCYo6C8VKiWJjYwQuQ= +github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.28/go.mod h1:3So8EA/aAYm36L7XIvCVwLa0s5N0P7o2b1oqnx/2R4g= +github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.28 h1:1mOW9zAUMhTSrMDssEHS/ajx8JcAj/IcftzcmNlmVLI= +github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.28/go.mod h1:kGlXVIWDfvt2Ox5zEaNglmq0hXPHgQFNMix33Tw22jA= +github.com/aws/aws-sdk-go-v2/internal/ini v1.8.1 h1:VaRN3TlFdd6KxX1x3ILT5ynH6HvKgqdiXoTxAF4HQcQ= +github.com/aws/aws-sdk-go-v2/internal/ini v1.8.1/go.mod h1:FbtygfRFze9usAadmnGJNc8KsP346kEe+y2/oyhGAGc= +github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.12.1 h1:iXtILhvDxB6kPvEXgsDhGaZCSC6LQET5ZHSdJozeI0Y= +github.com/aws/aws-sdk-go-v2/service/internal/accept-encoding v1.12.1/go.mod h1:9nu0fVANtYiAePIBh2/pFUSwtJ402hLnp854CNoDOeE= +github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.12.9 h1:TQmKDyETFGiXVhZfQ/I0cCFziqqX58pi4tKJGYGFSz0= +github.com/aws/aws-sdk-go-v2/service/internal/presigned-url v1.12.9/go.mod h1:HVLPK2iHQBUx7HfZeOQSEu3v2ubZaAY2YPbAm5/WUyY= +github.com/aws/aws-sdk-go-v2/service/sso v1.24.11 h1:kuIyu4fTT38Kj7YCC7ouNbVZSSpqkZ+LzIfhCr6Dg+I= +github.com/aws/aws-sdk-go-v2/service/sso v1.24.11/go.mod h1:Ro744S4fKiCCuZECXgOi760TiYylUM8ZBf6OGiZzJtY= +github.com/aws/aws-sdk-go-v2/service/ssooidc v1.28.10 h1:l+dgv/64iVlQ3WsBbnn+JSbkj01jIi+SM0wYsj3y/hY= +github.com/aws/aws-sdk-go-v2/service/ssooidc v1.28.10/go.mod h1:Fzsj6lZEb8AkTE5S68OhcbBqeWPsR8RnGuKPr8Todl8= +github.com/aws/aws-sdk-go-v2/service/sts v1.33.9 h1:BRVDbewN6VZcwr+FBOszDKvYeXY1kJ+GGMCcpghlw0U= +github.com/aws/aws-sdk-go-v2/service/sts v1.33.9/go.mod h1:f6vjfZER1M17Fokn0IzssOTMT2N8ZSq+7jnNF0tArvw= +github.com/aws/smithy-go v1.22.1 h1:/HPHZQ0g7f4eUeK6HKglFz8uwVfZKgoI25rb/J+dnro= +github.com/aws/smithy-go v1.22.1/go.mod h1:irrKGvNn1InZwb2d7fkIRNucdfwR8R+Ts3wxYa/cJHg= github.com/aymanbagabas/go-osc52/v2 v2.0.1 h1:HwpRHbFMcZLEVr42D4p7XBqjyuxQH5SMiErDT4WkJ2k= github.com/aymanbagabas/go-osc52/v2 v2.0.1/go.mod h1:uYgXzlJ7ZpABp8OJ+exZzJJhRNQ2ASbcXHWsFqH8hp8= github.com/aymanbagabas/go-udiff v0.2.0 h1:TK0fH4MteXUDspT88n8CKzvK0X9O2xu9yQjWpi6yML8= github.com/aymanbagabas/go-udiff v0.2.0/go.mod h1:RE4Ex0qsGkTAJoQdQQCA0uG+nAzJO/pI/QwceO5fgrA= github.com/aymerick/douceur v0.2.0 h1:Mv+mAeH1Q+n9Fr+oyamOlAkUNPWPlA8PPGR0QAaYuPk= github.com/aymerick/douceur v0.2.0/go.mod h1:wlT5vV2O3h55X9m7iVYN0TBM0NH/MmbLnd30/FjWUq4= +github.com/bitly/go-simplejson v0.5.0/go.mod h1:cXHtHw4XUPsvGaxgjIAn8PhEWG9NfngEKAMDJEczWVA= +github.com/bmizerany/assert v0.0.0-20160611221934-b7ed37b82869/go.mod h1:Ekp36dRnpXw/yCqJaO+ZrUyxD+3VXMFFr56k5XYrpB4= +github.com/bugsnag/bugsnag-go v1.4.0/go.mod h1:2oa8nejYd4cQ/b0hMIopN0lCRxU0bueqREvZLWFrtK8= +github.com/bugsnag/panicwrap v1.2.0/go.mod h1:D/8v3kj0zr8ZAKg1AQ6crr+5VwKN5eIywRkfhyM/+dE= +github.com/bytedance/mockey v1.2.13 h1:jokWZAm/pUEbD939Rhznz615MKUCZNuvCFQlJ2+ntoo= +github.com/bytedance/mockey v1.2.13/go.mod h1:1BPHF9sol5R1ud/+0VEHGQq/+i2lN+GTsr3O2Q9IENY= +github.com/bytedance/sonic v1.12.2 h1:oaMFuRTpMHYLpCntGca65YWt5ny+wAceDERTkT2L9lg= +github.com/bytedance/sonic v1.12.2/go.mod h1:B8Gt/XvtZ3Fqj+iSKMypzymZxw/FVwgIGKzMzT9r/rk= +github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU= +github.com/bytedance/sonic/loader v0.2.0 h1:zNprn+lsIP06C/IqCHs3gPQIvnvpKbbxyXQP1iU4kWM= +github.com/bytedance/sonic/loader v0.2.0/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU= github.com/catppuccin/go v0.3.0 h1:d+0/YicIq+hSTo5oPuRi5kOpqkVA5tAsU6dNhvRu+aY= github.com/catppuccin/go v0.3.0/go.mod h1:8IHJuMGaUUjQM82qBrGNBv7LFq6JI3NnQCF6MOlZjpc= +github.com/certifi/gocertifi v0.0.0-20190105021004-abcd57078448/go.mod h1:GJKEexRPVJrBSOjoqN5VNOIKJ5Q3RViH6eu3puDRwx4= github.com/charmbracelet/bubbles v0.20.0 h1:jSZu6qD8cRQ6k9OMfR1WlM+ruM8fkPWkHvQWD9LIutE= github.com/charmbracelet/bubbles v0.20.0/go.mod h1:39slydyswPy+uVOHZ5x/GjwVAFkCsV8IIVy+4MhzwwU= github.com/charmbracelet/bubbletea v1.3.4 h1:kCg7B+jSCFPLYRA52SDZjr51kG/fMUEoPoZrkaDHyoI= @@ -38,6 +81,18 @@ github.com/charmbracelet/x/exp/strings v0.0.0-20240722160745-212f7b056ed0 h1:qko github.com/charmbracelet/x/exp/strings v0.0.0-20240722160745-212f7b056ed0/go.mod h1:pBhA0ybfXv6hDjQUZ7hk1lVxBiUbupdw5R31yPUViVQ= github.com/charmbracelet/x/term v0.2.1 h1:AQeHeLZ1OqSXhrAWpYUtZyX1T3zVxfpZuEQMIQaGIAQ= github.com/charmbracelet/x/term v0.2.1/go.mod h1:oQ4enTYFV7QN4m0i9mzHrViD7TQKvNEEkHUMCmsxdUg= +github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y= +github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w= +github.com/cloudwego/eino v0.3.17 h1:cRQUCLU6897cautWe1u3les1H3OILasUIhnHzxr2QcE= +github.com/cloudwego/eino v0.3.17/go.mod h1:+kmJimGEcKuSI6OKhet7kBedkm1WUZS3H1QRazxgWUo= +github.com/cloudwego/eino-ext/components/model/claude v0.0.0-20250320062631-616205c32186 h1:GGneAI4dIuQaxXfgsaaytmFuIi57cFTQ0R7b8PDUOxg= +github.com/cloudwego/eino-ext/components/model/claude v0.0.0-20250320062631-616205c32186/go.mod h1:ABAc+C6D9zs6KZZKfuHUhsCi+YWFogbWlqsMkQ9wJYY= +github.com/cloudwego/eino-ext/components/model/openai v0.0.0-20250320062631-616205c32186 h1:4f7KLAI2/177oZZ2iJO2bpzeJU9KPg3TduVTR0ulzQk= +github.com/cloudwego/eino-ext/components/model/openai v0.0.0-20250320062631-616205c32186/go.mod h1:vs6irdysecD6QxymjsCTzAgN2JceOTPYPjTXfJCNcgQ= +github.com/cloudwego/eino-ext/libs/acl/openai v0.0.0-20250305023926-469de0301955 h1:fgvkmTqAalDfjdy3b6Ur2mh/KEwB9L2uvqS4MFgTOqc= +github.com/cloudwego/eino-ext/libs/acl/openai v0.0.0-20250305023926-469de0301955/go.mod h1:6CThw1XQx/ASXNt31yuvp0X4Yp4GprknQuIvP9VKDpw= +github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg= +github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY= github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= @@ -50,20 +105,40 @@ github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f h1:Y/CXytFA4m6 github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f/go.mod h1:vw97MGsxSvLiUE2X8qFplwetxpGLQrlU1Q9AUEIzCaM= github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0= +github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= github.com/fsnotify/fsnotify v1.8.0 h1:dAwr6QBTBZIkG8roQaJjGof0pp0EeF+tNV7YBP3F/8M= github.com/fsnotify/fsnotify v1.8.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0= +github.com/getkin/kin-openapi v0.118.0 h1:z43njxPmJ7TaPpMSCQb7PN0dEYno4tyBPQcrFdHoLuM= +github.com/getkin/kin-openapi v0.118.0/go.mod h1:l5e9PaFUo9fyLJCPGQeXI2ML8c3P8BHOEV2VaAVf/pc= +github.com/getsentry/raven-go v0.2.0/go.mod h1:KungGk8q33+aIAZUIVWZDr2OfAEBsO49PX4NzFV5kcQ= +github.com/go-check/check v0.0.0-20180628173108-788fd7840127 h1:0gkP6mzaMqkmpcJYCFOLkIBwI7xFExG03bbkOkCvUPI= +github.com/go-check/check v0.0.0-20180628173108-788fd7840127/go.mod h1:9ES+weclKsC9YodN5RgxqK/VD9HM9JsCSh7rNhMZE98= github.com/go-logfmt/logfmt v0.6.0 h1:wGYYu3uicYdqXVgoYbvnkrPVXkuLM1p1ifugDMEdRi4= github.com/go-logfmt/logfmt v0.6.0/go.mod h1:WYhtIu8zTZfxdn5+rREduYbwxfcBr/Vr6KEVveWlfTs= +github.com/go-openapi/jsonpointer v0.19.5 h1:gZr+CIYByUqjcgeLXnQu2gHYQC9o73G2XUeOFYEICuY= +github.com/go-openapi/jsonpointer v0.19.5/go.mod h1:Pl9vOtqEWErmShwVjC8pYs9cog34VGT37dQOVbmoatg= +github.com/go-openapi/swag v0.19.5 h1:lTz6Ys4CmqqCQmZPBlbQENR1/GucA2bzYTE12Pw4tFY= +github.com/go-openapi/swag v0.19.5/go.mod h1:POnQmlKehdgb5mhVOsnJFsivZCEZ/vjK9gh66Z9tfKk= +github.com/go-test/deep v1.0.8 h1:TDsG77qcSprGbC6vTN8OuXp5g+J+b5Pcguhf7Zt61VM= +github.com/go-test/deep v1.0.8/go.mod h1:5C2ZWiW0ErCdrYzpqxLbTX7MG14M9iiw8DgHncVwcsE= github.com/go-viper/mapstructure/v2 v2.2.1 h1:ZAaOCxANMuZx5RCeg0mBdEZk7DZasvvZIxtHqx8aGss= github.com/go-viper/mapstructure/v2 v2.2.1/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM= +github.com/gofrs/uuid v3.2.0+incompatible/go.mod h1:b2aQJv3Z4Fp6yNu3cdSllBxTCLRxnplIgP/c0N/04lM= github.com/golang-migrate/migrate/v4 v4.18.2 h1:2VSCMz7x7mjyTXx3m2zPokOY82LTRgxK1yQYKo6wWQ8= github.com/golang-migrate/migrate/v4 v4.18.2/go.mod h1:2CM6tJvn2kqPXwnXO/d3rAQYiyoIm180VsO8PRX6Rpk= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/goph/emperror v0.17.2 h1:yLapQcmEsO0ipe9p5TaN22djm3OFV/TfM/fcYP0/J18= +github.com/goph/emperror v0.17.2/go.mod h1:+ZbQ+fUNO/6FNiUo0ujtMjhgad9Xa6fQL9KhH4LNHic= +github.com/gopherjs/gopherjs v1.17.2 h1:fQnZVsXk8uxXIStYb0N4bGk7jeyTalG/wsZjQ25dO0g= +github.com/gopherjs/gopherjs v1.17.2/go.mod h1:pRRIvn/QzFLrKfvEz3qUuEhtE/zLCWfreZ6J5gM2i+k= github.com/gorilla/css v1.0.1 h1:ntNaBIghp6JmvWnxbZKANoLyuXTPZ4cAMlo6RyhlbO8= github.com/gorilla/css v1.0.1/go.mod h1:BvnYkspnSzMmwRK+b8/xgNPLiIuNZr6vbZBTPQ2A3b0= +github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So= github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4= github.com/hashicorp/errwrap v1.1.0 h1:OxrOeh75EUXMY8TBjag2fzXGZ40LB6IKw45YeGUDY2I= github.com/hashicorp/errwrap v1.1.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4= @@ -71,10 +146,27 @@ github.com/hashicorp/go-multierror v1.1.1 h1:H5DkEtf6CXdFp0N0Em5UCwQpXMWke8IA0+l github.com/hashicorp/go-multierror v1.1.1/go.mod h1:iw975J/qwKPdAO1clOe2L8331t/9/fmwbPZ6JB6eMoM= github.com/hexops/gotextdiff v1.0.3 h1:gitA9+qJrrTCsiCl7+kh75nPqQt1cx4ZkudSTLoUqJM= github.com/hexops/gotextdiff v1.0.3/go.mod h1:pSWU5MAI3yDq+fZBTazCSJysOMbxWL1BSow5/V2vxeg= +github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU= github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8= github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw= +github.com/invopop/yaml v0.1.0 h1:YW3WGUoJEXYfzWBjn00zIlrw7brGVD0fUKRYDPAPhrc= +github.com/invopop/yaml v0.1.0/go.mod h1:2XuRLgs/ouIrW3XNzuNj7J3Nvu/Dig5MXvbCEdiBN3Q= +github.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY= +github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo= +github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= +github.com/kardianos/osext v0.0.0-20190222173326-2bc1f35cddc0/go.mod h1:1NbS8ALrpOvjt0rHPNLyCIeMtbizbir8U//inJ+zuB8= +github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= +github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= github.com/kujtimiihoxha/vimtea v0.0.3-0.20250317175717-9d8ba9c69840 h1:AORwYXTzap8hg0zmTA5RWB/0fxv9F19dF42dCY0IsRc= @@ -85,6 +177,12 @@ github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw= github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o= github.com/lucasb-eyer/go-colorful v1.2.0 h1:1nnpGOrhyZZuNyfu1QjKiUICQ74+3FNCN69Aj6K7nkY= github.com/lucasb-eyer/go-colorful v1.2.0/go.mod h1:R4dSotOR9KMtayYi1e77YzuveK+i7ruzyGqttikkLy0= +github.com/mailru/easyjson v0.0.0-20190614124828-94de47d64c63/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc= +github.com/mailru/easyjson v0.0.0-20190626092158-b2ccc519800e/go.mod h1:C1wdFJiN94OJF2b5HbByQZoLdCWB1Yqtg26g4irojpc= +github.com/mailru/easyjson v0.7.7 h1:UGYAvKxe3sBsEDzO8ZeWOSlIQfWFlxbzLZe7hwFURr0= +github.com/mailru/easyjson v0.7.7/go.mod h1:xzfreul335JAWq5oZzymOObrkdz5UnU4kGfJJLY9Nlc= +github.com/mattn/go-colorable v0.1.6 h1:6Su7aK7lXmJ/U79bYtBjLNaha4Fs1Rg9plHpcH+vvnE= +github.com/mattn/go-colorable v0.1.6/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc= github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= github.com/mattn/go-localereader v0.0.1 h1:ygSAOl7ZXTx4RdPYinUpg6W99U8jWvWi9Ye2JC/oIi4= @@ -94,10 +192,19 @@ github.com/mattn/go-runewidth v0.0.16 h1:E5ScNMtiwvlvB5paMFdw9p4kSQzbXFikJ5SQO6T github.com/mattn/go-runewidth v0.0.16/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= github.com/mattn/go-sqlite3 v1.14.24 h1:tpSp2G2KyMnnQu99ngJ47EIkWVmliIizyZBfPrBWDRM= github.com/mattn/go-sqlite3 v1.14.24/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y= +github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b h1:j7+1HpAFS1zy5+Q4qx1fWh90gTKwiN4QCGoY9TWyyO4= +github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b/go.mod h1:01TrycV0kFyexm33Z7vhZRXopbI8J3TDReVlkTgMUxE= github.com/microcosm-cc/bluemonday v1.0.27 h1:MpEUotklkwCSLeH+Qdx1VJgNqLlpY2KXwXFM08ygZfk= github.com/microcosm-cc/bluemonday v1.0.27/go.mod h1:jFi9vgW+H7c3V0lb6nR74Ib/DIB5OBs92Dimizgw2cA= github.com/mitchellh/hashstructure/v2 v2.0.2 h1:vGKWl0YJqUNxE8d+h8f6NJLcCJrgbhC4NcD46KavDd4= github.com/mitchellh/hashstructure/v2 v2.0.2/go.mod h1:MG3aRVU/N29oo/V/IhBX8GR/zz4kQkprJgF2EVszyDE= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826 h1:RWengNIwukTxcDr9M+97sNutRR1RKhG96O6jWumTTnw= +github.com/mohae/deepcopy v0.0.0-20170929034955-c48cc78d4826/go.mod h1:TaXosZuwdSHYgviHp1DAtfrULt5eUgsSMsZf+YrPgl8= github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6 h1:ZK8zHtRHOkbHy6Mmr5D264iyp3TiX5OmNcI5cIARiQI= github.com/muesli/ansi v0.0.0-20230316100256-276c6243b2f6/go.mod h1:CJlz5H+gyd6CUWT45Oy4q24RdLyn7Md9Vj2/ldJBSIo= github.com/muesli/cancelreader v0.2.2 h1:3I4Kt4BQjOR54NavqnDogx/MIoWBFa0StPA8ELUXHmA= @@ -106,8 +213,18 @@ github.com/muesli/reflow v0.3.0 h1:IFsN6K9NfGtjeggFP+68I4chLZV2yIKsXJFNZ+eWh6s= github.com/muesli/reflow v0.3.0/go.mod h1:pbwTDkVPibjO2kyvBQRBxTWEEGDGq0FlB1BIKtnHY/8= github.com/muesli/termenv v0.16.0 h1:S5AlUN9dENB57rsbnkPyfdGuWIlkmzJjbFf0Tf5FWUc= github.com/muesli/termenv v0.16.0/go.mod h1:ZRfOIKPFDYQoDFF4Olj7/QJbW60Ol/kL1pU3VfY/Cnk= +github.com/nikolalohinski/gonja v1.5.3 h1:GsA+EEaZDZPGJ8JtpeGN78jidhOlxeJROpqMT9fTj9c= +github.com/nikolalohinski/gonja v1.5.3/go.mod h1:RmjwxNiXAEqcq1HeK5SSMmqFJvKOfTfXhkJv6YBtPa4= +github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= +github.com/onsi/ginkgo v1.8.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= +github.com/onsi/gomega v1.5.0/go.mod h1:ex+gbHU/CVuBBDIJjb2X0qEXbFg53c61hWP/1CpauHY= github.com/pelletier/go-toml/v2 v2.2.3 h1:YmeHyLY8mFWbdkNWwpr+qIL2bEqT0o95WSdkNHvL12M= github.com/pelletier/go-toml/v2 v2.2.3/go.mod h1:MfCQTFTvCcUyyvvwm1+G6H/jORL20Xlb6rzQu9GuUkc= +github.com/perimeterx/marshmallow v1.1.4 h1:pZLDH9RjlLGGorbXhcaQLhfuV0pFMNfPO55FuFkxqLw= +github.com/perimeterx/marshmallow v1.1.4/go.mod h1:dsXbUu8CRzfYP5a87xpp0xq9S3u0Vchtcl8we9tYaXw= +github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4= +github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= github.com/rivo/uniseg v0.1.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= @@ -116,11 +233,23 @@ github.com/rivo/uniseg v0.4.7 h1:WUdvkW8uEhrYfLC4ZzdpI2ztxP1I582+49Oc5Mq64VQ= github.com/rivo/uniseg v0.4.7/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88= github.com/rogpeppe/go-internal v1.12.0 h1:exVL4IDcn6na9z1rAb56Vxr+CgyK3nn3O+epU5NdKM8= github.com/rogpeppe/go-internal v1.12.0/go.mod h1:E+RYuTGaKKdloAfM02xzb0FW3Paa99yedzYV+kq4uf4= +github.com/rollbar/rollbar-go v1.0.2/go.mod h1:AcFs5f0I+c71bpHlXNNDbOWJiKwjFDtISeXco0L5PKQ= github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= github.com/sagikazarmark/locafero v0.7.0 h1:5MqpDsTGNDhY8sGp0Aowyf0qKsPrhewaLSsFaodPcyo= github.com/sagikazarmark/locafero v0.7.0/go.mod h1:2za3Cg5rMaTMoG/2Ulr9AwtFaIppKXTRYnozin4aB5k= github.com/sahilm/fuzzy v0.1.1 h1:ceu5RHF8DGgoi+/dR5PsECjCDH1BE3Fnmpo7aVXOdRA= github.com/sahilm/fuzzy v0.1.1/go.mod h1:VFvziUEIMCrT6A6tw2RFIXPXXmzXbOsSHF0DOI8ZK9Y= +github.com/sashabaranov/go-openai v1.32.5 h1:/eNVa8KzlE7mJdKPZDj6886MUzZQjoVHyn0sLvIt5qA= +github.com/sashabaranov/go-openai v1.32.5/go.mod h1:lj5b/K+zjTSFxVLijLSTDZuP7adOgerWeFyZLUhAKRg= +github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo= +github.com/sirupsen/logrus v1.9.3 h1:dueUQJ1C2q9oE3F7wvmSGAaVtTmUizReu6fjN8uqzbQ= +github.com/sirupsen/logrus v1.9.3/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ= +github.com/slongfield/pyfmt v0.0.0-20220222012616-ea85ff4c361f h1:Z2cODYsUxQPofhpYRMQVwWz4yUVpHF+vPi+eUdruUYI= +github.com/slongfield/pyfmt v0.0.0-20220222012616-ea85ff4c361f/go.mod h1:JqzWyvTuI2X4+9wOHmKSQCYxybB/8j6Ko43qVmXDuZg= +github.com/smarty/assertions v1.15.0 h1:cR//PqUBUiQRakZWqBiFFQ9wb8emQGDb0HeGdqGByCY= +github.com/smarty/assertions v1.15.0/go.mod h1:yABtdzeQs6l1brC900WlRNwj6ZR55d7B+E8C6HtKdec= +github.com/smartystreets/goconvey v1.8.1 h1:qGjIddxOk4grTu9JPOU31tVfq3cNdBlNa5sSznIX1xY= +github.com/smartystreets/goconvey v1.8.1/go.mod h1:+/u4qLyY6x1jReYOp7GOM2FSt8aP9CzCZL03bI28W60= github.com/sourcegraph/conc v0.3.0 h1:OQTbbt6P72L20UqAkXXuLOj79LfEanQ+YQFNpLA9ySo= github.com/sourcegraph/conc v0.3.0/go.mod h1:Sdozi7LEKbFPqYX2/J+iBAM6HpqSLTASQIKqDmF7Mt0= github.com/spf13/afero v1.12.0 h1:UcOPyRBYczmFn6yvphxkn9ZEOY65cpwGKb5mL36mrqs= @@ -134,13 +263,41 @@ github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An github.com/spf13/viper v1.20.0 h1:zrxIyR3RQIOsarIrgL8+sAvALXul9jeEPa06Y0Ph6vY= github.com/spf13/viper v1.20.0/go.mod h1:P9Mdzt1zoHIG8m2eZQinpiBjo6kCmZSKBClNNqjJvu4= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8= github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU= +github.com/tidwall/gjson v1.14.2/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk= +github.com/tidwall/gjson v1.18.0 h1:FIDeeyB800efLX89e5a8Y0BNH+LOngJyGrIWxG2FKQY= +github.com/tidwall/gjson v1.18.0/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk= +github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA= +github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM= +github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU= +github.com/tidwall/pretty v1.2.1 h1:qjsOFOWWQl+N3RsoF5/ssm1pHmJJwhjlSbZ51I6wMl4= +github.com/tidwall/pretty v1.2.1/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU= +github.com/tidwall/sjson v1.2.5 h1:kLy8mja+1c9jlljvWTlSazM7cKDRfJuR/bOJhcY5NcY= +github.com/tidwall/sjson v1.2.5/go.mod h1:Fvgq9kS/6ociJEDnK0Fk1cpYF4FIW6ZF7LAe+6jwd28= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/ugorji/go v1.2.7 h1:qYhyWUUd6WbiM+C6JZAUkIJt/1WrjzNHY9+KCIjVqTo= +github.com/ugorji/go v1.2.7/go.mod h1:nF9osbDWLy6bDVv/Rtoh6QgnvNDpmCalQV5urGCCS6M= +github.com/ugorji/go/codec v1.2.7 h1:YPXUKf7fYbp/y8xloBqZOw2qaVggbfwMlI8WM3wZUJ0= +github.com/ugorji/go/codec v1.2.7/go.mod h1:WGN1fab3R1fzQlVQTkfxVtIBhWDRqOviHU95kRgeqEY= +github.com/x-cray/logrus-prefixed-formatter v0.5.2 h1:00txxvfBM9muc0jiLIEAkAcIMJzfthRT6usrui8uGmg= +github.com/x-cray/logrus-prefixed-formatter v0.5.2/go.mod h1:2duySbKsL6M18s5GU7VPsoEPHyzalCE06qoARUCeBBE= github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no= github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM= +github.com/yargevad/filepathx v1.0.0 h1:SYcT+N3tYGi+NvazubCNlvgIPbzAk7i7y2dwg3I5FYc= +github.com/yargevad/filepathx v1.0.0/go.mod h1:BprfX/gpYNJHJfc35GjRRpVcwWXS89gGulUIU5tK3tA= github.com/yuin/goldmark v1.7.1/go.mod h1:uzxRWxtg69N339t3louHJ7+O03ezfj6PlliRlaOzY1E= github.com/yuin/goldmark v1.7.8 h1:iERMLn0/QJeHFhxSt3p6PeN9mGnvIKSpG9YYorDMnic= github.com/yuin/goldmark v1.7.8/go.mod h1:uzxRWxtg69N339t3louHJ7+O03ezfj6PlliRlaOzY1E= @@ -148,22 +305,47 @@ github.com/yuin/goldmark-emoji v1.0.5 h1:EMVWyCGPlXJfUXBXpuMu+ii3TIaxbVBnEX9uaDC github.com/yuin/goldmark-emoji v1.0.5/go.mod h1:tTkZEbwu5wkPmgTcitqddVxY9osFZiavD+r4AzQrh1U= go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE= go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc= +go.uber.org/mock v0.4.0 h1:VcM4ZOtdbR4f6VXfiOpwpVJDL6lCReaZ6mw31wqh7KU= +go.uber.org/mock v0.4.0/go.mod h1:a6FSlNadKUHUa9IP5Vyt1zh4fC7uAwxMutEAscFbkZc= go.uber.org/multierr v1.9.0 h1:7fIwc/ZtS0q++VgcfqFDxSBZVv/Xo49/SYnDFupUwlI= go.uber.org/multierr v1.9.0/go.mod h1:X2jQV1h+kxSjClGpnseKVIxpmcjrj7MNnI0bnlfKTVQ= -golang.org/x/exp v0.0.0-20230315142452-642cacee5cc0 h1:pVgRXcIictcr+lBQIFeiwuwtDIs4eL21OuM9nyAADmo= -golang.org/x/exp v0.0.0-20230315142452-642cacee5cc0/go.mod h1:CxIveKay+FTh1D0yPZemJVgC/95VzuuOLq5Qi4xnoYc= +golang.org/x/arch v0.11.0 h1:KXV8WWKCXm6tRpLirl2szsO5j/oOODwZf4hATmGVNs4= +golang.org/x/arch v0.11.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys= +golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= +golang.org/x/crypto v0.32.0 h1:euUpcYgM8WcP71gNpTqQCn6rC2t6ULUPiOzfWaXVVfc= +golang.org/x/crypto v0.32.0/go.mod h1:ZnnJkOaASj8g0AjIduWNlq2NRxL0PlBrbKVyZ6V/Ugc= +golang.org/x/exp v0.0.0-20230713183714-613f0c0eb8a1 h1:MGwJjxBy0HJshjDNfLsYO8xppfqWlA5ZT9OhtUUhTNw= +golang.org/x/exp v0.0.0-20230713183714-613f0c0eb8a1/go.mod h1:FXUEEKJgO7OQYeo8N01OfiKP8RXMtf6e8aTskBGqWdc= +golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I= golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= golang.org/x/sync v0.12.0 h1:MHc5BpPuC30uJk597Ri8TV3CNZcTLu6B6z4lJy+g6Jw= golang.org/x/sync v0.12.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA= +golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= golang.org/x/sys v0.0.0-20210809222454-d867a43fc93e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= golang.org/x/sys v0.30.0 h1:QjkSwP/36a20jFYWkSue1YwXzLmsV5Gfq7Eiy72C1uc= golang.org/x/sys v0.30.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/term v0.30.0 h1:PQ39fJZ+mfadBm0y5WlL4vlM7Sx1Hgf13sMIY2+QS9Y= +golang.org/x/term v0.30.0/go.mod h1:NYYFdzHoI5wRh/h5tDMdMqCqPJZEuNqVR5xJLd/n67g= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= golang.org/x/text v0.23.0 h1:D71I7dUrlY+VX0gQShAThNGHFxZ13dGLBHQLVl1mJlY= golang.org/x/text v0.23.0/go.mod h1:/BLNzu4aZCJ1+kcD0DNRotWKage4q2rGVAg4o22unh4= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo= gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys= +gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw= +gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= +gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.0/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50= diff --git a/internal/app/services.go b/internal/app/services.go index 60b0e32f..3c4f211a 100644 --- a/internal/app/services.go +++ b/internal/app/services.go @@ -5,7 +5,9 @@ import ( "database/sql" "github.com/kujtimiihoxha/termai/internal/db" + "github.com/kujtimiihoxha/termai/internal/llm" "github.com/kujtimiihoxha/termai/internal/logging" + "github.com/kujtimiihoxha/termai/internal/message" "github.com/kujtimiihoxha/termai/internal/session" "github.com/spf13/viper" ) @@ -14,6 +16,8 @@ type App struct { Context context.Context Sessions session.Service + Messages message.Service + LLM llm.Service Logger logging.Interface } @@ -23,9 +27,15 @@ func New(ctx context.Context, conn *sql.DB) *App { log := logging.NewLogger(logging.Options{ Level: viper.GetString("log.level"), }) + sessions := session.NewService(ctx, q) + messages := message.NewService(ctx, q) + llm := llm.NewService(ctx, log, sessions, messages) + return &App{ Context: ctx, - Sessions: session.NewService(ctx, q), + Sessions: sessions, + Messages: messages, + LLM: llm, Logger: log, } } diff --git a/internal/db/db.go b/internal/db/db.go index b97eb710..e882106c 100644 --- a/internal/db/db.go +++ b/internal/db/db.go @@ -24,15 +24,30 @@ func New(db DBTX) *Queries { func Prepare(ctx context.Context, db DBTX) (*Queries, error) { q := Queries{db: db} var err error + if q.createMessageStmt, err = db.PrepareContext(ctx, createMessage); err != nil { + return nil, fmt.Errorf("error preparing query CreateMessage: %w", err) + } if q.createSessionStmt, err = db.PrepareContext(ctx, createSession); err != nil { return nil, fmt.Errorf("error preparing query CreateSession: %w", err) } + if q.deleteMessageStmt, err = db.PrepareContext(ctx, deleteMessage); err != nil { + return nil, fmt.Errorf("error preparing query DeleteMessage: %w", err) + } if q.deleteSessionStmt, err = db.PrepareContext(ctx, deleteSession); err != nil { return nil, fmt.Errorf("error preparing query DeleteSession: %w", err) } + if q.deleteSessionMessagesStmt, err = db.PrepareContext(ctx, deleteSessionMessages); err != nil { + return nil, fmt.Errorf("error preparing query DeleteSessionMessages: %w", err) + } + if q.getMessageStmt, err = db.PrepareContext(ctx, getMessage); err != nil { + return nil, fmt.Errorf("error preparing query GetMessage: %w", err) + } if q.getSessionByIDStmt, err = db.PrepareContext(ctx, getSessionByID); err != nil { return nil, fmt.Errorf("error preparing query GetSessionByID: %w", err) } + if q.listMessagesBySessionStmt, err = db.PrepareContext(ctx, listMessagesBySession); err != nil { + return nil, fmt.Errorf("error preparing query ListMessagesBySession: %w", err) + } if q.listSessionsStmt, err = db.PrepareContext(ctx, listSessions); err != nil { return nil, fmt.Errorf("error preparing query ListSessions: %w", err) } @@ -44,21 +59,46 @@ func Prepare(ctx context.Context, db DBTX) (*Queries, error) { func (q *Queries) Close() error { var err error + if q.createMessageStmt != nil { + if cerr := q.createMessageStmt.Close(); cerr != nil { + err = fmt.Errorf("error closing createMessageStmt: %w", cerr) + } + } if q.createSessionStmt != nil { if cerr := q.createSessionStmt.Close(); cerr != nil { err = fmt.Errorf("error closing createSessionStmt: %w", cerr) } } + if q.deleteMessageStmt != nil { + if cerr := q.deleteMessageStmt.Close(); cerr != nil { + err = fmt.Errorf("error closing deleteMessageStmt: %w", cerr) + } + } if q.deleteSessionStmt != nil { if cerr := q.deleteSessionStmt.Close(); cerr != nil { err = fmt.Errorf("error closing deleteSessionStmt: %w", cerr) } } + if q.deleteSessionMessagesStmt != nil { + if cerr := q.deleteSessionMessagesStmt.Close(); cerr != nil { + err = fmt.Errorf("error closing deleteSessionMessagesStmt: %w", cerr) + } + } + if q.getMessageStmt != nil { + if cerr := q.getMessageStmt.Close(); cerr != nil { + err = fmt.Errorf("error closing getMessageStmt: %w", cerr) + } + } if q.getSessionByIDStmt != nil { if cerr := q.getSessionByIDStmt.Close(); cerr != nil { err = fmt.Errorf("error closing getSessionByIDStmt: %w", cerr) } } + if q.listMessagesBySessionStmt != nil { + if cerr := q.listMessagesBySessionStmt.Close(); cerr != nil { + err = fmt.Errorf("error closing listMessagesBySessionStmt: %w", cerr) + } + } if q.listSessionsStmt != nil { if cerr := q.listSessionsStmt.Close(); cerr != nil { err = fmt.Errorf("error closing listSessionsStmt: %w", cerr) @@ -106,23 +146,33 @@ func (q *Queries) queryRow(ctx context.Context, stmt *sql.Stmt, query string, ar } type Queries struct { - db DBTX - tx *sql.Tx - createSessionStmt *sql.Stmt - deleteSessionStmt *sql.Stmt - getSessionByIDStmt *sql.Stmt - listSessionsStmt *sql.Stmt - updateSessionStmt *sql.Stmt + db DBTX + tx *sql.Tx + createMessageStmt *sql.Stmt + createSessionStmt *sql.Stmt + deleteMessageStmt *sql.Stmt + deleteSessionStmt *sql.Stmt + deleteSessionMessagesStmt *sql.Stmt + getMessageStmt *sql.Stmt + getSessionByIDStmt *sql.Stmt + listMessagesBySessionStmt *sql.Stmt + listSessionsStmt *sql.Stmt + updateSessionStmt *sql.Stmt } func (q *Queries) WithTx(tx *sql.Tx) *Queries { return &Queries{ - db: tx, - tx: tx, - createSessionStmt: q.createSessionStmt, - deleteSessionStmt: q.deleteSessionStmt, - getSessionByIDStmt: q.getSessionByIDStmt, - listSessionsStmt: q.listSessionsStmt, - updateSessionStmt: q.updateSessionStmt, + db: tx, + tx: tx, + createMessageStmt: q.createMessageStmt, + createSessionStmt: q.createSessionStmt, + deleteMessageStmt: q.deleteMessageStmt, + deleteSessionStmt: q.deleteSessionStmt, + deleteSessionMessagesStmt: q.deleteSessionMessagesStmt, + getMessageStmt: q.getMessageStmt, + getSessionByIDStmt: q.getSessionByIDStmt, + listMessagesBySessionStmt: q.listMessagesBySessionStmt, + listSessionsStmt: q.listSessionsStmt, + updateSessionStmt: q.updateSessionStmt, } } diff --git a/internal/db/messages.sql.go b/internal/db/messages.sql.go new file mode 100644 index 00000000..d0f69458 --- /dev/null +++ b/internal/db/messages.sql.go @@ -0,0 +1,117 @@ +// Code generated by sqlc. DO NOT EDIT. +// versions: +// sqlc v1.27.0 +// source: messages.sql + +package db + +import ( + "context" +) + +const createMessage = `-- name: CreateMessage :one +INSERT INTO messages ( + id, + session_id, + message_data, + created_at, + updated_at +) VALUES ( + ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now') +) +RETURNING id, session_id, message_data, created_at, updated_at +` + +type CreateMessageParams struct { + ID string `json:"id"` + SessionID string `json:"session_id"` + MessageData string `json:"message_data"` +} + +func (q *Queries) CreateMessage(ctx context.Context, arg CreateMessageParams) (Message, error) { + row := q.queryRow(ctx, q.createMessageStmt, createMessage, arg.ID, arg.SessionID, arg.MessageData) + var i Message + err := row.Scan( + &i.ID, + &i.SessionID, + &i.MessageData, + &i.CreatedAt, + &i.UpdatedAt, + ) + return i, err +} + +const deleteMessage = `-- name: DeleteMessage :exec +DELETE FROM messages +WHERE id = ? +` + +func (q *Queries) DeleteMessage(ctx context.Context, id string) error { + _, err := q.exec(ctx, q.deleteMessageStmt, deleteMessage, id) + return err +} + +const deleteSessionMessages = `-- name: DeleteSessionMessages :exec +DELETE FROM messages +WHERE session_id = ? +` + +func (q *Queries) DeleteSessionMessages(ctx context.Context, sessionID string) error { + _, err := q.exec(ctx, q.deleteSessionMessagesStmt, deleteSessionMessages, sessionID) + return err +} + +const getMessage = `-- name: GetMessage :one +SELECT id, session_id, message_data, created_at, updated_at +FROM messages +WHERE id = ? LIMIT 1 +` + +func (q *Queries) GetMessage(ctx context.Context, id string) (Message, error) { + row := q.queryRow(ctx, q.getMessageStmt, getMessage, id) + var i Message + err := row.Scan( + &i.ID, + &i.SessionID, + &i.MessageData, + &i.CreatedAt, + &i.UpdatedAt, + ) + return i, err +} + +const listMessagesBySession = `-- name: ListMessagesBySession :many +SELECT id, session_id, message_data, created_at, updated_at +FROM messages +WHERE session_id = ? +ORDER BY created_at ASC +` + +func (q *Queries) ListMessagesBySession(ctx context.Context, sessionID string) ([]Message, error) { + rows, err := q.query(ctx, q.listMessagesBySessionStmt, listMessagesBySession, sessionID) + if err != nil { + return nil, err + } + defer rows.Close() + items := []Message{} + for rows.Next() { + var i Message + if err := rows.Scan( + &i.ID, + &i.SessionID, + &i.MessageData, + &i.CreatedAt, + &i.UpdatedAt, + ); err != nil { + return nil, err + } + items = append(items, i) + } + if err := rows.Close(); err != nil { + return nil, err + } + if err := rows.Err(); err != nil { + return nil, err + } + return items, nil +} diff --git a/internal/db/migrations/000001_initial.down.sql b/internal/db/migrations/000001_initial.down.sql index 7324ddb6..4f2712d8 100644 --- a/internal/db/migrations/000001_initial.down.sql +++ b/internal/db/migrations/000001_initial.down.sql @@ -1,4 +1,8 @@ --- sqlfluff:dialect:sqlite DROP TRIGGER IF EXISTS update_sessions_updated_at; +DROP TRIGGER IF EXISTS update_messages_updated_at; + +DROP TRIGGER IF EXISTS update_session_message_count_on_delete; +DROP TRIGGER IF EXISTS update_session_message_count_on_insert; DROP TABLE IF EXISTS sessions; +DROP TABLE IF EXISTS messages; diff --git a/internal/db/migrations/000001_initial.up.sql b/internal/db/migrations/000001_initial.up.sql index 115b6145..b87592bb 100644 --- a/internal/db/migrations/000001_initial.up.sql +++ b/internal/db/migrations/000001_initial.up.sql @@ -1,9 +1,10 @@ --- sqlfluff:dialect:sqlite +-- Sessions CREATE TABLE IF NOT EXISTS sessions ( id TEXT PRIMARY KEY, title TEXT NOT NULL, message_count INTEGER NOT NULL DEFAULT 0 CHECK (message_count >= 0), - tokens INTEGER NOT NULL DEFAULT 0 CHECK (tokens >= 0), + prompt_tokens INTEGER NOT NULL DEFAULT 0 CHECK (prompt_tokens >= 0), + completion_tokens INTEGER NOT NULL DEFAULT 0 CHECK (completion_tokens>= 0), cost REAL NOT NULL DEFAULT 0.0 CHECK (cost >= 0.0), updated_at INTEGER NOT NULL, -- Unix timestamp in milliseconds created_at INTEGER NOT NULL -- Unix timestamp in milliseconds @@ -15,3 +16,38 @@ BEGIN UPDATE sessions SET updated_at = strftime('%s', 'now') WHERE id = new.id; END; + +-- Messages +CREATE TABLE IF NOT EXISTS messages ( + id TEXT PRIMARY KEY, + session_id TEXT NOT NULL, + message_data TEXT NOT NULL, -- JSON string of message content + created_at INTEGER NOT NULL, -- Unix timestamp in milliseconds + updated_at INTEGER NOT NULL, -- Unix timestamp in milliseconds + FOREIGN KEY (session_id) REFERENCES sessions (id) ON DELETE CASCADE +); + +CREATE INDEX IF NOT EXISTS idx_messages_session_id ON messages (session_id); + +CREATE TRIGGER IF NOT EXISTS update_messages_updated_at +AFTER UPDATE ON messages +BEGIN +UPDATE messages SET updated_at = strftime('%s', 'now') +WHERE id = new.id; +END; + +CREATE TRIGGER IF NOT EXISTS update_session_message_count_on_insert +AFTER INSERT ON messages +BEGIN +UPDATE sessions SET + message_count = message_count + 1 +WHERE id = new.session_id; +END; + +CREATE TRIGGER IF NOT EXISTS update_session_message_count_on_delete +AFTER DELETE ON messages +BEGIN +UPDATE sessions SET + message_count = message_count - 1 +WHERE id = old.session_id; +END; diff --git a/internal/db/models.go b/internal/db/models.go index 834aaf4d..de9cd3cd 100644 --- a/internal/db/models.go +++ b/internal/db/models.go @@ -4,12 +4,21 @@ package db -type Session struct { - ID string `json:"id"` - Title string `json:"title"` - MessageCount int64 `json:"message_count"` - Tokens int64 `json:"tokens"` - Cost float64 `json:"cost"` - UpdatedAt int64 `json:"updated_at"` - CreatedAt int64 `json:"created_at"` +type Message struct { + ID string `json:"id"` + SessionID string `json:"session_id"` + MessageData string `json:"message_data"` + CreatedAt int64 `json:"created_at"` + UpdatedAt int64 `json:"updated_at"` +} + +type Session struct { + ID string `json:"id"` + Title string `json:"title"` + MessageCount int64 `json:"message_count"` + PromptTokens int64 `json:"prompt_tokens"` + CompletionTokens int64 `json:"completion_tokens"` + Cost float64 `json:"cost"` + UpdatedAt int64 `json:"updated_at"` + CreatedAt int64 `json:"created_at"` } diff --git a/internal/db/querier.go b/internal/db/querier.go index ac2374cf..626d206a 100644 --- a/internal/db/querier.go +++ b/internal/db/querier.go @@ -9,10 +9,14 @@ import ( ) type Querier interface { - // sqlfluff:dialect:sqlite + CreateMessage(ctx context.Context, arg CreateMessageParams) (Message, error) CreateSession(ctx context.Context, arg CreateSessionParams) (Session, error) + DeleteMessage(ctx context.Context, id string) error DeleteSession(ctx context.Context, id string) error + DeleteSessionMessages(ctx context.Context, sessionID string) error + GetMessage(ctx context.Context, id string) (Message, error) GetSessionByID(ctx context.Context, id string) (Session, error) + ListMessagesBySession(ctx context.Context, sessionID string) ([]Message, error) ListSessions(ctx context.Context) ([]Session, error) UpdateSession(ctx context.Context, arg UpdateSessionParams) (Session, error) } diff --git a/internal/db/sessions.sql.go b/internal/db/sessions.sql.go index 3e0b6eea..f3ee4ff4 100644 --- a/internal/db/sessions.sql.go +++ b/internal/db/sessions.sql.go @@ -14,7 +14,8 @@ INSERT INTO sessions ( id, title, message_count, - tokens, + prompt_tokens, + completion_tokens, cost, updated_at, created_at @@ -24,26 +25,28 @@ INSERT INTO sessions ( ?, ?, ?, + ?, strftime('%s', 'now'), strftime('%s', 'now') -) RETURNING id, title, message_count, tokens, cost, updated_at, created_at +) RETURNING id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at ` type CreateSessionParams struct { - ID string `json:"id"` - Title string `json:"title"` - MessageCount int64 `json:"message_count"` - Tokens int64 `json:"tokens"` - Cost float64 `json:"cost"` + ID string `json:"id"` + Title string `json:"title"` + MessageCount int64 `json:"message_count"` + PromptTokens int64 `json:"prompt_tokens"` + CompletionTokens int64 `json:"completion_tokens"` + Cost float64 `json:"cost"` } -// sqlfluff:dialect:sqlite func (q *Queries) CreateSession(ctx context.Context, arg CreateSessionParams) (Session, error) { row := q.queryRow(ctx, q.createSessionStmt, createSession, arg.ID, arg.Title, arg.MessageCount, - arg.Tokens, + arg.PromptTokens, + arg.CompletionTokens, arg.Cost, ) var i Session @@ -51,7 +54,8 @@ func (q *Queries) CreateSession(ctx context.Context, arg CreateSessionParams) (S &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, @@ -70,7 +74,7 @@ func (q *Queries) DeleteSession(ctx context.Context, id string) error { } const getSessionByID = `-- name: GetSessionByID :one -SELECT id, title, message_count, tokens, cost, updated_at, created_at +SELECT id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at FROM sessions WHERE id = ? LIMIT 1 ` @@ -82,7 +86,8 @@ func (q *Queries) GetSessionByID(ctx context.Context, id string) (Session, error &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, @@ -91,7 +96,7 @@ func (q *Queries) GetSessionByID(ctx context.Context, id string) (Session, error } const listSessions = `-- name: ListSessions :many -SELECT id, title, message_count, tokens, cost, updated_at, created_at +SELECT id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at FROM sessions ORDER BY created_at DESC ` @@ -109,7 +114,8 @@ func (q *Queries) ListSessions(ctx context.Context) ([]Session, error) { &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, @@ -131,23 +137,26 @@ const updateSession = `-- name: UpdateSession :one UPDATE sessions SET title = ?, - tokens = ?, + prompt_tokens = ?, + completion_tokens = ?, cost = ? WHERE id = ? -RETURNING id, title, message_count, tokens, cost, updated_at, created_at +RETURNING id, title, message_count, prompt_tokens, completion_tokens, cost, updated_at, created_at ` type UpdateSessionParams struct { - Title string `json:"title"` - Tokens int64 `json:"tokens"` - Cost float64 `json:"cost"` - ID string `json:"id"` + Title string `json:"title"` + PromptTokens int64 `json:"prompt_tokens"` + CompletionTokens int64 `json:"completion_tokens"` + Cost float64 `json:"cost"` + ID string `json:"id"` } func (q *Queries) UpdateSession(ctx context.Context, arg UpdateSessionParams) (Session, error) { row := q.queryRow(ctx, q.updateSessionStmt, updateSession, arg.Title, - arg.Tokens, + arg.PromptTokens, + arg.CompletionTokens, arg.Cost, arg.ID, ) @@ -156,7 +165,8 @@ func (q *Queries) UpdateSession(ctx context.Context, arg UpdateSessionParams) (S &i.ID, &i.Title, &i.MessageCount, - &i.Tokens, + &i.PromptTokens, + &i.CompletionTokens, &i.Cost, &i.UpdatedAt, &i.CreatedAt, diff --git a/internal/db/sql/messages.sql b/internal/db/sql/messages.sql new file mode 100644 index 00000000..db5e192f --- /dev/null +++ b/internal/db/sql/messages.sql @@ -0,0 +1,30 @@ +-- name: GetMessage :one +SELECT * +FROM messages +WHERE id = ? LIMIT 1; + +-- name: ListMessagesBySession :many +SELECT * +FROM messages +WHERE session_id = ? +ORDER BY created_at ASC; + +-- name: CreateMessage :one +INSERT INTO messages ( + id, + session_id, + message_data, + created_at, + updated_at +) VALUES ( + ?, ?, ?, strftime('%s', 'now'), strftime('%s', 'now') +) +RETURNING *; + +-- name: DeleteMessage :exec +DELETE FROM messages +WHERE id = ?; + +-- name: DeleteSessionMessages :exec +DELETE FROM messages +WHERE session_id = ?; diff --git a/internal/db/sql/sessions.sql b/internal/db/sql/sessions.sql index e90e5e32..2be8b7cc 100644 --- a/internal/db/sql/sessions.sql +++ b/internal/db/sql/sessions.sql @@ -1,10 +1,10 @@ --- sqlfluff:dialect:sqlite -- name: CreateSession :one INSERT INTO sessions ( id, title, message_count, - tokens, + prompt_tokens, + completion_tokens, cost, updated_at, created_at @@ -14,6 +14,7 @@ INSERT INTO sessions ( ?, ?, ?, + ?, strftime('%s', 'now'), strftime('%s', 'now') ) RETURNING *; @@ -32,7 +33,8 @@ ORDER BY created_at DESC; UPDATE sessions SET title = ?, - tokens = ?, + prompt_tokens = ?, + completion_tokens = ?, cost = ? WHERE id = ? RETURNING *; diff --git a/internal/llm/agent/agent.go b/internal/llm/agent/agent.go new file mode 100644 index 00000000..e02fc0f7 --- /dev/null +++ b/internal/llm/agent/agent.go @@ -0,0 +1,17 @@ +package agent + +import ( + "context" + "fmt" + + "github.com/cloudwego/eino/flow/agent/react" +) + +func GetAgent(ctx context.Context, name string) (*react.Agent, string, error) { + switch name { + case "coder": + agent, err := NewCoderAgent(ctx) + return agent, CoderSystemPrompt(), err + } + return nil, "", fmt.Errorf("agent %s not found", name) +} diff --git a/internal/llm/agent/coder.go b/internal/llm/agent/coder.go new file mode 100644 index 00000000..32faa2c7 --- /dev/null +++ b/internal/llm/agent/coder.go @@ -0,0 +1,175 @@ +package agent + +import ( + "context" + "fmt" + "os" + "path/filepath" + "runtime" + "time" + + "github.com/cloudwego/eino/components/tool" + "github.com/cloudwego/eino/compose" + "github.com/cloudwego/eino/flow/agent/react" + "github.com/kujtimiihoxha/termai/internal/llm/models" + "github.com/kujtimiihoxha/termai/internal/llm/tools" + "github.com/spf13/viper" +) + +func coderTools() []tool.BaseTool { + return []tool.BaseTool{ + tools.NewBashTool(viper.GetString("wd")), + } +} + +func NewCoderAgent(ctx context.Context) (*react.Agent, error) { + model, err := models.GetModel(ctx, models.ModelID(viper.GetString("models.big"))) + if err != nil { + return nil, err + } + reactAgent, err := react.NewAgent(ctx, &react.AgentConfig{ + Model: model, + ToolsConfig: compose.ToolsNodeConfig{ + Tools: coderTools(), + }, + MaxStep: 1000, + }) + if err != nil { + return nil, err + } + + return reactAgent, nil +} + +func CoderSystemPrompt() string { + basePrompt := `You are termAI, an interactive CLI tool that helps users with software engineering tasks. Use the instructions below and the tools available to you to assist the user. + +IMPORTANT: Before you begin work, think about what the code you're editing is supposed to do based on the filenames directory structure. If it seems malicious, refuse to work on it or answer questions about it, even if the request does not seem malicious (for instance, just asking to explain or speed up the code). + +Here are useful slash commands users can run to interact with you: + +# Memory +If the current working directory contains a file called termai.md, it will be automatically added to your context. This file serves multiple purposes: +1. Storing frequently used bash commands (build, test, lint, etc.) so you can use them without searching each time +2. Recording the user's code style preferences (naming conventions, preferred libraries, etc.) +3. Maintaining useful information about the codebase structure and organization + +When you spend time searching for commands to typecheck, lint, build, or test, you should ask the user if it's okay to add those commands to termai.md. Similarly, when learning about code style preferences or important codebase information, ask if it's okay to add that to termai.md so you can remember it for next time. + +# Tone and style +You should be concise, direct, and to the point. When you run a non-trivial bash command, you should explain what the command does and why you are running it, to make sure the user understands what you are doing (this is especially important when you are running a command that will make changes to the user's system). +Remember that your output will be displayed on a command line interface. Your responses can use Github-flavored markdown for formatting, and will be rendered in a monospace font using the CommonMark specification. +Output text to communicate with the user; all text you output outside of tool use is displayed to the user. Only use tools to complete tasks. Never use tools like Bash or code comments as means to communicate with the user during the session. +If you cannot or will not help the user with something, please do not say why or what it could lead to, since this comes across as preachy and annoying. Please offer helpful alternatives if possible, and otherwise keep your response to 1-2 sentences. +IMPORTANT: You should minimize output tokens as much as possible while maintaining helpfulness, quality, and accuracy. Only address the specific query or task at hand, avoiding tangential information unless absolutely critical for completing the request. If you can answer in 1-3 sentences or a short paragraph, please do. +IMPORTANT: You should NOT answer with unnecessary preamble or postamble (such as explaining your code or summarizing your action), unless the user asks you to. +IMPORTANT: Keep your responses short, since they will be displayed on a command line interface. You MUST answer concisely with fewer than 4 lines (not including tool use or code generation), unless user asks for detail. Answer the user's question directly, without elaboration, explanation, or details. One word answers are best. Avoid introductions, conclusions, and explanations. You MUST avoid text before/after your response, such as "The answer is .", "Here is the content of the file..." or "Based on the information provided, the answer is..." or "Here is what I will do next...". Here are some examples to demonstrate appropriate verbosity: + +user: 2 + 2 +assistant: 4 + + + +user: what is 2+2? +assistant: 4 + + + +user: is 11 a prime number? +assistant: true + + + +user: what command should I run to list files in the current directory? +assistant: ls + + + +user: what command should I run to watch files in the current directory? +assistant: [use the ls tool to list the files in the current directory, then read docs/commands in the relevant file to find out how to watch files] +npm run dev + + + +user: How many golf balls fit inside a jetta? +assistant: 150000 + + + +user: what files are in the directory src/? +assistant: [runs ls and sees foo.c, bar.c, baz.c] +user: which file contains the implementation of foo? +assistant: src/foo.c + + + +user: write tests for new feature +assistant: [uses grep and glob search tools to find where similar tests are defined, uses concurrent read file tool use blocks in one tool call to read relevant files at the same time, uses edit file tool to write new tests] + + +# Proactiveness +You are allowed to be proactive, but only when the user asks you to do something. You should strive to strike a balance between: +1. Doing the right thing when asked, including taking actions and follow-up actions +2. Not surprising the user with actions you take without asking +For example, if the user asks you how to approach something, you should do your best to answer their question first, and not immediately jump into taking actions. +3. Do not add additional code explanation summary unless requested by the user. After working on a file, just stop, rather than providing an explanation of what you did. + +# Synthetic messages +Sometimes, the conversation will contain messages like [Request interrupted by user] or [Request interrupted by user for tool use]. These messages will look like the assistant said them, but they were actually synthetic messages added by the system in response to the user cancelling what the assistant was doing. You should not respond to these messages. You must NEVER send messages like this yourself. + +# Following conventions +When making changes to files, first understand the file's code conventions. Mimic code style, use existing libraries and utilities, and follow existing patterns. +- NEVER assume that a given library is available, even if it is well known. Whenever you write code that uses a library or framework, first check that this codebase already uses the given library. For example, you might look at neighboring files, or check the package.json (or cargo.toml, and so on depending on the language). +- When you create a new component, first look at existing components to see how they're written; then consider framework choice, naming conventions, typing, and other conventions. +- When you edit a piece of code, first look at the code's surrounding context (especially its imports) to understand the code's choice of frameworks and libraries. Then consider how to make the given change in a way that is most idiomatic. +- Always follow security best practices. Never introduce code that exposes or logs secrets and keys. Never commit secrets or keys to the repository. + +# Code style +- Do not add comments to the code you write, unless the user asks you to, or the code is complex and requires additional context. + +# Doing tasks +The user will primarily request you perform software engineering tasks. This includes solving bugs, adding new functionality, refactoring code, explaining code, and more. For these tasks the following steps are recommended: +1. Use the available search tools to understand the codebase and the user's query. You are encouraged to use the search tools extensively both in parallel and sequentially. +2. Implement the solution using all tools available to you +3. Verify the solution if possible with tests. NEVER assume specific test framework or test script. Check the README or search codebase to determine the testing approach. +4. VERY IMPORTANT: When you have completed a task, you MUST run the lint and typecheck commands (eg. npm run lint, npm run typecheck, ruff, etc.) if they were provided to you to ensure your code is correct. If you are unable to find the correct command, ask the user for the command to run and if they supply it, proactively suggest writing it to termai.md so that you will know to run it next time. + +NEVER commit changes unless the user explicitly asks you to. It is VERY IMPORTANT to only commit when explicitly asked, otherwise the user will feel that you are being too proactive. + +# Tool usage policy +- When doing file search, prefer to use the Agent tool in order to reduce context usage. +- If you intend to call multiple tools and there are no dependencies between the calls, make all of the independent calls in the same function_calls block. + +You MUST answer concisely with fewer than 4 lines of text (not including tool use or code generation), unless user asks for detail.` + + envInfo := getEnvironmentInfo() + + return fmt.Sprintf("%s\n\n%s", basePrompt, envInfo) +} + +func getEnvironmentInfo() string { + cwd := viper.GetString("wd") + isGit := isGitRepo(cwd) + platform := runtime.GOOS + date := time.Now().Format("1/2/2006") + + return fmt.Sprintf(`Here is useful information about the environment you are running in: + +Working directory: %s +Is directory a git repo: %s +Platform: %s +Today's date: %s +`, cwd, boolToYesNo(isGit), platform, date) +} + +func isGitRepo(dir string) bool { + _, err := os.Stat(filepath.Join(dir, ".git")) + return err == nil +} + +func boolToYesNo(b bool) string { + if b { + return "Yes" + } + return "No" +} diff --git a/internal/llm/llm.go b/internal/llm/llm.go new file mode 100644 index 00000000..bbf9961e --- /dev/null +++ b/internal/llm/llm.go @@ -0,0 +1,206 @@ +package llm + +import ( + "context" + "log" + "sync" + "time" + + "github.com/cloudwego/eino/callbacks" + "github.com/cloudwego/eino/compose" + "github.com/cloudwego/eino/schema" + "github.com/google/uuid" + "github.com/kujtimiihoxha/termai/internal/llm/agent" + "github.com/kujtimiihoxha/termai/internal/logging" + "github.com/kujtimiihoxha/termai/internal/message" + "github.com/kujtimiihoxha/termai/internal/pubsub" + "github.com/kujtimiihoxha/termai/internal/session" + + eModel "github.com/cloudwego/eino/components/model" + enioAgent "github.com/cloudwego/eino/flow/agent" + "github.com/spf13/viper" +) + +const ( + AgentRequestoEvent pubsub.EventType = "agent_request" + AgentErrorEvent pubsub.EventType = "agent_error" + AgentResponseEvent pubsub.EventType = "agent_response" +) + +type AgentMessageType int + +const ( + AgentMessageTypeNewUserMessage AgentMessageType = iota + AgentMessageTypeAgentResponse + AgentMessageTypeError +) + +type agentID string + +const ( + RootAgent agentID = "root" + TaskAgent agentID = "task" +) + +type AgentEvent struct { + ID string `json:"id"` + Type AgentMessageType `json:"type"` + AgentID agentID `json:"agent_id"` + MessageID string `json:"message_id"` + SessionID string `json:"session_id"` + Content string `json:"content"` +} + +type Service interface { + pubsub.Suscriber[AgentEvent] + + SendRequest(sessionID string, content string) +} +type service struct { + *pubsub.Broker[AgentEvent] + Requests sync.Map + ctx context.Context + activeRequests sync.Map + messages message.Service + sessions session.Service + logger logging.Interface +} + +func (s *service) handleRequest(id string, sessionID string, content string) { + cancel, ok := s.activeRequests.Load(id) + if !ok { + return + } + defer cancel.(context.CancelFunc)() + defer s.activeRequests.Delete(id) + + history, err := s.messages.List(sessionID) + if err != nil { + s.Publish(AgentErrorEvent, AgentEvent{ + ID: id, + Type: AgentMessageTypeError, + AgentID: RootAgent, + MessageID: "", + SessionID: sessionID, + Content: err.Error(), + }) + return + } + + log.Printf("Request: %s", content) + agent, systemMessage, err := agent.GetAgent(s.ctx, viper.GetString("agents.default")) + if err != nil { + s.Publish(AgentErrorEvent, AgentEvent{ + ID: id, + Type: AgentMessageTypeError, + AgentID: RootAgent, + MessageID: "", + SessionID: sessionID, + Content: err.Error(), + }) + return + } + + messages := []*schema.Message{ + { + Role: schema.System, + Content: systemMessage, + }, + } + for _, m := range history { + messages = append(messages, &m.MessageData) + } + builder := callbacks.NewHandlerBuilder() + builder.OnStartFn(func(ctx context.Context, info *callbacks.RunInfo, input callbacks.CallbackInput) context.Context { + i, ok := input.(*eModel.CallbackInput) + if info.Component == "ChatModel" && ok { + if len(messages) < len(i.Messages) { + // find new messages + newMessages := i.Messages[len(messages):] + for _, m := range newMessages { + _, err = s.messages.Create(sessionID, *m) + if err != nil { + s.Publish(AgentErrorEvent, AgentEvent{ + ID: id, + Type: AgentMessageTypeError, + AgentID: RootAgent, + MessageID: "", + SessionID: sessionID, + Content: err.Error(), + }) + } + messages = append(messages, m) + } + } + } + + return ctx + }) + builder.OnEndFn(func(ctx context.Context, info *callbacks.RunInfo, output callbacks.CallbackOutput) context.Context { + return ctx + }) + + out, err := agent.Generate(s.ctx, messages, enioAgent.WithComposeOptions(compose.WithCallbacks(builder.Build()))) + if err != nil { + s.Publish(AgentErrorEvent, AgentEvent{ + ID: id, + Type: AgentMessageTypeError, + AgentID: RootAgent, + MessageID: "", + SessionID: sessionID, + Content: err.Error(), + }) + return + } + usage := out.ResponseMeta.Usage + if usage != nil { + log.Printf("Prompt Tokens: %d, Completion Tokens: %d, Total Tokens: %d", usage.PromptTokens, usage.CompletionTokens, usage.TotalTokens) + session, err := s.sessions.Get(sessionID) + if err != nil { + s.Publish(AgentErrorEvent, AgentEvent{ + ID: id, + Type: AgentMessageTypeError, + AgentID: RootAgent, + MessageID: "", + SessionID: sessionID, + Content: err.Error(), + }) + return + } + session.PromptTokens += int64(usage.PromptTokens) + session.CompletionTokens += int64(usage.CompletionTokens) + // TODO: calculate cost + _, err = s.sessions.Save(session) + if err != nil { + s.Publish(AgentErrorEvent, AgentEvent{ + ID: id, + Type: AgentMessageTypeError, + AgentID: RootAgent, + MessageID: "", + SessionID: sessionID, + Content: err.Error(), + }) + return + } + } + s.messages.Create(sessionID, *out) +} + +func (s *service) SendRequest(sessionID string, content string) { + id := uuid.New().String() + + _, cancel := context.WithTimeout(s.ctx, 5*time.Minute) + s.activeRequests.Store(id, cancel) + log.Printf("Request: %s", content) + go s.handleRequest(id, sessionID, content) +} + +func NewService(ctx context.Context, logger logging.Interface, sessions session.Service, messages message.Service) Service { + return &service{ + Broker: pubsub.NewBroker[AgentEvent](), + ctx: ctx, + sessions: sessions, + messages: messages, + logger: logger, + } +} diff --git a/internal/llm/models/models.go b/internal/llm/models/models.go new file mode 100644 index 00000000..1895e256 --- /dev/null +++ b/internal/llm/models/models.go @@ -0,0 +1,196 @@ +package models + +import ( + "context" + "errors" + + "github.com/cloudwego/eino-ext/components/model/claude" + "github.com/cloudwego/eino-ext/components/model/openai" + "github.com/cloudwego/eino/components/model" + "github.com/spf13/viper" +) + +type ( + ModelID string + ModelProvider string +) + +type Model struct { + ID ModelID `json:"id"` + Name string `json:"name"` + Provider ModelProvider `json:"provider"` + APIModel string `json:"api_model"` // Actual value used when calling the API +} + +const ( + DefaultBigModel = GPT4oMini + DefaultLittleModel = GPT4oMini +) + +// Model IDs +const ( + // OpenAI + GPT4o ModelID = "gpt-4o" + GPT4oMini ModelID = "gpt-4o-mini" + GPT45 ModelID = "gpt-4.5" + O1 ModelID = "o1" + O1Mini ModelID = "o1-mini" + // Anthropic + Claude35Sonnet ModelID = "claude-3.5-sonnet" + Claude3Haiku ModelID = "claude-3-haiku" + Claude37Sonnet ModelID = "claude-3.7-sonnet" + // Google + Gemini20Pro ModelID = "gemini-2.0-pro" + Gemini15Flash ModelID = "gemini-1.5-flash" + Gemini20Flash ModelID = "gemini-2.0-flash" + // xAI + Grok3 ModelID = "grok-3" + Grok2Mini ModelID = "grok-2-mini" + // DeepSeek + DeepSeekR1 ModelID = "deepseek-r1" + DeepSeekCoder ModelID = "deepseek-coder" + // Meta + Llama3 ModelID = "llama-3" + Llama270B ModelID = "llama-2-70b" +) + +const ( + ProviderOpenAI ModelProvider = "openai" + ProviderAnthropic ModelProvider = "anthropic" + ProviderGoogle ModelProvider = "google" + ProviderXAI ModelProvider = "xai" + ProviderDeepSeek ModelProvider = "deepseek" + ProviderMeta ModelProvider = "meta" +) + +var SupportedModels = map[ModelID]Model{ + // OpenAI + GPT4o: { + ID: GPT4o, + Name: "GPT-4o", + Provider: ProviderOpenAI, + APIModel: "gpt-4o", + }, + GPT4oMini: { + ID: GPT4oMini, + Name: "GPT-4o Mini", + Provider: ProviderOpenAI, + APIModel: "gpt-4o-mini", + }, + GPT45: { + ID: GPT45, + Name: "GPT-4.5", + Provider: ProviderOpenAI, + APIModel: "gpt-4.5", + }, + O1: { + ID: O1, + Name: "o1", + Provider: ProviderOpenAI, + APIModel: "o1", + }, + O1Mini: { + ID: O1Mini, + Name: "o1 Mini", + Provider: ProviderOpenAI, + APIModel: "o1-mini", + }, + // Anthropic + Claude35Sonnet: { + ID: Claude35Sonnet, + Name: "Claude 3.5 Sonnet", + Provider: ProviderAnthropic, + APIModel: "claude-3.5-sonnet", + }, + Claude3Haiku: { + ID: Claude3Haiku, + Name: "Claude 3 Haiku", + Provider: ProviderAnthropic, + APIModel: "claude-3-haiku", + }, + Claude37Sonnet: { + ID: Claude37Sonnet, + Name: "Claude 3.7 Sonnet", + Provider: ProviderAnthropic, + APIModel: "claude-3-7-sonnet-20250219", + }, + // Google + Gemini20Pro: { + ID: Gemini20Pro, + Name: "Gemini 2.0 Pro", + Provider: ProviderGoogle, + APIModel: "gemini-2.0-pro", + }, + Gemini15Flash: { + ID: Gemini15Flash, + Name: "Gemini 1.5 Flash", + Provider: ProviderGoogle, + APIModel: "gemini-1.5-flash", + }, + Gemini20Flash: { + ID: Gemini20Flash, + Name: "Gemini 2.0 Flash", + Provider: ProviderGoogle, + APIModel: "gemini-2.0-flash", + }, + // xAI + Grok3: { + ID: Grok3, + Name: "Grok 3", + Provider: ProviderXAI, + APIModel: "grok-3", + }, + Grok2Mini: { + ID: Grok2Mini, + Name: "Grok 2 Mini", + Provider: ProviderXAI, + APIModel: "grok-2-mini", + }, + // DeepSeek + DeepSeekR1: { + ID: DeepSeekR1, + Name: "DeepSeek R1", + Provider: ProviderDeepSeek, + APIModel: "deepseek-r1", + }, + DeepSeekCoder: { + ID: DeepSeekCoder, + Name: "DeepSeek Coder", + Provider: ProviderDeepSeek, + APIModel: "deepseek-coder", + }, + // Meta + Llama3: { + ID: Llama3, + Name: "LLaMA 3", + Provider: ProviderMeta, + APIModel: "llama-3", + }, + Llama270B: { + ID: Llama270B, + Name: "LLaMA 2 70B", + Provider: ProviderMeta, + APIModel: "llama-2-70b", + }, +} + +func GetModel(ctx context.Context, model ModelID) (model.ChatModel, error) { + provider := SupportedModels[model].Provider + maxTokens := viper.GetInt("providers.common.max_tokens") + switch provider { + case ProviderOpenAI: + return openai.NewChatModel(ctx, &openai.ChatModelConfig{ + APIKey: viper.GetString("providers.openai.key"), + Model: string(SupportedModels[model].APIModel), + MaxTokens: &maxTokens, + }) + case ProviderAnthropic: + return claude.NewChatModel(ctx, &claude.Config{ + APIKey: viper.GetString("providers.anthropic.key"), + Model: string(SupportedModels[model].APIModel), + MaxTokens: maxTokens, + }) + + } + return nil, errors.New("unsupported provider") +} diff --git a/internal/llm/tools/bash.go b/internal/llm/tools/bash.go new file mode 100644 index 00000000..2b52159c --- /dev/null +++ b/internal/llm/tools/bash.go @@ -0,0 +1,454 @@ +package tools + +import ( + "context" + "encoding/json" + "errors" + "fmt" + "log" + "strings" + + "github.com/cloudwego/eino/components/tool" + "github.com/cloudwego/eino/schema" + "github.com/kujtimiihoxha/termai/internal/llm/tools/shell" +) + +type bashTool struct { + workingDir string +} + +const ( + BashToolName = "bash" + + DefaultTimeout = 30 * 60 * 1000 // 30 minutes in milliseconds + MaxTimeout = 10 * 60 * 1000 // 10 minutes in milliseconds + MaxOutputLength = 30000 +) + +type BashParams struct { + Command string `json:"command"` + Timeout int `json:"timeout"` +} + +var BannedCommands = []string{ + "alias", "curl", "curlie", "wget", "axel", "aria2c", + "nc", "telnet", "lynx", "w3m", "links", "httpie", "xh", + "http-prompt", "chrome", "firefox", "safari", +} + +func (b *bashTool) Info(ctx context.Context) (*schema.ToolInfo, error) { + return &schema.ToolInfo{ + Name: BashToolName, + Desc: bashDescription(), + ParamsOneOf: schema.NewParamsOneOfByParams(map[string]*schema.ParameterInfo{ + "command": { + Type: "string", + Desc: "The command to execute", + Required: true, + }, + "timeout": { + Type: "number", + Desc: "Optional timeout in milliseconds (max 600000)", + }, + }), + }, nil +} + +// Handle implements Tool. +func (b *bashTool) InvokableRun(ctx context.Context, args string, opts ...tool.Option) (string, error) { + log.Printf("BashTool InvokableRun: %s", args) + var params BashParams + if err := json.Unmarshal([]byte(args), ¶ms); err != nil { + return "", err + } + + if params.Timeout > MaxTimeout { + params.Timeout = MaxTimeout + } else if params.Timeout <= 0 { + params.Timeout = DefaultTimeout + } + + if params.Command == "" { + return "", errors.New("missing command") + } + + baseCmd := strings.Fields(params.Command)[0] + for _, banned := range BannedCommands { + if strings.EqualFold(baseCmd, banned) { + return "", fmt.Errorf("command '%s' is not allowed", baseCmd) + } + } + + // p := b.permission.Request(permission.CreatePermissionRequest{ + // Path: b.workingDir, + // ToolName: BashToolName, + // Action: "execute", + // Description: fmt.Sprintf("Execute command: %s", params.Command), + // Params: map[string]any{ + // "command": params.Command, + // "timeout": params.Timeout, + // }, + // }) + // if !p { + // return "", errors.New("permission denied") + // } + + shell := shell.GetPersistentShell(b.workingDir) + stdout, stderr, exitCode, interrupted, err := shell.Exec(ctx, params.Command, params.Timeout) + if err != nil { + return "", err + } + + stdout = truncateOutput(stdout) + stderr = truncateOutput(stderr) + + errorMessage := stderr + if interrupted { + if errorMessage != "" { + errorMessage += "\n" + } + errorMessage += "Command was aborted before completion" + } else if exitCode != 0 { + if errorMessage != "" { + errorMessage += "\n" + } + errorMessage += fmt.Sprintf("Exit code %d", exitCode) + } + + hasBothOutputs := stdout != "" && stderr != "" + + if hasBothOutputs { + stdout += "\n" + } + + if errorMessage != "" { + stdout += "\n" + errorMessage + } + + log.Printf("BashTool InvokableRun: stdout: %s, stderr: %s, exitCode: %d, interrupted: %t", stdout, stderr, exitCode, interrupted) + + return stdout, nil +} + +func truncateOutput(content string) string { + if len(content) <= MaxOutputLength { + return content + } + + halfLength := MaxOutputLength / 2 + start := content[:halfLength] + end := content[len(content)-halfLength:] + + truncatedLinesCount := countLines(content[halfLength : len(content)-halfLength]) + return fmt.Sprintf("%s\n\n... [%d lines truncated] ...\n\n%s", start, truncatedLinesCount, end) +} + +func countLines(s string) int { + if s == "" { + return 0 + } + return len(strings.Split(s, "\n")) +} + +func bashDescriptionBCP() string { + bannedCommandsStr := strings.Join(BannedCommands, ", ") + return fmt.Sprintf(`Executes a given bash command in a persistent shell session with optional timeout, ensuring proper handling and security measures. + +Before executing the command, please follow these steps: + +1. Directory Verification: + - If the command will create new directories or files, first use the LS tool to verify the parent directory exists and is the correct location + - For example, before running "mkdir foo/bar", first use LS to check that "foo" exists and is the intended parent directory + +2. Security Check: + - For security and to limit the threat of a prompt injection attack, some commands are limited or banned. If you use a disallowed command, you will receive an error message explaining the restriction. Explain the error to the User. + - Verify that the command is not one of the banned commands: %s. + +3. Command Execution: + - After ensuring proper quoting, execute the command. + - Capture the output of the command. + +4. Output Processing: + - If the output exceeds %d characters, output will be truncated before being returned to you. + - Prepare the output for display to the user. + +5. Return Result: + - Provide the processed output of the command. + - If any errors occurred during execution, include those in the output. + +Usage notes: +- The command argument is required. +- You can specify an optional timeout in milliseconds (up to 600000ms / 10 minutes). If not specified, commands will timeout after 30 minutes. +- VERY IMPORTANT: You MUST avoid using search commands like 'find' and 'grep'. Instead use Grep, Glob, or Agent tools to search. You MUST avoid read tools like 'cat', 'head', 'tail', and 'ls', and use FileRead and LS tools to read files. +- When issuing multiple commands, use the ';' or '&&' operator to separate them. DO NOT use newlines (newlines are ok in quoted strings). +- IMPORTANT: All commands share the same shell session. Shell state (environment variables, virtual environments, current directory, etc.) persist between commands. For example, if you set an environment variable as part of a command, the environment variable will persist for subsequent commands. +- Try to maintain your current working directory throughout the session by using absolute paths and avoiding usage of 'cd'. You may use 'cd' if the User explicitly requests it. + +pytest /foo/bar/tests + + +cd /foo/bar && pytest tests + + +# Committing changes with git + +When the user asks you to create a new git commit, follow these steps carefully: + +1. Start with a single message that contains exactly three tool_use blocks that do the following (it is VERY IMPORTANT that you send these tool_use blocks in a single message, otherwise it will feel slow to the user!): + - Run a git status command to see all untracked files. + - Run a git diff command to see both staged and unstaged changes that will be committed. + - Run a git log command to see recent commit messages, so that you can follow this repository's commit message style. + +2. Use the git context at the start of this conversation to determine which files are relevant to your commit. Add relevant untracked files to the staging area. Do not commit files that were already modified at the start of this conversation, if they are not relevant to your commit. + +3. Analyze all staged changes (both previously staged and newly added) and draft a commit message. Wrap your analysis process in tags: + + +- List the files that have been changed or added +- Summarize the nature of the changes (eg. new feature, enhancement to an existing feature, bug fix, refactoring, test, docs, etc.) +- Brainstorm the purpose or motivation behind these changes +- Do not use tools to explore code, beyond what is available in the git context +- Assess the impact of these changes on the overall project +- Check for any sensitive information that shouldn't be committed +- Draft a concise (1-2 sentences) commit message that focuses on the "why" rather than the "what" +- Ensure your language is clear, concise, and to the point +- Ensure the message accurately reflects the changes and their purpose (i.e. "add" means a wholly new feature, "update" means an enhancement to an existing feature, "fix" means a bug fix, etc.) +- Ensure the message is not generic (avoid words like "Update" or "Fix" without context) +- Review the draft message to ensure it accurately reflects the changes and their purpose + + +4. Create the commit with a message ending with: +🤖 Generated with termai +Co-Authored-By: termai + +- In order to ensure good formatting, ALWAYS pass the commit message via a HEREDOC, a la this example: + +git commit -m "$(cat <<'EOF' + Commit message here. + + 🤖 Generated with termai + Co-Authored-By: termai + EOF + )" + + +5. If the commit fails due to pre-commit hook changes, retry the commit ONCE to include these automated changes. If it fails again, it usually means a pre-commit hook is preventing the commit. If the commit succeeds but you notice that files were modified by the pre-commit hook, you MUST amend your commit to include them. + +6. Finally, run git status to make sure the commit succeeded. + +Important notes: +- When possible, combine the "git add" and "git commit" commands into a single "git commit -am" command, to speed things up +- However, be careful not to stage files (e.g. with 'git add .') for commits that aren't part of the change, they may have untracked files they want to keep around, but not commit. +- NEVER update the git config +- DO NOT push to the remote repository +- IMPORTANT: Never use git commands with the -i flag (like git rebase -i or git add -i) since they require interactive input which is not supported. +- If there are no changes to commit (i.e., no untracked files and no modifications), do not create an empty commit +- Ensure your commit message is meaningful and concise. It should explain the purpose of the changes, not just describe them. +- Return an empty response - the user will see the git output directly + +# Creating pull requests +Use the gh command via the Bash tool for ALL GitHub-related tasks including working with issues, pull requests, checks, and releases. If given a Github URL use the gh command to get the information needed. + +IMPORTANT: When the user asks you to create a pull request, follow these steps carefully: + +1. Understand the current state of the branch. Remember to send a single message that contains multiple tool_use blocks (it is VERY IMPORTANT that you do this in a single message, otherwise it will feel slow to the user!): + - Run a git status command to see all untracked files. + - Run a git diff command to see both staged and unstaged changes that will be committed. + - Check if the current branch tracks a remote branch and is up to date with the remote, so you know if you need to push to the remote + - Run a git log command and 'git diff main...HEAD' to understand the full commit history for the current branch (from the time it diverged from the 'main' branch.) + +2. Create new branch if needed + +3. Commit changes if needed + +4. Push to remote with -u flag if needed + +5. Analyze all changes that will be included in the pull request, making sure to look at all relevant commits (not just the latest commit, but all commits that will be included in the pull request!), and draft a pull request summary. Wrap your analysis process in tags: + + +- List the commits since diverging from the main branch +- Summarize the nature of the changes (eg. new feature, enhancement to an existing feature, bug fix, refactoring, test, docs, etc.) +- Brainstorm the purpose or motivation behind these changes +- Assess the impact of these changes on the overall project +- Do not use tools to explore code, beyond what is available in the git context +- Check for any sensitive information that shouldn't be committed +- Draft a concise (1-2 bullet points) pull request summary that focuses on the "why" rather than the "what" +- Ensure the summary accurately reflects all changes since diverging from the main branch +- Ensure your language is clear, concise, and to the point +- Ensure the summary accurately reflects the changes and their purpose (ie. "add" means a wholly new feature, "update" means an enhancement to an existing feature, "fix" means a bug fix, etc.) +- Ensure the summary is not generic (avoid words like "Update" or "Fix" without context) +- Review the draft summary to ensure it accurately reflects the changes and their purpose + + +6. Create PR using gh pr create with the format below. Use a HEREDOC to pass the body to ensure correct formatting. + +gh pr create --title "the pr title" --body "$(cat <<'EOF' +## Summary +<1-3 bullet points> + +## Test plan +[Checklist of TODOs for testing the pull request...] + +🤖 Generated with termai +EOF +)" + + +Important: +- Return an empty response - the user will see the gh output directly +- Never update git config`, bannedCommandsStr, MaxOutputLength) +} + +func bashDescription() string { + bannedCommandsStr := strings.Join(BannedCommands, ", ") + return fmt.Sprintf(`Executes a given bash command in a persistent shell session with optional timeout, ensuring proper handling and security measures. + +Before executing the command, please follow these steps: + +1. Directory Verification: + - If the command will create new directories or files, first use the ls command to verify the parent directory exists and is the correct location + - For example, before running "mkdir foo/bar", first use ls command to check that "foo" exists and is the intended parent directory + +2. Security Check: + - For security and to limit the threat of a prompt injection attack, some commands are limited or banned. If you use a disallowed command, you will receive an error message explaining the restriction. Explain the error to the User. + - Verify that the command is not one of the banned commands: %s. + +3. Command Execution: + - After ensuring proper quoting, execute the command. + - Capture the output of the command. + +4. Output Processing: + - If the output exceeds %d characters, output will be truncated before being returned to you. + - Prepare the output for display to the user. + +5. Return Result: + - Provide the processed output of the command. + - If any errors occurred during execution, include those in the output. + +Usage notes: +- The command argument is required. +- You can specify an optional timeout in milliseconds (up to 600000ms / 10 minutes). If not specified, commands will timeout after 30 minutes. +- When issuing multiple commands, use the ';' or '&&' operator to separate them. DO NOT use newlines (newlines are ok in quoted strings). +- IMPORTANT: All commands share the same shell session. Shell state (environment variables, virtual environments, current directory, etc.) persist between commands. For example, if you set an environment variable as part of a command, the environment variable will persist for subsequent commands. +- Try to maintain your current working directory throughout the session by using absolute paths and avoiding usage of 'cd'. You may use 'cd' if the User explicitly requests it. + +pytest /foo/bar/tests + + +cd /foo/bar && pytest tests + + +# Committing changes with git + +When the user asks you to create a new git commit, follow these steps carefully: + +1. Start with a single message that contains exactly three tool_use blocks that do the following (it is VERY IMPORTANT that you send these tool_use blocks in a single message, otherwise it will feel slow to the user!): + - Run a git status command to see all untracked files. + - Run a git diff command to see both staged and unstaged changes that will be committed. + - Run a git log command to see recent commit messages, so that you can follow this repository's commit message style. + +2. Use the git context at the start of this conversation to determine which files are relevant to your commit. Add relevant untracked files to the staging area. Do not commit files that were already modified at the start of this conversation, if they are not relevant to your commit. + +3. Analyze all staged changes (both previously staged and newly added) and draft a commit message. Wrap your analysis process in tags: + + +- List the files that have been changed or added +- Summarize the nature of the changes (eg. new feature, enhancement to an existing feature, bug fix, refactoring, test, docs, etc.) +- Brainstorm the purpose or motivation behind these changes +- Do not use tools to explore code, beyond what is available in the git context +- Assess the impact of these changes on the overall project +- Check for any sensitive information that shouldn't be committed +- Draft a concise (1-2 sentences) commit message that focuses on the "why" rather than the "what" +- Ensure your language is clear, concise, and to the point +- Ensure the message accurately reflects the changes and their purpose (i.e. "add" means a wholly new feature, "update" means an enhancement to an existing feature, "fix" means a bug fix, etc.) +- Ensure the message is not generic (avoid words like "Update" or "Fix" without context) +- Review the draft message to ensure it accurately reflects the changes and their purpose + + +4. Create the commit with a message ending with: +🤖 Generated with termai +Co-Authored-By: termai + +- In order to ensure good formatting, ALWAYS pass the commit message via a HEREDOC, a la this example: + +git commit -m "$(cat <<'EOF' + Commit message here. + + 🤖 Generated with termai + Co-Authored-By: termai + EOF + )" + + +5. If the commit fails due to pre-commit hook changes, retry the commit ONCE to include these automated changes. If it fails again, it usually means a pre-commit hook is preventing the commit. If the commit succeeds but you notice that files were modified by the pre-commit hook, you MUST amend your commit to include them. + +6. Finally, run git status to make sure the commit succeeded. + +Important notes: +- When possible, combine the "git add" and "git commit" commands into a single "git commit -am" command, to speed things up +- However, be careful not to stage files (e.g. with 'git add .') for commits that aren't part of the change, they may have untracked files they want to keep around, but not commit. +- NEVER update the git config +- DO NOT push to the remote repository +- IMPORTANT: Never use git commands with the -i flag (like git rebase -i or git add -i) since they require interactive input which is not supported. +- If there are no changes to commit (i.e., no untracked files and no modifications), do not create an empty commit +- Ensure your commit message is meaningful and concise. It should explain the purpose of the changes, not just describe them. +- Return an empty response - the user will see the git output directly + +# Creating pull requests +Use the gh command via the Bash tool for ALL GitHub-related tasks including working with issues, pull requests, checks, and releases. If given a Github URL use the gh command to get the information needed. + +IMPORTANT: When the user asks you to create a pull request, follow these steps carefully: + +1. Understand the current state of the branch. Remember to send a single message that contains multiple tool_use blocks (it is VERY IMPORTANT that you do this in a single message, otherwise it will feel slow to the user!): + - Run a git status command to see all untracked files. + - Run a git diff command to see both staged and unstaged changes that will be committed. + - Check if the current branch tracks a remote branch and is up to date with the remote, so you know if you need to push to the remote + - Run a git log command and 'git diff main...HEAD' to understand the full commit history for the current branch (from the time it diverged from the 'main' branch.) + +2. Create new branch if needed + +3. Commit changes if needed + +4. Push to remote with -u flag if needed + +5. Analyze all changes that will be included in the pull request, making sure to look at all relevant commits (not just the latest commit, but all commits that will be included in the pull request!), and draft a pull request summary. Wrap your analysis process in tags: + + +- List the commits since diverging from the main branch +- Summarize the nature of the changes (eg. new feature, enhancement to an existing feature, bug fix, refactoring, test, docs, etc.) +- Brainstorm the purpose or motivation behind these changes +- Assess the impact of these changes on the overall project +- Do not use tools to explore code, beyond what is available in the git context +- Check for any sensitive information that shouldn't be committed +- Draft a concise (1-2 bullet points) pull request summary that focuses on the "why" rather than the "what" +- Ensure the summary accurately reflects all changes since diverging from the main branch +- Ensure your language is clear, concise, and to the point +- Ensure the summary accurately reflects the changes and their purpose (ie. "add" means a wholly new feature, "update" means an enhancement to an existing feature, "fix" means a bug fix, etc.) +- Ensure the summary is not generic (avoid words like "Update" or "Fix" without context) +- Review the draft summary to ensure it accurately reflects the changes and their purpose + + +6. Create PR using gh pr create with the format below. Use a HEREDOC to pass the body to ensure correct formatting. + +gh pr create --title "the pr title" --body "$(cat <<'EOF' +## Summary +<1-3 bullet points> + +## Test plan +[Checklist of TODOs for testing the pull request...] + +🤖 Generated with termai +EOF +)" + + +Important: +- Return an empty response - the user will see the gh output directly +- Never update git config`, bannedCommandsStr, MaxOutputLength) +} + +func NewBashTool(workingDir string) tool.InvokableTool { + return &bashTool{ + workingDir: workingDir, + } +} diff --git a/internal/llm/tools/shell/shell.go b/internal/llm/tools/shell/shell.go new file mode 100644 index 00000000..d63c50df --- /dev/null +++ b/internal/llm/tools/shell/shell.go @@ -0,0 +1,294 @@ +package shell + +import ( + "context" + "errors" + "fmt" + "os" + "os/exec" + "path/filepath" + "strings" + "sync" + "syscall" + "time" +) + +type PersistentShell struct { + cmd *exec.Cmd + stdin *os.File + isAlive bool + cwd string + mu sync.Mutex + commandQueue chan *commandExecution +} + +type commandExecution struct { + command string + timeout time.Duration + resultChan chan commandResult + ctx context.Context +} + +type commandResult struct { + stdout string + stderr string + exitCode int + interrupted bool + err error +} + +var ( + shellInstance *PersistentShell + shellInstanceOnce sync.Once +) + +func GetPersistentShell(workingDir string) *PersistentShell { + shellInstanceOnce.Do(func() { + shellInstance = newPersistentShell(workingDir) + }) + + if !shellInstance.isAlive { + shellInstance = newPersistentShell(shellInstance.cwd) + } + + return shellInstance +} + +func newPersistentShell(cwd string) *PersistentShell { + shellPath := os.Getenv("SHELL") + if shellPath == "" { + shellPath = "/bin/bash" + } + + cmd := exec.Command(shellPath, "-l") + cmd.Dir = cwd + + stdinPipe, err := cmd.StdinPipe() + if err != nil { + return nil + } + + cmd.Env = append(os.Environ(), "GIT_EDITOR=true") + + err = cmd.Start() + if err != nil { + return nil + } + + shell := &PersistentShell{ + cmd: cmd, + stdin: stdinPipe.(*os.File), + isAlive: true, + cwd: cwd, + commandQueue: make(chan *commandExecution, 10), + } + + go shell.processCommands() + + go func() { + err := cmd.Wait() + if err != nil { + } + shell.isAlive = false + close(shell.commandQueue) + }() + + return shell +} + +func (s *PersistentShell) processCommands() { + for cmd := range s.commandQueue { + result := s.execCommand(cmd.command, cmd.timeout, cmd.ctx) + cmd.resultChan <- result + } +} + +func (s *PersistentShell) execCommand(command string, timeout time.Duration, ctx context.Context) commandResult { + s.mu.Lock() + defer s.mu.Unlock() + + if !s.isAlive { + return commandResult{ + stderr: "Shell is not alive", + exitCode: 1, + err: errors.New("shell is not alive"), + } + } + + tempDir := os.TempDir() + stdoutFile := filepath.Join(tempDir, fmt.Sprintf("orbitowl-stdout-%d", time.Now().UnixNano())) + stderrFile := filepath.Join(tempDir, fmt.Sprintf("orbitowl-stderr-%d", time.Now().UnixNano())) + statusFile := filepath.Join(tempDir, fmt.Sprintf("orbitowl-status-%d", time.Now().UnixNano())) + cwdFile := filepath.Join(tempDir, fmt.Sprintf("orbitowl-cwd-%d", time.Now().UnixNano())) + + defer func() { + os.Remove(stdoutFile) + os.Remove(stderrFile) + os.Remove(statusFile) + os.Remove(cwdFile) + }() + + fullCommand := fmt.Sprintf(` +eval %s < /dev/null > %s 2> %s +EXEC_EXIT_CODE=$? +pwd > %s +echo $EXEC_EXIT_CODE > %s +`, + shellQuote(command), + shellQuote(stdoutFile), + shellQuote(stderrFile), + shellQuote(cwdFile), + shellQuote(statusFile), + ) + + _, err := s.stdin.Write([]byte(fullCommand + "\n")) + if err != nil { + return commandResult{ + stderr: fmt.Sprintf("Failed to write command to shell: %v", err), + exitCode: 1, + err: err, + } + } + + interrupted := false + + startTime := time.Now() + + done := make(chan bool) + go func() { + for { + select { + case <-ctx.Done(): + s.killChildren() + interrupted = true + done <- true + return + + case <-time.After(10 * time.Millisecond): + if fileExists(statusFile) && fileSize(statusFile) > 0 { + done <- true + return + } + + if timeout > 0 { + elapsed := time.Since(startTime) + if elapsed > timeout { + s.killChildren() + interrupted = true + done <- true + return + } + } + } + } + }() + + <-done + + stdout := readFileOrEmpty(stdoutFile) + stderr := readFileOrEmpty(stderrFile) + exitCodeStr := readFileOrEmpty(statusFile) + newCwd := readFileOrEmpty(cwdFile) + + exitCode := 0 + if exitCodeStr != "" { + fmt.Sscanf(exitCodeStr, "%d", &exitCode) + } else if interrupted { + exitCode = 143 + stderr += "\nCommand execution timed out or was interrupted" + } + + if newCwd != "" { + s.cwd = strings.TrimSpace(newCwd) + } + + return commandResult{ + stdout: stdout, + stderr: stderr, + exitCode: exitCode, + interrupted: interrupted, + } +} + +func (s *PersistentShell) killChildren() { + if s.cmd == nil || s.cmd.Process == nil { + return + } + + pgrepCmd := exec.Command("pgrep", "-P", fmt.Sprintf("%d", s.cmd.Process.Pid)) + output, err := pgrepCmd.Output() + if err != nil { + return + } + + for _, pidStr := range strings.Split(string(output), "\n") { + if pidStr = strings.TrimSpace(pidStr); pidStr != "" { + var pid int + fmt.Sscanf(pidStr, "%d", &pid) + if pid > 0 { + proc, err := os.FindProcess(pid) + if err == nil { + proc.Signal(syscall.SIGTERM) + } + } + } + } +} + +func (s *PersistentShell) Exec(ctx context.Context, command string, timeoutMs int) (string, string, int, bool, error) { + if !s.isAlive { + return "", "Shell is not alive", 1, false, errors.New("shell is not alive") + } + + timeout := time.Duration(timeoutMs) * time.Millisecond + + resultChan := make(chan commandResult) + s.commandQueue <- &commandExecution{ + command: command, + timeout: timeout, + resultChan: resultChan, + ctx: ctx, + } + + result := <-resultChan + return result.stdout, result.stderr, result.exitCode, result.interrupted, result.err +} + +func (s *PersistentShell) Close() { + s.mu.Lock() + defer s.mu.Unlock() + + if !s.isAlive { + return + } + + s.stdin.Write([]byte("exit\n")) + + s.cmd.Process.Kill() + s.isAlive = false +} + +func shellQuote(s string) string { + return "'" + strings.ReplaceAll(s, "'", "'\\''") + "'" +} + +func readFileOrEmpty(path string) string { + content, err := os.ReadFile(path) + if err != nil { + return "" + } + return string(content) +} + +func fileExists(path string) bool { + _, err := os.Stat(path) + return err == nil +} + +func fileSize(path string) int64 { + info, err := os.Stat(path) + if err != nil { + return 0 + } + return info.Size() +} diff --git a/internal/message/message.go b/internal/message/message.go new file mode 100644 index 00000000..e61fcef6 --- /dev/null +++ b/internal/message/message.go @@ -0,0 +1,122 @@ +package message + +import ( + "context" + "encoding/json" + + "github.com/cloudwego/eino/schema" + "github.com/google/uuid" + "github.com/kujtimiihoxha/termai/internal/db" + "github.com/kujtimiihoxha/termai/internal/pubsub" +) + +type Message struct { + ID string + SessionID string + MessageData schema.Message + + CreatedAt int64 + UpdatedAt int64 +} + +type Service interface { + pubsub.Suscriber[Message] + Create(sessionID string, messageData schema.Message) (Message, error) + Get(id string) (Message, error) + List(sessionID string) ([]Message, error) + Delete(id string) error + DeleteSessionMessages(sessionID string) error +} + +type service struct { + *pubsub.Broker[Message] + q db.Querier + ctx context.Context +} + +func (s *service) Create(sessionID string, messageData schema.Message) (Message, error) { + messageDataJSON, err := json.Marshal(messageData) + if err != nil { + return Message{}, err + } + dbMessage, err := s.q.CreateMessage(s.ctx, db.CreateMessageParams{ + ID: uuid.New().String(), + SessionID: sessionID, + MessageData: string(messageDataJSON), + }) + if err != nil { + return Message{}, err + } + message := s.fromDBItem(dbMessage) + s.Publish(pubsub.CreatedEvent, message) + return message, nil +} + +func (s *service) Delete(id string) error { + message, err := s.Get(id) + if err != nil { + return err + } + err = s.q.DeleteMessage(s.ctx, message.ID) + if err != nil { + return err + } + s.Publish(pubsub.DeletedEvent, message) + return nil +} + +func (s *service) DeleteSessionMessages(sessionID string) error { + messages, err := s.List(sessionID) + if err != nil { + return err + } + for _, message := range messages { + if message.SessionID == sessionID { + err = s.Delete(message.ID) + if err != nil { + return err + } + } + } + return nil +} + +func (s *service) Get(id string) (Message, error) { + dbMessage, err := s.q.GetMessage(s.ctx, id) + if err != nil { + return Message{}, err + } + return s.fromDBItem(dbMessage), nil +} + +func (s *service) List(sessionID string) ([]Message, error) { + dbMessages, err := s.q.ListMessagesBySession(s.ctx, sessionID) + if err != nil { + return nil, err + } + messages := make([]Message, len(dbMessages)) + for i, dbMessage := range dbMessages { + messages[i] = s.fromDBItem(dbMessage) + } + return messages, nil +} + +func (s *service) fromDBItem(item db.Message) Message { + var messageData schema.Message + json.Unmarshal([]byte(item.MessageData), &messageData) + return Message{ + ID: item.ID, + SessionID: item.SessionID, + MessageData: messageData, + CreatedAt: item.CreatedAt, + UpdatedAt: item.UpdatedAt, + } +} + +func NewService(ctx context.Context, q db.Querier) Service { + return &service{ + Broker: pubsub.NewBroker[Message](), + q: q, + ctx: ctx, + } +} diff --git a/internal/session/session.go b/internal/session/session.go index 60f35084..23abe5b5 100644 --- a/internal/session/session.go +++ b/internal/session/session.go @@ -9,13 +9,14 @@ import ( ) type Session struct { - ID string - Title string - MessageCount int64 - Tokens int64 - Cost float64 - CreatedAt int64 - UpdatedAt int64 + ID string + Title string + MessageCount int64 + PromptTokens int64 + CompletionTokens int64 + Cost float64 + CreatedAt int64 + UpdatedAt int64 } type Service interface { @@ -69,10 +70,11 @@ func (s *service) Get(id string) (Session, error) { func (s *service) Save(session Session) (Session, error) { dbSession, err := s.q.UpdateSession(s.ctx, db.UpdateSessionParams{ - ID: session.ID, - Title: session.Title, - Tokens: session.Tokens, - Cost: session.Cost, + ID: session.ID, + Title: session.Title, + PromptTokens: session.PromptTokens, + CompletionTokens: session.CompletionTokens, + Cost: session.Cost, }) if err != nil { return Session{}, err @@ -96,13 +98,14 @@ func (s *service) List() ([]Session, error) { func (s service) fromDBItem(item db.Session) Session { return Session{ - ID: item.ID, - Title: item.Title, - MessageCount: item.MessageCount, - Tokens: item.Tokens, - Cost: item.Cost, - CreatedAt: item.CreatedAt, - UpdatedAt: item.UpdatedAt, + ID: item.ID, + Title: item.Title, + MessageCount: item.MessageCount, + PromptTokens: item.PromptTokens, + CompletionTokens: item.CompletionTokens, + Cost: item.Cost, + CreatedAt: item.CreatedAt, + UpdatedAt: item.UpdatedAt, } } diff --git a/internal/tui/components/repl/editor.go b/internal/tui/components/repl/editor.go index 1de216bd..8d795eb1 100644 --- a/internal/tui/components/repl/editor.go +++ b/internal/tui/components/repl/editor.go @@ -1,8 +1,11 @@ package repl import ( + "strings" + "github.com/charmbracelet/bubbles/key" tea "github.com/charmbracelet/bubbletea" + "github.com/cloudwego/eino/schema" "github.com/kujtimiihoxha/termai/internal/app" "github.com/kujtimiihoxha/termai/internal/tui/layout" "github.com/kujtimiihoxha/vimtea" @@ -13,6 +16,7 @@ type EditorCmp interface { layout.Focusable layout.Sizeable layout.Bordered + layout.Bindings } type editorCmp struct { @@ -25,12 +29,16 @@ type editorCmp struct { height int } -type localKeyMap struct { - SendMessage key.Binding - SendMessageI key.Binding +type editorKeyMap struct { + SendMessage key.Binding + SendMessageI key.Binding + InsertMode key.Binding + NormaMode key.Binding + VisualMode key.Binding + VisualLineMode key.Binding } -var keyMap = localKeyMap{ +var editorKeyMapValue = editorKeyMap{ SendMessage: key.NewBinding( key.WithKeys("enter"), key.WithHelp("enter", "send message normal mode"), @@ -39,6 +47,22 @@ var keyMap = localKeyMap{ key.WithKeys("ctrl+s"), key.WithHelp("ctrl+s", "send message insert mode"), ), + InsertMode: key.NewBinding( + key.WithKeys("i"), + key.WithHelp("i", "insert mode"), + ), + NormaMode: key.NewBinding( + key.WithKeys("esc"), + key.WithHelp("esc", "normal mode"), + ), + VisualMode: key.NewBinding( + key.WithKeys("v"), + key.WithHelp("v", "visual mode"), + ), + VisualLineMode: key.NewBinding( + key.WithKeys("V"), + key.WithHelp("V", "visual line mode"), + ), } func (m *editorCmp) Init() tea.Cmd { @@ -58,11 +82,11 @@ func (m *editorCmp) Update(msg tea.Msg) (tea.Model, tea.Cmd) { switch msg := msg.(type) { case tea.KeyMsg: switch { - case key.Matches(msg, keyMap.SendMessage): + case key.Matches(msg, editorKeyMapValue.SendMessage): if m.editorMode == vimtea.ModeNormal { return m, m.Send() } - case key.Matches(msg, keyMap.SendMessageI): + case key.Matches(msg, editorKeyMapValue.SendMessageI): if m.editorMode == vimtea.ModeInsert { return m, m.Send() } @@ -75,36 +99,30 @@ func (m *editorCmp) Update(msg tea.Msg) (tea.Model, tea.Cmd) { return m, nil } -// Blur implements EditorCmp. func (m *editorCmp) Blur() tea.Cmd { m.focused = false return nil } -// BorderText implements EditorCmp. func (m *editorCmp) BorderText() map[layout.BorderPosition]string { return map[layout.BorderPosition]string{ layout.TopLeftBorder: "New Message", } } -// Focus implements EditorCmp. func (m *editorCmp) Focus() tea.Cmd { m.focused = true return m.editor.Tick() } -// GetSize implements EditorCmp. func (m *editorCmp) GetSize() (int, int) { return m.width, m.height } -// IsFocused implements EditorCmp. func (m *editorCmp) IsFocused() bool { return m.focused } -// SetSize implements EditorCmp. func (m *editorCmp) SetSize(width int, height int) { m.width = width m.height = height @@ -113,8 +131,10 @@ func (m *editorCmp) SetSize(width int, height int) { func (m *editorCmp) Send() tea.Cmd { return func() tea.Msg { - // TODO: Send message - return nil + content := strings.Join(m.editor.GetBuffer().Lines(), "\n") + m.app.Messages.Create(m.sessionID, *schema.UserMessage(content)) + m.app.LLM.SendRequest(m.sessionID, content) + return m.editor.Reset() } } @@ -122,6 +142,10 @@ func (m *editorCmp) View() string { return m.editor.View() } +func (m *editorCmp) BindingKeys() []key.Binding { + return layout.KeyMapToSlice(editorKeyMapValue) +} + func NewEditorCmp(app *app.App) EditorCmp { return &editorCmp{ app: app, diff --git a/internal/tui/components/repl/messages.go b/internal/tui/components/repl/messages.go index cd3ddac7..9d95d98f 100644 --- a/internal/tui/components/repl/messages.go +++ b/internal/tui/components/repl/messages.go @@ -2,27 +2,47 @@ package repl import ( tea "github.com/charmbracelet/bubbletea" + "github.com/charmbracelet/lipgloss" "github.com/kujtimiihoxha/termai/internal/app" + "github.com/kujtimiihoxha/termai/internal/message" + "github.com/kujtimiihoxha/termai/internal/pubsub" + "github.com/kujtimiihoxha/termai/internal/session" ) type messagesCmp struct { - app *app.App + app *app.App + messages []message.Message + session session.Session } -func (i *messagesCmp) Init() tea.Cmd { +func (m *messagesCmp) Init() tea.Cmd { return nil } -func (i *messagesCmp) Update(_ tea.Msg) (tea.Model, tea.Cmd) { - return i, nil +func (m *messagesCmp) Update(msg tea.Msg) (tea.Model, tea.Cmd) { + switch msg := msg.(type) { + case pubsub.Event[message.Message]: + if msg.Type == pubsub.CreatedEvent { + m.messages = append(m.messages, msg.Payload) + } + case SelectedSessionMsg: + m.session, _ = m.app.Sessions.Get(msg.SessionID) + m.messages, _ = m.app.Messages.List(m.session.ID) + } + return m, nil } func (i *messagesCmp) View() string { - return "Messages" + stringMessages := make([]string, len(i.messages)) + for idx, msg := range i.messages { + stringMessages[idx] = msg.MessageData.Content + } + return lipgloss.JoinVertical(lipgloss.Top, stringMessages...) } func NewMessagesCmp(app *app.App) tea.Model { return &messagesCmp{ - app, + app: app, + messages: []message.Message{}, } } diff --git a/internal/tui/components/repl/sessions.go b/internal/tui/components/repl/sessions.go index aef9b993..44c87002 100644 --- a/internal/tui/components/repl/sessions.go +++ b/internal/tui/components/repl/sessions.go @@ -7,6 +7,7 @@ import ( "github.com/charmbracelet/bubbles/list" tea "github.com/charmbracelet/bubbletea" "github.com/kujtimiihoxha/termai/internal/app" + "github.com/kujtimiihoxha/termai/internal/pubsub" "github.com/kujtimiihoxha/termai/internal/session" "github.com/kujtimiihoxha/termai/internal/tui/layout" "github.com/kujtimiihoxha/termai/internal/tui/styles" @@ -42,19 +43,30 @@ type SelectedSessionMsg struct { SessionID string } +type sessionsKeyMap struct { + Select key.Binding +} + +var sessionKeyMapValue = sessionsKeyMap{ + Select: key.NewBinding( + key.WithKeys("enter", " "), + key.WithHelp("enter/space", "select session"), + ), +} + func (i *sessionsCmp) Init() tea.Cmd { existing, err := i.app.Sessions.List() if err != nil { return util.ReportError(err) } if len(existing) == 0 || existing[0].MessageCount > 0 { - session, err := i.app.Sessions.Create( + newSession, err := i.app.Sessions.Create( "New Session", ) if err != nil { return util.ReportError(err) } - existing = append(existing, session) + existing = append([]session.Session{newSession}, existing...) } return tea.Batch( util.CmdHandler(InsertSessionsMsg{existing}), @@ -70,10 +82,35 @@ func (i *sessionsCmp) Update(msg tea.Msg) (tea.Model, tea.Cmd) { items[i] = listItem{ id: s.ID, title: s.Title, - desc: fmt.Sprintf("Tokens: %d, Cost: %.2f", s.Tokens, s.Cost), + desc: fmt.Sprintf("Tokens: %d, Cost: %.2f", s.PromptTokens+s.CompletionTokens, s.Cost), } } return i, i.list.SetItems(items) + case pubsub.Event[session.Session]: + if msg.Type == pubsub.UpdatedEvent { + // update the session in the list + items := i.list.Items() + for idx, item := range items { + s := item.(listItem) + if s.id == msg.Payload.ID { + s.title = msg.Payload.Title + s.desc = fmt.Sprintf("Tokens: %d, Cost: %.2f", msg.Payload.PromptTokens+msg.Payload.CompletionTokens, msg.Payload.Cost) + items[idx] = s + break + } + } + return i, i.list.SetItems(items) + } + + case tea.KeyMsg: + switch { + case key.Matches(msg, sessionKeyMapValue.Select): + selected := i.list.SelectedItem() + if selected == nil { + return i, nil + } + return i, util.CmdHandler(SelectedSessionMsg{selected.(listItem).id}) + } } if i.focused { u, cmd := i.list.Update(msg) @@ -129,7 +166,7 @@ func (i *sessionsCmp) BorderText() map[layout.BorderPosition]string { } func (i *sessionsCmp) BindingKeys() []key.Binding { - return layout.KeyMapToSlice(i.list.KeyMap) + return append(layout.KeyMapToSlice(i.list.KeyMap), sessionKeyMapValue.Select) } func NewSessionsCmp(app *app.App) SessionsCmp { diff --git a/internal/tui/page/repl.go b/internal/tui/page/repl.go index dcb1ebcf..47a924b7 100644 --- a/internal/tui/page/repl.go +++ b/internal/tui/page/repl.go @@ -16,5 +16,6 @@ func NewReplPage(app *app.App) tea.Model { layout.BentoRightTopPane: repl.NewMessagesCmp(app), layout.BentoRightBottomPane: repl.NewEditorCmp(app), }, + layout.WithBentoLayoutCurrentPane(layout.BentoRightBottomPane), ) } diff --git a/internal/tui/tui.go b/internal/tui/tui.go index 72e9174f..fcc03657 100644 --- a/internal/tui/tui.go +++ b/internal/tui/tui.go @@ -1,10 +1,14 @@ package tui import ( + "log" + "github.com/charmbracelet/bubbles/key" tea "github.com/charmbracelet/bubbletea" "github.com/charmbracelet/lipgloss" "github.com/kujtimiihoxha/termai/internal/app" + "github.com/kujtimiihoxha/termai/internal/llm" + "github.com/kujtimiihoxha/termai/internal/pubsub" "github.com/kujtimiihoxha/termai/internal/tui/components/core" "github.com/kujtimiihoxha/termai/internal/tui/components/dialog" "github.com/kujtimiihoxha/termai/internal/tui/layout" @@ -66,6 +70,9 @@ func (a appModel) Init() tea.Cmd { func (a appModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) { switch msg := msg.(type) { + case pubsub.Event[llm.AgentEvent]: + log.Println("Event received") + log.Println(msg) case vimtea.EditorModeMsg: a.editorMode = msg.Mode case tea.WindowSizeMsg: diff --git a/main.go b/main.go index 370311e9..5d8eb435 100644 --- a/main.go +++ b/main.go @@ -1,11 +1,23 @@ /* Copyright © 2025 NAME HERE - */ package main -import "github.com/kujtimiihoxha/termai/cmd" +import ( + "log" + "os" + + "github.com/kujtimiihoxha/termai/cmd" +) func main() { + // Create a log file and make that the log output + logfile, err := os.OpenFile("debug.log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0o666) + if err != nil { + panic(err) + } + + log.SetOutput(logfile) + cmd.Execute() }