fix: [2.5] bulk insert should use function runner's input field list instead schema's (#41561)

aoiasd · web-flow · commit 8af350d9db11 · 2025-04-27T22:16:40.000+08:00
relate: #41213 pr: #41560 Signed-off-by: aoiasd <zhicheng.yue@zilliz.com>
diff --git a/internal/datanode/importv2/util.go b/internal/datanode/importv2/util.go
@@ -214,8 +214,10 @@ func RunEmbeddingFunction(task *ImportTask, data *storage.InsertData) error {
 		if err != nil {
 			return err
 		}
-		inputDatas := make([]any, 0, len(fn.InputFieldIds))
-		for _, inputFieldID := range fn.InputFieldIds {
+
+		inputFieldIDs := lo.Map(runner.GetInputFields(), func(field *schemapb.FieldSchema, _ int) int64 { return field.GetFieldID() })
+		inputDatas := make([]any, 0, len(inputFieldIDs))
+		for _, inputFieldID := range inputFieldIDs {
 			inputDatas = append(inputDatas, data.Data[inputFieldID].GetDataRows())
 		}
 		outputFieldData, err := runner.BatchRun(inputDatas...)

Original file line number	Diff line number	Diff line change
`@@ -214,8 +214,10 @@ func RunEmbeddingFunction(task ImportTask, data storage.InsertData) error {`
`214`	`214`	`if err != nil {`
`215`	`215`	`return err`
`216`	`216`	`}`
`217`		`- inputDatas := make([]any, 0, len(fn.InputFieldIds))`
`218`		`- for _, inputFieldID := range fn.InputFieldIds {`
	`217`	`+`
	`218`	`+ inputFieldIDs := lo.Map(runner.GetInputFields(), func(field *schemapb.FieldSchema, _ int) int64 { return field.GetFieldID() })`
	`219`	`+ inputDatas := make([]any, 0, len(inputFieldIDs))`
	`220`	`+ for _, inputFieldID := range inputFieldIDs {`
`219`	`221`	`inputDatas = append(inputDatas, data.Data[inputFieldID].GetDataRows())`
`220`	`222`	`}`
`221`	`223`	`outputFieldData, err := runner.BatchRun(inputDatas...)`