feat: checking users limits when AI scans their images

backend/agents/list_agent.go

@@ -3,8 +3,10 @@ package agents
 import (
 	"context"
 	"encoding/json"
+	"fmt"
 	"screenmark/screenmark/.gen/haystack/haystack/model"
 	"screenmark/screenmark/agents/client"
+	"screenmark/screenmark/limits"
 	"screenmark/screenmark/models"
 
 	"github.com/charmbracelet/log"
@@ -174,7 +176,7 @@ type addToListArguments struct {
 	Schema []models.IDValue
 }
 
-func NewListAgent(log *log.Logger, listModel models.ListModel) client.AgentClient {
+func NewListAgent(log *log.Logger, listModel models.ListModel, limitsMethods limits.LimitsManagerMethods) client.AgentClient {
 	agentClient := client.CreateAgentClient(client.CreateAgentClientOptions{
 		SystemPrompt: listPrompt,
 		JsonTools:    listTools,
@@ -193,6 +195,16 @@ func NewListAgent(log *log.Logger, listModel models.ListModel) client.AgentClien
 			return "", err
 		}
 
+		hasReachedLimit, err := limitsMethods.HasReachedStackLimit(info.UserId)
+		if err != nil {
+			return "", fmt.Errorf("error checking stack limits: %w", err)
+		}
+
+		if hasReachedLimit {
+			log.Warn("User has reached limits", "userID", info.UserId)
+			return "", fmt.Errorf("reached stack limits")
+		}
+
 		ctx := context.Background()
 		savedList, err := listModel.Save(ctx, info.UserId, args.Name, args.Desription, args.Schema)
 
@@ -201,8 +213,6 @@ func NewListAgent(log *log.Logger, listModel models.ListModel) client.AgentClien
 			return "", err
 		}
 
-		log.Debug(savedList)
-
 		return savedList, nil
 	})