types

type Document ¶

type Document struct {
	Content    string            // The document text/chunk content
	SourcePath string            // File path or document identifier
	Metadata   map[string]string // Custom metadata (e.g., date, author, type, tags)
	ChunkIndex int               // Position of this chunk within the source document (0-based)
}

Document represents a document for reranking with content and metadata This is a generic structure that can be used across the RAG system

type Event ¶

type Event struct {
	Type         EventTye
	StrategyName string // Name of the component emitting the event (strategy name, "reranker", "fusion", etc.)
	Message      string
	Progress     *Progress
	Error        error
	TotalTokens  int64   // For usage events
	Cost         float64 // For usage events
}

Event represents a RAG operation lifecycle event. This is the canonical RAG event type used by strategies, reranking, fusion, the RAG manager, and the runtime.

type EventCallback ¶ added in v1.83.0

type EventCallback func(event Event)

EventCallback receives RAG manager events during initialization.

type EventForwarder ¶ added in v1.83.0

type EventForwarder interface {
	// Name returns the toolset's user-facing name.
	Name() string
	// SetEventCallback registers a single callback, replacing any previous
	// one. Hosts that may share the toolset should use EventSubscriber.
	SetEventCallback(callback EventCallback)
}

EventForwarder is implemented by RAG toolsets that can stream lifecycle events (indexing progress, usage, errors) to a callback. The runtime type-asserts to this interface to wire event forwarding without importing the concrete rag toolset package — which keeps the cgo tree-sitter dependency out of the runtime's import graph.

type EventSubscriber ¶ added in v1.140.0

type EventSubscriber interface {
	Name() string
	SubscribeEvents(callback EventCallback) (unsubscribe func())
}

EventSubscriber is implemented by RAG toolsets that fan lifecycle events out to any number of hosts (one per runtime stream sharing the toolset). Subscriptions may be added before or after Start; every subscriber receives every event emitted while it is registered.

type EventTye ¶

type EventTye string

const (
	EventTypeIndexingStarted  EventTye = "indexing_started"
	EventTypeIndexingProgress EventTye = "indexing_progress"
	EventTypeIndexingComplete EventTye = "indexing_complete"
	EventTypeUsage            EventTye = "usage"
	EventTypeError            EventTye = "error"
)

type Progress ¶

type Progress struct {
	Current int
	Total   int
}

Progress represents progress within a multi-step operation (e.g., indexing, reranking).

type Usage ¶ added in v1.112.0

type Usage struct {
	TotalTokens int64
	Cost        float64
	ModelID     string
}

func (u *Usage) Add(other Usage)