From 85ad276a29b75b24903ab925956118d537afd52d Mon Sep 17 00:00:00 2001 From: sigoden Date: Tue, 7 May 2024 09:12:18 +0800 Subject: feat: support playground/arena webui (#487) --- src/client/model.rs | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/client/model.rs') 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 { self.max_output_tokens.or(self.ref_max_output_tokens) } -- cgit v1.2.3