From 767f51adced9da33e1e1a04d4aead30630933a31 Mon Sep 17 00:00:00 2001 From: Xijun Dai Date: Tue, 27 May 2025 18:59:07 +0800 Subject: [PATCH] feat(ai-proxy): add doubao Image Generation support (#2331) Signed-off-by: Xijun Dai --- .../extensions/ai-proxy/provider/doubao.go | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go b/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go index cfdb7dca3..c367acb30 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/doubao.go @@ -11,9 +11,10 @@ import ( ) const ( - doubaoDomain = "ark.cn-beijing.volces.com" - doubaoChatCompletionPath = "/api/v3/chat/completions" - doubaoEmbeddingsPath = "/api/v3/embeddings" + doubaoDomain = "ark.cn-beijing.volces.com" + doubaoChatCompletionPath = "/api/v3/chat/completions" + doubaoEmbeddingsPath = "/api/v3/embeddings" + doubaoImageGenerationPath = "/api/v3/images/generations" ) type doubaoProviderInitializer struct{} @@ -27,8 +28,9 @@ func (m *doubaoProviderInitializer) ValidateConfig(config *ProviderConfig) error func (m *doubaoProviderInitializer) DefaultCapabilities() map[string]string { return map[string]string{ - string(ApiNameChatCompletion): doubaoChatCompletionPath, - string(ApiNameEmbeddings): doubaoEmbeddingsPath, + string(ApiNameChatCompletion): doubaoChatCompletionPath, + string(ApiNameEmbeddings): doubaoEmbeddingsPath, + string(ApiNameImageGeneration): doubaoImageGenerationPath, } } @@ -75,5 +77,8 @@ func (m *doubaoProvider) GetApiName(path string) ApiName { if strings.Contains(path, doubaoEmbeddingsPath) { return ApiNameEmbeddings } + if strings.Contains(path, doubaoImageGenerationPath) { + return ApiNameImageGeneration + } return "" }