draftrag

type AnthropicLLMOptions struct {
	// BaseURL — базовый URL Anthropic API (например, "https://api.anthropic.com").
	BaseURL string
	// APIKey — ключ доступа. Передаётся в заголовке X-API-Key.
	APIKey string
	// Model — имя модели. Если пустая строка, используется claude-3-haiku-20240307.
	Model string
	// AnthropicVersion — версия API. Если пустая строка, используется "2023-06-01".
	AnthropicVersion string

	// Temperature — параметр генерации; если nil, параметр не передаётся в запросе.
	Temperature *float64
	// MaxTokens — лимит выходных токенов; если nil, используется дефолт (1024).
	MaxTokens *int

	// HTTPClient — опциональный клиент; если nil, используется http.DefaultClient.
	HTTPClient *http.Client
	// Timeout — опциональный таймаут на один вызов Generate (не применяется к GenerateStream).
	Timeout time.Duration
}

type BasicChunkerOptions struct {
	// ChunkSize — целевой размер чанка в рунах (обязательно > 0).
	ChunkSize int
	// Overlap — перекрытие между чанками в рунах (обязательно >= 0 и < ChunkSize).
	Overlap int
	// MaxChunks — максимальное количество возвращаемых чанков (>= 0). 0 означает “без лимита”.
	//
	// Если MaxChunks > 0, чанкер возвращает префикс первых MaxChunks чанков (best-effort, без ошибки).
	MaxChunks int
}

type CacheOptions struct {
	// MaxSize — максимальное количество записей в LRU-кэше.
	// 0 → 1000.
	MaxSize int

	// Redis — опциональный Redis L2 кэш.
	Redis RedisCacheOptions

	// Logger — опциональный структурированный логгер для событий кэша.
	// nil означает no-op.
	Logger Logger
}

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

type ChromaDBOptions struct {
	// BaseURL — базовый URL ChromaDB HTTP API. Если пустая строка, используется http://localhost:8000.
	BaseURL string
	// Collection — имя коллекции (обязательно).
	Collection string
	// Dimension — фиксированная размерность embedding-векторов (обязательно > 0).
	Dimension int
	// HTTP таймаут (по умолчанию: 10s)
	Timeout time.Duration
}

type Chunk = domain.Chunk

type Chunker = domain.Chunker

type CircuitBreakerStats = resilience.Stats

type CircuitState = resilience.CircuitState

type Document = domain.Document

type DocumentStore = domain.DocumentStore

type EmbedCacheStats = cache.Stats

type Embedder = domain.Embedder

type HookStage = domain.HookStage

type Hooks = domain.Hooks

type HybridConfig = domain.HybridConfig

type HybridSearcher = domain.HybridSearcher

type IndexBatchError = domain.IndexBatchError

type IndexBatchResult = domain.IndexBatchResult

type InlineCitation = domain.InlineCitation

type LLMProvider = domain.LLMProvider

type LogField = domain.LogField

type LogLevel = domain.LogLevel

type Logger = domain.Logger

type MetadataFilter = domain.MetadataFilter

type OllamaEmbedderOptions struct {
	// BaseURL — базовый URL Ollama API. Если пустая строка, используется http://localhost:11434.
	BaseURL string
	// Model — имя модели эмбеддингов (обязательно).
	Model string
	// APIKey — опциональный ключ доступа (для кастомных инстансов с авторизацией).
	APIKey string

	// HTTPClient — опциональный клиент; если nil, используется http.DefaultClient.
	HTTPClient *http.Client
	// Timeout — опциональный таймаут на один вызов Embed.
	Timeout time.Duration
}

type OllamaLLMOptions struct {
	// BaseURL — базовый URL Ollama API. Если пустая строка, используется http://localhost:11434.
	BaseURL string
	// Model — имя модели (обязательно).
	Model string
	// APIKey — опциональный ключ доступа (для кастомных инстансов с авторизацией).
	APIKey string

	// Temperature — параметр генерации; если nil, параметр не передаётся в запросе.
	Temperature *float64
	// MaxTokens — лимит выходных токенов; если nil, параметр не передаётся в запросе.
	MaxTokens *int

	// HTTPClient — опциональный клиент; если nil, используется http.DefaultClient.
	HTTPClient *http.Client
	// Timeout — опциональный таймаут на один вызов Generate.
	Timeout time.Duration
}

type OpenAICompatibleEmbedderOptions struct {
	// BaseURL — базовый URL провайдера (например, "https://api.openai.com").
	BaseURL string
	// APIKey — ключ доступа. Передаётся в заголовке Authorization: Bearer.
	APIKey string
	// Model — имя embeddings модели.
	Model string

	// HTTPClient — опциональный клиент; если nil, используется http.DefaultClient.
	HTTPClient *http.Client
	// Timeout — опциональный таймаут на один вызов Embed.
	Timeout time.Duration
}

type OpenAICompatibleLLMOptions struct {
	// BaseURL — базовый URL провайдера (например, "https://api.openai.com").
	BaseURL string
	// APIKey — ключ доступа. Передаётся в заголовке Authorization: Bearer.
	APIKey string
	// Model — имя модели.
	Model string

	// Temperature — параметр генерации; если nil, параметр не передаётся в запросе.
	Temperature *float64
	// MaxOutputTokens — лимит выходных токенов; если nil, параметр не передаётся в запросе.
	MaxOutputTokens *int

	// HTTPClient — опциональный клиент; если nil, используется http.DefaultClient.
	HTTPClient *http.Client
	// Timeout — опциональный таймаут на один вызов Generate.
	Timeout time.Duration
}

type PGVectorMigrateOptions struct {
	PGVectorOptions

	// DDLTimeout — дефолтный таймаут для миграций, если у ctx нет deadline.
	// Если 0 — используется 30s.
	DDLTimeout time.Duration
}

type PGVectorOptions struct {
	// TableName — имя таблицы для хранения чанков.
	// В v1 поддерживается только простой идентификатор без схемы (например, "draftrag_chunks").
	TableName string

	// EmbeddingDimension — фиксированная размерность embedding-векторов.
	// При несоответствии размерности операции store возвращают ErrEmbeddingDimensionMismatch (errors.Is).
	EmbeddingDimension int

	// CreateExtension включает попытку выполнить `CREATE EXTENSION IF NOT EXISTS vector`.
	// Часто требует повышенных прав; при отсутствии прав будет возвращена ошибка.
	CreateExtension bool

	// IndexMethod — метод индекса: "ivfflat" (по умолчанию) или "hnsw".
	IndexMethod string

	// Lists — параметр ivfflat индекса (WITH (lists = N)).
	Lists int
}

type PGVectorRuntimeOptions struct {
	// SearchTimeout — дефолтный таймаут для Search*, если у ctx нет deadline.
	SearchTimeout time.Duration
	// UpsertTimeout — дефолтный таймаут для Upsert, если у ctx нет deadline.
	UpsertTimeout time.Duration
	// DeleteTimeout — дефолтный таймаут для Delete, если у ctx нет deadline.
	DeleteTimeout time.Duration

	// MaxTopK ограничивает topK в Search*. 0 означает “без лимита”.
	MaxTopK int
	// MaxParentIDs ограничивает количество ParentIDs в фильтре. 0 означает “без лимита”.
	MaxParentIDs int
	// MaxContentBytes ограничивает размер chunk.Content в байтах. 0 означает “без лимита”.
	MaxContentBytes int
}

type PGVectorStoreOptions struct {
	PGVectorOptions
	Runtime PGVectorRuntimeOptions
}

type ParentIDFilter = domain.ParentIDFilter

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

