feat(replicate): support creating prediction from official model

examples/replicate-basic/README.md

@@ -0,0 +1,3 @@
+# Replicate Basic Example
+
+![spartan](../../assets/dragon.webp)

examples/replicate-basic/src/main.rs

@@ -1,5 +1,5 @@
 use replic::{
-    client::{Client, CreatePrediction},
+    client::{Client, CreateModelPrediction},
     config::Config,
 };
 
@@ -8,16 +8,17 @@ async fn main() {
     let config = Config::from_env().unwrap();
     let client = Client::new(config).unwrap();
 
-    let collections = client
-        .create_prediction(CreatePrediction {
-            version: "f2ab8a5bfe79f02f0789a146cf5e73d2a4ff2684a98c2b303d1e1ff3814271db".to_string(),
-            input: serde_json::json!({
-                "prompt": "black forest gateau cake spelling out the words \"FLUX SCHNELL\", tasty, food photography, dynamic shot"
-            }),
-            webhook: None,
-            webhook_event_filters: None,
-        })
-        .await
-        .unwrap();
-    println!("{:?}", collections);
+    let payload = CreateModelPrediction {
+        owner: "black-forest-labs".to_string(),
+        name: "flux-schnell".to_string(),
+        input: serde_json::json!({
+            "prompt": "3D model of a baby dragon",
+            "num_outputs": 1,
+            "aspect_ratio": "1:1",
+            "output_format": "webp",
+            "output_quality": 100
+        }),
+    };
+    let prediction = client.create_model_prediction(payload).await.unwrap();
+    println!("{:?}", prediction);
 }

replicate/src/client.rs

@@ -179,6 +179,22 @@ impl Client {
         Ok(response)
     }
 
+    /// Create a prediction from an official model
+    pub async fn create_model_prediction(
+        &self,
+        payload: CreateModelPrediction,
+    ) -> Result<Prediction, Error> {
+        let path = format!("models/{}/{}/predictions", payload.owner, payload.name);
+        let response = self
+            .request(Method::POST, path.as_str())?
+            .json(&serde_json::json!({ "input": payload.input }))
+            .send()
+            .await?
+            .json::<Prediction>()
+            .await?;
+        Ok(response)
+    }
+
     /// Cancel a prediction.
     pub async fn cancel_prediction(&self, prediction_id: String) -> Result<(), Error> {
         let path = format!("predictions/{}/cancel", prediction_id);
@@ -491,6 +507,18 @@ pub struct CreatePrediction {
     pub webhook_event_filters: Option<Vec<WebHookEvent>>,
 }
 
+#[derive(Debug, Clone, Serialize, Deserialize)]
+pub struct CreateModelPrediction {
+    /// Model owner
+    pub owner: String,
+
+    /// Model name
+    pub name: String,
+
+    /// The model's input as a JSON object.
+    pub input: serde_json::Value,
+}
+
 #[derive(Debug, Clone, Serialize, Deserialize)]
 #[serde(rename_all = "lowercase")]
 pub enum WebHookEvent {