stroma

module
v2.3.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: May 2, 2026 License: MIT

README

Stroma

Stroma is a neutral corpus and indexing substrate.

It owns the lowest-level operations needed to ingest text artifacts, chunk them, embed them, persist them in SQLite plus sqlite-vec, retrieve semantically close sections, and call OpenAI-compatible embedding and chat completion endpoints over a shared HTTP substrate. Callers consume Stroma through its APIs and treat the SQLite snapshot as an opaque local artifact. It does not own governance, specifications, compliance, drift analysis, prompt templates, product-specific output semantics, MCP, or CLI workflows.

Scope

Stroma is for products that need a reusable text corpus layer with:

  • canonical records with deterministic content fingerprints
  • pluggable chunking strategies (chunk.PolicyMarkdownPolicy default, KindRouterPolicy for per-record-kind dispatch, LateChunkPolicy for parent/leaf hierarchy)
  • pluggable embedders (Embedder / ContextualEmbedder) with a deterministic fixture and an OpenAI-compatible HTTP embedder
  • OpenAI-compatible chat completion client (chat.OpenAI) sharing the same substrate as embed.OpenAI: retry with Retry-After (capped), classified failures (auth / rate_limit / timeout / server / transport / schema_mismatch / dependency_unavailable), preserved lower-level causes on provider errors, APIToken redaction, and a product-neutral structured JSON helper
  • hybrid retrieval: dense vector + FTS5, fused via a pluggable FusionStrategy (RRFFusion by default) with per-arm provenance surfaced to downstream rerankers
  • quantization knobs: float32 (default), int8 (4× smaller), binary (1-bit sign-packed vec0 prefilter that is 32× smaller for the prefilter representation; full-precision vectors are retained in a companion table for cosine rescoring, so total snapshot size is not 32× smaller)
  • optional Matryoshka prefilter at a truncated dimension with full-dim cosine rescore (SearchParams.SearchDimension)
  • atomic rebuilds and incremental Update with embedding reuse at the section level, chaining schema migrations v2 → v3 → v4 → v5 in one transaction

Stroma is not for:

  • spec governance
  • source discovery or repository scanning
  • code compliance or doc drift analysis
  • prompt templates, system prompts, or semantic interpretation of structured chat responses
  • product-specific adapters and transports

Packages

  • corpus — canonical record model, NewRecord helper, Normalize, deterministic Fingerprint
  • chunkPolicy interface with MarkdownPolicy, KindRouterPolicy, LateChunkPolicy; MarkdownWithOptions returns ErrTooManySections when a body exceeds the DoS cap
  • embedEmbedder and ContextualEmbedder interfaces; deterministic Fixture; OpenAI-compatible HTTP embedder with MaxBatchSize batching, deadline scaling across batches, and APIToken redaction in String/GoString/LogValue
  • chat — OpenAI-compatible chat completion client (chat.OpenAI, chat.Message, ChatCompletionText, ChatCompletionJSON); tolerates string and multi-part array content; structured JSON responses decode into caller-owned targets and malformed JSON returns schema_mismatch; APIToken redaction parity with embed.OpenAIConfig
  • provider — shared HTTP substrate used by embed and chat: retry with capped Retry-After, response-size bounding, negative MaxRetries normalization to zero, and a stable FailureClass taxonomy surfaced via *provider.Error. Callers branch on FailureClass to retry / degrade / propagate, and can unwrap lower-level transport/decode causes where available
  • store — SQLite readiness probes, sqlite-vec readiness, quantization blob helpers (QuantizationFloat32 / QuantizationInt8 / QuantizationBinary)
  • index — atomic Rebuild plus streaming RebuildFromSource with embedding reuse and explicit reuse diagnostics, incremental Update and UpdateFromSource with MaxPlannedRecords batching guard, long-lived Snapshot readers, Stats, hybrid Search with provenance and explicit MaxSearchLimit, ExpandContext for parent/neighbor walks

