langgraph-api 0.0.9__tar.gz → 0.0.11__tar.gz

{langgraph_api-0.0.9 → langgraph_api-0.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.9
+Version: 0.0.11
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -16,6 +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: langgraph-sdk (>=0.1.47,<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)

{langgraph_api-0.0.9 → langgraph_api-0.0.11}/langgraph_api/auth/custom.py

@@ -3,13 +3,13 @@ import copy
 import functools
 import importlib.util
 import inspect
-import logging
 import os
 import sys
 from collections.abc import Awaitable, Callable, Mapping
 from contextlib import AsyncExitStack
 from typing import Any, get_args
 
+import structlog
 from langgraph_sdk import Auth
 from starlette.authentication import (
     AuthCredentials,
@@ -26,7 +26,7 @@ from starlette.responses import Response
 from langgraph_api.auth.langsmith.backend import LangsmithAuthBackend
 from langgraph_api.config import LANGGRAPH_AUTH
 
-logger = logging.getLogger(__name__)
+logger = structlog.stdlib.get_logger(__name__)
 
 SUPPORTED_PARAMETERS = {
     "request": Request,
@@ -50,11 +50,15 @@ def get_custom_auth_middleware() -> AuthenticationBackend:
             "LANGGRAPH_AUTH must be set to a Python file path or a config dict"
             " to use custom authentication."
         )
+    logger.info("Using custom authentication", langgraph_auth=LANGGRAPH_AUTH)
     return _get_custom_auth_middleware(LANGGRAPH_AUTH)
 
 
 @functools.lru_cache(maxsize=1)
 def get_auth_instance() -> Auth | None:
+    logger.info(
+        f"Getting auth instance: {LANGGRAPH_AUTH}", langgraph_auth=str(LANGGRAPH_AUTH)
+    )
     if not LANGGRAPH_AUTH:
         return None
     path = LANGGRAPH_AUTH.get("path")
@@ -116,7 +120,6 @@ class CustomAuthBackend(AuthenticationBackend):
         ) = None,
         disable_studio_auth: bool = False,
     ):
-        assert fn is not None
         if fn is None:
             self.fn = None
         elif not inspect.iscoroutinefunction(fn):
@@ -133,6 +136,14 @@ class CustomAuthBackend(AuthenticationBackend):
         else:
             self.ls_auth = None
 
+    def __str__(self):
+        return (
+            f"CustomAuthBackend(fn={self.fn}, "
+            f"ls_auth={self.ls_auth}, "
+            f"param_names={self._param_names}"
+            ")"
+        )
+
     async def authenticate(
         self, conn: HTTPConnection
     ) -> tuple[AuthCredentials, BaseUser] | None:
@@ -147,8 +158,8 @@ class CustomAuthBackend(AuthenticationBackend):
             args = _extract_arguments_from_scope(
                 conn.scope, self._param_names, request=Request(conn.scope)
             )
-            scopes, user = await self.fn(**args)
-            return AuthCredentials(scopes), _normalize_user(user)
+            response = await self.fn(**args)
+            return _normalize_auth_response(response)
         except AssertionError as e:
             raise AuthenticationError(str(e)) from None
 
