embedding

type CacheKey struct {
	Text   string
	Model  string
	Method string
}

type CacheStats struct {
	Enabled    bool
	RedisKeys  int // Number of keys in Redis
	MemoryKeys int // Number of keys in memory
	TotalKeys  int // Total number of keys
	TTL        time.Duration
}

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

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

type EmbeddingService interface {
	// Embed generates vector embedding for a query text (uses "query" prefix).
	// For document storage, use EmbedWithPrefix with "passage:" prefix.
	//
	// Args:
	//   ctx - operation context.
	//   text - text to embed.
	//
	// Returns:
	//   []float64 - embedding vector.
	//   error - any error encountered.
	Embed(ctx context.Context, text string) ([]float64, error)

	// EmbedWithPrefix generates vector embedding with custom prefix.
	// Use "query:" for search queries and "passage:" for document storage.
	//
	// Args:
	//   ctx - operation context.
	//   text - text to embed.
	//   prefix - prefix to add before text (e.g., "query:", "passage:").
	//
	// Returns:
	//   []float64 - embedding vector.
	//   error - any error encountered.
	EmbedWithPrefix(ctx context.Context, text, prefix string) ([]float64, error)

	// EmbedBatch generates vector embeddings for multiple texts.
	//
	// Args:
	//   ctx - operation context.
	//   texts - texts to embed.
	//
	// Returns:
	//   [][]float64 - embedding vectors for each text.
	//   error - any error encountered.
	EmbedBatch(ctx context.Context, texts []string) ([][]float64, error)

	// HealthCheck checks if the embedding service is healthy.
	//
	// Args:
	//   ctx - operation context.
	//
	// Returns:
	//   error - any error encountered, nil if healthy.
	HealthCheck(ctx context.Context) error

	// GetModel returns the embedding model name.
	//
	// Returns:
	//   string - the model name.
	GetModel() string

	// GetTimeout returns the embedding timeout.
	//
	// Returns:
	//   time.Duration - the timeout duration.
	GetTimeout() time.Duration
}

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

type FallbackStrategy int

type HTTPError struct {
	StatusCode int
	Message    string
}

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

type RedisClient interface {
	Get(ctx context.Context, key string) (string, error)
	Set(ctx context.Context, key string, value interface{}, expiration time.Duration) error
	Del(ctx context.Context, keys ...string) error
	Keys(ctx context.Context, pattern string) ([]string, error)
}