image generation wip

models.go

@@ -18,6 +18,7 @@ type Model struct {
 	Vision    bool `json:"-"`
 	JSON      bool `json:"-"`
 	Tools     bool `json:"-"`
+	Images    bool `json:"-"`
 }
 
 var ModelMap = make(map[string]*Model)
@@ -89,5 +90,13 @@ func GetModelTags(model openrouter.Model, m *Model) {
 		}
 	}
 
+	for _, modality := range model.Architecture.OutputModalities {
+		if modality == "image" {
+			m.Images = true
+
+			m.Tags = append(m.Tags, "image")
+		}
+	}
+
 	sort.Strings(m.Tags)
 }