From fb5c89219e72c97232f790e71a905b4336933991 Mon Sep 17 00:00:00 2001
From: kujtimiihoxha <kujtimii.h@gmail.com>
Date: Mon, 29 Sep 2025 09:04:43 +0200
Subject: [PATCH] chore: small fixes

---
 internal/event/logger.go     | 9 +++++----
 internal/llm/agent/agent.go  | 2 +-
 internal/llm/agent/errors.go | 1 +
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/internal/event/logger.go b/internal/event/logger.go
index 7648ae2c2cca91ed20535c0d65a677cd4db84500..7581676b018f5ac6001827db851a132792d21985 100644
--- a/internal/event/logger.go
+++ b/internal/event/logger.go
@@ -1,6 +1,7 @@
 package event
 
 import (
+	"fmt"
 	"log/slog"
 
 	"github.com/posthog/posthog-go"
@@ -11,17 +12,17 @@ var _ posthog.Logger = logger{}
 type logger struct{}
 
 func (logger) Debugf(format string, args ...any) {
-	slog.Debug(format, args...)
+	slog.Debug(fmt.Sprintf(format, args...))
 }
 
 func (logger) Logf(format string, args ...any) {
-	slog.Info(format, args...)
+	slog.Info(fmt.Sprintf(format, args...))
 }
 
 func (logger) Warnf(format string, args ...any) {
-	slog.Warn(format, args...)
+	slog.Warn(fmt.Sprintf(format, args...))
 }
 
 func (logger) Errorf(format string, args ...any) {
-	slog.Error(format, args...)
+	slog.Error(fmt.Sprintf(format, args...))
 }
diff --git a/internal/llm/agent/agent.go b/internal/llm/agent/agent.go
index 1f260f7c4134a3561f34aa64aa2096a9cd14641d..ce2a08d81ef9494e57739061f85696f771163991 100644
--- a/internal/llm/agent/agent.go
+++ b/internal/llm/agent/agent.go
@@ -578,7 +578,7 @@ loop:
 			}
 		case <-timer.C:
 			a.finishMessage(ctx, &assistantMsg, message.FinishReasonError, "Stream timeout", "No chunk received within timeout")
-			return assistantMsg, nil, fmt.Errorf("stream chunk timeout")
+			return assistantMsg, nil, ErrStreamTimeout
 		case <-ctx.Done():
 			a.finishMessage(context.Background(), &assistantMsg, message.FinishReasonCanceled, "Request cancelled", "")
 			return assistantMsg, nil, ctx.Err()
diff --git a/internal/llm/agent/errors.go b/internal/llm/agent/errors.go
index 0e2f983d64b42b93ad3a51f32ce0335b0374a613..943918390c6708b5ae6ea5e40e50ebce209cc263 100644
--- a/internal/llm/agent/errors.go
+++ b/internal/llm/agent/errors.go
@@ -7,6 +7,7 @@ import (
 
 var (
 	ErrRequestCancelled = errors.New("request canceled by user")
+	ErrStreamTimeout    = errors.New("stream chunk timeout")
 	ErrSessionBusy      = errors.New("session is currently processing another request")
 )