Method Initialize

pkg/rag/manager.go:155–235

Initialize indexes all documents using all configured strategies Each strategy indexes its own document set (shared + strategy-specific) Strategies are initialized in parallel for better performance

(ctx context.Context)

// Each strategy indexes its own document set (shared + strategy-specific)
// Strategies are initialized in parallel for better performance
func (m *Manager) Initialize(ctx context.Context) (err error) {
	tracer := otel.Tracer("github.com/docker/docker-agent/pkg/rag")
	ctx, span := tracer.Start(ctx, "rag.initialize",
		trace.WithSpanKind(trace.SpanKindInternal),
		trace.WithAttributes(
			attribute.String(genai.AttrDataSourceID, m.name),
			attribute.Int("cagent.rag.num_strategies", len(m.strategies)),
		),
	)
	defer func() {
		if err != nil {
			span.RecordError(err)
			span.SetStatus(codes.Error, err.Error())
		}
		span.End()
	}()

	slog.DebugContext(ctx, "[RAG Manager] Starting initialization",
		"rag_name", m.name,
		"num_strategies", len(m.strategies))

	// Initialize strategies in parallel to avoid blocking
	type result struct {
		strategyName string
		err          error
	}

	resultsChan := make(chan result, len(m.strategies))

	for strategyName, strategyImpl := range m.strategies {
		strategyCfg := m.strategyConfigs[strategyName]

		go func() {
			slog.DebugContext(ctx, "[RAG Manager] Initializing strategy",
				"rag_name", m.name,
				"strategy", strategyName,
				"num_docs", len(strategyCfg.Docs),
				"chunk_size", strategyCfg.Chunking.Size,
				"chunk_overlap", strategyCfg.Chunking.Overlap,
				"respect_word_boundaries", strategyCfg.Chunking.RespectWordBoundaries,
				"code_aware", strategyCfg.Chunking.CodeAware)

			start := time.Now()
			err := strategyImpl.Initialize(ctx, strategyCfg.Docs, strategyCfg.Chunking)
			indexDuration := time.Since(start)
			slog.DebugContext(ctx, "[RAG Manager] Strategy indexing duration",
				"rag_name", m.name,
				"strategy", strategyName,
				"duration", indexDuration)
			if err != nil {
				slog.ErrorContext(ctx, "[RAG Manager] Strategy initialization failed",
					"rag_name", m.name,
					"strategy", strategyName,
					"error", err)
			} else {
				slog.InfoContext(ctx, "[RAG Manager] Strategy initialized successfully",
					"rag_name", m.name,
					"strategy", strategyName)