summaryrefslogtreecommitdiff
path: root/models/models.go
diff options
context:
space:
mode:
Diffstat (limited to 'models/models.go')
-rw-r--r--models/models.go14
1 files changed, 14 insertions, 0 deletions
diff --git a/models/models.go b/models/models.go
index 973eb3d..97d0272 100644
--- a/models/models.go
+++ b/models/models.go
@@ -608,6 +608,20 @@ func (lcp *LCPModels) ListModels() []string {
return resp
}
+func (lcp *LCPModels) HasVision(modelID string) bool {
+ for _, m := range lcp.Data {
+ if m.ID == modelID {
+ args := m.Status.Args
+ for i := 0; i < len(args)-1; i++ {
+ if args[i] == "--mmproj" {
+ return true
+ }
+ }
+ }
+ }
+ return false
+}
+
type ResponseStats struct {
Tokens int
Duration float64