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

This commit is contained in:
CaIon committed 2025-07-05 14:14:40 +08:00
1 parent 14cca53ec8
commit 95775a5a52
3 files changed
+32 -6

No files matched your search

+4
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
}