langgraph-runtime-pg 0.11.1__py3-none-any.whl

langgraph_runtime_pg/retry.py

@@ -0,0 +1,39 @@
+"""DB retry helpers matching the inmem surface."""
+
+from __future__ import annotations
+
+import asyncio
+import functools
+from collections.abc import Awaitable, Callable
+from typing import ParamSpec, TypeVar
+
+P = ParamSpec("P")
+T = TypeVar("T")
+
+
+class RetryableException(Exception):
+    pass
+
+
+RETRIABLE_EXCEPTIONS: tuple[type[BaseException], ...] = (RetryableException,)
+OVERLOADED_EXCEPTIONS: tuple[type[BaseException], ...] = ()
+
+
+def retry_db(func: Callable[P, Awaitable[T]]) -> Callable[P, Awaitable[T]]:
+    """Retry ``RETRIABLE_EXCEPTIONS`` up to 3 times."""
+
+    @functools.wraps(func)
+    async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
+        last: BaseException | None = None
+        for i in range(3):
+            try:
+                return await func(*args, **kwargs)
+            except RETRIABLE_EXCEPTIONS as exc:
+                last = exc
+                if i == 2:
+                    raise
+                await asyncio.sleep(0.01)
+        assert last is not None  # pragma: no cover
+        raise last  # pragma: no cover
+
+    return wrapper

langgraph_runtime_pg/routes.py

@@ -0,0 +1,9 @@
+"""Internal routes stub."""
+
+from __future__ import annotations
+
+from typing import Any
+
+
+def get_internal_routes() -> list[Any]:
+    return []

langgraph_runtime_pg/store.py

@@ -0,0 +1,183 @@
+"""Store factory: AsyncPostgresStore for the API, PgStore for tests."""
+
+from __future__ import annotations
+
+import asyncio
+from datetime import UTC, datetime
+from typing import Any, cast
+
+from langgraph.store.postgres.aio import AsyncPostgresStore
+from psycopg.rows import dict_row
+from psycopg_pool import AsyncConnectionPool
+from sqlalchemy import delete, select as sa_select
+from sqlalchemy.dialects.postgresql import insert as pg_insert
+from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
+
+from langgraph_runtime_pg.database import to_psycopg_uri
+from langgraph_runtime_pg.models import StoreItemRow
+
+_STORE_CONFIG: dict | None = None
+_STORE: AsyncPostgresStore | None = None
+_STORE_POOL: AsyncConnectionPool[Any] | None = None
+_STORE_INSTANCE: PgStore | None = None
+_SETUP_LOCK = asyncio.Lock()
+
+
+def set_store_config(config: dict) -> None:
+    global _STORE_CONFIG
+    _STORE_CONFIG = config
+
+
+async def setup_store() -> AsyncPostgresStore:
+    """Open AsyncPostgresStore and initialize its tables."""
+    global _STORE, _STORE_POOL
+    async with _SETUP_LOCK:
+        if _STORE is not None:
+            return _STORE
+        if _STORE_POOL is not None:
+            try:
+                await _STORE_POOL.close()
+            except Exception:
+                pass
+            _STORE_POOL = None
+
+        uri = to_psycopg_uri()
+        pool = AsyncConnectionPool(
+            conninfo=uri,
+            min_size=1,
+            max_size=10,
+            open=False,
+            kwargs={
+                "autocommit": True,
+                "prepare_threshold": 0,
+                "row_factory": dict_row,
+            },
+        )
+        try:
+            await pool.open()
+            store = AsyncPostgresStore(cast(Any, pool))
+            await store.setup()
+        except Exception:
+            try:
+                await pool.close()
+            except Exception:
+                pass
+            raise
+        _STORE_POOL = pool
+        _STORE = store
+        return _STORE
+
+
+async def teardown_store() -> None:
+    global _STORE, _STORE_POOL
+    store, pool = _STORE, _STORE_POOL
+    _STORE = None
+    _STORE_POOL = None
+    if store is not None:
+        try:
+            stop = getattr(store, "stop_ttl_sweeper", None)
+            if stop is not None:
+                await asyncio.wait_for(stop(), timeout=2.0)
+        except Exception:
+            pass
+    if pool is not None:
+        try:
+            await asyncio.wait_for(pool.close(), timeout=5.0)
+        except Exception:
+            pass
+
+
+class PgStore:
+    """Minimal prefix/key store over the store_items table (tests)."""
+
+    def __init__(self, session_factory: async_sessionmaker[AsyncSession]):
+        self._sf = session_factory
+
+    async def start_ttl_sweeper(self) -> asyncio.Task:
+        return asyncio.create_task(asyncio.sleep(0))
+
+    async def aput(self, prefix: str, key: str, value: dict) -> None:
+        now = datetime.now(UTC)
+        async with self._sf() as session:
+            await session.execute(
+                pg_insert(StoreItemRow)
+                .values(
+                    prefix=prefix,
+                    key=key,
+                    value=value,
+                    created_at=now,
+                    updated_at=now,
+                )
+                .on_conflict_do_update(
+                    index_elements=["prefix", "key"],
+                    set_={"value": value, "updated_at": now},
+                )
+            )
+            await session.commit()
+
+    async def aget(self, prefix: str, key: str) -> dict | None:
+        async with self._sf() as session:
+            row = (
+                await session.execute(
+                    sa_select(StoreItemRow).where(
+                        StoreItemRow.prefix == prefix,
+                        StoreItemRow.key == key,
+                    )
+                )
+            ).scalar_one_or_none()
+            return row.value if row else None
+
+    async def adelete(self, prefix: str, key: str) -> None:
+        async with self._sf() as session:
+            await session.execute(
+                delete(StoreItemRow).where(
+                    StoreItemRow.prefix == prefix,
+                    StoreItemRow.key == key,
+                )
+            )
+            await session.commit()
+
+    async def alist(self, prefix: str) -> list[dict]:
+        async with self._sf() as session:
+            rows = (
+                (
+                    await session.execute(
+                        sa_select(StoreItemRow).where(StoreItemRow.prefix == prefix)
+                    )
+                )
+                .scalars()
+                .all()
+            )
+            return [{"key": r.key, "value": r.value} for r in rows]
+
+    def close(self) -> None:
+        pass
+
+
+def Store(*args: Any, **kwargs: Any) -> AsyncPostgresStore:
+    """Return the process-wide AsyncPostgresStore (requires setup_store)."""
+    if _STORE is None:
+        raise RuntimeError(
+            "Store requires setup_store()/start_pool() first (DATABASE_URI required)"
+        )
+    return _STORE
+
+
+def pg_store_for_tests() -> PgStore:
+    """Return the test PgStore singleton over store_items."""
+    global _STORE_INSTANCE
+    if _STORE_INSTANCE is None:
+        from langgraph_runtime_pg.database import _SESSION_FACTORY
+
+        if _SESSION_FACTORY is None:
+            raise RuntimeError("Call start_pool() first")
+        _STORE_INSTANCE = PgStore(_SESSION_FACTORY)
+    return _STORE_INSTANCE
+
+
+def reset_store() -> None:
+    """Reset the test PgStore singleton."""
+    global _STORE_INSTANCE
+    if _STORE_INSTANCE is not None:
+        _STORE_INSTANCE.close()
+    _STORE_INSTANCE = None

