Added new known safety text error

igolaizola · Sep 1, 2024 · c912fe3 · c912fe3
1 parent 7e2356f
commit c912fe3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pkg/runway/runway.go b/pkg/runway/runway.go
@@ -376,7 +376,7 @@ func (e *Error) Reason() string {
 func (e *Error) Temporary() bool {
 	r := e.data.Error.Reason
 	switch {
-	case r == "SAFETY.INPUT.TEXT", r == "SAFETY.INPUT.IMAGE":
+	case r == "SAFETY.INPUT.TEXT", r == "SAFETY.INPUT.IMAGE", r == "INPUT_PREPROCESSING.SAFETY.TEXT":
 		return false
 	case strings.HasPrefix(r, "INTERNAL.BAD_OUTPUT."), r == "SAFETY.OUTPUT.VIDEO":
 		return true
@@ -394,6 +394,7 @@ func (e *Error) Unknown() bool {
 		"SAFETY.INPUT.TEXT",
 		"SAFETY.INPUT.IMAGE",
 		"SAFETY.OUTPUT.VIDEO",
+		"INPUT_PREPROCESSING.SAFETY.TEXT",
 		"":
 		return false
 	default: