From 0a2fb35ae23463f9445d153719af2bde4d994f17 Mon Sep 17 00:00:00 2001
From: johnlanni <zty98751@alibaba-inc.com>
Date: Wed, 7 May 2025 16:54:40 +0800
Subject: [PATCH] fix gemini provider in ai-proxy

---
 plugins/wasm-go/extensions/ai-proxy/provider/gemini.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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) {