github/new-api
1
0
Fork 0
mirror of https://github.com/QuantumNous/new-api.git synced 2026-04-30 23:41:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ feat(endpoint types): add support for image generation models in endpoint type handling

Browse Source
This commit is contained in:
CaIon
2025-07-05 14:14:40 +08:00
parent 6ac7878863
commit 30fb349d91
3 changed files with 32 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
common/endpoint_type.go
View File

@@ -33,5 +33,9 @@ func GetEndpointTypesByChannelType(channelType int, modelName string) []constant
endpointTypes = []constant.EndpointType{constant.EndpointTypeOpenAI}
}
}
if IsImageGenerationModel(modelName) {
// add to first
endpointTypes = append([]constant.EndpointType{constant.EndpointTypeImageGeneration}, endpointTypes...)
}
return endpointTypes
}