middleware

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

type Cache interface {
	Get(ctx context.Context, key string) (*Response, bool)
	Set(ctx context.Context, key string, response *Response, ttl time.Duration) error
}

type CacheMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

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

type Logger interface {
	Info(ctx context.Context, message string, fields map[string]interface{})
	Debug(ctx context.Context, message string, fields map[string]interface{})
	Error(ctx context.Context, message string, err error, fields map[string]interface{})
}

type LoggingMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

type Metrics interface {
	RecordIndexLatency(ctx context.Context, duration time.Duration)
	RecordIndexCount(ctx context.Context, status string)
	RecordQueryLatency(ctx context.Context, duration time.Duration)
	RecordQueryCount(ctx context.Context, status string)
}

type MetricsMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

type Middleware interface {
	// Name returns the middleware name
	Name() string

	// BeforeIndex is called before indexing a document
	BeforeIndex(ctx context.Context, source *Source) error

	// AfterIndex is called after indexing a document
	AfterIndex(ctx context.Context, chunks []core.Chunk) error

	// BeforeQuery is called before executing a query
	BeforeQuery(ctx context.Context, query *Query) error

	// AfterQuery is called after executing a query
	AfterQuery(ctx context.Context, response *Response) error
}

type Query struct {
	Question string
	Options  map[string]interface{}
}

type RateLimitMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

type RateLimiter interface {
	Allow(ctx context.Context, key string) (bool, error)
}

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

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

type TransformMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

type ValidationMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}