Snider
abe74a1a3d
- Change module from forge.lthn.ai/core/go to forge.lthn.ai/core/cli - Remove pkg/ directory (now served from core/go) - Add require + replace for forge.lthn.ai/core/go => ../go - Update go.work to include ../go workspace module - Fix all internal/cmd/* imports: pkg/ refs → forge.lthn.ai/core/go/pkg/ - Rename internal/cmd/sdk package to sdkcmd (avoids conflict with pkg/sdk) - Remove SDK library files from internal/cmd/sdk/ (now in core/go/pkg/sdk/) - Remove duplicate RAG helper functions from internal/cmd/rag/ - Remove stale cmd/core-ide/ (now in core/ide repo) - Update IDE variant to remove core-ide import - Fix test assertion for new module name - Run go mod tidy to sync dependencies core/cli is now a pure CLI application importing core/go for packages. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> Co-authored-by: Claude <developers@lethean.io> Reviewed-on: #1
84 lines
2.9 KiB
Go
84 lines
2.9 KiB
Go
package rag
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
|
|
"forge.lthn.ai/core/go/pkg/i18n"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// Shared flags
|
|
var (
|
|
qdrantHost string
|
|
qdrantPort int
|
|
ollamaHost string
|
|
ollamaPort int
|
|
model string
|
|
verbose bool
|
|
)
|
|
|
|
var ragCmd = &cobra.Command{
|
|
Use: "rag",
|
|
Short: i18n.T("cmd.rag.short"),
|
|
Long: i18n.T("cmd.rag.long"),
|
|
}
|
|
|
|
func initFlags() {
|
|
// Qdrant connection flags (persistent) - defaults to localhost for local development
|
|
qHost := "localhost"
|
|
if v := os.Getenv("QDRANT_HOST"); v != "" {
|
|
qHost = v
|
|
}
|
|
ragCmd.PersistentFlags().StringVar(&qdrantHost, "qdrant-host", qHost, i18n.T("cmd.rag.flag.qdrant_host"))
|
|
|
|
qPort := 6334
|
|
if v := os.Getenv("QDRANT_PORT"); v != "" {
|
|
if p, err := strconv.Atoi(v); err == nil {
|
|
qPort = p
|
|
}
|
|
}
|
|
ragCmd.PersistentFlags().IntVar(&qdrantPort, "qdrant-port", qPort, i18n.T("cmd.rag.flag.qdrant_port"))
|
|
|
|
// Ollama connection flags (persistent) - defaults to localhost for local development
|
|
oHost := "localhost"
|
|
if v := os.Getenv("OLLAMA_HOST"); v != "" {
|
|
oHost = v
|
|
}
|
|
ragCmd.PersistentFlags().StringVar(&ollamaHost, "ollama-host", oHost, i18n.T("cmd.rag.flag.ollama_host"))
|
|
|
|
oPort := 11434
|
|
if v := os.Getenv("OLLAMA_PORT"); v != "" {
|
|
if p, err := strconv.Atoi(v); err == nil {
|
|
oPort = p
|
|
}
|
|
}
|
|
ragCmd.PersistentFlags().IntVar(&ollamaPort, "ollama-port", oPort, i18n.T("cmd.rag.flag.ollama_port"))
|
|
|
|
m := "nomic-embed-text"
|
|
if v := os.Getenv("EMBEDDING_MODEL"); v != "" {
|
|
m = v
|
|
}
|
|
ragCmd.PersistentFlags().StringVar(&model, "model", m, i18n.T("cmd.rag.flag.model"))
|
|
|
|
// Verbose flag (persistent)
|
|
ragCmd.PersistentFlags().BoolVarP(&verbose, "verbose", "v", false, i18n.T("common.flag.verbose"))
|
|
|
|
// Ingest command flags
|
|
ingestCmd.Flags().StringVar(&collection, "collection", "hostuk-docs", i18n.T("cmd.rag.ingest.flag.collection"))
|
|
ingestCmd.Flags().BoolVar(&recreate, "recreate", false, i18n.T("cmd.rag.ingest.flag.recreate"))
|
|
ingestCmd.Flags().IntVar(&chunkSize, "chunk-size", 500, i18n.T("cmd.rag.ingest.flag.chunk_size"))
|
|
ingestCmd.Flags().IntVar(&chunkOverlap, "chunk-overlap", 50, i18n.T("cmd.rag.ingest.flag.chunk_overlap"))
|
|
|
|
// Query command flags
|
|
queryCmd.Flags().StringVar(&queryCollection, "collection", "hostuk-docs", i18n.T("cmd.rag.query.flag.collection"))
|
|
queryCmd.Flags().IntVar(&limit, "top", 5, i18n.T("cmd.rag.query.flag.top"))
|
|
queryCmd.Flags().Float32Var(&threshold, "threshold", 0.5, i18n.T("cmd.rag.query.flag.threshold"))
|
|
queryCmd.Flags().StringVar(&category, "category", "", i18n.T("cmd.rag.query.flag.category"))
|
|
queryCmd.Flags().StringVar(&format, "format", "text", i18n.T("cmd.rag.query.flag.format"))
|
|
|
|
// Collections command flags
|
|
collectionsCmd.Flags().BoolVar(&listCollections, "list", false, i18n.T("cmd.rag.collections.flag.list"))
|
|
collectionsCmd.Flags().BoolVar(&showStats, "stats", false, i18n.T("cmd.rag.collections.flag.stats"))
|
|
collectionsCmd.Flags().StringVar(&deleteCollection, "delete", "", i18n.T("cmd.rag.collections.flag.delete"))
|
|
}
|