langgraph-api 0.16.0.dev6__py3-none-any.whl → 0.16.0.dev7__py3-none-any.whl

langgraph_api/__init__.py

@@ -1 +1 @@
-__version__ = "0.16.0.dev6"
+__version__ = "0.16.0.dev7"

langgraph_api/api/threads.py

@@ -4,6 +4,7 @@ from starlette.exceptions import HTTPException
 from starlette.responses import Response
 from starlette.routing import BaseRoute
 
+from langgraph_api.config import USE_CUSTOM_CHECKPOINTER
 from langgraph_api.encryption.middleware import (
     decrypt_response,
     decrypt_responses,
@@ -45,6 +46,7 @@ from langgraph_api.validation import (
     ThreadStateSearch,
     ThreadStateUpdate,
 )
+from langgraph_runtime import checkpoint as runtime_checkpoint
 from langgraph_runtime.database import connect
 from langgraph_runtime.retry import retry_db
 
@@ -376,6 +378,25 @@ async def get_thread_history_post(
     return ApiResponse(states)
 
 
+@retry_db
+async def get_thread_storage(request: ApiRequest):
+    """Bytes a thread's checkpoints occupy, per table and per channel."""
+    thread_id = request.path_params["thread_id"]
+    validate_uuid(thread_id, "Invalid thread ID: must be a UUID")
+    thread_storage = getattr(runtime_checkpoint, "thread_storage", None)
+    if USE_CUSTOM_CHECKPOINTER or thread_storage is None:
+        raise HTTPException(
+            status_code=501,
+            detail="Checkpoint storage is only reported for the built-in checkpointer.",
+        )
+    async with connect() as conn:
+        await fetchone(await Threads.get(conn, thread_id))
+    # The built-in tables are read directly; the core-api stand-in cannot run SQL.
+    async with connect(supports_core_api=False) as conn:
+        storage = await thread_storage(conn, thread_id)
+    return ApiResponse({"thread_id": thread_id, **storage})
+
+
 @retry_db
 async def get_thread(
     request: ApiRequest,
@@ -547,6 +568,9 @@ threads_routes: list[BaseRoute] = [
     ApiRoute("/threads/{thread_id}", endpoint=patch_thread, methods=["PATCH"]),
     ApiRoute("/threads/{thread_id}", endpoint=delete_thread, methods=["DELETE"]),
     ApiRoute("/threads/{thread_id}/state", endpoint=get_thread_state, methods=["GET"]),
+    ApiRoute(
+        "/threads/{thread_id}/storage", endpoint=get_thread_storage, methods=["GET"]
+    ),
     ApiRoute(
         "/threads/{thread_id}/state", endpoint=update_thread_state, methods=["POST"]
     ),

langgraph_api/grpc/servicers/checkpointer.py

@@ -13,6 +13,7 @@ import orjson
 import structlog
 from google.protobuf.empty_pb2 import Empty  # ty: ignore[unresolved-import]
 from langgraph_grpc_common.conversion.checkpoint import (
+    checkpoint_config_to_proto,
     checkpoint_from_proto,
     checkpoint_metadata_from_proto,
     checkpoint_tuple_to_proto,
@@ -22,7 +23,6 @@ from langgraph_grpc_common.conversion.checkpoint import (
 from langgraph_grpc_common.conversion.config import (
     config_from_proto,
     config_from_proto_optional,
-    config_to_proto,
 )
 from langgraph_grpc_common.proto import checkpointer_pb2
 from langgraph_grpc_common.proto.checkpointer_pb2_grpc import CheckpointerServicer
@@ -64,7 +64,7 @@ class CheckpointerServicerImpl(CheckpointerServicer):
             next_config = await checkpointer.aput(
                 config, checkpoint, metadata, new_versions
             )
-            next_config_pb = config_to_proto(next_config)
+            next_config_pb = checkpoint_config_to_proto(next_config)
             if next_config_pb is None:
                 return checkpointer_pb2.PutResponse()
             return checkpointer_pb2.PutResponse(next_config=next_config_pb)

{langgraph_api-0.16.0.dev6.dist-info → langgraph_api-0.16.0.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.16.0.dev6
+Version: 0.16.0.dev7
 Author-email: Will Fu-Hinthorn <will@langchain.dev>, Josh Rogers <josh@langchain.dev>, Parker Rule <parker@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE

{langgraph_api-0.16.0.dev6.dist-info → langgraph_api-0.16.0.dev7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langgraph_api/__init__.py,sha256=Tea7ymkB4nX_zD1WSeuGpkEk3r2zpOb-cvWZRXmsogA,28
+langgraph_api/__init__.py,sha256=aBRzlOKsAMKJlCRnhr4-7ukE_yGNccJoqTq2ODz7v1k,28
 langgraph_api/_factory_utils.py,sha256=_FqW0olMu2PGLsnRq0ylt3a4jDCe12YIFHExMXI6Nyo,5976
 langgraph_api/asgi_transport.py,sha256=XApY3lIWBZTMbbsl8dDJzl0cLGirmAGE0SifqZUnXvs,11896
 langgraph_api/asyncio.py,sha256=smqyAoO9nIyPPQRw1IVSCh85fwoZP4PEo-5dDhxGruc,10676
@@ -51,7 +51,7 @@ langgraph_api/api/openapi.py,sha256=Zkdlb9mjrQyHro1TtrDIWVuaBDovxx-uGWJ1fZMOg54,
 langgraph_api/api/profile.py,sha256=CA1ZkHALOuP8orYTICnEhcG_JnnA2wnyjbWyeb117jA,3455
 langgraph_api/api/runs.py,sha256=Po9sBkjTImo1lnKKfgFmYJRtXIGk0Mriqm5_9YC77Kw,36068
 langgraph_api/api/store.py,sha256=mvFEGlrTN77bxH3rLqreRJ1iWvG0r6iv88nQTAkpuAQ,7244
-langgraph_api/api/threads.py,sha256=n2cn0FZr_b1jKq7McX6HHuYn80Z0x89wNH4zLs5Co6Y,19649
+langgraph_api/api/threads.py,sha256=1VaWwUJvCIiCkjHHsaS6l1R-ErlBs3Y-Jzyy954e-WM,20764
 langgraph_api/api/ui.py,sha256=A1cRxaMK_4pb192mrZ_LMxLSFb8NriYwwvjeDdNDD7k,2803
 langgraph_api/api/a2ui_schemas/README.md,sha256=fZCc9rSK-qeUHsbJ17EYxMZjS3yUrYQdoQLX7ltgP8M,298
 langgraph_api/api/a2ui_schemas/client_capabilities.json,sha256=3YAHlRfeOn9dUH6uAek1loCL4caeSUuQzhrtWKWW9nU,1906
@@ -102,7 +102,7 @@ langgraph_api/grpc/ops/crons.py,sha256=BzFTBo0C45oTLzFqxVsR15Fy1RvCoGGEL9rYiI6Za
 langgraph_api/grpc/ops/runs.py,sha256=oOSxzgubnPvRRoRb1OKMDzdcmXve4V6Zh5AWS4sc4Rs,42401
 langgraph_api/grpc/ops/threads.py,sha256=SJraV838nZ0m6wRrCia7sLYh8KK1WfY7x5xmcspZLek,48140
 langgraph_api/grpc/servicers/__init__.py,sha256=l8yqTA_gMbIj0xHw5-RZYo0JbjL-EpFD8RxPMgW_-bA,422
-langgraph_api/grpc/servicers/checkpointer.py,sha256=aeB2hzsDxbb9i-ildx-kylpDxaYtpAUSG6rphOzbDOs,10482
+langgraph_api/grpc/servicers/checkpointer.py,sha256=0NeBZDrRiMTpUElu0ss60FTZ8LDjgnSVRdzwWVnUYF4,10504
 langgraph_api/grpc/servicers/encryption.py,sha256=qmov9a4gBJqdUNZkFASf6BLKOWce0Juwd_650G-DyPs,16924
 langgraph_api/js/.gitignore,sha256=l5yI6G_V6F1600I1IjiUKn87f4uYIrBAYU1MOyBBhg4,59
 langgraph_api/js/.prettierrc,sha256=0es3ovvyNIqIw81rPQsdt1zCQcOdBqyR_DMbFE4Ifms,19
@@ -173,15 +173,15 @@ langgraph_runtime/store.py,sha256=7mowndlsIroGHv3NpTSOZDJR0lCuaYMBoTnTrewjslw,11
 LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 hatch_build.py,sha256=Nn9N-EFfVqmH5q3iCUrhqX6zc7qT1jLI3gWSBN2xgYQ,1738
 logging.json,sha256=sBy8HDDPuucpLbLUD6e8t5fsiQjJoklZsbVHi2qQ7aQ,367
-openapi.json,sha256=C009Hi2gZaUKU4OOf2P12Qdjodx1mxZd16RAm1lZiIw,219991
+openapi.json,sha256=ZrgPaq1MMyscZiLqi3F09rWPUN67YRoT-5GHPcdN3HU,224041
 langgraph_grpc_common/__init__.py,sha256=sO7jFHekQxt6zZPhTPZAWLYLB_S8w7-VuJ7NyWFlFNo,97
-langgraph_grpc_common/checkpointer.py,sha256=-F51xCHjP3A2gWfwVTTm7x7w9YsH78qJnZ57zTYAwdI,15344
+langgraph_grpc_common/checkpointer.py,sha256=GoSI_wFBRxI-ltJKARI84qPOYT_qH__d0fsMwM8E6BE,15394
 langgraph_grpc_common/serde.py,sha256=8q290nZ6TdxbEhCTzwEs44XGi2b-3pfrUeMYY5WGlTY,6179
 langgraph_grpc_common/store.py,sha256=ae34gPQS260oBh2hcNLzI7-c1xKMs6FVvsJExqFRbhk,4743
 langgraph_grpc_common/conversion/__init__.py,sha256=pLsOrRtmF9YkgmCPhcfym6lYOB7PCJTRItyo1YK_AT8,208
 langgraph_grpc_common/conversion/_compat.py,sha256=MIEo69I7tfaPskunPGwBCKheksLwxpw-z-4O3hvO9og,4010
-langgraph_grpc_common/conversion/checkpoint.py,sha256=0Wyjyt4jL7BrwjbxWXnrEH869KhQJ0IOKI26NfJsv9g,13185
-langgraph_grpc_common/conversion/config.py,sha256=4QESPHOoYBmCg46ZbK4ulFAq1WBlha3fabLeXI7PBcU,13094
+langgraph_grpc_common/conversion/checkpoint.py,sha256=BENL92Wn67Puedf6hWmb6eT_wtuurqd6l0KvICMM5x8,13490
+langgraph_grpc_common/conversion/config.py,sha256=LmPNpMiJ5JFTAHI_Z9A_zjbxPru5gCbIa-wgs2L9HSk,13458
 langgraph_grpc_common/conversion/durability.py,sha256=FWhqjzVN4JrrsAL9AwsJkvIorQGNRl14w97Va-WZRk0,991
 langgraph_grpc_common/conversion/store.py,sha256=eQQ7YHrMYRsd1jN9LbqvCxGQGErySV_07n3meADiYIE,15237
 langgraph_grpc_common/conversion/struct.py,sha256=yk4jpPdywzn8UosAohun21lU5D_IIglOY9rkBhiG-gA,2072
@@ -251,8 +251,8 @@ langgraph_grpc_common/proto/store_pb2.py,sha256=JKcuJK0YUC1BSwLUZJWHCQ7AWkzVS9Kf
 langgraph_grpc_common/proto/store_pb2.pyi,sha256=ItSuXbA2X9bQlHtuEjDGi0fT_gJj2b0sr7XuQkybe3s,28983
 langgraph_grpc_common/proto/store_pb2_grpc.py,sha256=Kyt8eV2NXkjh3be_BYytGdD19kNf74961cruonMvCrw,3300
 langgraph_grpc_common/proto/store_pb2_grpc.pyi,sha256=oS1h2cDoq2OjyM8xazcmc8LwRDC5oCAD0zto2QUmPQw,2027
-langgraph_api-0.16.0.dev6.dist-info/METADATA,sha256=mV3FLySPe4k0GO_pVdWvPeIcLNEhmvKFcoTCqNktyR8,5850
-langgraph_api-0.16.0.dev6.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-langgraph_api-0.16.0.dev6.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
-langgraph_api-0.16.0.dev6.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.16.0.dev6.dist-info/RECORD,,
+langgraph_api-0.16.0.dev7.dist-info/METADATA,sha256=O4v84kkO5EJfHWbc_4YRpJP86LiHIG14qlxnrJGZ9Vw,5850
+langgraph_api-0.16.0.dev7.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+langgraph_api-0.16.0.dev7.dist-info/entry_points.txt,sha256=hGedv8n7cgi41PypMfinwS_HfCwA7xJIfS0jAp8htV8,78
+langgraph_api-0.16.0.dev7.dist-info/licenses/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.16.0.dev7.dist-info/RECORD,,

{langgraph_api-0.16.0.dev6.dist-info → langgraph_api-0.16.0.dev7.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.32.0
+Generator: hatchling 1.32.4
 Root-Is-Purelib: true
 Tag: py3-none-any

langgraph_grpc_common/checkpointer.py

@@ -29,7 +29,6 @@ from langgraph_grpc_common import serde as _grpc_serde
 from langgraph_grpc_common.conversion import checkpoint as ckpt_conv
 from langgraph_grpc_common.conversion.config import (
     config_from_proto,
-    config_to_proto,
     convert_dict_to_json_bytes,
 )
 from langgraph_grpc_common.proto import checkpointer_pb2
@@ -146,7 +145,9 @@ class GrpcCheckpointer(BaseCheckpointSaver):
 
     async def aget_tuple(self, config: RunnableConfig) -> CheckpointTuple | None:
         with self._scoped():
-            request = checkpointer_pb2.GetTupleRequest(config=config_to_proto(config))
+            request = checkpointer_pb2.GetTupleRequest(
+                config=ckpt_conv.checkpoint_config_to_proto(config)
+            )
 
         async def _request() -> CheckpointTuple | None:
             response = await (await self._stub()).GetTuple(request)
@@ -183,7 +184,7 @@ class GrpcCheckpointer(BaseCheckpointSaver):
     ) -> RunnableConfig:
         with self._scoped():
             request = checkpointer_pb2.PutRequest(
-                config=config_to_proto(config),
+                config=ckpt_conv.checkpoint_config_to_proto(config),
                 checkpoint=ckpt_conv.checkpoint_to_proto(checkpoint),
                 metadata=ckpt_conv.checkpoint_metadata_to_proto(metadata),
                 new_versions={k: str(v) for k, v in new_versions.items()},
@@ -218,7 +219,7 @@ class GrpcCheckpointer(BaseCheckpointSaver):
     ) -> None:
         with self._scoped():
             request = checkpointer_pb2.PutWritesRequest(
-                config=config_to_proto(config),
+                config=ckpt_conv.checkpoint_config_to_proto(config),
                 writes=ckpt_conv.writes_to_proto(writes),
                 task_id=task_id,
                 task_path=task_path,
@@ -272,9 +273,9 @@ class GrpcCheckpointer(BaseCheckpointSaver):
     ) -> AsyncIterator[CheckpointTuple]:
         with self._scoped():
             request = checkpointer_pb2.ListRequest(
-                config=config_to_proto(config) if config is not None else None,
+                config=ckpt_conv.checkpoint_config_to_proto(config),
                 filter_json=convert_dict_to_json_bytes(filter) or b"",
-                before=config_to_proto(before) if before is not None else None,
+                before=ckpt_conv.checkpoint_config_to_proto(before),
             )
         if limit is not None:
             request.limit = limit

langgraph_grpc_common/conversion/checkpoint.py

@@ -2,6 +2,7 @@ from collections.abc import Mapping
 from collections.abc import Sequence as SequenceType
 from typing import Any, Literal, cast
 
+from langchain_core.runnables import RunnableConfig
 from langgraph.checkpoint.base import (
     Checkpoint,
     CheckpointMetadata,
@@ -35,6 +36,14 @@ SOURCE_MAP = {
 }
 
 
+def checkpoint_config_to_proto(
+    config: RunnableConfig | None,
+) -> engine_common_pb2.EngineRunnableConfig | None:
+    if config is None:
+        return None
+    return config_to_proto(config, skip_private_configurable=True)
+
+
 def checkpoint_from_proto(
     request_checkpoint: engine_common_pb2.Checkpoint,
 ) -> Checkpoint:
@@ -135,13 +144,13 @@ def checkpoint_tuple_to_proto(
     checkpoint_tuple: CheckpointTuple,
 ) -> engine_common_pb2.CheckpointTuple:
     proto = engine_common_pb2.CheckpointTuple()
-    if config_pb := config_to_proto(checkpoint_tuple.config):
+    if config_pb := checkpoint_config_to_proto(checkpoint_tuple.config):
         proto.config.CopyFrom(config_pb)
     proto.checkpoint.CopyFrom(checkpoint_to_proto(checkpoint_tuple.checkpoint))
     if checkpoint_tuple.metadata:
         proto.metadata.CopyFrom(checkpoint_metadata_to_proto(checkpoint_tuple.metadata))
     if checkpoint_tuple.parent_config:
-        if parent_pb := config_to_proto(checkpoint_tuple.parent_config):
+        if parent_pb := checkpoint_config_to_proto(checkpoint_tuple.parent_config):
             proto.parent_config.CopyFrom(parent_pb)
     if checkpoint_tuple.pending_writes:
         for task_id, channel, value in checkpoint_tuple.pending_writes:

langgraph_grpc_common/conversion/config.py

@@ -30,6 +30,7 @@ from langgraph_grpc_common.conversion.struct import _default_serializer
 from langgraph_grpc_common.proto import engine_common_pb2, enum_stream_mode_pb2
 
 CONFIG_KEY_GRAPH_ID = "graph_id"
+CONFIG_KEY_NODE_FINISHED = "__pregel_node_finished"
 # DR-specific key for storing root graph stream modes, required for patched subgraph streaming.
 CONFIG_KEY_ROOT_STREAM_MODES = "__pregel_root_stream_modes"
 CONFIG_KEY_REPLAY_STATE = "__pregel_replay_state"
@@ -185,6 +186,8 @@ KNOWN_CONFIG_KEYS = {
 
 def config_to_proto(
     config: RunnableConfig,
+    *,
+    skip_private_configurable: bool = False,
 ) -> engine_common_pb2.EngineRunnableConfig | None:
     # Prepare kwargs for construction
     if not config:
@@ -228,7 +231,11 @@ def config_to_proto(
         pb_config.tags.extend(tags)
 
     if configurable := config.get("configurable"):
-        _inject_configurable_into_proto(configurable, pb_config)
+        _inject_configurable_into_proto(
+            configurable,
+            pb_config,
+            skip_private_configurable=skip_private_configurable,
+        )
 
     # Preserve extra top-level keys (not in KNOWN_CONFIG_KEYS)
     extra = {k: v for k, v in config.items() if k not in KNOWN_CONFIG_KEYS}
@@ -252,6 +259,8 @@ def config_to_proto(
     return pb_config
 
 
+# Known process-local framework values excluded from every proto conversion.
+# Checkpoint conversion also skips arbitrary private keys to match Python savers.
 RESTRICTED_RESERVED_CONFIGURABLE_KEYS = {
     CONFIG_KEY_SEND,
     CONFIG_KEY_READ,
@@ -261,13 +270,17 @@ RESTRICTED_RESERVED_CONFIGURABLE_KEYS = {
     CONFIG_KEY_STREAM,
     CONFIG_KEY_CACHE,
     CONFIG_KEY_RUNNER_SUBMIT,
+    CONFIG_KEY_NODE_FINISHED,
     CONFIG_KEY_ROOT_STREAM_MODES,
     CONFIG_KEY_REPLAY_STATE,
 }
 
 
 def _inject_configurable_into_proto(
-    configurable: dict[str, Any], proto: engine_common_pb2.EngineRunnableConfig
+    configurable: dict[str, Any],
+    proto: engine_common_pb2.EngineRunnableConfig,
+    *,
+    skip_private_configurable: bool = False,
 ) -> None:
     extra = {}
     for key, value in configurable.items():
@@ -313,26 +326,24 @@ def _inject_configurable_into_proto(
         elif key == CONFIG_KEY_TRACING_EXAMPLE_ID:
             if value is not None:
                 proto.tracing_example_id = str(value)
+        elif skip_private_configurable and key.startswith("__"):
+            continue
         elif key not in RESTRICTED_RESERVED_CONFIGURABLE_KEYS:
             extra[key] = value
-    if extra:
-        extra_configurable_json = {}
-        for k, v in extra.items():
-            try:
-                extra_configurable_json[k] = orjson.dumps(
-                    v, default=_default_serializer
-                )
-            except Exception:
-                logger.warning(
-                    "Failed to serialize extra configurable value",
-                    extra={
-                        "configurable_key": str(k),
-                        "configurable_value_type": str(type(v)),
-                    },
-                )
-                raise
-
-        proto.extra_configurable_json.update(extra_configurable_json)
+    for k, v in extra.items():
+        try:
+            proto.extra_configurable_json[k] = orjson.dumps(
+                v, default=_default_serializer
+            )
+        except Exception:
+            logger.warning(
+                "Failed to serialize extra configurable value",
+                extra={
+                    "configurable_key": str(k),
+                    "configurable_value_type": str(type(v)),
+                },
+            )
+            raise
 
 
 def runtime_to_proto(runtime: Runtime) -> engine_common_pb2.Runtime:

openapi.json

@@ -948,6 +948,60 @@
         }
       }
     },
+    "/threads/{thread_id}/storage": {
+      "get": {
+        "tags": ["Threads"],
+        "summary": "Get Thread Storage",
+        "description": "Bytes the thread's checkpoints occupy in the built-in checkpointer, as stored: checkpoints, channel values (per channel) and pending writes. Returns 501 when a custom checkpointer is configured.",
+        "operationId": "get_thread_storage_threads__thread_id__storage_get",
+        "parameters": [
+          {
+            "description": "The ID of the thread.",
+            "required": true,
+            "schema": {
+              "type": "string",
+              "format": "uuid",
+              "title": "Thread Id",
+              "description": "The ID of the thread."
+            },
+            "name": "thread_id",
+            "in": "path"
+          }
+        ],
+        "responses": {
+          "200": {
+            "description": "Success",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/ThreadStorage"
+                }
+              }
+            }
+          },
+          "404": {
+            "description": "Thread not found",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/ErrorResponse"
+                }
+              }
+            }
+          },
+          "422": {
+            "description": "Validation Error",
+            "content": {
+              "application/json": {
+                "schema": {
+                  "$ref": "#/components/schemas/ErrorResponse"
+                }
+              }
+            }
+          }
+        }
+      }
+    },
     "/threads/{thread_id}/state": {
       "get": {
         "tags": ["Threads"],
@@ -5940,6 +5994,72 @@
         "title": "ThreadStateCheckpointRequest",
         "description": "Payload for getting the state of a thread at a checkpoint."
       },
+      "ThreadStorage": {
+        "properties": {
+          "thread_id": {
+            "type": "string",
+            "format": "uuid",
+            "title": "Thread Id"
+          },
+          "total_bytes": {
+            "type": "integer",
+            "title": "Total Bytes",
+            "description": "Sum of the three sections below."
+          },
+          "checkpoints": {
+            "type": "object",
+            "title": "Checkpoints",
+            "description": "Checkpoint rows: serialized checkpoint plus metadata.",
+            "properties": {
+              "count": { "type": "integer", "title": "Count" },
+              "bytes": { "type": "integer", "title": "Bytes" }
+            },
+            "required": ["count", "bytes"]
+          },
+          "channel_values": {
+            "type": "object",
+            "title": "Channel Values",
+            "description": "Stored channel values, one row per channel version.",
+            "properties": {
+              "count": { "type": "integer", "title": "Count" },
+              "bytes": { "type": "integer", "title": "Bytes" },
+              "by_channel": {
+                "type": "object",
+                "title": "By Channel",
+                "additionalProperties": {
+                  "type": "object",
+                  "properties": {
+                    "versions": { "type": "integer", "title": "Versions" },
+                    "bytes": { "type": "integer", "title": "Bytes" }
+                  },
+                  "required": ["versions", "bytes"]
+                }
+              }
+            },
+            "required": ["count", "bytes", "by_channel"]
+          },
+          "pending_writes": {
+            "type": "object",
+            "title": "Pending Writes",
+            "description": "Task writes not yet folded into a checkpoint.",
+            "properties": {
+              "count": { "type": "integer", "title": "Count" },
+              "bytes": { "type": "integer", "title": "Bytes" }
+            },
+            "required": ["count", "bytes"]
+          }
+        },
+        "type": "object",
+        "required": [
+          "thread_id",
+          "total_bytes",
+          "checkpoints",
+          "channel_values",
+          "pending_writes"
+        ],
+        "title": "ThreadStorage",
+        "description": "Bytes a thread occupies in the checkpointer, as stored."
+      },
       "ThreadState": {
         "properties": {
           "values": {