feat: Enhance ai-cache Plugin with Vector Similarity-Based LLM Cache Recall and Multi-DB Support

plugins/wasm-go/extensions/ai-cache/README.md

@@ -101,29 +101,35 @@ LLM 结果缓存插件，默认配置方式可以直接用于 openai 协议的
 
 # 向量数据库提供商特有配置
 ## Chroma
-Chroma 所对应的 `vector.type` 为 `chroma`。它并无特有的配置字段。
+Chroma 所对应的 `vector.type` 为 `chroma`。它并无特有的配置字段。需要提前创建 Collection。
 
 ## DashVector
-DashVector 所对应的 `vector.type` 为 `dashvector`。它并无特有的配置字段。
+DashVector 所对应的 `vector.type` 为 `dashvector`。它并无特有的配置字段。需要提前创建 Collection。
 
 ## ElasticSearch
-ElasticSearch 所对应的 `vector.type` 为 `elasticsearch`。它特有的配置字段如下：
+ElasticSearch 所对应的 `vector.type` 为 `elasticsearch`。需要提前创建 Index 并填入在 `vector.collectionID` 中。当前依赖于 [KNN](https://www.elastic.co/guide/en/elasticsearch/reference/current/knn-search.html) 方法，请保证 ES 版本支持 `KNN`，当前已在 `8.16` 版本测试。
+它特有的配置字段如下：
 | 名称              | 数据类型 | 填写要求 | 默认值 | 描述                                                                          |
 |-------------------|----------|----------|--------|-------------------------------------------------------------------------------|
 | `vector.esUsername` | string   | 非必填   | -      | ElasticSearch 用户名 |
 | `vector.esPassword` | string | 非必填 | - | ElasticSearch 密码 |
 
+`vector.esUsername` 和 `vector.esPassword` 用于 Basic 认证。同时也支持 Api Key 认证，当填写了 `vector.apiKey` 时，则启用 Api Key 认证，如果使用 SaaS 版本需要填写 `encoded` 的值。
+
 ## Milvus
-Milvus 所对应的 `vector.type` 为 `milvus`。它并无特有的配置字段。
+Milvus 所对应的 `vector.type` 为 `milvus`。它并无特有的配置字段。需要提前创建 Collection。
 
 ## Pinecone
-Pinecone 所对应的 `vector.type` 为 `pinecone`。它并无特有的配置字段。
+Pinecone 所对应的 `vector.type` 为 `pinecone`。它并无特有的配置字段。需要提前创建 Index，并填写 Index 访问域名至 `serviceHost`。
+Pinecone 中的 `Namespace` 参数通过插件的 `vector.collectionID` 进行配置。
 
 ## Qdrant
-Qdrant 所对应的 `vector.type` 为 `qdrant`。它并无特有的配置字段。
+Qdrant 所对应的 `vector.type` 为 `qdrant`。它并无特有的配置字段。需要提前创建 Collection。
 
 ## Weaviate
 Weaviate 所对应的 `vector.type` 为 `weaviate`。它并无特有的配置字段。
+需要提前创建 Collection。需要注意的是 Weaviate 会设置首字母自动大写，在填写配置 `collectionID` 的时候需要将首字母设置为大写。
+如果使用 SaaS 需要填写 `serviceHost` 参数。
 
 ## 配置示例
 ### 基础配置

plugins/wasm-go/extensions/ai-cache/vector/elasticsearch.go

@@ -82,11 +82,16 @@ func (d *ESProvider) QueryEmbedding(
 	)
 }
 
-// base64 编码 ES 身份认证字符串
+// base64 编码 ES 身份认证字符串或使用 Apikey
 func (d *ESProvider) getCredentials() string {
-	credentials := fmt.Sprintf("%s:%s", d.config.esUsername, d.config.esPassword)
-	encodedCredentials := base64.StdEncoding.EncodeToString([]byte(credentials))
-	return fmt.Sprintf("Basic %s", encodedCredentials)
+	if len(d.config.apiKey) != 0 {
+		return fmt.Sprintf("ApiKey %s", d.config.apiKey)
+	} else {
+		credentials := fmt.Sprintf("%s:%s", d.config.esUsername, d.config.esPassword)
+		encodedCredentials := base64.StdEncoding.EncodeToString([]byte(credentials))
+		return fmt.Sprintf("Basic %s", encodedCredentials)
+	}
+
 }
 
 func (d *ESProvider) UploadAnswerAndEmbedding(

plugins/wasm-go/extensions/ai-cache/vector/pinecone.go

@@ -15,17 +15,14 @@ type pineconeProviderInitializer struct{}
 
 func (c *pineconeProviderInitializer) ValidateConfig(config ProviderConfig) error {
 	if len(config.serviceHost) == 0 {
-		return errors.New("[Pinecone] serviceDomain is required")
+		return errors.New("[Pinecone] serviceHost is required")
 	}
 	if len(config.serviceName) == 0 {
 		return errors.New("[Pinecone] serviceName is required")
 	}
 	if len(config.apiKey) == 0 {
 		return errors.New("[Pinecone] apiKey is required")
 	}
-	if len(config.collectionID) == 0 {
-		return errors.New("[Pinecone] collectionID is required")
-	}
 	return nil
 }
 

plugins/wasm-go/extensions/ai-cache/vector/provider.go

@@ -26,8 +26,8 @@ var (
 	providerInitializers = map[string]providerInitializer{
 		PROVIDER_TYPE_DASH_VECTOR: &dashVectorProviderInitializer{},
 		PROVIDER_TYPE_CHROMA:      &chromaProviderInitializer{},
-		PROVIDER_TYPE_ES:          &weaviateProviderInitializer{},
-		PROVIDER_TYPE_WEAVIATE:    &esProviderInitializer{},
+		PROVIDER_TYPE_ES:          &esProviderInitializer{},
+		PROVIDER_TYPE_WEAVIATE:    &weaviateProviderInitializer{},
 		PROVIDER_TYPE_PINECONE:    &pineconeProviderInitializer{},
 		PROVIDER_TYPE_QDRANT:      &qdrantProviderInitializer{},
 		PROVIDER_TYPE_MILVUS:      &milvusProviderInitializer{},

plugins/wasm-go/extensions/ai-cache/vector/weaviate.go

@@ -88,6 +88,7 @@ func (d *WeaviateProvider) QueryEmbedding(
 		"/v1/graphql",
 		[][2]string{
 			{"Content-Type", "application/json"},
+			{"Authorization", fmt.Sprintf("Bearer %s", d.config.apiKey)},
 		},
 		requestBody,
 		func(statusCode int, responseHeaders http.Header, responseBody []byte) {
@@ -128,6 +129,7 @@ func (d *WeaviateProvider) UploadAnswerAndEmbedding(
 		"/v1/objects",
 		[][2]string{
 			{"Content-Type", "application/json"},
+			{"Authorization", fmt.Sprintf("Bearer %s", d.config.apiKey)},
 		},
 		requestBody,
 		func(statusCode int, responseHeaders http.Header, responseBody []byte) {