From 8c03f2c3e63a2a0f3667cbfe7fad534555e41e62 Mon Sep 17 00:00:00 2001
From: igolaizola <11333576+igolaizola@users.noreply.github.com>
Date: Sat, 10 Aug 2024 22:10:06 +0200
Subject: [PATCH] Added safety output video reason and unknown command

---
 pkg/runway/runway.go | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/pkg/runway/runway.go b/pkg/runway/runway.go
index 56548ef..f3fd612 100644
--- a/pkg/runway/runway.go
+++ b/pkg/runway/runway.go
@@ -378,7 +378,7 @@ func (e *Error) Temporary() bool {
 	switch {
 	case r == "SAFETY.INPUT.TEXT", r == "SAFETY.INPUT.IMAGE":
 		return false
-	case strings.HasPrefix(r, "INTERNAL.BAD_OUTPUT."):
+	case strings.HasPrefix(r, "INTERNAL.BAD_OUTPUT."), r == "SAFETY.OUTPUT.VIDEO":
 		return true
 	case r == "":
 		return true
@@ -387,6 +387,19 @@ func (e *Error) Temporary() bool {
 	}
 }
 
+func (e *Error) Unknown() bool {
+	switch e.data.Error.Reason {
+	case "INTERNAL.BAD_OUTPUT.CODE01",
+		"INTERNAL.BAD_OUTPUT.CODE03",
+		"SAFETY.INPUT.TEXT",
+		"SAFETY.INPUT.IMAGE",
+		"SAFETY.OUTPUT.VIDEO":
+		return false
+	default:
+		return true
+	}
+}
+
 func (c *Client) Generate(ctx context.Context, cfg *GenerateRequest) (*Generation, error) {
 	// Load team ID
 	if err := c.loadTeamID(ctx); err != nil {