@@ -163,10 +174,12 @@ def _get_custom_auth_middleware(
         path = config.get("path")
         disable_studio_auth = config.get("disable_studio_auth", disable_studio_auth)
     auth_instance = _get_auth_instance(path)
-    return CustomAuthBackend(
+    result = CustomAuthBackend(
         auth_instance._authenticate_handler if auth_instance else None,
         disable_studio_auth,
     )
+    logger.info(f"Loaded custom auth middleware: {str(result)}")
+    return result
 
 
 @functools.lru_cache(maxsize=1)
@@ -182,7 +195,7 @@ def _get_auth_instance(path: str | None = None) -> Auth | None:
         auth_instance._authenticate_handler = _solve_fastapi_dependencies(
             auth_instance._authenticate_handler, deps
         )
-
+    logger.info(f"Loaded auth instance from path {path}: {auth_instance}")
     return auth_instance
 
 
@@ -216,10 +229,11 @@ def _extract_arguments_from_scope(
     if "headers" in param_names:
         args["headers"] = dict(scope.get("headers", {}))
     if "authorization" in param_names:
-        headers = scope.get("headers", {})
-        args["authorization"] = headers.get("authorization") or headers.get(
-            "Authorization"
-        )
+        headers = dict(scope.get("headers", {}))
+        authorization = headers.get(b"authorization") or headers.get(b"Authorization")
+        if isinstance(authorization, bytes):
+            authorization = authorization.decode(encoding="utf-8")
+        args["authorization"] = authorization
     if "method" in param_names:
         args["method"] = scope.get("method")
 
@@ -254,6 +268,7 @@ def _solve_fastapi_dependencies(
     fn: Callable[..., Any], deps: Mapping[str, Any]
 ) -> Callable:
     """Solve FastAPI dependencies for a given function."""
+    logger.info("Solving FastAPI dependencies", fn=str(fn), deps=str(deps))
     from fastapi.dependencies.utils import (
         get_parameterless_sub_dependant,
         solve_dependencies,
@@ -400,6 +415,28 @@ class ProxyUser(BaseUser):
         return getattr(self._user, name)
 
 
+def _normalize_auth_response(
+    response: Any,
+) -> tuple[AuthCredentials, BaseUser]:
+    if isinstance(response, tuple):
+        if len(response) != 2:
+            raise ValueError(
+                f"Expected a tuple with two elements (permissions, user), got {len(response)}"
+            )
+        permissions, user = response
+    elif hasattr(response, "permissions"):
+        permissions = response.permissions
+        user = response
+    elif isinstance(response, dict | Mapping) and "permissions" in response:
+        permissions = response["permissions"]
+        user = response
+    else:
+        user = response
+        permissions = []
+
+    return AuthCredentials(permissions), _normalize_user(user)
+
+
 def _normalize_user(user: Any) -> BaseUser:
     """Normalize user into a BaseUser instance."""
     if isinstance(user, BaseUser):

{langgraph_api-0.0.9 → langgraph_api-0.0.11}/langgraph_api/auth/middleware.py

@@ -1,3 +1,4 @@
+import structlog
 from starlette.middleware import Middleware
 from starlette.middleware.authentication import (
     AuthenticationError,
@@ -9,8 +10,11 @@ from starlette.types import Receive, Scope, Send
 
 from langgraph_api.config import LANGGRAPH_AUTH_TYPE
 
+logger = structlog.stdlib.get_logger(__name__)
+
 
 def get_auth_backend():
+    logger.info(f"Using auth of type={LANGGRAPH_AUTH_TYPE}")
     if LANGGRAPH_AUTH_TYPE == "langsmith":
         from langgraph_api.auth.langsmith.backend import LangsmithAuthBackend
 

{langgraph_api-0.0.9 → langgraph_api-0.0.11}/langgraph_api/utils.py

@@ -6,7 +6,7 @@ from datetime import datetime
 from typing import Any, Protocol, TypeAlias, TypeVar
 
 from langgraph_sdk import Auth
-from starlette.authentication import AuthCredentials, BaseUser
+from starlette.authentication import AuthCredentials, BaseUser, SimpleUser
 from starlette.exceptions import HTTPException
 
 T = TypeVar("T")
@@ -32,8 +32,8 @@ def set_auth_ctx(user: BaseUser | None, auth: AuthCredentials | None) -> None:
     else:
         AuthContext.set(
             Auth.types.BaseAuthContext(
-                scopes=auth.scopes,
-                user=user,
+                permissions=auth.scopes,
+                user=user or SimpleUser(""),
             )
         )
 

{langgraph_api-0.0.9 → langgraph_api-0.0.11}/langgraph_storage/ops.py

@@ -83,7 +83,7 @@ class Authenticated:
             return
         return Auth.types.AuthContext(
             user=ctx.user,
-            scopes=ctx.scopes,
+            permissions=ctx.permissions,
             resource=cls.resource,
             action=action,
         )
@@ -846,7 +846,6 @@ class Threads(Authenticated):
             raise HTTPException(
                 status_code=404, detail=f"Thread with ID {thread_id} not found"
             )
-
         # Delete the thread
         conn.locks.pop(thread_id, None)
         # Cascade delete all runs associated with this thread

{langgraph_api-0.0.9 → langgraph_api-0.0.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.9"
+version = "0.0.11"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -35,6 +35,7 @@ jsonschema-rs = "^0.25.0"
 structlog = "^24.4.0"
 pyjwt = "^2.9.0"
 cryptography = "^43.0.3"
+langgraph-sdk = "^0.1.47"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.6.2"
@@ -42,8 +43,7 @@ codespell = "^2.2.0"
 pytest = "^7.4.4"
 anyio = "^4.4.0"
 pytest-watcher = "^0.4.2"
-langgraph-cli = "^0.1.62"
-langgraph-sdk = "^0.1.45"
+langgraph-cli = "^0.1.63"
 pytest-repeat = "^0.9.3"
 pytest-retry = "^1.6.3"