ben
/
coze-studio
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: ark embedding norm

This commit is contained in:
xuzhaonan 2025-07-23 11:07:31 +08:00 committed by 徐兆楠
parent 5d1c55ab1a
commit dc984f8afb
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/infra/impl/embedding/ark/ark.go
View File

@ -76,9 +76,8 @@ func (d embWrap) slicedNormL2(vectors [][]float64) ([][]float64, error) {
} }
if curDims := len(vectors[0]); curDims < int(d.dims) { if curDims := len(vectors[0]); curDims < int(d.dims) {
return nil, fmt.Errorf("[slicedNormL2] got dims=%d less than %d", curDims, d.dims) return nil, fmt.Errorf("[slicedNormL2] got dims=%d less than %d", curDims, d.dims)
} else if curDims == int(d.dims) {
return vectors, nil
} }
result := make([][]float64, len(vectors)) result := make([][]float64, len(vectors))
for i, vec := range vectors { for i, vec := range vectors {
sliced := vec[:d.dims] sliced := vec[:d.dims]