From 3716d22ecacb8b43f1495c0f08d117f0bda128dd Mon Sep 17 00:00:00 2001
From: John Costa <johncosta027@gmail.com>
Date: Thu, 17 Apr 2025 11:07:37 +0100
Subject: [PATCH] fix(logger): nil pointer error + log debug level clean

---
 backend/agents/client/client.go | 3 ---
 backend/agents/event_agent.go   | 2 +-
 backend/agents/note_agent.go    | 1 +
 backend/events.go               | 6 +++++-
 4 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/backend/agents/client/client.go b/backend/agents/client/client.go
index 29370c1..42be680 100644
--- a/backend/agents/client/client.go
+++ b/backend/agents/client/client.go
@@ -160,8 +160,6 @@ func (client AgentClient) Request(req *AgentRequestBody) (AgentResponse, error)
 		return AgentResponse{}, errors.New("Unsupported. We currently only accept 1 choice from AI.")
 	}
 
-	client.Log.SetLevel(log.DebugLevel)
-
 	msg := agentResponse.Choices[0].Message
 
 	if len(msg.Content) > 0 {
@@ -229,7 +227,6 @@ func (client *AgentClient) Process(info ToolHandlerInfo, req *AgentRequestBody)
 			client.Reply = toolCall.Function.Arguments
 		}
 
-		client.Log.SetLevel(log.DebugLevel)
 		client.Log.Debugf("Response: %s", toolResponse.Content)
 
 		req.Chat.AddToolResponse(toolResponse)
diff --git a/backend/agents/event_agent.go b/backend/agents/event_agent.go
index 2224988..e9575a3 100644
--- a/backend/agents/event_agent.go
+++ b/backend/agents/event_agent.go
@@ -116,7 +116,7 @@ func NewEventAgent(log *log.Logger, eventsModel models.EventModel, locationModel
 		EndToolCall:  "finish",
 	})
 
-	locationAgent := NewLocationAgent(log.With("Locations 📍"), locationModel)
+	locationAgent := NewLocationAgent(log.WithPrefix("Events 📅 > Locations 📍"), locationModel)
 	locationQuery := "Can you get me the ID of the location present in this image?"
 	locationAgent.Options.Query = &locationQuery
 
diff --git a/backend/agents/note_agent.go b/backend/agents/note_agent.go
index 33b080f..3b06c59 100644
--- a/backend/agents/note_agent.go
+++ b/backend/agents/note_agent.go
@@ -71,6 +71,7 @@ func (agent NoteAgent) GetNotes(userId uuid.UUID, imageId uuid.UUID, imageName s
 func NewNoteAgent(log *log.Logger, noteModel models.NoteModel) NoteAgent {
 	client := client.CreateAgentClient(client.CreateAgentClientOptions{
 		SystemPrompt: noteAgentPrompt,
+		Log:          log,
 	})
 
 	agent := NoteAgent{
diff --git a/backend/events.go b/backend/events.go
index cfb2c90..5d4a06e 100644
--- a/backend/events.go
+++ b/backend/events.go
@@ -14,11 +14,15 @@ import (
 )
 
 func createLogger(prefix string) *log.Logger {
-	return log.NewWithOptions(os.Stdout, log.Options{
+	logger := log.NewWithOptions(os.Stdout, log.Options{
 		ReportTimestamp: true,
 		TimeFormat:      time.Kitchen,
 		Prefix:          prefix,
 	})
+
+	logger.SetLevel(log.DebugLevel)
+
+	return logger
 }
 
 func ListenNewImageEvents(db *sql.DB, eventManager *EventManager) {