langgraph_runtime_pg-0.11.1.dist-info/METADATA

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: langgraph-runtime-pg
+Version: 0.11.1
+Summary: Self-hosted LangSmith Deployments alternative: MIT Postgres+Redis runtime (edition=pg) for stock langgraph-api
+Project-URL: Homepage, https://github.com/langhost/langhost
+Project-URL: Repository, https://github.com/langhost/langhost
+Project-URL: Issues, https://github.com/langhost/langhost/issues
+Project-URL: Changelog, https://github.com/langhost/langhost/releases
+Author-email: Mohankumar Ramachandran <mail@mohanram.dev>
+License-Expression: MIT
+License-File: LICENSE
+Keywords: agent-protocol,agent-server,agents,ai-agents,asgi,checkpoint,langchain,langgraph,langgraph-api,langgraph-checkpoint,langgraph-sdk,langsmith,langsmith-deployments,llm,multi-agent,open-source,postgres,postgresql,redis,runtime,self-hosted,uvicorn
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Database
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
+Requires-Python: >=3.11
+Requires-Dist: alembic>=1.14
+Requires-Dist: asyncpg>=0.29
+Requires-Dist: croniter>=1.0
+Requires-Dist: langgraph-api==0.11.1
+Requires-Dist: langgraph-checkpoint-postgres==3.1.0
+Requires-Dist: orjson>=3.9
+Requires-Dist: psycopg[binary]>=3.3.4
+Requires-Dist: redis>=5.0
+Requires-Dist: sqlalchemy[asyncio]>=2.0.51
+Requires-Dist: starlette>=0.37
+Requires-Dist: structlog>=24.0
+Description-Content-Type: text/markdown
+
+# langgraph-runtime-pg
+
+MIT Postgres + Redis runtime (`LANGGRAPH_RUNTIME_EDITION=pg`) for stock
+[`langgraph-api`](https://pypi.org/project/langgraph-api/).
+
+See the [repository README](https://github.com/langhost/langhost)
+for setup, CLI (`langhost`), and architecture.

langgraph_runtime_pg-0.11.1.dist-info/RECORD

@@ -0,0 +1,23 @@
+langgraph_runtime_pg/__init__.py,sha256=ZUcjd1kcny4doy9RCGYC660IGUY4Kncym5kNOsoLCpo,697
+langgraph_runtime_pg/checkpoint.py,sha256=CDbfl0rpiU5obyWN5844qchR3yUxuredYiQ04-X0SW4,2413
+langgraph_runtime_pg/database.py,sha256=aKuaURsEZOH5MMwBBCDrSrHDgPedmOz1D-S4S9i1wbg,14389
+langgraph_runtime_pg/lifespan.py,sha256=j7RIbIE-9ASfL60dG8AYF1DitLZOsOpw2Sgfirz4Xdg,6603
+langgraph_runtime_pg/metrics.py,sha256=IUwlI2xAhwnYwiOR-ltg-aO655aEX0RouHKNJbRWlBc,128
+langgraph_runtime_pg/migrate.py,sha256=eq0GcE6EhapEOYv6jeNyQlm--G7qcwvN5AJbXqDAJHs,3061
+langgraph_runtime_pg/models.py,sha256=Fblz8JONqGpkPCaiU6kk4yPyrOUdh99MbkULhFcBons,8474
+langgraph_runtime_pg/ops.py,sha256=JAtM02a-81ALZrMocftvvin_jawr2bUG7vHHgrDoAPM,133432
+langgraph_runtime_pg/queue.py,sha256=MjDRKtEGUp9Sk-n6caQATiYnH7e-_7Dw_o8ofMVY9oo,10485
+langgraph_runtime_pg/redis_stream.py,sha256=Z-UQNyNRHkdOlZSMP2EigE6CvNooUE8Kp-2SAc57l6E,31813
+langgraph_runtime_pg/retry.py,sha256=uRx9QtuWKVdw4KGyfiI8NBklqVensGXYcTiun8joMYQ,1074
+langgraph_runtime_pg/routes.py,sha256=r2hK0E1RKyi1D4PBF3-dsk7_otu0BZZZfNzaa3TPEwo,144
+langgraph_runtime_pg/store.py,sha256=hFbQTVJLAPhFELwVdABAHvShENDztppD8VJ1v5QyaS4,5654
+langgraph_runtime_pg/migrations/__init__.py,sha256=M-gt0gXAGxVsYXJgZouNInvcPmk291foMrYnhooVaS0,56
+langgraph_runtime_pg/migrations/env.py,sha256=fWLULVAKqqmSvCpVHJ61WyJzuBYnqCSm8eCWgUbjHCY,1329
+langgraph_runtime_pg/migrations/versions/001_initial_schema.py,sha256=MnzQ8mEl0xCMhT5RBC4NujyEKUIkC8rhxHM_HWoXXYc,11747
+langgraph_runtime_pg/migrations/versions/__init__.py,sha256=4paf33VBK_l7fudrFVmQbL68tQNDGVNFUGgdjxBSq5s,32
+langgraph_runtime_pg/migrations/script.py.mako,sha256=kyVbIw4oMjKYXbLX7wA29--0PK_UU--SdXx-my0Znls,645
+langgraph_runtime_pg-0.11.1.dist-info/METADATA,sha256=y7Ysm6w5tXlDcsGiZFDD3pQg-C41X7lQINCUBLdzY5k,2194
+langgraph_runtime_pg-0.11.1.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+langgraph_runtime_pg-0.11.1.dist-info/entry_points.txt,sha256=8S_Q8wEvQtqbTxqgbHazjiCnmY2NDOEH2dLDVIHd3WA,83
+langgraph_runtime_pg-0.11.1.dist-info/licenses/LICENSE,sha256=VOzjTuqsyEVtwd7ylP1eoLjPAAvdB6iC9a7JBGFJUgE,1080
+langgraph_runtime_pg-0.11.1.dist-info/RECORD,,

langgraph_runtime_pg-0.11.1.dist-info/WHEEL

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

langgraph_runtime_pg-0.11.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+langgraph-runtime-pg-migrate = langgraph_runtime_pg.migrate:main

langgraph_runtime_pg-0.11.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Mohankumar Ramachandran
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.