feat: add error detection and handling for LLM API responses

1 files changed, 6 insertions, 1 deletions

diff --git a/bot.go b/bot.go
index aecc0b9..409a97b 100644
--- a/bot.go
+++ b/bot.go
@@ -191,7 +191,12 @@ func sendMsgToLLM(body io.Reader) {
 			streamDone <- true
 			break
 		}
-		// here line can contain some error msg; if it does we should log it and break from loop; ai!
+		// Handle error messages in response content
+		if content != "" && strings.Contains(strings.ToLower(content), "error") {
+			logger.Error("API error response detected", "content", content, "url", cfg.CurrentAPI)
+			streamDone <- true
+			break
+		}
 		if stop {
 			if content != "" {
 				logger.Warn("text inside of finish llmchunk", "chunk", content, "counter", counter)