From 38b63f00a3f51e02b5bbe6be6cabe0e1fc0e8a68 Mon Sep 17 00:00:00 2001 From: frankj Date: Mon, 4 Aug 2025 21:30:10 +0800 Subject: [PATCH] fix(knowledge): add missing progress bar update (#510) --- .../document/searchstore/vikingdb/vikingdb_searchstore.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/backend/infra/impl/document/searchstore/vikingdb/vikingdb_searchstore.go b/backend/infra/impl/document/searchstore/vikingdb/vikingdb_searchstore.go index e71fe278..ca3b4067 100644 --- a/backend/infra/impl/document/searchstore/vikingdb/vikingdb_searchstore.go +++ b/backend/infra/impl/document/searchstore/vikingdb/vikingdb_searchstore.go @@ -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 })