langgraph-api 0.0.36__py3-none-any.whl → 0.0.38__py3-none-any.whl

langgraph_api/api/mcp.py

@@ -351,7 +351,9 @@ async def handle_tools_list(
 
     # Get assistants from the API
     # For now set a large limit to get all assistants
-    assistants = await client.assistants.search(offset=cursor, limit=DEFAULT_PAGE_SIZE)
+    assistants = await client.assistants.search(
+        offset=cursor, limit=DEFAULT_PAGE_SIZE, headers=request.headers
+    )
 
     if len(assistants) == DEFAULT_PAGE_SIZE:
         next_cursor = cursor + DEFAULT_PAGE_SIZE
@@ -370,7 +372,7 @@ async def handle_tools_list(
         else:
             seen_names.add(name)
 
-        schemas = await client.assistants.get_schemas(id_)
+        schemas = await client.assistants.get_schemas(id_, headers=request.headers)
         tools.append(
             {
                 "name": name,
@@ -408,7 +410,9 @@ async def handle_tools_call(
         }
 
     arguments = params.get("arguments", {})
-    assistants = await client.assistants.search(limit=MAX_ASSISTANTS)
+    assistants = await client.assistants.search(
+        limit=MAX_ASSISTANTS, headers=request.headers
+    )
     matching_assistant = [
         assistant for assistant in assistants if assistant["name"] == tool_name
     ]
@@ -437,7 +441,11 @@ async def handle_tools_call(
         tool_name = matching_assistant[0]["assistant_id"]
 
     value = await client.runs.wait(
-        thread_id=None, assistant_id=tool_name, input=arguments, raise_error=False
+        thread_id=None,
+        assistant_id=tool_name,
+        input=arguments,
+        raise_error=False,
+        headers=request.headers,
     )
 
     if "__error__" in value:

langgraph_api/auth/custom.py

@@ -379,13 +379,16 @@ def _depends() -> Any:
         return None
 
 
+_EXCLUDED = ("values", "keys", "items", "dict")
+
+
 class DotDict:
     def __init__(self, dictionary: dict[str, Any]):
         self._dict = dictionary
         for key, value in dictionary.items():
             if isinstance(value, dict):
                 setattr(self, key, DotDict(value))
-            else:
+            elif key not in _EXCLUDED:
                 setattr(self, key, value)
 
     def __getattr__(self, name):
@@ -393,6 +396,9 @@ class DotDict:
             raise AttributeError(f"'DotDict' object has no attribute '{name}'")
         return self._dict[name]
 
+    def __contains__(self, key: str) -> bool:
+        return key in self._dict
+
     def __getitem__(self, key):
         return self._dict[key]
 
@@ -409,6 +415,21 @@ class DotDict:
     def dict(self):
         return self._dict
 
+    def items(self):
+        return self._dict.items()
+
+    def values(self):
+        return self._dict.values()
+
+    def keys(self):
+        return self._dict.keys()
+
+    def __iter__(self):
+        return iter(self._dict)
+
+    def __len__(self):
+        return len(self._dict)
+
 
 class ProxyUser(BaseUser):
     """A proxy that wraps a user object to ensure it has all BaseUser properties.
@@ -462,6 +483,9 @@ class ProxyUser(BaseUser):
             **d,
         }
 
+    def __contains__(self, key: str) -> bool:
+        return key in self._user
+
     def __getitem__(self, key):
         return self._user[key]
 

langgraph_api/cli.py

@@ -130,6 +130,7 @@ def run_server(
     auth: AuthConfig | None = None,
     http: typing.Optional["HttpConfig"] = None,
     studio_url: str | None = None,
+    disable_persistence: bool = False,
     **kwargs: typing.Any,
 ):
     """Run the LangGraph API server."""
@@ -189,6 +190,7 @@ def run_server(
         LANGGRAPH_AUTH=json.dumps(auth) if auth else None,
         LANGGRAPH_HTTP=json.dumps(http) if http else None,
         LANGGRAPH_API_URL=local_url,
+        LANGGRAPH_DISABLE_FILE_PERSISTENCE=str(disable_persistence).lower(),
         # See https://developer.chrome.com/blog/private-network-access-update-2024-03
         ALLOW_PRIVATE_NETWORK="true",
     )

langgraph_api/models/run.py

@@ -1,9 +1,11 @@
 import asyncio
+import urllib.parse
 import uuid
 from collections.abc import Mapping, Sequence
 from typing import Any, NamedTuple, TypedDict
 from uuid import UUID
 
+import orjson
 from langgraph.checkpoint.base.id import uuid6
 from starlette.authentication import BaseUser
 from starlette.exceptions import HTTPException
@@ -152,6 +154,11 @@ def get_user_id(user: BaseUser | None) -> str | None:
             pass
 
 
+LANGSMITH_METADATA = "langsmith-metadata"
+LANGSMITH_TAGS = "langsmith-tags"
+LANGSMITH_PROJECT = "langsmith-project"
+
+
 async def create_valid_run(
     conn: AsyncConnectionProto,
     thread_id: str | None,
@@ -174,14 +181,13 @@ async def create_valid_run(
     stream_mode, multitask_strategy, prevent_insert_if_inflight = assign_defaults(
         payload
     )
-
     # assign custom headers and checkpoint to config
     config = payload.get("config") or {}
-    config.setdefault("configurable", {})
+    configurable = config.setdefault("configurable", {})
     if checkpoint_id:
-        config["configurable"]["checkpoint_id"] = str(checkpoint_id)
+        configurable["checkpoint_id"] = str(checkpoint_id)
     if checkpoint := payload.get("checkpoint"):
-        config["configurable"].update(checkpoint)
+        configurable.update(checkpoint)
     for key, value in headers.items():
         if key.startswith("x-"):
             if key in (
@@ -190,14 +196,27 @@ async def create_valid_run(
                 "x-service-key",
             ):
                 continue
-            config["configurable"][key] = value
+            configurable[key] = value
+        elif key == "langsmith-trace":
+            configurable[key] = value
+            if baggage := headers.get("baggage"):
+                for item in baggage.split(","):
+                    key, value = item.split("=")
+                    if key == LANGSMITH_METADATA and key not in configurable:
+                        configurable[key] = orjson.loads(urllib.parse.unquote(value))
+                    elif key == LANGSMITH_TAGS:
+                        configurable[key] = urllib.parse.unquote(value).split(",")
+                    elif key == LANGSMITH_PROJECT:
+                        configurable[key] = urllib.parse.unquote(value)
+        elif key == "user-agent":
+            configurable[key] = value
     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
-        config["configurable"]["langgraph_auth_permissions"] = ctx.permissions
+        configurable["langgraph_auth_user"] = user
+        configurable["langgraph_auth_user_id"] = user_id
+        configurable["langgraph_auth_permissions"] = ctx.permissions
     else:
         user_id = None
     run_coro = Runs.put(

langgraph_api/serde.py

@@ -1,5 +1,4 @@
 import asyncio
-import pickle
 import uuid
 from base64 import b64encode
 from collections import deque
@@ -18,9 +17,13 @@ from re import Pattern
 from typing import Any, NamedTuple
 from zoneinfo import ZoneInfo
 
+import cloudpickle
 import orjson
+import structlog
 from langgraph.checkpoint.serde.jsonplus import JsonPlusSerializer
 
+logger = structlog.stdlib.get_logger(__name__)
+
 
 class Fragment(NamedTuple):
     buf: bytes
@@ -122,9 +125,15 @@ class Serializer(JsonPlusSerializer):
         try:
             return super().dumps_typed(obj)
         except TypeError:
-            return "pickle", pickle.dumps(obj)
+            return "pickle", cloudpickle.dumps(obj)
 
     def loads_typed(self, data: tuple[str, bytes]) -> Any:
         if data[0] == "pickle":
-            return pickle.loads(data[1])
+            try:
+                return cloudpickle.loads(data[1])
+            except Exception as e:
+                logger.warning(
+                    "Failed to unpickle object, replacing w None", exc_info=e
+                )
+                return None
         return super().loads_typed(data)

{langgraph_api-0.0.36.dist-info → langgraph_api-0.0.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.36
+Version: 0.0.38
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -10,13 +10,14 @@ Classifier: License :: Other/Proprietary License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: cloudpickle (>=3.0.0,<4.0.0)
 Requires-Dist: cryptography (>=43.0.3,<44.0.0)
 Requires-Dist: httpx (>=0.25.0)
 Requires-Dist: jsonschema-rs (>=0.20.0,<0.30)
 Requires-Dist: langchain-core (>=0.2.38,<0.4.0)
 Requires-Dist: langgraph (>=0.2.56,<0.4.0)
 Requires-Dist: langgraph-checkpoint (>=2.0.23,<3.0)
-Requires-Dist: langgraph-sdk (>=0.1.58,<0.2.0)
+Requires-Dist: langgraph-sdk (>=0.1.59,<0.2.0)
 Requires-Dist: langsmith (>=0.1.63,<0.4.0)
 Requires-Dist: orjson (>=3.9.7)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)

{langgraph_api-0.0.36.dist-info → langgraph_api-0.0.38.dist-info}/RECORD

@@ -2,7 +2,7 @@ LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
 langgraph_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/api/__init__.py,sha256=iDxgg0arqzZ0-roCSuWo3f8zXmKcl4fuU3Jw0-lqtZs,5447
 langgraph_api/api/assistants.py,sha256=nU6tnbgdr_6Utlq0A9nw2a6xxpUM_DNuCFI42_Kcs_o,14233
-langgraph_api/api/mcp.py,sha256=dpKT9DgIoLERTmYZ4sSOPyHbfGbm7hCyb2MrMS_ol18,13593
+langgraph_api/api/mcp.py,sha256=KbR19dtFCpJEiKYj3IfepAuJij8YZVELuVp7JY_yu_o,13754
 langgraph_api/api/meta.py,sha256=ifJ_Ki0Qf2DYbmY6OKlqKhLGxbt55gm0lEqH1A0cJbw,2790
 langgraph_api/api/openapi.py,sha256=f9gfmWN2AMKNUpLCpSgZuw_aeOF9jCXPdOtFT5PaTWM,10960
 langgraph_api/api/runs.py,sha256=uijgtrw_FSf1lZHsx8pM-CIj_ur2O88Y7ys-CJZ4SNQ,17988
@@ -11,14 +11,14 @@ langgraph_api/api/threads.py,sha256=kvv8pmRoUIvPFEuAhJpMC6qMv7Xo5itrzb5EzJFyWMg,
 langgraph_api/api/ui.py,sha256=LiOZVewKOPbKEykCm30hCEaOA7vuS_Ti5hB32EEy4vw,2082
 langgraph_api/asyncio.py,sha256=hVuAxWTHoUyNqTzcIEKTkAvh7HFrdGK6WIDowDxORAE,8397
 langgraph_api/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/auth/custom.py,sha256=Gays0bYwZmRQA8-LSmQSNp1zS2_NG_xybG2cW79YweU,20890
+langgraph_api/auth/custom.py,sha256=sPYkF-xTdGIn_WMSE2tfR_azg_azzXPRWwUfCye6GQ8,21401
 langgraph_api/auth/langsmith/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langgraph_api/auth/langsmith/backend.py,sha256=InScaL-HYCnxYEauhxU198gRZV9pJn9SzzBoR9Edn7g,2654
 langgraph_api/auth/langsmith/client.py,sha256=eKchvAom7hdkUXauD8vHNceBDDUijrFgdTV8bKd7x4Q,3998
 langgraph_api/auth/middleware.py,sha256=jU8aDSIZHdzCGdifejRF7ndHkSjBtqIHcBwFIuUdHEA,1875
 langgraph_api/auth/noop.py,sha256=Bk6Nf3p8D_iMVy_OyfPlyiJp_aEwzL-sHrbxoXpCbac,586
 langgraph_api/auth/studio_user.py,sha256=FzFQRROKDlA9JjtBuwyZvk6Mbwno5M9RVYjDO6FU3F8,186
-langgraph_api/cli.py,sha256=cSRKZzaxSbfDQO3-2YWAo2NgeE5IK0mabYWt4m5q-dU,12047
+langgraph_api/cli.py,sha256=76wJLx1ekaRmXR5DxGy1_RuA0lQq2jBUzHAAVUKV2Ho,12163
 langgraph_api/command.py,sha256=3O9v3i0OPa96ARyJ_oJbLXkfO8rPgDhLCswgO9koTFA,768
 langgraph_api/config.py,sha256=MwTyJUr2wwZdUbL0gIvol_cCGi24Py9DvBnSozE1ai8,9342
 langgraph_api/cron_scheduler.py,sha256=9yzbbGxzNgJdIg4ZT7yu2oTwT_wRuPxD1c2sbbd52xs,2630
@@ -67,12 +67,12 @@ langgraph_api/middleware/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZ
 langgraph_api/middleware/http_logger.py,sha256=yuFPNFIWwn-4AE1CogBfWlo8KytzywLi_Bd4ccsyVQE,3150
 langgraph_api/middleware/private_network.py,sha256=eYgdyU8AzU2XJu362i1L8aSFoQRiV7_aLBPw7_EgeqI,2111
 langgraph_api/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langgraph_api/models/run.py,sha256=J1DpfMwv7W3zX9WtCANyObaCuZWBVfz2JWgdIBa9VzU,10180
+langgraph_api/models/run.py,sha256=85-pyyvosFAot8WrWl2QjCje-2c4lhattYvoAEMqztA,11000
 langgraph_api/patch.py,sha256=82xjuFqY7tgrUm-k1XWHI6k8S6QovSD0zhe--8_xW4o,1296
 langgraph_api/queue_entrypoint.py,sha256=4xICUxXarNV8DhnaqAMhVi3xCmyVKCL3J5NzHxPA9Xc,1835
 langgraph_api/route.py,sha256=fM4qYCGbmH0a3_cV8uKocb1sLklehxO6HhdRXqLK6OM,4421
 langgraph_api/schema.py,sha256=hNbg6ep2wiGBBtBJVNBgMYA8uC33AfaqhRXXVUY_At8,5361
-langgraph_api/serde.py,sha256=HfaIBNfdKg2W_mIpw_er5irG3GlIiiVqkqfbDgnYHpg,3671
+langgraph_api/serde.py,sha256=D5t_HeABMYKRAsoXWaWG0IsdaYM8yOXIez2wJUTIgT8,3963
 langgraph_api/server.py,sha256=bnXOOYztQmqR-QVpEFoRWB5Fzd33PuEIlwBK2R7W8NE,4849
 langgraph_api/sse.py,sha256=2wNodCOP2eg7a9mpSu0S3FQ0CHk2BBV_vv0UtIgJIcc,4034
 langgraph_api/state.py,sha256=8jx4IoTCOjTJuwzuXJKKFwo1VseHjNnw_CCq4x1SW14,2284
@@ -86,18 +86,18 @@ langgraph_license/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 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=X6YHrCwIPVE0iEQy16GhleyyHUdXdCkC-c-Q1SW-rAM,3884
+langgraph_storage/checkpoint.py,sha256=Qq0y6vdh27qdF3h5nOLT5CcX9Rj2bcFqkVOMeCaGoK4,4036
 langgraph_storage/database.py,sha256=I0AgFeJ-NSTT34vxKxQBUf1z2syFP0S8QpKCqTixrzY,5652
 langgraph_storage/inmem_stream.py,sha256=8bxkILIuFpr7P7RQ37SQAxrpRKvmbHdRB_nbfFiomlk,3263
 langgraph_storage/ops.py,sha256=S0QHbnquEXvAM63NQMgqtyTjm-8nlq0pOvgJgFD38Z8,75312
 langgraph_storage/queue.py,sha256=UDgsUTtUMfBSRDrQ8Onis-FJO4n7KTsX6sdpbY8Hs0A,5055
 langgraph_storage/retry.py,sha256=XmldOP4e_H5s264CagJRVnQMDFcEJR_dldVR1Hm5XvM,763
-langgraph_storage/store.py,sha256=8QSM0gm6ZV40antWw0YHfnU71swZRsozjd85R-MSrjg,3100
+langgraph_storage/store.py,sha256=JB9jZ87GE19MVN9wgl3-esgR2eIkeipws9q6qsPWkgc,3399
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=-25y3NRQ88e_944UXo76Goa34HJhC7pj6I9tjYUwvuE,131492
-langgraph_api-0.0.36.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.36.dist-info/METADATA,sha256=T5nIH34PPCdvY8DcFJWKDgrrNutgpgwXGmv52fX_2AI,4027
-langgraph_api-0.0.36.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.36.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.36.dist-info/RECORD,,
+langgraph_api-0.0.38.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.38.dist-info/METADATA,sha256=kv_c0bBL9Vp6pM0NQOUv1uyruDkf5K67X74nSICrekw,4071
+langgraph_api-0.0.38.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.38.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.38.dist-info/RECORD,,

langgraph_storage/checkpoint.py

@@ -16,6 +16,9 @@ from langgraph_api.serde import Serializer
 logger = logging.getLogger(__name__)
 
 _EXCLUDED_KEYS = {"checkpoint_ns", "checkpoint_id", "run_id", "thread_id"}
+DISABLE_FILE_PERSISTENCE = (
+    os.getenv("LANGGRAPH_DISABLE_FILE_PERSISTENCE", "false").lower() == "true"
+)
 
 
 class InMemorySaver(MemorySaver):
@@ -56,7 +59,7 @@ class InMemorySaver(MemorySaver):
 
         super().__init__(
             serde=serde if serde is not None else Serializer(),
-            factory=factory,
+            factory=factory if not DISABLE_FILE_PERSISTENCE else None,
         )
 
     def put(

langgraph_storage/store.py

@@ -13,13 +13,22 @@ from langgraph.store.memory import InMemoryStore
 from langgraph_api.graph import resolve_embeddings
 
 _STORE_CONFIG = None
+DISABLE_FILE_PERSISTENCE = (
+    os.getenv("LANGGRAPH_DISABLE_FILE_PERSISTENCE", "false").lower() == "true"
+)
 
 
 class DiskBackedInMemStore(InMemoryStore):
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
-        self._data = PersistentDict(dict, filename=_STORE_FILE)
-        self._vectors = PersistentDict(lambda: defaultdict(dict), filename=_VECTOR_FILE)
+        if not DISABLE_FILE_PERSISTENCE:
+            self._data = PersistentDict(dict, filename=_STORE_FILE)
+            self._vectors = PersistentDict(
+                lambda: defaultdict(dict), filename=_VECTOR_FILE
+            )
+        else:
+            self._data = InMemoryStore._data
+            self._vectors = InMemoryStore._vectors
         self._load_data(self._data, which="data")
         self._load_data(self._vectors, which="vectors")