nullkey
0e081aec5c
Operability surface and the bulk of the jopts→fopts migration: * --log-level error|warn|info|debug + WEKNORA_LOG_LEVEL env, wired to the SDK via client.SetDebugLevel. Invalid --log-level returns FlagError (exit 2). * kb status <kb-id> / kb check <kb-id> verb split (1 HTTP vs 1+N for failed_count aggregation). * agent status <agent-id> / agent check <agent-id> verb split (probes kb_scope_all_reachable via 1+N HTTP). * kb create <name> positional (matches agent create). * Positional id help strings namespaced (<kb-id> / <agent-id>). * All auth / context / link / doctor / kb / agent CRUD commands migrated to the FormatOptions API. * root.go Execute(ctx) takes a context so signal-cancellation propagates via cmd.Context() into long-running commands. * Pagination termination uses len(accum) >= total (not page*pageSize) so server-capped page sizes do not truncate aggregations.
120 lines
3.7 KiB
Go
120 lines
3.7 KiB
Go
package kb
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/Tencent/WeKnora/cli/internal/cmdutil"
|
|
"github.com/Tencent/WeKnora/cli/internal/iostreams"
|
|
sdk "github.com/Tencent/WeKnora/client"
|
|
)
|
|
|
|
// StatusResult is the health-oriented response for `kb status <id>`.
|
|
// Shallow read only: 1 HTTP call, no failed-doc aggregation.
|
|
// For deep verification including failed_count, use `kb check <id>`.
|
|
type StatusResult struct {
|
|
ID string `json:"id"`
|
|
Reachable bool `json:"reachable"`
|
|
KnowledgeCount int64 `json:"knowledge_count,omitempty"`
|
|
ChunkCount int64 `json:"chunk_count,omitempty"`
|
|
IsProcessing bool `json:"is_processing,omitempty"`
|
|
ProcessingCount int64 `json:"processing_count,omitempty"`
|
|
}
|
|
|
|
// StatusService is the narrow SDK surface needed for kb status.
|
|
type StatusService interface {
|
|
GetKnowledgeBase(ctx context.Context, id string) (*sdk.KnowledgeBase, error)
|
|
}
|
|
|
|
var kbStatusFields = []string{
|
|
"id", "reachable", "knowledge_count", "chunk_count",
|
|
"is_processing", "processing_count",
|
|
}
|
|
|
|
// NewCmdStatus builds `weknora kb status <id>`.
|
|
func NewCmdStatus(f *cmdutil.Factory) *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "status <kb-id>",
|
|
Short: "Show health status of a knowledge base (shallow, 1 HTTP)",
|
|
Long: `Show health-oriented fields for a KB.
|
|
|
|
1 HTTP call:
|
|
reachable / knowledge_count / chunk_count / is_processing / processing_count
|
|
|
|
For deep verification including failed_count, use 'weknora kb check <id>'
|
|
(1 + N HTTP, pages the doc list with parse_status=failed).
|
|
|
|
For full metadata (config / pinned / tenant), use 'weknora kb view <id>'.`,
|
|
Example: ` weknora kb status kb_abc
|
|
weknora kb status kb_abc --format json`,
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(c *cobra.Command, args []string) error {
|
|
fopts, err := cmdutil.CheckFormatFlag(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fopts.ResolveDefault(iostreams.IO.IsStdoutTTY())
|
|
cli, err := f.Client()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
res, err := runStatus(c.Context(), cli, args[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return emitStatus(res, fopts, iostreams.IO.Out)
|
|
},
|
|
}
|
|
cmdutil.AddFormatFlag(cmd, kbStatusFields...)
|
|
return cmd
|
|
}
|
|
|
|
// runStatus is the testable core: fetch KB metadata and return a StatusResult.
|
|
// Never returns an error for "kb not reachable" (Reachable=false carries
|
|
// that signal).
|
|
func runStatus(ctx context.Context, svc StatusService, id string) (*StatusResult, error) {
|
|
kb, err := svc.GetKnowledgeBase(ctx, id)
|
|
if err != nil {
|
|
return &StatusResult{ID: id, Reachable: false}, nil
|
|
}
|
|
return &StatusResult{
|
|
ID: kb.ID,
|
|
Reachable: true,
|
|
KnowledgeCount: kb.KnowledgeCount,
|
|
ChunkCount: kb.ChunkCount,
|
|
IsProcessing: kb.IsProcessing,
|
|
ProcessingCount: kb.ProcessingCount,
|
|
}, nil
|
|
}
|
|
|
|
// emitStatus renders res using --format options. Mirrors emitWaitResult
|
|
// pattern from cli/cmd/doc/wait.go for consistency.
|
|
func emitStatus(res *StatusResult, fopts *cmdutil.FormatOptions, w io.Writer) error {
|
|
switch fopts.Mode {
|
|
case cmdutil.FormatJSON, cmdutil.FormatNDJSON:
|
|
return fopts.Emit(w, res)
|
|
case cmdutil.FormatText, "":
|
|
return writeStatusText(w, res)
|
|
default:
|
|
return fmt.Errorf("unsupported --format %q for kb status", fopts.Mode)
|
|
}
|
|
}
|
|
|
|
func writeStatusText(w io.Writer, res *StatusResult) error {
|
|
fmt.Fprintf(w, "ID: %s\n", res.ID)
|
|
fmt.Fprintf(w, "Reachable: %v\n", res.Reachable)
|
|
if !res.Reachable {
|
|
return nil
|
|
}
|
|
fmt.Fprintf(w, "Knowledge: %d\n", res.KnowledgeCount)
|
|
fmt.Fprintf(w, "Chunks: %d\n", res.ChunkCount)
|
|
fmt.Fprintf(w, "Processing: %v (%d active)\n", res.IsProcessing, res.ProcessingCount)
|
|
return nil
|
|
}
|
|
|
|
// compile-time check: SDK client satisfies StatusService.
|
|
var _ StatusService = (*sdk.Client)(nil)
|