From c35af037203ac5c39a4f704d5343bc2b5cc56a0c Mon Sep 17 00:00:00 2001
From: Grail Finder <wohilas@gmail.com>
Date: Thu, 21 Nov 2024 18:15:01 +0300
Subject: Feat: add tools to tool map

---
 bot.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'bot.go')

diff --git a/bot.go b/bot.go
index 66303a2..1d7133d 100644
--- a/bot.go
+++ b/bot.go
@@ -33,11 +33,12 @@ var (
 	historyDir    = "./history/"
 	// TODO: pass as an cli arg
 	showSystemMsgs  bool
+	chunkLimit      = 1000
 	activeChatName  string
 	chunkChan       = make(chan string, 10)
 	streamDone      = make(chan bool, 1)
 	chatBody        *models.ChatBody
-	store           storage.ChatHistory
+	store           storage.FullRepo
 	defaultFirstMsg = "Hello! What can I do for you?"
 	defaultStarter  = []models.MessagesStory{
 		{Role: "system", Content: systemMsg},
@@ -89,14 +90,15 @@ func sendMsgToLLM(body io.Reader) (any, error) {
 			break
 		}
 		llmchunk := models.LLMRespChunk{}
-		if counter > 2000 {
+		if counter > chunkLimit {
+			logger.Warn("response hit chunk limit", "limit", chunkLimit)
 			streamDone <- true
 			break
 		}
 		line, err := reader.ReadBytes('\n')
 		if err != nil {
 			streamDone <- true
-			panic(err)
+			logger.Error("error reading response body", "error", err)
 		}
 		// logger.Info("linecheck", "line", string(line), "len", len(line), "counter", counter)
 		if len(line) <= 1 {
-- 
cgit v1.2.3