From fc98f000d86f1cca7d452bc21fee35d559e85556 Mon Sep 17 00:00:00 2001 From: wizardchen Date: Wed, 10 Jun 2026 15:17:51 +0800 Subject: [PATCH] feat(upload): add process configuration support to CreateKnowledgeFromFile - Updated the CreateKnowledgeFromFile function to accept an optional processConfig parameter, enhancing flexibility in knowledge file uploads. - Modified related service and test files to accommodate the new parameter, ensuring consistent handling across the upload process. --- cli/cmd/doc/upload.go | 3 ++- cli/cmd/doc/upload_recursive.go | 2 +- cli/cmd/doc/upload_recursive_test.go | 1 + cli/cmd/doc/upload_test.go | 1 + 4 files changed, 5 insertions(+), 2 deletions(-) diff --git a/cli/cmd/doc/upload.go b/cli/cmd/doc/upload.go index 822dea82..d733dc9d 100644 --- a/cli/cmd/doc/upload.go +++ b/cli/cmd/doc/upload.go @@ -62,6 +62,7 @@ type UploadService interface { metadata map[string]string, enableMultimodel *bool, customFileName, channel string, + processConfig *sdk.KnowledgeProcessOverrides, ) (*sdk.Knowledge, error) } @@ -294,7 +295,7 @@ func runUpload(ctx context.Context, opts *UploadOptions, fopts *cmdutil.FormatOp if err != nil { return err } - k, err := svc.CreateKnowledgeFromFile(ctx, kbID, path, meta, opts.EnableMultimodel, opts.Name, cmp.Or(opts.Channel, uploadChannel)) + k, err := svc.CreateKnowledgeFromFile(ctx, kbID, path, meta, opts.EnableMultimodel, opts.Name, cmp.Or(opts.Channel, uploadChannel), nil) if err != nil { if errors.Is(err, sdk.ErrDuplicateFile) { // SDK returns sentinel without an "HTTP error :" prefix diff --git a/cli/cmd/doc/upload_recursive.go b/cli/cmd/doc/upload_recursive.go index 97cf7f03..3c4cdbc4 100644 --- a/cli/cmd/doc/upload_recursive.go +++ b/cli/cmd/doc/upload_recursive.go @@ -85,7 +85,7 @@ func runUploadRecursive(ctx context.Context, opts *UploadOptions, fopts *cmdutil channel := cmp.Or(opts.Channel, uploadChannel) outcomes, runErr := cmdutil.RunBatch(ctx, matches, func(ctx context.Context, p string) error { - k, err := svc.CreateKnowledgeFromFile(ctx, kbID, p, meta, opts.EnableMultimodel, "", channel) + k, err := svc.CreateKnowledgeFromFile(ctx, kbID, p, meta, opts.EnableMultimodel, "", channel, nil) if err != nil { code := cmdutil.ClassifyHTTPError(err) if firstFailCode == "" { diff --git a/cli/cmd/doc/upload_recursive_test.go b/cli/cmd/doc/upload_recursive_test.go index 6a7dc502..d4a6454d 100644 --- a/cli/cmd/doc/upload_recursive_test.go +++ b/cli/cmd/doc/upload_recursive_test.go @@ -40,6 +40,7 @@ func (s *scriptedUploadSvc) CreateKnowledgeFromFile( metadata map[string]string, enableMultimodel *bool, _, channel string, + _ *sdk.KnowledgeProcessOverrides, ) (*sdk.Knowledge, error) { s.called = append(s.called, filepath.Base(filePath)) s.lastMetadata = metadata diff --git a/cli/cmd/doc/upload_test.go b/cli/cmd/doc/upload_test.go index 3058936a..f481fb10 100644 --- a/cli/cmd/doc/upload_test.go +++ b/cli/cmd/doc/upload_test.go @@ -34,6 +34,7 @@ func (f *fakeUploadSvc) CreateKnowledgeFromFile( metadata map[string]string, enableMultimodel *bool, customFileName, channel string, + _ *sdk.KnowledgeProcessOverrides, ) (*sdk.Knowledge, error) { f.got.kbID = kbID f.got.filePath = filePath