c5bc97de19
Ports all remaining LEM pipeline commands from pkg/lem into core ml, eliminating the standalone LEM CLI dependency. Each command is split into reusable business logic (pkg/ml/) and a thin cobra wrapper (internal/cmd/ml/). New commands: query, inventory, metrics, ingest, normalize, seed-influx, consolidate, import-all, approve, publish, coverage. Adds Path(), Exec(), QueryRowScan() convenience methods to DB type. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package ml
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"forge.lthn.ai/core/cli/pkg/cli"
|
|
"forge.lthn.ai/core/cli/pkg/ml"
|
|
)
|
|
|
|
var importCmd = &cli.Command{
|
|
Use: "import-all",
|
|
Short: "Import all LEM data into DuckDB",
|
|
Long: "Imports golden set, training examples, benchmark results, benchmark questions, and seeds into DuckDB from M3 and local files.",
|
|
RunE: runImportAll,
|
|
}
|
|
|
|
var (
|
|
importSkipM3 bool
|
|
importDataDir string
|
|
importM3Host string
|
|
)
|
|
|
|
func init() {
|
|
importCmd.Flags().BoolVar(&importSkipM3, "skip-m3", false, "Skip pulling data from M3")
|
|
importCmd.Flags().StringVar(&importDataDir, "data-dir", "", "Local data directory (defaults to db directory)")
|
|
importCmd.Flags().StringVar(&importM3Host, "m3-host", "m3", "M3 SSH host alias")
|
|
}
|
|
|
|
func runImportAll(cmd *cli.Command, args []string) error {
|
|
path := dbPath
|
|
if path == "" {
|
|
path = os.Getenv("LEM_DB")
|
|
}
|
|
if path == "" {
|
|
return fmt.Errorf("--db or LEM_DB required")
|
|
}
|
|
|
|
dataDir := importDataDir
|
|
if dataDir == "" {
|
|
dataDir = filepath.Dir(path)
|
|
}
|
|
|
|
db, err := ml.OpenDBReadWrite(path)
|
|
if err != nil {
|
|
return fmt.Errorf("open db: %w", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
cfg := ml.ImportConfig{
|
|
SkipM3: importSkipM3,
|
|
DataDir: dataDir,
|
|
M3Host: importM3Host,
|
|
}
|
|
|
|
return ml.ImportAll(db, cfg, cmd.OutOrStdout())
|
|
}
|