rag support

Signed-off-by: Christopher Petito <chrisjpetito@gmail.com>

Author: krissetto

cmd/root/new.go

@@ -68,7 +68,7 @@ func (f *newFlags) runNewCommand(cmd *cobra.Command, args []string) error {
 
 	sess := session.New(opts...)
 
-	a := app.New("", rt, sess, prompt)
+	a := app.New(ctx, "", rt, sess, prompt)
 	m := tui.New(a)
 
 	progOpts := []tea.ProgramOption{

cmd/root/run.go

@@ -240,7 +240,7 @@ func handleRunMode(ctx context.Context, agentFilename string, rt runtime.Runtime
 		return err
 	}
 
-	a := app.New(agentFilename, rt, sess, firstMessage)
+	a := app.New(ctx, agentFilename, rt, sess, firstMessage)
 	m := tui.New(a)
 
 	progOpts := []tea.ProgramOption{

pkg/app/app.go

@@ -22,15 +22,26 @@ type App struct {
 	cancel           context.CancelFunc
 }
 
-func New(agentFilename string, rt runtime.Runtime, sess *session.Session, firstMessage *string) *App {
-	return &App{
+func New(ctx context.Context, agentFilename string, rt runtime.Runtime, sess *session.Session, firstMessage *string) *App {
+	app := &App{
 		agentFilename:    agentFilename,
 		runtime:          rt,
 		session:          sess,
 		firstMessage:     firstMessage,
 		events:           make(chan tea.Msg, 128),
 		throttleDuration: 50 * time.Millisecond, // Throttle rapid events
 	}
+
+	// If the runtime supports background RAG initialization, start it
+	// and forward events to the TUI. Remote runtimes typically handle RAG server-side
+	// and won't implement this optional interface.
+	if ragRuntime, ok := rt.(runtime.RAGInitializer); ok {
+		go ragRuntime.StartBackgroundRAGInit(ctx, func(event runtime.Event) {
+			app.events <- event
+		})
+	}
+
+	return app
 }
 
 func (a *App) FirstMessage() *string {

pkg/config/auto.go

@@ -56,3 +56,22 @@ func PreferredMaxTokens(provider string) int {
 	}
 	return 64000
 }
+
+// AutoEmbeddingModelConfigs returns the ordered list of embedding-capable models
+// to try when a RAG strategy uses `model: auto` for embeddings.
+//
+// The priority is:
+//  1. OpenAI -> text-embedding-3-small model
+//  2. DMR -> Google's embeddinggemma model (via Docker Model Runner)
+func AutoEmbeddingModelConfigs() []latest.ModelConfig {
+	return []latest.ModelConfig{
+		{
+			Provider: "openai",
+			Model:    "text-embedding-3-small",
+		},
+		{
+			Provider: "dmr",
+			Model:    "ai/embeddinggemma",
+		},
+	}
+}

pkg/config/overrides.go

@@ -81,29 +81,63 @@ func ensureModelsExist(cfg *v2.Config) error {
 		cfg.Models = map[string]v2.ModelConfig{}
 	}
 
+	// Ensure models referenced by agents exist
 	for agentName := range cfg.Agents {
 		agentConfig := cfg.Agents[agentName]
 
 		modelNames := strings.SplitSeq(agentConfig.Model, ",")
 		for modelName := range modelNames {
-			if modelName == "auto" {
-				continue
+			if err := ensureSingleModelExists(cfg, modelName, fmt.Sprintf("agent '%s'", agentName)); err != nil {
+				return err
 			}
-			if _, exists := cfg.Models[modelName]; exists {
+		}
+	}
+
+	// Ensure models referenced by RAG strategies exist
+	for ragName, ragCfg := range cfg.RAG {
+		for _, stratCfg := range ragCfg.Strategies {
+			rawModel, ok := stratCfg.Params["model"]
+			if !ok {
 				continue
 			}
 
-			providerName, model, ok := strings.Cut(modelName, "/")
+			modelName, ok := rawModel.(string)
 			if !ok {
-				return fmt.Errorf("agent '%s' references non-existent model '%s'", agentName, modelName)
+				return fmt.Errorf("RAG strategy '%s' in RAG '%s' has non-string model value", stratCfg.Type, ragName)
 			}
 
-			cfg.Models[modelName] = v2.ModelConfig{
-				Provider: providerName,
-				Model:    model,
+			if err := ensureSingleModelExists(cfg, modelName, fmt.Sprintf("RAG strategy '%s' in RAG '%s'", stratCfg.Type, ragName)); err != nil {
+				return err
 			}
 		}
 	}
 
 	return nil
 }
+
+// ensureSingleModelExists normalizes shorthand model IDs like "openai/gpt-5-mini"
+// into full entries in cfg.Models so they can be reused by agents, RAG, and other
+// subsystems without duplicating parsing logic.
+func ensureSingleModelExists(cfg *v2.Config, modelName, context string) error {
+	modelName = strings.TrimSpace(modelName)
+	if modelName == "" || modelName == "auto" {
+		// "auto" is handled dynamically at runtime and does not need a config entry.
+		return nil
+	}
+
+	if _, exists := cfg.Models[modelName]; exists {
+		return nil
+	}
+
+	providerName, model, ok := strings.Cut(modelName, "/")
+	if !ok || providerName == "" || model == "" {
+		return fmt.Errorf("%s references non-existent model '%s'", context, modelName)
+	}
+
+	cfg.Models[modelName] = v2.ModelConfig{
+		Provider: providerName,
+		Model:    model,
+	}
+
+	return nil
+}