admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix voyageai model int8 bug (#46819)

Browse Source 
https://github.com/milvus-io/milvus/issues/46815

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
This commit is contained in:
junjiejiangjjj 2026-01-07 10:41:29 +08:00 committed by GitHub
parent 53a300db83
commit 537199fb80
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
internal/util/function/models/voyageai/voyageai_client.go
View File

@ -128,10 +128,11 @@ func (c *voyageAIEmbedding) embedding(modelName string, texts []string, dim int,
return nil, fmt.Errorf("Voyageai: unsupport output type: [%s], only support float and int8", outputType)
}
r := EmbeddingRequest{
Model: modelName,
Input: texts,
InputType: textType,
Truncation: truncation,
Model: modelName,
Input: texts,
InputType: textType,
Truncation: truncation,
OutputDtype: outputType,
}
if dim != 0 {
r.OutputDimension = int64(dim)