--- # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 title: Standalone KV Indexer subtitle: Run the KV cache indexer as an independent HTTP service for querying block state --- ## Overview The standalone KV indexer (`dynamo-kv-indexer`) is a lightweight HTTP binary that subscribes to ZMQ KV event streams from workers, maintains a radix tree of cached blocks, and exposes HTTP endpoints for querying and managing workers. This is distinct from the [Standalone Router](../../../components/src/dynamo/router/README.md), which is a full routing service. The standalone indexer provides only the indexing and query layer without routing logic. The HTTP API follows the [Mooncake KV Indexer RFC](https://github.com/kvcache-ai/Mooncake/issues/1403) conventions. ## Multi-Model and Multi-Tenant Support The indexer maintains one radix tree per `(model_name, tenant_id)` pair. Workers registered with different model names or tenant IDs are isolated into separate indexers — queries against one model/tenant never return scores from another. - **`model_name`** (required on `/register` and `/query`): Identifies the model. Workers serving different models get separate radix trees. - **`tenant_id`** (optional, defaults to `"default"`): Enables multi-tenant isolation within the same model. Omit for single-tenant deployments. - **`block_size`** is per-indexer: the first `/register` call for a given `(model_name, tenant_id)` sets the block size. Subsequent registrations for the same pair must use the same block size or the request will fail. ## Compatibility The standalone indexer works with any engine that publishes KV cache events over ZMQ in the expected msgpack format. This includes bare vLLM and SGLang engines, which emit ZMQ KV events natively — no Dynamo-specific wrapper is required. ## Use Cases - **Debugging**: Inspect the radix tree state to verify which blocks are cached on which workers. - **State verification**: Confirm that the indexer's view of KV cache state matches the router's internal state (used in integration tests). - **Custom routing**: Build external routing logic that queries the indexer for overlap scores and makes its own worker selection decisions. - **Monitoring**: Observe KV cache distribution across workers without running a full router. ## P2P Recovery Multiple indexer replicas can subscribe to the same ZMQ worker endpoints for fault tolerance. When a replica starts (or restarts after a crash), it bootstraps its radix tree state from a healthy peer before processing live events. ### How It Works 1. Workers are registered via `--workers` CLI, which connects ZMQ SUB sockets immediately. 2. A 1-second delay ensures the peer's tree state has advanced past the ZMQ connection point, so the dump covers any events that would otherwise be lost to the slow-joiner window. 3. The indexer fetches a `/dump` from the first reachable peer in `--peers`. 4. Dump events are applied to populate the radix tree. 5. ZMQ listeners are unblocked and begin draining any events that buffered during recovery. If no peers are reachable, the indexer starts with an empty state. ### Example: Two-Replica Setup ```bash # Replica A (first instance, no peers) dynamo-kv-indexer --port 8090 --block-size 16 \ --workers "1=tcp://worker1:5557,2=tcp://worker2:5558" # Replica B (recovers from A on startup) dynamo-kv-indexer --port 8091 --block-size 16 \ --workers "1=tcp://worker1:5557,2=tcp://worker2:5558" \ --peers "http://localhost:8090" ``` Both replicas subscribe to the same workers. Replica B recovers A's tree state on startup, then both independently process live ZMQ events going forward. ### Consistency The dump is a weakly consistent BFS snapshot of the radix tree — concurrent writes may race with the traversal. This is acceptable because: - **Stale blocks** (partially removed branches): live `Remove` events will clean them up. - **Missing blocks** (partially added branches): live `Stored` events will add them. - The tree converges to the correct state after live events catch up. ### Peer Management Peers can be registered at startup via `--peers` or dynamically via the HTTP API. The peer list is used for recovery only — peers do not synchronize state in real time. ## Building The binary is a feature-gated target in the `dynamo-kv-router` crate: ```bash cargo build -p dynamo-kv-router --features indexer-bin --bin dynamo-kv-indexer ``` ## CLI ```bash dynamo-kv-indexer --port 8090 [--threads 4] [--block-size 16 --model-name my-model --tenant-id default --workers "1=tcp://host:5557,2:1=tcp://host:5558"] [--peers "http://peer1:8090,http://peer2:8091"] ``` | Flag | Default | Description | |------|---------|-------------| | `--block-size` | (none) | KV cache block size for initial `--workers` (required when `--workers` is set) | | `--port` | `8090` | HTTP server listen port | | `--threads` | `4` | Number of indexer threads (1 = single-threaded, >1 = thread pool) | | `--workers` | (none) | Initial workers as `instance_id[:dp_rank]=zmq_address,...` pairs (dp_rank defaults to 0) | | `--model-name` | `default` | Model name for initial `--workers` | | `--tenant-id` | `default` | Tenant ID for initial `--workers` | | `--peers` | (none) | Comma-separated peer indexer URLs for P2P recovery on startup | ## HTTP API ### `POST /register` — Register an endpoint Register a ZMQ endpoint for an instance. Each call creates or reuses the indexer for the given `(model_name, tenant_id)` pair. ```bash # Single model, default tenant curl -X POST http://localhost:8090/register \ -H 'Content-Type: application/json' \ -d '{ "instance_id": 1, "endpoint": "tcp://127.0.0.1:5557", "model_name": "llama-3-8b", "block_size": 16 }' # With tenant isolation curl -X POST http://localhost:8090/register \ -H 'Content-Type: application/json' \ -d '{ "instance_id": 2, "endpoint": "tcp://127.0.0.1:5558", "model_name": "llama-3-8b", "tenant_id": "customer-a", "block_size": 16, "dp_rank": 0 }' ``` | Field | Required | Default | Description | |-------|----------|---------|-------------| | `instance_id` | yes | — | Worker instance identifier | | `endpoint` | yes | — | ZMQ PUB address to subscribe to | | `model_name` | yes | — | Model name (used to select the indexer) | | `block_size` | yes | — | KV cache block size (must match the engine) | | `tenant_id` | no | `"default"` | Tenant identifier for isolation | | `dp_rank` | no | `0` | Data parallel rank | ### `POST /unregister` — Deregister an instance Remove an instance. Omitting `tenant_id` removes the instance from **all** tenants for the given model; providing it targets only that tenant's indexer. ```bash # Remove from all tenants curl -X POST http://localhost:8090/unregister \ -H 'Content-Type: application/json' \ -d '{"instance_id": 1, "model_name": "llama-3-8b"}' # Remove from a specific tenant curl -X POST http://localhost:8090/unregister \ -H 'Content-Type: application/json' \ -d '{"instance_id": 1, "model_name": "llama-3-8b", "tenant_id": "customer-a"}' # Remove a specific dp_rank curl -X POST http://localhost:8090/unregister \ -H 'Content-Type: application/json' \ -d '{"instance_id": 1, "model_name": "llama-3-8b", "tenant_id": "default", "dp_rank": 0}' ``` | Field | Required | Default | Description | |-------|----------|---------|-------------| | `instance_id` | yes | — | Worker instance to remove | | `model_name` | yes | — | Model name (identifies the indexer) | | `tenant_id` | no | — | Tenant identifier (omit to remove from all tenants) | | `dp_rank` | no | — | Specific dp_rank to remove (omit to remove all) | ### `GET /workers` — List registered instances ```bash curl http://localhost:8090/workers ``` Returns: ```json [{"instance_id": 1, "endpoints": {"0": "tcp://127.0.0.1:5557", "1": "tcp://127.0.0.1:5558"}}] ``` ### `POST /query` — Query overlap for token IDs Given raw token IDs, compute block hashes and return per-instance overlap scores (in matched tokens): ```bash curl -X POST http://localhost:8090/query \ -H 'Content-Type: application/json' \ -d '{"token_ids": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16], "model_name": "llama-3-8b"}' ``` Returns: ```json { "scores": {"1": {"0": 32}, "2": {"1": 0}}, "frequencies": [1, 1], "tree_sizes": {"1": {"0": 5}, "2": {"1": 3}} } ``` Scores are in **matched tokens** (block overlap count × block size). Nested by `instance_id` then `dp_rank`. | Field | Required | Default | Description | |-------|----------|---------|-------------| | `token_ids` | yes | — | Token sequence to query | | `model_name` | yes | — | Model name (selects the indexer) | | `tenant_id` | no | `"default"` | Tenant identifier | | `lora_name` | no | — | LoRA adapter (overrides indexer-level lora_name for this query) | ### `POST /query_by_hash` — Query overlap for pre-computed hashes ```bash curl -X POST http://localhost:8090/query_by_hash \ -H 'Content-Type: application/json' \ -d '{"block_hashes": [123456, 789012], "model_name": "llama-3-8b"}' ``` Same response format as `/query`. Scores are in matched tokens. | Field | Required | Default | Description | |-------|----------|---------|-------------| | `block_hashes` | yes | — | Pre-computed block hash array | | `model_name` | yes | — | Model name (selects the indexer) | | `tenant_id` | no | `"default"` | Tenant identifier | ### `GET /dump` — Dump all radix tree events Returns the full radix tree state as a JSON object keyed by `model_name:tenant_id`: ```bash curl http://localhost:8090/dump ``` Returns: ```json { "llama-3-8b:default": { "block_size": 16, "events": [, ...] }, "mistral-7b:customer-a": { "block_size": 16, "events": [, ...] } } ``` Each indexer is dumped concurrently. The `block_size` field lets recovering peers create indexers with the correct block size without requiring `--block-size` on every replica. ### `POST /register_peer` — Register a peer indexer ```bash curl -X POST http://localhost:8090/register_peer \ -H 'Content-Type: application/json' \ -d '{"url": "http://peer:8091"}' ``` ### `POST /deregister_peer` — Remove a peer indexer ```bash curl -X POST http://localhost:8090/deregister_peer \ -H 'Content-Type: application/json' \ -d '{"url": "http://peer:8091"}' ``` ### `GET /peers` — List registered peers ```bash curl http://localhost:8090/peers ``` Returns: ```json ["http://peer:8091"] ``` ## Limitations - **ZMQ only**: Workers must publish KV events via ZMQ PUB sockets. The standalone indexer does not subscribe to NATS event streams. - **No routing logic**: The indexer only maintains the radix tree and answers queries. It does not track active blocks, manage request lifecycle, or perform worker selection. ## Architecture ```mermaid graph TD subgraph Workers W1[Worker 1
ZMQ PUB] W2[Worker 2
ZMQ PUB] end subgraph "Standalone Indexer (HTTP)" REG[Worker Registry] ZMQ[ZMQ SUB Listeners] IDX["Indexer Map
(model, tenant) → Radix Tree"] HTTP[HTTP API
/query /dump /register] end CLIENT[External Client] W1 -->|ZMQ events| ZMQ W2 -->|ZMQ events| ZMQ CLIENT -->|POST /register| REG REG -->|spawn listeners| ZMQ ZMQ -->|apply events| IDX CLIENT -->|POST /query, GET /dump| HTTP HTTP -->|query| IDX style W1 fill:#f3e5f5,stroke:#333,color:#333 style W2 fill:#f3e5f5,stroke:#333,color:#333 style IDX fill:#2e8b57,stroke:#333,color:#fff style ZMQ fill:#2e8b57,stroke:#333,color:#fff style REG fill:#2e8b57,stroke:#333,color:#fff style HTTP fill:#2e8b57,stroke:#333,color:#fff style CLIENT fill:#fff3e0,stroke:#333,color:#333 ``` ### P2P Recovery Flow ```mermaid sequenceDiagram participant B as Replica B (new) participant A as Replica A (healthy) participant W as Workers (ZMQ PUB) B->>W: Connect ZMQ SUB sockets Note over B,W: 1s delay for peer tree to advance past connection point B->>A: GET /dump A-->>B: Radix tree snapshot + block sizes Note over B: Apply dump events Note over B: Unblock ZMQ listeners B->>W: Start draining buffered events Note over B: Ready to serve queries ``` ## See Also - **[Mooncake KV Indexer RFC](https://github.com/kvcache-ai/Mooncake/issues/1403)**: Community API standardization for KV cache indexers - **[Router Guide](router-guide.md)**: Full KV router configuration and tuning - **[Router Design](../../design-docs/router-design.md)**: Architecture and event transport modes - **[Standalone Router](../../../components/src/dynamo/router/README.md)**: Full routing service (routes requests to workers)