1bf130b25a
Sync CLI module imports across all command packages. Refactor daemon command with expanded functionality. Update go.mod and go.work dependencies. Co-Authored-By: Virgil <virgil@lethean.io>
41 lines
1.4 KiB
Go
41 lines
1.4 KiB
Go
package ml
|
|
|
|
import (
|
|
"forge.lthn.ai/core/go/pkg/cli"
|
|
"forge.lthn.ai/core/go-ai/ml"
|
|
)
|
|
|
|
var (
|
|
consolidateM3Host string
|
|
consolidateRemoteDir string
|
|
consolidatePattern string
|
|
consolidateOutputDir string
|
|
consolidateMergedOut string
|
|
)
|
|
|
|
var consolidateCmd = &cli.Command{
|
|
Use: "consolidate",
|
|
Short: "Pull and merge response JSONL files from M3",
|
|
Long: "Pulls JSONL response files from M3 via SSH/SCP, merges them by idx, deduplicates, and writes a single merged JSONL output.",
|
|
RunE: runConsolidate,
|
|
}
|
|
|
|
func init() {
|
|
consolidateCmd.Flags().StringVar(&consolidateM3Host, "m3-host", "m3", "M3 SSH host")
|
|
consolidateCmd.Flags().StringVar(&consolidateRemoteDir, "remote", "/Volumes/Data/lem/responses", "Remote response directory")
|
|
consolidateCmd.Flags().StringVar(&consolidatePattern, "pattern", "gold*.jsonl", "File glob pattern")
|
|
consolidateCmd.Flags().StringVar(&consolidateOutputDir, "output", "", "Local output directory (default: responses)")
|
|
consolidateCmd.Flags().StringVar(&consolidateMergedOut, "merged", "", "Merged output path (default: gold-merged.jsonl in parent of output dir)")
|
|
}
|
|
|
|
func runConsolidate(cmd *cli.Command, args []string) error {
|
|
cfg := ml.ConsolidateConfig{
|
|
M3Host: consolidateM3Host,
|
|
RemoteDir: consolidateRemoteDir,
|
|
Pattern: consolidatePattern,
|
|
OutputDir: consolidateOutputDir,
|
|
MergedOut: consolidateMergedOut,
|
|
}
|
|
|
|
return ml.Consolidate(cfg, cmd.OutOrStdout())
|
|
}
|