embed

type Config struct {
	ModelName string
	CacheDir  string
	BatchSize int
	Log       *slog.Logger
}

type Embedder interface {
	Dimension() int
	Embed(ctx context.Context, texts []string) ([][]float32, error)
	Close() error
}

type ModelSpec struct {
	Name         string
	Dim          int
	MaxTokens    int
	ModelURL     string
	TokenizerURL string
	// ModelSHA256/TokenizerSHA256 pin the expected content of the downloads.
	// DECISION(2026-07): URLs point at immutable HF revision commits and the
	// hash is verified after download — /main/ URLs plus a merely-logged
	// checksum meant a silent upstream change (or truncated download) would
	// be trusted. Empty is now REFUSED at download time unless the operator sets
	// CONTEXTMAXXER_ALLOW_UNVERIFIED_DOWNLOAD; the locally-trained ft specs carry
	// no URLs at all and never reach that path.
	ModelSHA256     string
	TokenizerSHA256 string
	InputNames      []string
	// Pooling collapses per-token states to one vector: "mean" (default,
	// encoder models) or "last" (decoder-based embedding models).
	Pooling string
	// QueryPrefix/DocPrefix are prepended to query/passage texts for
	// instruction-tuned models; empty for symmetric encoders.
	QueryPrefix string
	DocPrefix   string
	// EOSTokenID, when >0, is enforced as the final token of every sequence
	// (required for last-token pooling to land on the trained position, and
	// must survive truncation).
	EOSTokenID int64
}

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

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