Retrieval Evidence And Batch Use

Use OpenSnapshot when issuing many searches against one built index. A Snapshot is safe for concurrent reads; callers own the concurrency limit, so use a bounded worker pool or semaphore sized for the host and workload, then close the snapshot after all searches and context expansions finish.

For durable evidence handles, persist at least:

  • Stats.ContentFingerprint from the opened snapshot, identifying the indexed content generation
  • SearchHit.ChunkID, identifying a chunk only within that snapshot generation
  • SearchHit.Ref plus any caller-needed record metadata or SourceRef

ChunkID is not a cross-rebuild identity. Before expanding a previously saved hit, reopen the snapshot, compare Stats.ContentFingerprint with the saved value, and rerun search if it differs. SearchHit.Score and HitProvenance are ranking evidence for the query that produced the hit; keep them for audit/debugging, but do not use them as identity fields.

ExpandContext(hit.ChunkID, opts) returns the hit chunk plus requested parent/neighbor sections in document order. On flat snapshots, parent expansion is a no-op and neighbors are same-record chunks. On hierarchical snapshots, parent expansion follows parent_chunk_id one level and neighbors stay in the same sibling group. A missing chunk returns an empty slice and nil error, which lets wrappers treat stale handles as "not found" after they have already checked the content fingerprint.

Search Filtering

SearchParams.Kinds, SearchParams.Refs, and SearchParams.Metadata restrict candidate records before each retrieval arm ranks and truncates its own shortlist. The vector arm applies those filters inside the vector prefilter stage; when any record filter is present, Stroma scans only chunks that satisfy the record predicates instead of taking a vec0 MATCH k shortlist and filtering it afterward. Ref and kind filters use normal indexed table predicates before vector blobs are scored. The FTS arm applies the same filters in the fts_chunks query before ordering by FTS rank and LIMIT. This avoids under-filled results when a small collection would otherwise be filtered out after unrelated high-ranking chunks consume each arm's candidate window.

Metadata filters are exact string matches against stored record metadata: values within one key are ORed (item_id IN (...)), and multiple keys are ANDed together. Empty kind/ref filter values, empty metadata keys, duplicate metadata keys after trimming, empty metadata value lists, and whitespace-only metadata values reject instead of becoming accidental unfiltered searches. Empty metadata values are valid exact matches. Kinds remains the kind allow-list, and Refs expresses ref IN (...). Metadata predicates are evaluated from stored JSON metadata rather than a separate inverted metadata index, so prefer Kinds or Refs for high-volume hot filters when possible.

Rebuild Streaming, Update Memory And Batching

Use RebuildFromSource when the corpus lives behind a filesystem, database, or other lazy loader and callers should not build one full []corpus.Record before indexing. A RecordSource returns one corpus.Record at a time; Stroma normalizes each record, chunks and embeds records in bounded internal batches, writes them to the staging snapshot, rejects duplicate refs through the snapshot's primary key, and computes the final content fingerprint from persisted (ref, content_hash) pairs. This keeps record bodies bounded to the current source record plus the current planned batch; Stroma retains that batch's chunks/vectors until it is flushed. Rebuild remains the slice convenience API and still sorts the provided records before delegating to the same write path. RebuildFromSource preserves source order, so source order determines snapshot-local ChunkID assignment; callers that need repeatable chunk IDs across streaming rebuilds should emit records in a stable order.

RebuildFromSource keeps the same atomic staging-file contract as Rebuild: a source, chunking, embedding, or duplicate-ref error discards the staged file and leaves the destination snapshot unchanged. It is not a resumable checkpointing API. The staging SQLite transaction stays open while the source is consumed and embeddings are produced, so callers should keep RecordSource.Next responsive to context cancellation and avoid doing unrelated slow work inside it.

Update chunks, contextualizes, reuse-plans, and embeds added/replaced records before opening its SQLite write transaction. That keeps external embedder latency out of the transaction and preserves stale-plan rollback semantics, but the pre-transaction plan retains each added record's chunks, reuse decisions, and new vectors until the write phase.

