Function ragEventForwarder

pkg/runtime/rag.go:11–39

ragEventForwarder returns a callback that converts RAG manager events to runtime events.

(ragName string, r *LocalRuntime, sendEvent func(Event))

// ragEventForwarder returns a callback that converts RAG manager events to runtime events.
func ragEventForwarder(ragName string, r *LocalRuntime, sendEvent func(Event)) ragtypes.EventCallback {
	return func(ragEvent ragtypes.Event) {
		agentName := r.currentAgentName()
		slog.Debug("Forwarding RAG event", "type", ragEvent.Type, "rag", ragName, "agent", agentName)

		switch ragEvent.Type {
		case ragtypes.EventTypeIndexingStarted:
			sendEvent(RAGIndexingStarted(ragName, ragEvent.StrategyName))
		case ragtypes.EventTypeIndexingProgress:
			if ragEvent.Progress != nil {
				sendEvent(RAGIndexingProgress(ragName, ragEvent.StrategyName, ragEvent.Progress.Current, ragEvent.Progress.Total, agentName))
			}
		case ragtypes.EventTypeIndexingComplete:
			sendEvent(RAGIndexingCompleted(ragName, ragEvent.StrategyName))
		case ragtypes.EventTypeUsage:
			sendEvent(NewTokenUsageEvent("", agentName, &Usage{
				InputTokens:   ragEvent.TotalTokens,
				ContextLength: ragEvent.TotalTokens,
				Cost:          ragEvent.Cost,
			}))
		case ragtypes.EventTypeError:
			if ragEvent.Error != nil {
				sendEvent(Error(fmt.Sprintf("RAG %s error: %v", ragName, ragEvent.Error)))
			}
		default:
			slog.Debug("Unhandled RAG event type", "type", ragEvent.Type, "rag", ragName)
		}
	}
}