Add error handling for Generate method in NewIgnorePolicy

secmon-lab · Dec 23, 2024 · 7777054 · 7777054
1 parent 369c75f
commit 7777054
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/usecase/new_ignore.go b/pkg/usecase/new_ignore.go
@@ -131,6 +131,9 @@ func NewIgnorePolicy(ctx context.Context,
 
 	alertSlugPrompts := append([]string{alertSlugPrompt}, alertDataSet...)
 	slugResp, err := genAI.Generate(ctx, alertSlugPrompts...)
+	if err != nil {
+		return err
+	}
 	alertSlug := strings.TrimSpace(slugResp[0])
 	logger.Info("Generated slug", "slug", alertSlug)