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

fix(knowledge): add missing progress bar update (#510)

This commit is contained in:
frankj 2025-08-04 21:30:10 +08:00 committed by GitHub
parent 72656e4fd1
commit 38b63f00a3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/infra/impl/document/searchstore/vikingdb/vikingdb_searchstore.go
View File

@ -72,6 +72,12 @@ func (v *vkSearchStore) Store(ctx context.Context, docs []*schema.Document, opts
if err := v.collection.UpsertData(docsWithVector); err != nil {
return nil, err
}
if implSpecOptions.ProgressBar != nil {
if err = implSpecOptions.ProgressBar.AddN(len(part)); err != nil {
return nil, err
}
}
}
ids = slices.Transform(docs, func(a *schema.Document) string { return a.ID })