diff --git a/plugins/wasm-go/extensions/ai-proxy/provider/gemini.go b/plugins/wasm-go/extensions/ai-proxy/provider/gemini.go index 1fb2ae667..b7b8496c4 100644 --- a/plugins/wasm-go/extensions/ai-proxy/provider/gemini.go +++ b/plugins/wasm-go/extensions/ai-proxy/provider/gemini.go @@ -36,7 +36,10 @@ func (g *geminiProviderInitializer) ValidateConfig(config *ProviderConfig) error } func (g *geminiProviderInitializer) DefaultCapabilities() map[string]string { - return map[string]string{} + return map[string]string{ + string(ApiNameChatCompletion): "", + string(ApiNameEmbeddings): "", + } } func (g *geminiProviderInitializer) CreateProvider(config ProviderConfig) (Provider, error) {