From 14d706f94a45ab43c51f9c03abea7b3588c571c7 Mon Sep 17 00:00:00 2001 From: Grail Finder Date: Wed, 27 Nov 2024 20:16:58 +0300 Subject: Feat: add toml config --- bot.go | 67 +++++++++++++++++++++++++---------------------------- config.example.toml | 9 +++++++ config/config.go | 37 +++++++++++++++++++++++++++++ go.mod | 1 + go.sum | 2 ++ tools.go | 6 ++--- tui.go | 24 +++++++++---------- 7 files changed, 95 insertions(+), 51 deletions(-) create mode 100644 config.example.toml create mode 100644 config/config.go diff --git a/bot.go b/bot.go index 40706c3..6820b64 100644 --- a/bot.go +++ b/bot.go @@ -3,6 +3,7 @@ package main import ( "bufio" "bytes" + "elefant/config" "elefant/models" "elefant/storage" "encoding/json" @@ -22,27 +23,16 @@ var httpClient = http.Client{ } var ( - logger *slog.Logger - userRole = "user" - assistantRole = "assistant" - toolRole = "tool" - assistantIcon = "<🤖>: " - userIcon = ": " - // TODO: pass as an cli arg or have config - APIURL = "http://localhost:8080/v1/chat/completions" - logFileName = "log.txt" - showSystemMsgs = true - chunkLimit = 1000 - activeChatName string - chunkChan = make(chan string, 10) - streamDone = make(chan bool, 1) - chatBody *models.ChatBody - store storage.FullRepo - defaultFirstMsg = "Hello! What can I do for you?" - defaultStarter = []models.MessagesStory{ - {Role: "system", Content: systemMsg}, - {Role: assistantRole, Content: defaultFirstMsg}, - } + cfg *config.Config + logger *slog.Logger + chunkLimit = 1000 + activeChatName string + chunkChan = make(chan string, 10) + streamDone = make(chan bool, 1) + chatBody *models.ChatBody + store storage.FullRepo + defaultFirstMsg = "Hello! What can I do for you?" + defaultStarter = []models.MessagesStory{} defaultStarterBytes = []byte{} interruptResp = false ) @@ -64,14 +54,14 @@ func formMsg(chatBody *models.ChatBody, newMsg, role string) io.Reader { // func sendMsgToLLM(body io.Reader) (*models.LLMRespChunk, error) { func sendMsgToLLM(body io.Reader) (any, error) { - resp, err := httpClient.Post(APIURL, "application/json", body) + resp, err := httpClient.Post(cfg.APIURL, "application/json", body) if err != nil { logger.Error("llamacpp api", "error", err) return nil, err } defer resp.Body.Close() llmResp := []models.LLMRespChunk{} - // chunkChan <- assistantIcon + // chunkChan <- cfg.AssistantIcon reader := bufio.NewReader(resp.Body) counter := 0 for { @@ -128,7 +118,7 @@ func chatRound(userMsg, role string, tv *tview.TextView) { go sendMsgToLLM(reader) if userMsg != "" { // no need to write assistant icon since we continue old message fmt.Fprintf(tv, fmt.Sprintf("(%d) ", len(chatBody.Messages))) - fmt.Fprintf(tv, assistantIcon) + fmt.Fprintf(tv, cfg.AssistantIcon) } respText := strings.Builder{} out: @@ -145,7 +135,7 @@ out: } botRespMode = false chatBody.Messages = append(chatBody.Messages, models.MessagesStory{ - Role: assistantRole, Content: respText.String(), + Role: cfg.AssistantRole, Content: respText.String(), }) // bot msg is done; // now check it for func call @@ -174,18 +164,18 @@ func findCall(msg string, tv *tview.TextView) { f, ok := fnMap[fc.Name] if !ok { m := fmt.Sprintf("%s is not implemented", fc.Name) - chatRound(m, toolRole, tv) + chatRound(m, cfg.ToolRole, tv) return } resp := f(fc.Args...) toolMsg := fmt.Sprintf("tool response: %+v", string(resp)) - chatRound(toolMsg, toolRole, tv) + chatRound(toolMsg, cfg.ToolRole, tv) } func chatToTextSlice(showSys bool) []string { resp := make([]string, len(chatBody.Messages)) for i, msg := range chatBody.Messages { - if !showSys && (msg.Role != assistantRole && msg.Role != userRole) { + if !showSys && (msg.Role != cfg.AssistantRole && msg.Role != cfg.UserRole) { continue } resp[i] = msg.ToText(i) @@ -201,14 +191,14 @@ func chatToText(showSys bool) string { func textToMsg(rawMsg string) models.MessagesStory { msg := models.MessagesStory{} // system and tool? - if strings.HasPrefix(rawMsg, assistantIcon) { - msg.Role = assistantRole - msg.Content = strings.TrimPrefix(rawMsg, assistantIcon) + if strings.HasPrefix(rawMsg, cfg.AssistantIcon) { + msg.Role = cfg.AssistantRole + msg.Content = strings.TrimPrefix(rawMsg, cfg.AssistantIcon) return msg } - if strings.HasPrefix(rawMsg, userIcon) { - msg.Role = userRole - msg.Content = strings.TrimPrefix(rawMsg, userIcon) + if strings.HasPrefix(rawMsg, cfg.UserIcon) { + msg.Role = cfg.UserRole + msg.Content = strings.TrimPrefix(rawMsg, cfg.UserIcon) return msg } return msg @@ -224,9 +214,14 @@ func textSliceToChat(chat []string) []models.MessagesStory { } func init() { - file, err := os.OpenFile(logFileName, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644) + cfg = config.LoadConfigOrDefault("config.example.toml") + defaultStarter = []models.MessagesStory{ + {Role: "system", Content: systemMsg}, + {Role: cfg.AssistantRole, Content: defaultFirstMsg}, + } + file, err := os.OpenFile(cfg.LogFile, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644) if err != nil { - logger.Error("failed to open log file", "error", err, "filename", logFileName) + logger.Error("failed to open log file", "error", err, "filename", cfg.LogFile) return } defaultStarterBytes, err = json.Marshal(defaultStarter) diff --git a/config.example.toml b/config.example.toml new file mode 100644 index 0000000..d1388e5 --- /dev/null +++ b/config.example.toml @@ -0,0 +1,9 @@ +APIURL = "http://localhost:8080/v1/chat/completions" +ShowSys = true +LogFile = "log.txt" +UserRole = "user" +ToolRole = "tool" +AssistantRole = "assistant" +AssistantIcon = "<🤖>: " +UserIcon = ": " +ToolIcon = "<>>: " diff --git a/config/config.go b/config/config.go new file mode 100644 index 0000000..27f8c66 --- /dev/null +++ b/config/config.go @@ -0,0 +1,37 @@ +package config + +import ( + "fmt" + + "github.com/BurntSushi/toml" +) + +type Config struct { + APIURL string `toml:"APIURL"` + ShowSys bool `toml:"ShowSys"` + LogFile string `toml:"LogFile"` + UserRole string `toml:"UserRole"` + ToolRole string `toml:"ToolRole"` + AssistantRole string `toml:"AssistantRole"` + AssistantIcon string `toml:"AssistantIcon"` + UserIcon string `toml:"UserIcon"` + ToolIcon string `toml:"ToolIcon"` +} + +func LoadConfigOrDefault(fn string) *Config { + if fn == "" { + fn = "config.toml" + } + config := &Config{} + _, err := toml.DecodeFile(fn, &config) + if err != nil { + fmt.Println("failed to read config from file, loading default") + config.APIURL = "http://localhost:8080/v1/chat/completions" + config.ShowSys = true + config.LogFile = "log.txt" + config.UserRole = "user" + config.ToolRole = "tool" + config.AssistantRole = "assistant" + } + return config +} diff --git a/go.mod b/go.mod index 687511d..0413db5 100644 --- a/go.mod +++ b/go.mod @@ -3,6 +3,7 @@ module elefant go 1.23.2 require ( + github.com/BurntSushi/toml v1.4.0 github.com/gdamore/tcell/v2 v2.7.4 github.com/glebarez/go-sqlite v1.22.0 github.com/jmoiron/sqlx v1.4.0 diff --git a/go.sum b/go.sum index e4a23b5..087628e 100644 --- a/go.sum +++ b/go.sum @@ -1,5 +1,7 @@ filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA= filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4= +github.com/BurntSushi/toml v1.4.0 h1:kuoIxZQy2WRRk1pttg9asf+WVv6tWQuBNVmK8+nqPr0= +github.com/BurntSushi/toml v1.4.0/go.mod h1:ukJfTF/6rtPPRCnwkur4qwRxa8vTRFBF0uk2lLoLwho= github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY= github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto= github.com/gdamore/encoding v1.0.0 h1:+7OoQ1Bc6eTm5niUzBa0Ctsh6JbMW6Ra+YNuAtDBdko= diff --git a/tools.go b/tools.go index b32d8cb..af4356e 100644 --- a/tools.go +++ b/tools.go @@ -61,7 +61,7 @@ also: - others do; */ func memorise(args ...string) []byte { - agent := assistantRole + agent := cfg.AssistantRole if len(args) < 2 { msg := "not enough args to call memorise tool; need topic and data to remember" logger.Error(msg) @@ -79,7 +79,7 @@ func memorise(args ...string) []byte { } func recall(args ...string) []byte { - agent := assistantRole + agent := cfg.AssistantRole if len(args) < 1 { logger.Warn("not enough args to call recall tool") return nil @@ -94,7 +94,7 @@ func recall(args ...string) []byte { } func recallTopics(args ...string) []byte { - agent := assistantRole + agent := cfg.AssistantRole topics, err := store.RecallTopics(agent) if err != nil { logger.Error("failed to use tool", "error", err, "args", args) diff --git a/tui.go b/tui.go index 33ecd65..7cf87cc 100644 --- a/tui.go +++ b/tui.go @@ -82,7 +82,7 @@ func init() { } // set chat body chatBody.Messages = defaultStarter - textView.SetText(chatToText(showSystemMsgs)) + textView.SetText(chatToText(cfg.ShowSys)) newChat := &models.Chat{ ID: id + 1, Name: fmt.Sprintf("%v_%v", "new", time.Now().Unix()), @@ -111,7 +111,7 @@ func init() { return } chatBody.Messages = history - textView.SetText(chatToText(showSystemMsgs)) + textView.SetText(chatToText(cfg.ShowSys)) activeChatName = fn pages.RemovePage("history") return @@ -134,7 +134,7 @@ func init() { } chatBody.Messages[0].Content = sysMsg // replace textview - textView.SetText(chatToText(showSystemMsgs)) + textView.SetText(chatToText(cfg.ShowSys)) pages.RemovePage("sys") } }) @@ -152,7 +152,7 @@ func init() { } chatBody.Messages[selectedIndex].Content = editedMsg // change textarea - textView.SetText(chatToText(showSystemMsgs)) + textView.SetText(chatToText(cfg.ShowSys)) pages.RemovePage("editArea") editMode = false return nil @@ -233,7 +233,7 @@ func init() { // textArea.SetMovedFunc(updateStatusLine) updateStatusLine() - textView.SetText(chatToText(showSystemMsgs)) + textView.SetText(chatToText(cfg.ShowSys)) textView.ScrollToEnd() app.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey { if event.Key() == tcell.KeyF1 { @@ -251,14 +251,14 @@ func init() { if event.Key() == tcell.KeyF2 { // regen last msg chatBody.Messages = chatBody.Messages[:len(chatBody.Messages)-1] - textView.SetText(chatToText(showSystemMsgs)) - go chatRound("", userRole, textView) + textView.SetText(chatToText(cfg.ShowSys)) + go chatRound("", cfg.UserRole, textView) return nil } if event.Key() == tcell.KeyF3 && !botRespMode { // delete last msg chatBody.Messages = chatBody.Messages[:len(chatBody.Messages)-1] - textView.SetText(chatToText(showSystemMsgs)) + textView.SetText(chatToText(cfg.ShowSys)) return nil } if event.Key() == tcell.KeyF4 { @@ -268,9 +268,9 @@ func init() { return nil } if event.Key() == tcell.KeyF5 { - // switch showSystemMsgs - showSystemMsgs = !showSystemMsgs - textView.SetText(chatToText(showSystemMsgs)) + // switch cfg.ShowSys + cfg.ShowSys = !cfg.ShowSys + textView.SetText(chatToText(cfg.ShowSys)) } if event.Key() == tcell.KeyF6 { interruptResp = true @@ -317,7 +317,7 @@ func init() { textView.ScrollToEnd() } // update statue line - go chatRound(msgText, userRole, textView) + go chatRound(msgText, cfg.UserRole, textView) return nil } if event.Key() == tcell.KeyPgUp || event.Key() == tcell.KeyPgDn { -- cgit v1.2.3