langgraph-api 0.0.37__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/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-0.0.37.dist-info → langgraph_api-0.0.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.37
+Version: 0.0.38
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -17,7 +17,7 @@ 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.37.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,7 +11,7 @@ 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
@@ -67,7 +67,7 @@ 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
@@ -96,8 +96,8 @@ langgraph_storage/store.py,sha256=JB9jZ87GE19MVN9wgl3-esgR2eIkeipws9q6qsPWkgc,33
 langgraph_storage/ttl_dict.py,sha256=FlpEY8EANeXWKo_G5nmIotPquABZGyIJyk6HD9u6vqY,1533
 logging.json,sha256=3RNjSADZmDq38eHePMm1CbP6qZ71AmpBtLwCmKU9Zgo,379
 openapi.json,sha256=-25y3NRQ88e_944UXo76Goa34HJhC7pj6I9tjYUwvuE,131492
-langgraph_api-0.0.37.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
-langgraph_api-0.0.37.dist-info/METADATA,sha256=zawmEdZt_TgVBU6iHpMFrfdREjAYy9PqXU6lyy1dBb8,4071
-langgraph_api-0.0.37.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-langgraph_api-0.0.37.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
-langgraph_api-0.0.37.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,,