langgraph-api 0.0.37__tar.gz → 0.0.38__tar.gz

{langgraph_api-0.0.37 → langgraph_api-0.0.38}/PKG-INFO

@@ -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 → langgraph_api-0.0.38}/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-0.0.37 → langgraph_api-0.0.38}/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-0.0.37 → langgraph_api-0.0.38}/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 → langgraph_api-0.0.38}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.37"
+version = "0.0.38"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -35,7 +35,7 @@ jsonschema-rs = ">=0.20.0,<0.30"
 structlog = ">=24.1.0,<26"
 pyjwt = "^2.9.0"
 cryptography = "^43.0.3"
-langgraph-sdk = "^0.1.58"
+langgraph-sdk = "^0.1.59"
 cloudpickle = "^3.0.0"
 
 [tool.poetry.group.dev.dependencies]