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

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

@@ -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 → langgraph_api-0.0.10}/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-0.0.9 → langgraph_api-0.0.10}/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.10}/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

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-api"
-version = "0.0.9"
+version = "0.0.10"
 description = ""
 authors = [
     "Nuno Campos <nuno@langchain.dev>",
@@ -42,7 +42,7 @@ codespell = "^2.2.0"
 pytest = "^7.4.4"
 anyio = "^4.4.0"
 pytest-watcher = "^0.4.2"
-langgraph-cli = "^0.1.62"
+langgraph-cli = "^0.1.63"
 langgraph-sdk = "^0.1.45"
 pytest-repeat = "^0.9.3"
 pytest-retry = "^1.6.3"