1 files changed, 4 insertions, 0 deletions
diff --git a/src/client/model.rs b/src/client/model.rs
index 90e46e4..b24a546 100644
--- a/src/client/model.rs
+++ b/src/client/model.rs
@@ -111,6 +111,10 @@ impl Model {
         )
     }
 
+    pub fn supports_vision(&self) -> bool {
+        self.capabilities.contains(ModelCapabilities::Vision)
+    }
+
     pub fn show_max_output_tokens(&self) -> Option<isize> {
         self.max_output_tokens.or(self.ref_max_output_tokens)
     }