langgraph-api 0.0.8__py3-none-any.whl → 0.0.9__py3-none-any.whl

langgraph_api/cron_scheduler.py

@@ -3,9 +3,10 @@ from random import random
 
 import structlog
 from langchain_core.runnables.config import run_in_executor
+from starlette.authentication import SimpleUser
 
 from langgraph_api.models.run import create_valid_run
-from langgraph_api.utils import next_cron_date
+from langgraph_api.utils import next_cron_date, with_user
 from langgraph_storage.database import connect
 from langgraph_storage.ops import Crons
 from langgraph_storage.retry import retry_db
@@ -22,35 +23,39 @@ async def cron_scheduler():
         try:
             async with connect() as conn:
                 async for cron in Crons.next(conn):
-                    logger.debug(f"Scheduling cron run {cron}")
-                    try:
-                        run_payload = cron["payload"]
-                        run = await create_valid_run(
-                            conn,
-                            thread_id=str(cron.get("thread_id"))
-                            if cron.get("thread_id")
-                            else None,
-                            payload=run_payload,
-                            user_id=cron.get("user_id"),
-                            headers={},
-                        )
-                        if not run:
-                            logger.error(
-                                "Run not created for cron_id={} payload".format(
-                                    cron["cron_id"],
+                    async with with_user(SimpleUser(str(cron["user_id"])), None):
+                        logger.debug(f"Scheduling cron run {cron}")
+                        try:
+                            run_payload = cron["payload"]
+                            run = await create_valid_run(
+                                conn,
+                                thread_id=(
+                                    str(cron.get("thread_id"))
+                                    if cron.get("thread_id")
+                                    else None
+                                ),
+                                payload=run_payload,
+                                headers={},
+                            )
+                            if not run:
+                                logger.error(
+                                    "Run not created for cron_id={} payload".format(
+                                        cron["cron_id"],
+                                    )
                                 )
+                        except Exception as e:
+                            logger.error(
+                                "Error scheduling cron run cron_id={}".format(
+                                    cron["cron_id"]
+                                ),
+                                exc_info=e,
                             )
-                    except Exception as e:
-                        logger.error(
-                            "Error scheduling cron run cron_id={}".format(
-                                cron["cron_id"]
-                            ),
-                            exc_info=e,
+                        next_run_date = await run_in_executor(
+                            None, next_cron_date, cron["schedule"], cron["now"]
+                        )
+                        await Crons.set_next_run_date(
+                            conn, cron["cron_id"], next_run_date
                         )
-                    next_run_date = await run_in_executor(
-                        None, next_cron_date, cron["schedule"], cron["now"]
-                    )
-                    await Crons.set_next_run_date(conn, cron["cron_id"], next_run_date)
 
             await asyncio.sleep(SLEEP_TIME)
         except asyncio.CancelledError:

langgraph_api/graph.py

@@ -366,17 +366,6 @@ def _graph_from_spec(spec: GraphSpec) -> GraphValue:
     return graph
 
 
-def _load(path: str, variable: str) -> None:
-    modname = "".join(choice("abcdefghijklmnopqrstuvwxyz") for _ in range(24))
-    modspec = importlib.util.spec_from_file_location(modname, path)
-    if modspec is None:
-        raise ValueError(f"Could not find python file for embeddings: {path}")
-    module = importlib.util.module_from_spec(modspec)
-    sys.modules[modname] = module
-    modspec.loader.exec_module(module)
-    return module.__dict__[variable]
-
-
 @functools.lru_cache
 def _get_init_embeddings() -> Callable[[str, ...], "Embeddings"] | None:
     try:

langgraph_api/models/run.py

@@ -1,10 +1,11 @@
 import asyncio
 import uuid
 from collections.abc import Mapping, Sequence
-from typing import Any, TypedDict
+from typing import Any, NamedTuple, TypedDict
 from uuid import UUID
 
 from langgraph.checkpoint.base.id import uuid6
+from starlette.authentication import BaseUser
 from starlette.exceptions import HTTPException
 
 from langgraph_api.graph import get_assistant_id
@@ -19,7 +20,7 @@ from langgraph_api.schema import (
     RunCommand,
     StreamMode,
 )
-from langgraph_api.utils import AsyncConnectionProto
+from langgraph_api.utils import AsyncConnectionProto, get_auth_ctx
 from langgraph_storage.ops import Runs, logger
 
 
@@ -82,24 +83,32 @@ class RunCreateDict(TypedDict):
 
 
 def ensure_ids(
-    assistant_id: str,
-    thread_id: str | None,
+    assistant_id: str | UUID,
+    thread_id: str | UUID | None,
     payload: RunCreateDict,
 ) -> tuple[uuid.UUID, uuid.UUID | None, uuid.UUID | None]:
     try:
-        results = [UUID(assistant_id)]
+        results = [
+            assistant_id if isinstance(assistant_id, UUID) else UUID(assistant_id)
+        ]
     except ValueError:
         raise HTTPException(status_code=422, detail="Invalid assistant ID") from None
     if thread_id:
         try:
-            results.append(UUID(thread_id))
+            results.append(
+                thread_id if isinstance(thread_id, UUID) else UUID(thread_id)
+            )
         except ValueError:
             raise HTTPException(status_code=422, detail="Invalid thread ID") from None
     else:
         results.append(None)
-    if payload.get("checkpoint_id"):
+    if checkpoint_id := payload.get("checkpoint_id"):
         try:
-            results.append(UUID(payload.get("checkpoint_id")))
+            results.append(
+                checkpoint_id
+                if isinstance(checkpoint_id, UUID)
+                else UUID(checkpoint_id)
+            )
         except ValueError:
             raise HTTPException(
                 status_code=422, detail="Invalid checkpoint ID"
@@ -125,27 +134,40 @@ def assign_defaults(
     return stream_mode, multitask_strategy, prevent_insert_if_inflight
 
 
+def get_user_id(user: BaseUser | None) -> str | None:
+    if user is None:
+        return None
+    try:
+        return user.identity
+    except NotImplementedError:
+        try:
+            return user.display_name
+        except NotImplementedError:
+            pass
+
+
 async def create_valid_run(
     conn: AsyncConnectionProto,
     thread_id: str | None,
     payload: RunCreateDict,
-    user_id: str | None,
     headers: Mapping[str, str],
     barrier: asyncio.Barrier | None = None,
     run_id: UUID | None = None,
 ) -> Run:
-    # get assistant_id
-    assistant_id = get_assistant_id(payload["assistant_id"])
-
-    # ensure UUID validity defaults
-    assistant_id, thread_id, checkpoint_id = ensure_ids(
-        assistant_id, thread_id, payload
+    (
+        assistant_id,
+        thread_id,
+        checkpoint_id,
+        run_id,
+    ) = _get_ids(
+        thread_id,
+        payload,
+        run_id=run_id,
     )
+    temporary = thread_id is None and payload.get("on_completion", "delete") == "delete"
     stream_mode, multitask_strategy, prevent_insert_if_inflight = assign_defaults(
         payload
     )
-    temporary = thread_id is None and payload.get("on_completion", "delete") == "delete"
-    run_id = run_id or uuid6()
 
     # assign custom headers and checkpoint to config
     config = payload.get("config") or {}
@@ -163,8 +185,14 @@ async def create_valid_run(
             ):
                 continue
             config["configurable"][key] = value
-
-    # try to insert
+    ctx = get_auth_ctx()
+    if ctx:
+        user = ctx.user
+        user_id = get_user_id(user)
+        config["configurable"]["langgraph_auth_user"] = user
+        config["configurable"]["langgraph_auth_user_id"] = user_id
+    else:
+        user_id = None
     run_coro = Runs.put(
         conn,
         assistant_id,
@@ -227,3 +255,33 @@ async def create_valid_run(
         )
     else:
         raise NotImplementedError
+
+
+class _Ids(NamedTuple):
+    assistant_id: uuid.UUID
+    thread_id: uuid.UUID | None
+    checkpoint_id: uuid.UUID | None
+    run_id: uuid.UUID
+
+
+def _get_ids(
+    thread_id: str | None,
+    payload: RunCreateDict,
+    run_id: UUID | None = None,
+) -> _Ids:
+    # get assistant_id
+    assistant_id = get_assistant_id(payload["assistant_id"])
+
+    # ensure UUID validity defaults
+    assistant_id, thread_id, checkpoint_id = ensure_ids(
+        assistant_id, thread_id, payload
+    )
+
+    run_id = run_id or uuid6()
+
+    return _Ids(
+        assistant_id,
+        thread_id,
+        checkpoint_id,
+        run_id,
+    )

langgraph_api/route.py

@@ -14,6 +14,7 @@ from starlette.routing import Route, compile_path, get_name
 from starlette.types import ASGIApp, Receive, Scope, Send
 
 from langgraph_api.serde import json_dumpb
+from langgraph_api.utils import set_auth_ctx
 
 
 def api_request_response(
@@ -115,4 +116,5 @@ class ApiRoute(Route):
     async def handle(self, scope: Scope, receive: Receive, send: Send) -> None:
         # https://asgi.readthedocs.io/en/latest/specs/www.html#http-connection-scope
         scope["route"] = self.path
+        set_auth_ctx(scope.get("user"), scope.get("auth"))
         return await super().handle(scope, receive, send)

langgraph_api/utils.py

@@ -1,13 +1,45 @@
+import contextvars
 import uuid
 from collections.abc import AsyncIterator
 from contextlib import asynccontextmanager
 from datetime import datetime
 from typing import Any, Protocol, TypeAlias, TypeVar
 
+from langgraph_sdk import Auth
+from starlette.authentication import AuthCredentials, BaseUser
 from starlette.exceptions import HTTPException
 
 T = TypeVar("T")
 Row: TypeAlias = dict[str, Any]
+AuthContext = contextvars.ContextVar[Auth.types.BaseAuthContext | None](
+    "AuthContext", default=None
+)
+
+
+@asynccontextmanager
+async def with_user(user: BaseUser | None = None, auth: AuthCredentials | None = None):
+    current = get_auth_ctx()
+    set_auth_ctx(user, auth)
+    yield
+    if current is None:
+        return
+    set_auth_ctx(current.user, AuthCredentials(scopes=current.scopes))
+
+
+def set_auth_ctx(user: BaseUser | None, auth: AuthCredentials | None) -> None:
+    if not user or not auth:
+        AuthContext.set(None)
+    else:
+        AuthContext.set(
+            Auth.types.BaseAuthContext(
+                scopes=auth.scopes,
+                user=user,
+            )
+        )
+
+
+def get_auth_ctx() -> Auth.types.BaseAuthContext | None:
+    return AuthContext.get()
 
 
 class AsyncCursorProto(Protocol):

{langgraph_api-0.0.8.dist-info → langgraph_api-0.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.8
+Version: 0.0.9
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -16,7 +16,7 @@ Requires-Dist: jsonschema-rs (>=0.25.0,<0.26.0)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.56,<0.3.0)
 Requires-Dist: langgraph-checkpoint (>=2.0.7,<3.0)
-Requires-Dist: langsmith (>=0.1.63,<0.2.0)
+Requires-Dist: langsmith (>=0.1.63,<0.3.0)
 Requires-Dist: orjson (>=3.10.1)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)
 Requires-Dist: sse-starlette (>=2.1.0,<3.0.0)

{langgraph_api-0.0.8.dist-info → langgraph_api-0.0.9.dist-info}/RECORD

@@ -3,22 +3,23 @@ langgraph_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/api/__init__.py,sha256=tlMXuqnyJt99aSlUXwR-dS3w5X6sDDczJu4hbm2LP30,2057
 langgraph_api/api/assistants.py,sha256=3v4v7kLmlb2meSAnNt73toijnaSikbJgH9Jhb6IL04g,11270
 langgraph_api/api/meta.py,sha256=hueasWpTDQ6xYLo9Bzt2jhNH8XQRzreH8FTeFfnRoxQ,2700
-langgraph_api/api/openapi.py,sha256=zvKMMebBP28hLPMNt1kTtFK17aV9Tz-ok2okEM45XjA,906
-langgraph_api/api/runs.py,sha256=X3fPWVjj4IUGWMIkjVkddVSqBaNh2f9HqtXNVLol8-Q,15961
+langgraph_api/api/openapi.py,sha256=3eRo3V6Pni8DjqHI_MNLn8F5CEOOGVBMXPEffft8Ubo,2781
+langgraph_api/api/runs.py,sha256=wAzPXi_kcYB9BcLBL4FXgkBohWwCPIpe4XERnsnWnsA,16042
 langgraph_api/api/store.py,sha256=y7VIejpsE7rpPF-tiMGBqqBwWPZ1wb3o48th6NUvb5I,3849
 langgraph_api/api/threads.py,sha256=taU61XPcCEhBPCYPZcMDsgVDwwWUWJs8p-PrXFXWY48,8661
 langgraph_api/asyncio.py,sha256=XiFEllu-Kg4zAO084npHPYOPnLQRire3V75XrVQYMxE,6023
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langgraph_api/auth/custom.py,sha256=zBJlCjHBr-yAP5Cw-9Q2wa2HVMSsm7BWCqYB8dni-R8,16785
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/langsmith/backend.py,sha256=uHeb5-h13NIjrX_LDAvfWYr3zpbJvlvbdUffch48hbM,2571
 langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdTV8bKd7x4Q,3998
-langgraph_api/auth/middleware.py,sha256=_gJTOskEaln4RUT2rVYdQGPJVAyAiq-orsL_eQ3CynE,1369
+langgraph_api/auth/middleware.py,sha256=4RhtYv1eBA57irNUT3nXc5vpgBTKysaOKBnjyCMSuCY,1511
 langgraph_api/auth/noop.py,sha256=vDJmzG2vArJxVzdHePvrJWahEa0dvGnhc2LEMMeiFz0,391
-langgraph_api/cli.py,sha256=qQIJHw8doFE-sLebnqnUZjukeYp_WW4GDQ443l3z0ag,10281
-langgraph_api/config.py,sha256=cG6eO4P_SZ2pKedb2b4n4vnBHRQr0aiECvGvOA8ZlJA,2259
-langgraph_api/cron_scheduler.py,sha256=DAzY2DsADzEpPVbG2BOSLTIufI93yeRswd71Aby_lV0,2257
+langgraph_api/cli.py,sha256=7vQQiD3F50r-8KkbuFjwIz8LLbdKUTd4xZGUJPiO3yQ,11688
+langgraph_api/config.py,sha256=m3cf13AS9HVF84deJ3-9mmNTJ5GSVqcukES-FNbDbvo,2794
+langgraph_api/cron_scheduler.py,sha256=CybK-9Jwopi_scObTHRyB7gyb0KjC4gqaT2GLe-WOFg,2587
 langgraph_api/errors.py,sha256=Bu_i5drgNTyJcLiyrwVE_6-XrSU50BHf9TDpttki9wQ,1690
-langgraph_api/graph.py,sha256=gpLblFbqtT2VQU2I69KvRX8mqOJRJDxQW5Sh1fW8FXY,16268
+langgraph_api/graph.py,sha256=LXTo2lGnku3KPBiC9ssh4NcSvieT4Ks2vJuhPaqBCpY,15807
 langgraph_api/http.py,sha256=XrbyxpjtfSvnaWWh5ZLGpgZmY83WoDCrP_1GPguNiXI,4712
 langgraph_api/http_logger.py,sha256=Sxo_q-65tElauRvkzVLt9lJojgNdgtcHGBYD0IRyX7M,3146
 langgraph_api/js/.gitignore,sha256=qAah3Fq0HWAlfRj5ktZyC6QRQIsAolGLRGcRukA1XJI,33
@@ -54,33 +55,33 @@ langgraph_api/lifespan.py,sha256=Uj7NV-NqxxD1fgx_umM9pVqclcy-VlqrIxDljyj2he0,182
 langgraph_api/logging.py,sha256=tiDNrEFwqaIdL5ywZv908OXlzzfXsPCws9GXeoFtBV8,3367
 langgraph_api/metadata.py,sha256=mih2G7ScQxiqyUlbksVXkqR3Oo-pM1b6lXtzOsgR1sw,3044
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/models/run.py,sha256=nNhlG-l2P-DThFWYmCaar0UuPctadB9sH3CCWJfLqVc,8178
+langgraph_api/models/run.py,sha256=qGTcgQjnzmXrxer5VPaGHoGwifV1nyE4ZCXtBiZH1bc,9546
 langgraph_api/patch.py,sha256=94ddcTSZJe22JcpjxiSNjFZdYVnmeoWjk4IX4iBSoyk,1249
 langgraph_api/queue.py,sha256=7tsbDgv4GlUYieJsrvIJDMQUEok4Eu-n_PIQ93rwKjk,9810
-langgraph_api/route.py,sha256=Dzje_dSigJramglqkt4ERT9-cb2xFli7dx25ZV6B6mI,4147
+langgraph_api/route.py,sha256=HnAcWeStCrWNe37YUXIqEsjsrCPCIPGijHG7pM9ga28,4251
 langgraph_api/schema.py,sha256=EiCWRR2GmGrBrOYcuK9SeVQS5b98SdaJlKaqOL7t-WQ,5263
 langgraph_api/serde.py,sha256=VoJ7Z1IuqrQGXFzEP1qijAITtWCrmjtVqlCRuScjXJI,3533
 langgraph_api/server.py,sha256=afHDnL6b_fAIu_q4icnK60a74lHTTZOMIe1egdhRXIk,1522
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
 langgraph_api/stream.py,sha256=u0gjbCrmYvVw8Ux6DgsYTojLCHSwM4Pi-0LhSLGY4HM,11546
-langgraph_api/utils.py,sha256=FI50tOFMVidV4-1TefouL1N-OJX41qD_fSEoWigTtf0,1575
+langgraph_api/utils.py,sha256=17AZrdVzGHUQ7Y8b82Mknxe1LD5JMo4tZsb6TCcXEW0,2474
 langgraph_api/validation.py,sha256=McizHlz-Ez8Jhdbc79mbPSde7GIuf2Jlbjx2yv_l6dA,4475
 langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_license/middleware.py,sha256=_ODIYzQkymr6W9_Fp9wtf1kAQspnpsmr53xuzyF2GA0,612
 langgraph_license/validation.py,sha256=Uu_G8UGO_WTlLsBEY0gTVWjRR4czYGfw5YAD3HLZoj0,203
 langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_storage/checkpoint.py,sha256=55J7W0s5Z7svqqan9wphubSgCphWUItq98j-iWVwbH8,2774
-langgraph_storage/database.py,sha256=jYAC1tDijO7jTPABxAKB2GutTKLf3EkbMI9VAIM3dT8,5186
-langgraph_storage/ops.py,sha256=Uev3bLTrEP53vX9POHIEo0eYEVH0pyZiLmliE04VCIM,52708
+langgraph_storage/checkpoint.py,sha256=V4t2GwYEJdPCHbhq_4Udhlv0TWKDzlMu_rlNPdTDc50,3589
+langgraph_storage/database.py,sha256=Nr5zE9Fur3-tESkqe7xNXMf2QlBuw3H0CUie7jVa6Q4,6003
+langgraph_storage/ops.py,sha256=34zhzc1ISESlsfM-QlHaUlSNIvqSIgmN3aJHTQMb-xA,67670
 langgraph_storage/queue.py,sha256=6cTZ0ubHu3S1T43yxHMVOwsQsDaJupByiU0sTUFFls8,3261
 langgraph_storage/retry.py,sha256=uvYFuXJ-T6S1QY1ZwkZHyZQbsvS-Ab68LSbzbUUSI2E,696
 langgraph_storage/store.py,sha256=D-p3cWc_umamkKp-6Cz3cAriSACpvM5nxUIvND6PuxE,2710
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=UxAGHZYM4PgNd48TSZt7f2lVuyPUkDadxBBhRy5jcmk,124512
-langgraph_api-0.0.8.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.8.dist-info/METADATA,sha256=A_zUXmA2dKkblSdlXqHFDeKePEWLja9Yenl57LleJKs,3993
-langgraph_api-0.0.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.8.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.8.dist-info/RECORD,,
+langgraph_api-0.0.9.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.9.dist-info/METADATA,sha256=dfqbTUQL54HjgW4XZThB8H-bJALao6b5Fhi01JXQwE4,3993
+langgraph_api-0.0.9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.9.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.9.dist-info/RECORD,,

langgraph_storage/checkpoint.py

@@ -1,3 +1,4 @@
+import logging
 import os
 import uuid
 
@@ -12,6 +13,8 @@ from langgraph.checkpoint.memory import MemorySaver, PersistentDict
 
 from langgraph_api.serde import Serializer
 
+logger = logging.getLogger(__name__)
+
 _EXCLUDED_KEYS = {"checkpoint_ns", "checkpoint_id", "run_id", "thread_id"}
 
 
@@ -36,6 +39,19 @@ class InMemorySaver(MemorySaver):
                 d.load()
             except FileNotFoundError:
                 pass
+            except ModuleNotFoundError:
+                logger.error(
+                    "Unable to load cached data - your code has changed in a way that's incompatible with the cache."
+                    "\nThis usually happens when you've:"
+                    "\n  - Renamed or moved classes"
+                    "\n  - Changed class structures"
+                    "\n  - Pulled updates that modified class definitions in a way that's incompatible with the cache"
+                    "\n\nRemoving invalid cache data stored at path: .langgraph_api"
+                )
+                os.remove(self.filename)
+            except Exception as e:
+                logger.error("Failed to load cached data: %s", str(e))
+                os.remove(self.filename)
             return d
 
         super().__init__(

langgraph_storage/database.py

@@ -164,7 +164,23 @@ async def start_pool() -> None:
     if not os.path.exists(".langgraph_api"):
         os.mkdir(".langgraph_api")
     if os.path.exists(OPS_FILENAME):
-        GLOBAL_STORE.load()
+        try:
+            GLOBAL_STORE.load()
+        except ModuleNotFoundError:
+            logger.error(
+                "Unable to load cached data - your code has changed in a way that's incompatible with the cache."
+                "\nThis usually happens when you've:"
+                "\n  - Renamed or moved classes"
+                "\n  - Changed class structures"
+                "\n  - Pulled updates that modified class definitions in a way that's incompatible with the cache"
+                "\n\nRemoving invalid cache data stored at path: .langgraph_api"
+            )
+            os.remove(OPS_FILENAME)
+            os.remove(RETRY_COUNTER_FILENAME)
+        except Exception as e:
+            logger.error("Failed to load cached data: %s", str(e))
+            os.remove(OPS_FILENAME)
+            os.remove(RETRY_COUNTER_FILENAME)
     for k in ["runs", "threads", "assistants", "assistant_versions"]:
         if not GLOBAL_STORE.get(k):
             GLOBAL_STORE[k] = []