mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-29 15:05:31 +08:00
a789c60380
issue: #22837 contain https://github.com/milvus-io/milvus/pull/33625 https://github.com/milvus-io/milvus/pull/33867 https://github.com/milvus-io/milvus/pull/33911 which already merged to 2.4 branch Signed-off-by: chasingegg <chao.gao@zilliz.com> Co-authored-by: foxspy <xianliang.li@zilliz.com>
45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package indexparamcheck
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
|
|
"github.com/milvus-io/milvus/pkg/common"
|
|
"github.com/milvus-io/milvus/pkg/util/typeutil"
|
|
)
|
|
|
|
type floatVectorBaseChecker struct {
|
|
baseChecker
|
|
}
|
|
|
|
func (c floatVectorBaseChecker) staticCheck(params map[string]string) error {
|
|
if !CheckStrByValues(params, Metric, FloatVectorMetrics) {
|
|
return fmt.Errorf("metric type %s not found or not supported, supported: %v", params[Metric], FloatVectorMetrics)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c floatVectorBaseChecker) CheckTrain(params map[string]string) error {
|
|
if err := c.baseChecker.CheckTrain(params); err != nil {
|
|
return err
|
|
}
|
|
|
|
return c.staticCheck(params)
|
|
}
|
|
|
|
func (c floatVectorBaseChecker) CheckValidDataType(dType schemapb.DataType) error {
|
|
if !typeutil.IsDenseFloatVectorType(dType) {
|
|
return fmt.Errorf("data type should be FloatVector, Float16Vector or BFloat16Vector")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c floatVectorBaseChecker) SetDefaultMetricTypeIfNotExist(params map[string]string, dType schemapb.DataType) {
|
|
setDefaultIfNotExist(params, common.MetricTypeKey, FloatVectorDefaultMetricType)
|
|
}
|
|
|
|
func newFloatVectorBaseChecker() IndexChecker {
|
|
return &floatVectorBaseChecker{}
|
|
}
|