type PipelineOptions struct {
	// DefaultTopK — значение topK по умолчанию для Query/Answer.
	// Если 0, используется значение по умолчанию (5).
	// Если < 0, это считается ошибкой конфигурации (panic).
	DefaultTopK int
	// SystemPrompt — переопределение system prompt для Answer*. Пустая строка означает дефолт v1.
	SystemPrompt string
	// Chunker — опциональный чанкер; если не nil, Index индексирует чанки (Chunk→Embed→Upsert).
	Chunker Chunker
	// Hooks — опциональные хуки наблюдаемости для стадий pipeline (chunking/embed/search/generate).
	// nil означает no-op.
	Hooks Hooks

	// MaxContextChars — лимит размера секции “Контекст:” в prompt для Answer* (в символах).
	// 0 означает “без лимита”.
	MaxContextChars int
	// MaxContextChunks — лимит количества чанков, попадающих в секцию “Контекст:” в prompt для Answer*.
	// 0 означает “без лимита”.
	MaxContextChunks int

	// DedupSourcesByParentID включает дедупликацию retrieval sources по ParentID.
	// По умолчанию выключено (backward compatibility).
	DedupSourcesByParentID bool

	// MMREnabled включает MMR rerank/selection для retrieval sources (диверсификация контекста).
	// По умолчанию выключено (backward compatibility).
	MMREnabled bool
	// MMRLambda задаёт баланс релевантность/разнообразие в диапазоне [0..1].
	// Если 0 и MMR включён — используется значение по умолчанию (0.5).
	MMRLambda float64
	// MMRCandidatePool задаёт сколько кандидатов запросить у VectorStore до отбора.
	// Если 0 — используется topK запроса.
	MMRCandidatePool int

	// IndexConcurrency задаёт количество workers для параллельной индексации в IndexBatch.
	// Если 0 — используется значение по умолчанию (4).
	IndexConcurrency int
	// IndexBatchRateLimit задаёт максимальное количество вызовов Embed в секунду в IndexBatch.
	// Если 0 — без ограничений.
	IndexBatchRateLimit int

	// Reranker — опциональный reranker, применяется после retrieval.
	// nil означает "без reranking".
	Reranker Reranker
}

type QdrantOptions struct {
	// URL адрес Qdrant сервера (по умолчанию: http://localhost:6333)
	URL string
	// Имя коллекции для хранения чанков
	Collection string
	// Размерность векторов (обязательно)
	Dimension int
	// HTTP таймаут (по умолчанию: 10s)
	Timeout time.Duration
}

type RedisCacheClient interface {
	GetBytes(ctx context.Context, key string) ([]byte, error)
	SetBytes(ctx context.Context, key string, value []byte, ttl time.Duration) error
}

type RedisCacheOptions struct {
	// Client — Redis клиент; nil отключает Redis кэш.
	Client RedisCacheClient

	// TTL — время жизни записей в Redis (0 → без TTL).
	TTL time.Duration

	// KeyPrefix — префикс ключей в Redis ("" → дефолт `draftrag:embedder:`).
	KeyPrefix string
}

type Reranker = domain.Reranker

type RetrievalResult = domain.RetrievalResult

type RetryEmbedder struct {
	*resilience.RetryEmbedder
}

type RetryLLMProvider struct {
	*resilience.RetryLLMProvider
}

type RetryOptions struct {
	// MaxRetries — максимальное количество повторных попыток (0 → 3).
	MaxRetries int

	// BaseDelay — начальная задержка перед первым retry (0 → 100ms).
	BaseDelay time.Duration

	// MaxDelay — максимальная задержка (0 → 10s).
	MaxDelay time.Duration

	// Multiplier — множитель exponential backoff (0 → 2.0).
	Multiplier float64

	// JitterFactor — доля случайной составляющей (0 → 0.25).
	JitterFactor float64

	// CBThreshold — порог ошибок для перехода circuit breaker в open (0 → 5).
	CBThreshold int

	// CBTimeout — время восстановления circuit breaker (0 → 30s).
	CBTimeout time.Duration

	// Logger — опциональный структурированный логгер для retry/CB событий.
	// nil означает no-op.
	Logger Logger
}

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

type StageEndEvent = domain.StageEndEvent

type StageStartEvent = domain.StageStartEvent

type StreamingLLMProvider = domain.StreamingLLMProvider

type VectorStore = domain.VectorStore

type VectorStoreWithFilters = domain.VectorStoreWithFilters

type WeaviateOptions struct {
	// Host — адрес Weaviate, например "localhost:8080" (обязательно).
	Host string
	// Scheme — протокол: "http" или "https". По умолчанию "http".
	Scheme string
	// Collection — имя коллекции (Weaviate class, обязательно).
	Collection string
	// APIKey — API ключ для Weaviate Cloud (опционально).
	APIKey string
	// Timeout — HTTP таймаут (по умолчанию 10s).
	Timeout time.Duration
}