rag

type Cache interface {
	// Get retrieves a cached response for the given key
	Get(ctx context.Context, key string) (*Response, bool)
	// Set stores a response in the cache with the given key and expiration
	Set(ctx context.Context, key string, value *Response, expiration time.Duration)
	// Delete removes a cached response for the given key
	Delete(ctx context.Context, key string)
	// Clear removes all cached responses
	Clear(ctx context.Context)
}

type DefaultRouter struct{}

type Engine struct {
	// contains filtered or unexported fields
}

type MemoryCache struct {
	// contains filtered or unexported fields
}

type Option func(*Engine)

type QueryOptions struct {
	TopK           int
	PromptTemplate string
	Stream         bool
}

type Response struct {
	Answer  string
	Sources []vectorstore.Result
}

type RouteResult struct {
	// Type indicates the type of route (e.g., "vector", "keyword", "hybrid")
	Type string
	// Params contains additional parameters for the route
	Params map[string]interface{}
}

type Router interface {
	// Route determines the appropriate routing for a given query
	Route(ctx context.Context, query string) (RouteResult, error)
}

type Source struct {
	Type    string
	Path    string
	Content string
	Reader  interface{}
}

type StreamResponse struct {
	Chunk   string
	Sources []vectorstore.Result
	Done    bool
	Error   error
}