mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-07 01:28:27 +08:00
e56adc121b
https://github.com/milvus-io/milvus/issues/35856 Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
95 lines
3.2 KiB
Go
95 lines
3.2 KiB
Go
// Licensed to the LF AI & Data foundation under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package paramtable
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
type functionConfig struct {
|
|
TextEmbeddingProviders ParamGroup `refreshable:"true"`
|
|
}
|
|
|
|
func (p *functionConfig) init(base *BaseTable) {
|
|
p.TextEmbeddingProviders = ParamGroup{
|
|
KeyPrefix: "function.textEmbedding.providers.",
|
|
Version: "2.6.0",
|
|
Export: true,
|
|
DocFunc: func(key string) string {
|
|
switch key {
|
|
case "tei.enable":
|
|
return "Whether to enable TEI model service"
|
|
case "tei.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "azure_openai.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "azure_openai.url":
|
|
return "Your azure openai embedding url, Default is the official embedding url"
|
|
case "azure_openai.resource_name":
|
|
return "Your azure openai resource name"
|
|
case "openai.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "openai.url":
|
|
return "Your openai embedding url, Default is the official embedding url"
|
|
case "dashscope.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "dashscope.url":
|
|
return "Your dashscope embedding url, Default is the official embedding url"
|
|
case "cohere.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "cohere.url":
|
|
return "Your cohere embedding url, Default is the official embedding url"
|
|
case "voyageai.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "voyageai.url":
|
|
return "Your voyageai embedding url, Default is the official embedding url"
|
|
case "siliconflow.url":
|
|
return "Your siliconflow embedding url, Default is the official embedding url"
|
|
case "siliconflow.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "bedrock.credential":
|
|
return "The name in the crendential configuration item"
|
|
case "vertexai.url":
|
|
return "Your VertexAI embedding url"
|
|
case "vertexai.credential":
|
|
return "The name in the crendential configuration item"
|
|
default:
|
|
return ""
|
|
}
|
|
},
|
|
}
|
|
p.TextEmbeddingProviders.Init(base.mgr)
|
|
}
|
|
|
|
const (
|
|
textEmbeddingKey string = "textEmbedding"
|
|
)
|
|
|
|
func (p *functionConfig) GetTextEmbeddingProviderConfig(providerName string) map[string]string {
|
|
matchedParam := make(map[string]string)
|
|
|
|
params := p.TextEmbeddingProviders.GetValue()
|
|
prefix := providerName + "."
|
|
|
|
for k, v := range params {
|
|
if strings.HasPrefix(k, prefix) {
|
|
matchedParam[strings.TrimPrefix(k, prefix)] = v
|
|
}
|
|
}
|
|
return matchedParam
|
|
}
|