1 files changed, 9 insertions, 1 deletions
diff --git a/src/config/input.rs b/src/config/input.rs
index fc29c49..c3e0de3 100644
--- a/src/config/input.rs
+++ b/src/config/input.rs
@@ -1,4 +1,4 @@
-use crate::client::{ImageUrl, MessageContent, MessageContentPart};
+use crate::client::{ImageUrl, MessageContent, MessageContentPart, ModelCapabilities};
 use crate::utils::sha256sum;
 
 use anyhow::{bail, Context, Result};
@@ -119,6 +119,14 @@ impl Input {
             MessageContent::Array(list)
         }
     }
+
+    pub fn required_capabilities(&self) -> ModelCapabilities {
+        if !self.medias.is_empty() {
+            ModelCapabilities::Vision
+        } else {
+            ModelCapabilities::Text
+        }
+    }
 }
 
 pub fn resolve_data_url(data_urls: &HashMap<String, String>, data_url: String) -> String {