first commit

core/providers/runway/utils.go

@@ -0,0 +1,34 @@
+package runway
+
+import (
+	"strings"
+
+	schemas "github.com/maximhq/bifrost/core/schemas"
+)
+
+// getRunwayEndpoint determines which Runway API endpoint to use based on the request parameters.
+// Returns the appropriate endpoint path:
+// - /v1/text_to_video: when only text prompt is provided
+// - /v1/video_to_video: when video URI is provided
+// - /v1/image_to_video: when image input reference is provided
+func getRunwayEndpoint(req *schemas.BifrostVideoGenerationRequest) string {
+	if req.Params != nil && req.Params.VideoURI != nil && *req.Params.VideoURI != "" {
+		return "/v1/video_to_video"
+	}
+	if req.Input != nil && req.Input.InputReference != nil && *req.Input.InputReference != "" {
+		return "/v1/image_to_video"
+	}
+	return "/v1/text_to_video"
+}
+
+func isRunwayGenModel(model string) bool {
+	return strings.Contains(model, "gen")
+}
+
+func isRunwayVeoModel(model string) bool {
+	return strings.Contains(model, "veo")
+}
+
+func supportsVideoToVideo(model string) bool {
+	return model == "gen4_aleph"
+}