PNArchives/Memoh
1
0
Fork 0
mirror of https://github.com/memohai/Memoh.git synced 2026-04-25 07:00:48 +09:00
Code Issues Packages Projects Releases Wiki Activity

feat(models): add image model type support

Browse Source 
Add a dedicated image model type so bots can use image API models without overloading chat model capabilities, while keeping existing chat-based image generation selectable.
This commit is contained in:
Acbox
2026-04-16 16:00:22 +08:00
parent 33e18e7e64
commit ddda00f980
23 changed files with 326 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/handlers/providers.go
+10 -2
View File
@@ -332,12 +332,20 @@ func (h *ProvidersHandler) ImportModels(c echo.Context) error {
for _, m := range remoteModels {
modelType := models.ModelTypeChat
if strings.TrimSpace(m.Type) == string(models.ModelTypeEmbedding) {
switch strings.TrimSpace(m.Type) {
case string(models.ModelTypeEmbedding):
modelType = models.ModelTypeEmbedding
case string(models.ModelTypeImage):
modelType = models.ModelTypeImage
}
compatibilities := m.Compatibilities
if len(compatibilities) == 0 {
compatibilities = []string{models.CompatVision, models.CompatToolCall, models.CompatReasoning}
switch modelType {
case models.ModelTypeImage:
compatibilities = []string{models.CompatGenerate, models.CompatEdit}
case models.ModelTypeChat:
compatibilities = []string{models.CompatVision, models.CompatToolCall, models.CompatReasoning}
}
}
name := strings.TrimSpace(m.Name)
if name == "" {