langgraph-runtime-inmem 0.36.0.dev6__py3-none-any.whl → 0.36.0.dev7__py3-none-any.whl

langgraph_runtime_inmem/__init__.py

@@ -10,7 +10,7 @@ from langgraph_runtime_inmem import (
     store,
 )
 
-__version__ = "0.36.0.dev6"
+__version__ = "0.36.0.dev7"
 __all__ = [
     "ops",
     "database",

langgraph_runtime_inmem/checkpoint.py

@@ -299,3 +299,50 @@ def Checkpointer(*args, unpack_hook=None, **kwargs):
 
 
 __all__ = ["Checkpointer"]
+
+
+def _nbytes(value: Any) -> int:
+    if isinstance(value, bytes | bytearray | memoryview):
+        return len(value)
+    if isinstance(value, tuple | list):
+        return sum(_nbytes(item) for item in value)
+    return 0
+
+
+async def thread_storage(conn: Any, thread_id: str) -> dict[str, Any]:
+    """Bytes a thread's checkpoints occupy in memory, per table and per channel.
+
+    Same shape as the postgres runtime's report, so one endpoint serves both.
+    """
+    saver = Checkpointer()
+    by_channel: dict[str, dict[str, int]] = {}
+    for (tid, _ns, channel, _version), blob in list(saver.blobs.items()):
+        if tid != thread_id:
+            continue
+        entry = by_channel.setdefault(channel, {"versions": 0, "bytes": 0})
+        entry["versions"] += 1
+        entry["bytes"] += _nbytes(blob)
+    checkpoint_count = checkpoint_bytes = 0
+    if thread_id in saver.storage:
+        for checkpoints in saver.storage[thread_id].values():
+            for stored in checkpoints.values():
+                checkpoint_count += 1
+                checkpoint_bytes += _nbytes(stored)
+    write_count = write_bytes = 0
+    for (tid, _ns, _checkpoint_id), writes in list(saver.writes.items()):
+        if tid != thread_id:
+            continue
+        for write in writes.values():
+            write_count += 1
+            write_bytes += _nbytes(write)
+    channel_bytes = sum(entry["bytes"] for entry in by_channel.values())
+    return {
+        "total_bytes": checkpoint_bytes + channel_bytes + write_bytes,
+        "checkpoints": {"count": checkpoint_count, "bytes": checkpoint_bytes},
+        "channel_values": {
+            "count": sum(entry["versions"] for entry in by_channel.values()),
+            "bytes": channel_bytes,
+            "by_channel": by_channel,
+        },
+        "pending_writes": {"count": write_count, "bytes": write_bytes},
+    }

{langgraph_runtime_inmem-0.36.0.dev6.dist-info → langgraph_runtime_inmem-0.36.0.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.36.0.dev6
+Version: 0.36.0.dev7
 Summary: Inmem implementation for the LangGraph API server.
 Author-email: Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0

{langgraph_runtime_inmem-0.36.0.dev6.dist-info → langgraph_runtime_inmem-0.36.0.dev7.dist-info}/RECORD

@@ -1,6 +1,6 @@
-langgraph_runtime_inmem/__init__.py,sha256=M6VflFHbXEgvO81PfF83fzpAfAihXRl6PsaNTu0_dBw,342
+langgraph_runtime_inmem/__init__.py,sha256=Nbur-y5GQywaBa_M7pVAY7kDOVBl7kMyL_2b9uMlqRk,342
 langgraph_runtime_inmem/_persistence.py,sha256=0h_HLXc079TG0rXoiQjICPZLhhBpTsmZ2PG8UbB5YY4,1879
-langgraph_runtime_inmem/checkpoint.py,sha256=YWypCmwCkuG3vbgKh36Snurgig5vXyjeL0cOOfh-hkM,10313
+langgraph_runtime_inmem/checkpoint.py,sha256=oC8MILNJVmrv3Ph2XyRJ49Wb883yuv2tlsOCuPDRin0,12173
 langgraph_runtime_inmem/database.py,sha256=BdEPPBn2N-g5GsqWcia96SrHLKqocXRMYAO6pHlGcHM,6779
 langgraph_runtime_inmem/inmem_stream.py,sha256=ZKwPFtSFcQfdDLSIkEBpUXv692YdPpzpcqy5wzJjcUY,9207
 langgraph_runtime_inmem/lifespan.py,sha256=F93NL8ckeJGe78WhNMeC0ysOchSBym69KwyHIa_yqkY,6736
@@ -10,6 +10,6 @@ langgraph_runtime_inmem/queue.py,sha256=CtsVVtxj3-QQ2OeXia2CcasNGLov02kfg0q0yQMy
 langgraph_runtime_inmem/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
 langgraph_runtime_inmem/routes.py,sha256=QtG9KPForGQHwMZYzl3BRV4Zy1pt0zJxPCX2hO0gaqY,18693
 langgraph_runtime_inmem/store.py,sha256=7hZnVHp097RYAen7SmTQSl-9tiiDCQfLehVxzxU2xAo,3811
-langgraph_runtime_inmem-0.36.0.dev6.dist-info/METADATA,sha256=k24YgOxlogp3M8l8bE4q1WidBDR5noc6_duPmuwNJAw,607
-langgraph_runtime_inmem-0.36.0.dev6.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
-langgraph_runtime_inmem-0.36.0.dev6.dist-info/RECORD,,
+langgraph_runtime_inmem-0.36.0.dev7.dist-info/METADATA,sha256=883jlRlsJoTng30krA-uY9Ls-0ZLgVcJiB9U4rggbcY,607
+langgraph_runtime_inmem-0.36.0.dev7.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+langgraph_runtime_inmem-0.36.0.dev7.dist-info/RECORD,,

{langgraph_runtime_inmem-0.36.0.dev6.dist-info → langgraph_runtime_inmem-0.36.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