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

2025-07-05 14:14:40 +08:00
parent 6ac7878863
commit 30fb349d91
3 changed files with 32 additions and 6 deletions
--- a/common/endpoint_type.go
+++ b/common/endpoint_type.go
@@ -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
 }