jiazhizhong/higress
1
0
Fork 0
mirror of https://github.com/alibaba/higress.git synced 2026-06-09 20:57:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(ai-proxy): add doubao Image Generation support (#2331)

Browse Source 
Signed-off-by: Xijun Dai <daixijun1990@gmail.com>
This commit is contained in:
Xijun Dai
2025-05-27 18:59:07 +08:00
committed by GitHub
parent 168cb04c61
commit 767f51adce
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
plugins/wasm-go/extensions/ai-proxy/provider/doubao.go
View File

@@ -14,6 +14,7 @@ const (
doubaoDomain = "ark.cn-beijing.volces.com" doubaoDomain = "ark.cn-beijing.volces.com"
doubaoChatCompletionPath = "/api/v3/chat/completions" doubaoChatCompletionPath = "/api/v3/chat/completions"
doubaoEmbeddingsPath = "/api/v3/embeddings" doubaoEmbeddingsPath = "/api/v3/embeddings"
doubaoImageGenerationPath = "/api/v3/images/generations"
) )
type doubaoProviderInitializer struct{} type doubaoProviderInitializer struct{}
@@ -29,6 +30,7 @@ func (m *doubaoProviderInitializer) DefaultCapabilities() map[string]string {
return map[string]string{ return map[string]string{
string(ApiNameChatCompletion): doubaoChatCompletionPath, string(ApiNameChatCompletion): doubaoChatCompletionPath,
string(ApiNameEmbeddings): doubaoEmbeddingsPath, string(ApiNameEmbeddings): doubaoEmbeddingsPath,
string(ApiNameImageGeneration): doubaoImageGenerationPath,
} }
} }
@@ -75,5 +77,8 @@ func (m *doubaoProvider) GetApiName(path string) ApiName {
if strings.Contains(path, doubaoEmbeddingsPath) { if strings.Contains(path, doubaoEmbeddingsPath) {
return ApiNameEmbeddings return ApiNameEmbeddings
} }
if strings.Contains(path, doubaoImageGenerationPath) {
return ApiNameImageGeneration
}
return "" return ""
} }