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

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:
@@ -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,

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.dist-info → langgraph_api-0.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langgraph-api
-Version: 0.0.9
+Version: 0.0.10
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos

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

@@ -9,11 +9,11 @@ langgraph_api/api/store.py,sha256=y7VIejpsE7rpPF-tiMGBqqBwWPZ1wb3o48th6NUvb5I,38
 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/custom.py,sha256=nE8dgtSuX4W8UoByEmr5aH0yK6H_rk0VOoETET9BqJQ,17528
 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=4RhtYv1eBA57irNUT3nXc5vpgBTKysaOKBnjyCMSuCY,1511
+langgraph_api/auth/middleware.py,sha256=eHj4D2p2ASpMKzoTgMkEUXMhuubQdg9Jnq0gQS9qbvE,1637
 langgraph_api/auth/noop.py,sha256=vDJmzG2vArJxVzdHePvrJWahEa0dvGnhc2LEMMeiFz0,391
 langgraph_api/cli.py,sha256=7vQQiD3F50r-8KkbuFjwIz8LLbdKUTd4xZGUJPiO3yQ,11688
 langgraph_api/config.py,sha256=m3cf13AS9HVF84deJ3-9mmNTJ5GSVqcukES-FNbDbvo,2794
@@ -73,15 +73,15 @@ langgraph_license/validation.py,sha256=Uu_G8UGO_WTlLsBEY0gTVWjRR4czYGfw5YAD3HLZo
 langgraph_storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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/ops.py,sha256=V0gnBlpzW3AE9kBNcT_uONYYEHEdgTR6a4mA9Owz6S4,67669
 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.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_api-0.0.10.dist-info/LICENSE,sha256=ZPwVR73Biwm3sK6vR54djCrhaRiM4cAD2zvOQZV8Xis,3859
+langgraph_api-0.0.10.dist-info/METADATA,sha256=VgLW5dnP15w_SuuNkgpgYVjDrBGxxuKGvvEaDOkxMhk,3994
+langgraph_api-0.0.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+langgraph_api-0.0.10.dist-info/entry_points.txt,sha256=3EYLgj89DfzqJHHYGxPH4A_fEtClvlRbWRUHaXO7hj4,77
+langgraph_api-0.0.10.dist-info/RECORD,,

langgraph_storage/ops.py

@@ -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