diff --git a/crates/lib/artcraft_router/src/generate/generate_video/generate_video.rs b/crates/lib/artcraft_router/src/generate/generate_video/generate_video.rs index 5aa9e0884b..c4525b02ad 100644 --- a/crates/lib/artcraft_router/src/generate/generate_video/generate_video.rs +++ b/crates/lib/artcraft_router/src/generate/generate_video/generate_video.rs @@ -9,11 +9,23 @@ use tokens::tokens::generic_inference_jobs::InferenceJobToken; pub struct GenerateVideoArgs<'a> { pub client: &'a RouterClient, + pub provider: Provider, + + /// Which model to use. pub model: CommonVideoModel, - pub resolution: Option, - pub aspect_ratio: Option, + + /// The prompt for the video generation pub prompt: Option, + + // /// Some models support negative prompts + // pub negative_prompt: Option, + + /// The resolution to use + pub resolution: Option, + + /// The aspect ratio to use + pub aspect_ratio: Option, } pub struct GenerateVideoResponse {