Add chat history

Author: chand1012

cmd/chat.go

@@ -7,9 +7,12 @@ import (
 	"context"
 	"fmt"
 	"os"
+	"strconv"
+	"strings"
 
 	"github.com/chand1012/ottodocs/pkg/calc"
 	"github.com/chand1012/ottodocs/pkg/config"
+	"github.com/chand1012/ottodocs/pkg/history"
 	"github.com/chand1012/ottodocs/pkg/utils"
 	l "github.com/charmbracelet/log"
 	"github.com/sashabaranov/go-openai"
@@ -42,14 +45,175 @@ No code context is passed in this mode. Emulates web chat.`,
 			os.Exit(1)
 		}
 
+		// error if read only mode set without a history file
+		if readOnly && loadHistory == "" {
+			log.Error("Read only mode requires a history file. Use --history to specify a history file.")
+			os.Exit(1)
+		}
+
 		// all messages in the conversation
 		var messages []openai.ChatCompletionMessage
+		var fileName string // history file name
 
 		client := openai.NewClient(conf.APIKey)
 
+		if displayHistory {
+			files, err := history.ListHistoryFiles()
+			if err != nil {
+				log.Error(err)
+				os.Exit(1)
+			}
+
+			if len(files) == 0 {
+				fmt.Println("No chat history found.")
+				os.Exit(0)
+			}
+
+			for i, file := range files {
+				// load each file
+				historyFile, err := history.LoadHistoryFile(file)
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+				fmt.Printf("%d. %s\n", i, historyFile.DisplayName)
+			}
+			os.Exit(0)
+		}
+
+		if deleteHistory != "" {
+			if strings.HasSuffix(deleteHistory, ".json") {
+				// load the file
+				_, err = history.LoadHistoryFile(deleteHistory)
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+			} else {
+				i, err := strconv.Atoi(deleteHistory)
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+				files, err := history.ListHistoryFiles()
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+
+				if i >= len(files) {
+					log.Error("Index out of bounds")
+					os.Exit(1)
+				}
+
+				// load the file
+				_, err = history.LoadHistoryFile(files[i])
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+			}
+
+			confirm, err := utils.Input("Are you sure you want to delete this history file? (y/N): ")
+			if err != nil {
+				log.Error(err)
+				os.Exit(1)
+			}
+
+			confirm = strings.ToLower(confirm)
+			if confirm != "y" {
+				os.Exit(0)
+			}
+
+			err = history.DeleteHistoryFile(deleteHistory)
+			if err != nil {
+				log.Error(err)
+				os.Exit(1)
+			}
+
+			fmt.Println("Deleted history file.")
+			os.Exit(0)
+		}
+
+		if clearHistory {
+			// make sure the user wants to clear the history
+			log.Warn("This will clear all chat history. This operation cannot be undone.")
+			confirm, err := utils.Input("Are you sure you want to clear the chat history? (y/N): ")
+			if err != nil {
+				log.Error(err)
+				os.Exit(1)
+			}
+
+			confirm = strings.ToLower(confirm)
+
+			if confirm != "y" {
+				os.Exit(0)
+			}
+
+			err = history.ClearHistory()
+			if err != nil {
+				log.Error(err)
+				os.Exit(1)
+			}
+
+			fmt.Println("Cleared chat history.")
+			os.Exit(0)
+		}
+
 		utils.PrintColoredText("Otto: ", conf.OttoColor)
 		fmt.Println("Hello! I am Otto. Use Ctrl+C to exit at any time.")
 
+		if loadHistory != "" {
+			var historyFile *history.HistoryFile
+			if strings.HasSuffix(loadHistory, ".json") {
+				// load the file
+				historyFile, err = history.LoadHistoryFile(loadHistory)
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+			} else {
+				i, err := strconv.Atoi(loadHistory)
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+				files, err := history.ListHistoryFiles()
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+
+				if i >= len(files) {
+					log.Error("Index out of bounds")
+					os.Exit(1)
+				}
+
+				// load the file
+				historyFile, err = history.LoadHistoryFile(files[i])
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+			}
+
+			messages = historyFile.Messages
+			fileName = loadHistory
+
+			for _, message := range messages {
+				if message.Role == openai.ChatMessageRoleUser {
+					utils.PrintColoredText("You: ", conf.UserColor)
+				} else {
+					utils.PrintColoredText("Otto: ", conf.OttoColor)
+				}
+				fmt.Println(message.Content)
+			}
+
+			if readOnly {
+				os.Exit(0)
+			}
+		}
+
 		for {
 			question, err := utils.InputWithColor("You: ", conf.UserColor)
 			if err != nil {
@@ -110,6 +274,20 @@ No code context is passed in this mode. Emulates web chat.`,
 				Content: completeStream,
 				Role:    openai.ChatMessageRoleAssistant,
 			})
+
+			if fileName == "" {
+				fileName, err = history.SaveInitialHistory(messages, conf)
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+			} else {
+				err = history.UpdateHistoryFile(messages, fileName)
+				if err != nil {
+					log.Error(err)
+					os.Exit(1)
+				}
+			}
 		}
 	},
 }
@@ -118,4 +296,9 @@ func init() {
 	RootCmd.AddCommand(chatCmd)
 
 	chatCmd.Flags().BoolVarP(&verbose, "verbose", "v", false, "Verbose output")
+	chatCmd.Flags().BoolVarP(&displayHistory, "history", "H", false, "Display chat history")
+	chatCmd.Flags().BoolVarP(&readOnly, "read", "r", false, "Read the history file and exit")
+	chatCmd.Flags().BoolVar(&clearHistory, "clear", false, "Clear chat history")
+	chatCmd.Flags().StringVarP(&loadHistory, "load", "l", "", "Load chat history from file. Can either be a file path or an index of the chat history")
+	chatCmd.Flags().StringVarP(&deleteHistory, "delete", "d", "", "Delete chat history from file. Can either be a file path or an index of the chat history")
 }

cmd/vars.go

@@ -51,6 +51,12 @@ var currentTag string
 
 var contextFiles []string
 
+var displayHistory bool
+var loadHistory string
+var deleteHistory string
+var readOnly bool
+var clearHistory bool
+
 var log = l.NewWithOptions(os.Stderr, l.Options{
 	Level:           l.InfoLevel,
 	ReportTimestamp: false,