diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-07 09:12:18 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-07 09:12:18 +0800 |
| commit | 85ad276a29b75b24903ab925956118d537afd52d (patch) | |
| tree | 5f5632dd946c5d2a0390ce7fb266e7551fbe7031 /src/client | |
| parent | 7c6f75a139128e77f6c0e16ee176192dfef75599 (diff) | |
| download | aichat-85ad276a29b75b24903ab925956118d537afd52d.tar.gz | |
feat: support playground/arena webui (#487)
Diffstat (limited to 'src/client')
| -rw-r--r-- | src/client/model.rs | 4 |
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) } |