Use UpdateFromSource when a caller has the desired current corpus behind a lazy loader and wants Stroma to own the incremental diff. The RecordSource stream is interpreted as the complete desired record set: stored refs missing from the stream are removed, new refs are added, refs whose normalized ContentHash changed are replaced, and unchanged refs are counted as reused without loading their full stored bodies. Duplicate source refs and over-cap changed-record plans fail before embedding or opening the write transaction. Added/replaced records preserve source order for their new snapshot-local ChunkID assignment; callers that need repeatable chunk IDs for changed records should emit changed records in a stable order.

UpdateFromSource consumes record bodies one at a time, but it is not a constant-memory full-corpus diff. It keeps the snapshot's (ref, content_hash) pairs and the full source ref set in memory to infer removals and no-ops. It retains only added/replaced record bodies plus their planned chunks/vectors until commit. MaxPlannedRecords caps that changed-record plan and is checked while the source is consumed, so an over-cap source update fails before embedding and before the write handle is opened. The returned UnchangedRecordCount / UnchangedChunkCount identify fully unchanged source records separately from ReusedRecordCount / ReusedChunkCount, which also include section-level embedding reuse inside changed records.

For large ingests, split added records into caller-sized batches and set UpdateOptions.MaxPlannedRecords to that batch size. A batch above the cap fails before embedding and before the write transaction starts with an error wrapping index.ErrUpdatePlanTooLarge, so callers can retry smaller batches without changing the on-disk snapshot. MaxChunkSections still bounds per-record section expansion.

Example

ctx := context.Background()

records := []corpus.Record{
    corpus.NewRecord(
        "widget-overview",
        "Widget Overview",
        "# Overview\n\nWidgets are synchronized in batches.",
    ),
}

fixture, err := embed.NewFixture("fixture-demo", 16)
if err != nil {
    log.Fatal(err)
}

if _, err := index.Rebuild(ctx, records, index.BuildOptions{
    Path:     "stroma.db",
    Embedder: fixture,
}); err != nil {
    log.Fatal(err)
}

hits, err := index.Search(ctx, index.SearchQuery{
    Path: "stroma.db",
    SearchParams: index.SearchParams{
        Text:     "synchronized batches",
        Limit:    5,
        Embedder: fixture,
        // Fusion / Reranker / SearchDimension are optional; zero values
        // give hybrid RRF over vector+FTS with the full stored dimension.
    },
})
if err != nil {
    log.Fatal(err)
}

fmt.Println(hits[0].Ref)

See the v2.0.0 release notes for the full API surface.

Status

v2.0.0 (current) ships the stable substrate: hybrid retrieval, pluggable fusion, quantization, matryoshka, contextual retrieval, adaptive chunking, and incremental update. Higher-order products should consume the library rather than re-embedding their own indexing substrate.

Directories

Path Synopsis
Package chat is stroma's OpenAI-compatible chat completion primitive, the sibling of embed.OpenAI.
Package chat is stroma's OpenAI-compatible chat completion primitive, the sibling of embed.OpenAI.
Package chunk splits text bodies into heading-aware sections for indexing.
Package chunk splits text bodies into heading-aware sections for indexing.
Package corpus defines the neutral record unit that Stroma indexes.
Package corpus defines the neutral record unit that Stroma indexes.
Package embed defines embedder interfaces and a deterministic test fixture.
Package embed defines embedder interfaces and a deterministic test fixture.
Package index orchestrates atomic Stroma index rebuilds and searches.
Package index orchestrates atomic Stroma index rebuilds and searches.
Package provider is stroma's shared OpenAI-compatible HTTP substrate.
Package provider is stroma's shared OpenAI-compatible HTTP substrate.
Package store provides SQLite-backed storage primitives for Stroma indexes.
Package store provides SQLite-backed storage primitives for Stroma indexes.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL