langgraph-api 0.2.16__tar.gz → 0.2.18__tar.gz

{langgraph_api-0.2.16 → langgraph_api-0.2.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langgraph-api
-Version: 0.2.16
+Version: 0.2.18
 Summary: 
 License: Elastic-2.0
 Author: Nuno Campos
@@ -27,6 +27,7 @@ Requires-Dist: sse-starlette (>=2.1.0,<2.2.0)
 Requires-Dist: starlette (>=0.38.6)
 Requires-Dist: structlog (>=24.1.0,<26)
 Requires-Dist: tenacity (>=8.0.0)
+Requires-Dist: truststore (>=0.1)
 Requires-Dist: uvicorn (>=0.26.0)
 Requires-Dist: watchfiles (>=0.13)
 Description-Content-Type: text/markdown

langgraph_api-0.2.18/langgraph_api/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.18"

{langgraph_api-0.2.16 → langgraph_api-0.2.18}/langgraph_api/api/openapi.py

@@ -24,119 +24,6 @@ def set_custom_spec(spec: dict):
     CUSTOM_OPENAPI_SPEC = spec
 
 
-def get_mcp_openapi_paths() -> dict:
-    """Returns the OpenAPI path definitions for the /mcp/ endpoint."""
-    return {
-        "/mcp/": {
-            "get": {
-                "operationId": "get_mcp",
-                "summary": "MCP Get",
-                "description": (
-                    "Implemented according to the Streamable HTTP Transport "
-                    "specification."
-                ),
-                "responses": {
-                    "405": {
-                        "description": "GET method not allowed; streaming not "
-                        "supported.",
-                    }
-                },
-                "tags": ["MCP"],
-            },
-            "post": {
-                "operationId": "post_mcp",
-                "summary": "MCP Post",
-                "description": (
-                    "Implemented according to the Streamable HTTP Transport "
-                    "specification.\n"
-                    "Sends a JSON-RPC 2.0 message to the server.\n\n"
-                    "- **Request**: Provide an object with `jsonrpc`, `id`, `method`, "
-                    "and optional `params`.\n"
-                    "- **Response**: Returns a JSON-RPC response or acknowledgment.\n\n"
-                    "**Notes:**\n"
-                    "- Stateless: Sessions are not persisted across requests.\n"
-                ),
-                "parameters": [
-                    {
-                        "name": "Accept",
-                        "in": "header",
-                        "required": True,
-                        "schema": {
-                            "type": "string",
-                            "enum": [
-                                "application/json, text/event-stream",
-                            ],
-                        },
-                        "description": (
-                            "Accept header must include both "
-                            "'application/json' and 'text/event-stream' "
-                            "media types."
-                        ),
-                    }
-                ],
-                "requestBody": {
-                    "required": True,
-                    "content": {
-                        "application/json": {
-                            "schema": {"type": "object"},
-                            "description": "A JSON-RPC 2.0 request, notification, "
-                            "or response object.",
-                            "example": {
-                                "jsonrpc": "2.0",
-                                "id": "1",
-                                "method": "initialize",
-                                "params": {
-                                    "clientInfo": {
-                                        "name": "test_client",
-                                        "version": "1.0.0",
-                                    },
-                                    "protocolVersion": "2024-11-05",
-                                    "capabilities": {},
-                                },
-                            },
-                        }
-                    },
-                },
-                "responses": {
-                    "200": {
-                        "description": "Successful JSON-RPC response.",
-                        "content": {"application/json": {"schema": {"type": "object"}}},
-                    },
-                    "202": {
-                        "description": "Notification or response accepted; no content "
-                        "body.",
-                    },
-                    "400": {
-                        "description": "Bad request: invalid JSON or message format, "
-                        "or unacceptable Accept header.",
-                    },
-                    "405": {
-                        "description": "HTTP method not allowed.",
-                    },
-                    "500": {
-                        "description": "Internal server error or unexpected failure.",
-                    },
-                },
-                "tags": ["MCP"],
-            },
-            "delete": {
-                "operationId": "delete_mcp",
-                "summary": "Terminate Session",
-                "description": (
-                    "Implemented according to the Streamable HTTP Transport "
-                    "specification.\n"
-                    "Terminate an MCP session. The server implementation is stateless, "
-                    "so this is a no-op.\n\n"
-                ),
-                "responses": {
-                    "404": {},
-                },
-                "tags": ["MCP"],
-            },
-        }
-    }
-
-
 @lru_cache(maxsize=1)
 def get_openapi_spec() -> str:
     # patch the graph_id enums
@@ -201,10 +88,9 @@ def get_openapi_spec() -> str:
 
     MCP_ENABLED = HTTP_CONFIG is None or not HTTP_CONFIG.get("disable_mcp")
 
-    if MCP_ENABLED:
-        # Add the MCP paths to the OpenAPI spec
-        mcp_path = get_mcp_openapi_paths()
-        final["paths"]["/mcp/"] = mcp_path["/mcp/"]
+    if not MCP_ENABLED:
+        # Remove the MCP paths from the OpenAPI spec
+        final["paths"].pop("/mcp/", None)
 
     return orjson.dumps(final)
 

{langgraph_api-0.2.16 → langgraph_api-0.2.18}/langgraph_api/queue_entrypoint.py

@@ -1,3 +1,8 @@
+# ruff: noqa: E402
+import truststore  # noqa: F401
+
+truststore.inject_into_ssl()  # noqa: F401
+
 import asyncio
 import http.server
 import json

{langgraph_api-0.2.16 → langgraph_api-0.2.18}/langgraph_api/server.py

@@ -1,4 +1,5 @@
 # MONKEY PATCH: Patch Starlette to fix an error in the library
+# ruff: noqa: E402
 import langgraph_api.patch  # noqa: F401,I001
 import sys
 
@@ -6,34 +7,37 @@ import sys
 # patches an error in the Starlette library.
 import logging
 import typing
+import truststore  # noqa: F401
+
+truststore.inject_into_ssl()  # noqa: F401
+
+from contextlib import asynccontextmanager
 
 import jsonschema_rs
 import structlog
-from contextlib import asynccontextmanager
 from langgraph.errors import EmptyInputError, InvalidUpdateError
+from langgraph_sdk.client import configure_loopback_transports
 from starlette.applications import Starlette
 from starlette.middleware import Middleware
 from starlette.middleware.cors import CORSMiddleware
-
-from langgraph_api.api.openapi import set_custom_spec
+from starlette.routing import Mount
 from starlette.types import Receive, Scope, Send
 
 import langgraph_api.config as config
-from langgraph_api.api import routes, meta_routes, user_router
-from starlette.routing import Mount
+from langgraph_api.api import meta_routes, routes, user_router
+from langgraph_api.api.openapi import set_custom_spec
 from langgraph_api.errors import (
     overloaded_error_handler,
     validation_error_handler,
     value_error_handler,
 )
-from langgraph_runtime.lifespan import lifespan
+from langgraph_api.js.base import is_js_path
 from langgraph_api.middleware.http_logger import AccessLoggerMiddleware
 from langgraph_api.middleware.private_network import PrivateNetworkMiddleware
 from langgraph_api.middleware.request_id import RequestIdMiddleware
 from langgraph_api.utils import SchemaGenerator
+from langgraph_runtime.lifespan import lifespan
 from langgraph_runtime.retry import OVERLOADED_EXCEPTIONS
-from langgraph_api.js.base import is_js_path
-from langgraph_sdk.client import configure_loopback_transports
 
 logging.captureWarnings(True)
 logger = structlog.stdlib.get_logger(__name__)

{langgraph_api-0.2.16 → langgraph_api-0.2.18}/langgraph_api/stream.py

@@ -163,6 +163,7 @@ async def astream_state(
                         ns, mode, chunk = event["data"]["chunk"]
                     else:
                         mode, chunk = event["data"]["chunk"]
+                        ns = None
                     # --- begin shared logic with astream ---
                     if mode == "debug":
                         if chunk["type"] == "checkpoint":
@@ -172,7 +173,10 @@ async def astream_state(
                             on_task_result(chunk["payload"])
                     if mode == "messages":
                         if "messages-tuple" in stream_mode:
-                            yield "messages", chunk
+                            if subgraphs and ns:
+                                yield f"messages|{'|'.join(ns)}", chunk
+                            else:
+                                yield "messages", chunk
                         else:
                             msg, meta = cast(tuple[BaseMessage, dict[str, Any]], chunk)
                             if msg.id in messages:
@@ -214,6 +218,7 @@ async def astream_state(
                     ns, mode, chunk = event
                 else:
                     mode, chunk = event
+                    ns = None
                 # --- begin shared logic with astream_events ---
                 if mode == "debug":
                     if chunk["type"] == "checkpoint":
@@ -223,7 +228,10 @@ async def astream_state(
                         on_task_result(chunk["payload"])
                 if mode == "messages":
                     if "messages-tuple" in stream_mode:
-                        yield "messages", chunk
+                        if subgraphs and ns:
+                            yield f"messages|{'|'.join(ns)}", chunk
+                        else:
+                            yield "messages", chunk
                     else:
                         msg, meta = cast(tuple[BaseMessage, dict[str, Any]], chunk)
                         if msg.id in messages: