langgraph-api 0.4.28__tar.gz → 0.4.29__tar.gz

{langgraph_api-0.4.28 → langgraph_api-0.4.29}/Makefile

@@ -1,22 +1,39 @@
-.PHONY: build release lint format test test_watch start start-inmem start-inmem-license-oss start check-version check-base-imports
+.PHONY: build release lint format test test_watch start start-inmem start-inmem-license-oss start start-js-server build-go-server start-go-server stop-go-server check-version check-base-imports
+
+# Environment variables
+FF_USE_CORE_API ?= false
+CORE_API_LANGSERVE_GRAPHS = '{"agent": "./tests/graphs/agent.py:graph", "custom_lifespan": "./tests/graphs/my_router.py:graph", "single_node": "./tests/graphs/single_node.py:graph", "benchmark": "./tests/graphs/benchmark.py:graph", "config_graph": "./tests/graphs/config_graph.py:graph", "other": "./tests/graphs/other.py:make_graph", "weather": "./tests/graphs/weather.py:mk_weather_graph", "searchy": "./tests/graphs/searchy.py:graph", "agent_simple": "./tests/graphs/agent_simple.py:graph", "simple_runtime": "./tests/graphs/simple_runtime.py:graph", "agent_interrupt": "./tests/graphs/agent_interrupt.py:graph", "message_type_test": "./tests/graphs/message_type_test.py:graph", "remote_subgraph_parent": "./tests/graphs/remote_subgraph_parent.py:graph", "simple_remote": "./tests/graphs/simple_remote.py:graph"}'
+# Go server management
+build-go-server:
+	@echo "Building Go gRPC server..."
+	cd ../core/core-api/cmd/sqlite-test-server && go build -o cmd main.go
+
+start-go-server:
+	@echo "Starting Go gRPC server on port 50051..."
+	cd ../core/core-api/cmd/sqlite-test-server && LANGSERVE_GRAPHS=$(CORE_API_LANGSERVE_GRAPHS) ./cmd &
+	@sleep 2
+
+stop-go-server:
+	@echo "Stopping Go gRPC server on port 50051..."
+	@lsof -ti:50051 | xargs -r kill -9 || echo "No process found on port 50051"
 
 # lint commands
 
 lint:
-	uv run ruff check .
-	uv run ruff format . --diff
-	uvx ty check --exclude "**/pb/**" --exclude "**/*_test.py" --exclude "**/test_*.py" --exclude "**/tests/**" --exclude "venv/**" --exclude ".venv/**" --exclude "build/**" --exclude "dist/**" .
+	uv run ruff check . --exclude "**/generated/**"
+	uv run ruff format . --diff --exclude "**/generated/**"
+	uvx ty check --exclude "**/pb/**" --exclude "**/*_test.py" --exclude "**/test_*.py" --exclude "**/tests/**" --exclude "venv/**" --exclude ".venv/**" --exclude "build/**" --exclude "dist/**" --exclude "**/generated/**" .
 
 format:
-	uv run ruff check --fix .
-	uv run ruff format .
+	uv run ruff check --fix . --exclude "**/generated/**"
+	uv run ruff format . --exclude "**/generated/**"
 
 check-base-imports:
 	LANGGRAPH_RUNTIME_EDITION=inmem DATABASE_URI=:memory: REDIS_URI=_FAKE uv run python -c "from langgraph_api.config import *; from langgraph_runtime import *"
 
 # test commands
 
-TEST ?= tests/
+TEST ?= tests/ ../runtime_inmem/tests/
 AUTH_TEST ?= "tests/integration_tests/test_custom_auth.py"
 LANGGRAPH_HTTP ?= {"disable_mcp": false, "disable_a2a": false}
 LANGGRAPH_AES_KEY ?= '1234567890123456'
@@ -37,7 +54,11 @@ endif
 REVISION ?= $(shell git rev-parse --short HEAD)
 
 test-license-oss:
-	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' LANGGRAPH_STORE='$(STORE_CONFIG)' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ uv run pytest -v $(TEST)
+	@if [ "$(FF_USE_CORE_API)" = "true" ]; then \
+		LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' LANGGRAPH_STORE='$(STORE_CONFIG)' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ FF_USE_CORE_API=true uv run pytest -m grpc -v $(TEST); \
+	else \
+		LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' LANGGRAPH_STORE='$(STORE_CONFIG)' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ uv run pytest -v $(TEST); \
+	fi
 
 test-watch-oss:
 	LANGGRAPH_RUNTIME_EDITION=inmem LANGGRAPH_HTTP='$(HTTP_CONFIG)' REDIS_URI=_FAKE DATABASE_URI=:memory: MIGRATIONS_PATH=__inmem__ uv run --no-sync ptw . -- -x -vv --ff --capture=no $(TEST)
@@ -56,6 +77,10 @@ test-auth-watch:
 # dev commands
 
 start:
+	@if [ "$(FF_USE_CORE_API)" = "true" ]; then \
+		$(MAKE) build-go-server && $(MAKE) start-go-server; \
+		trap '$(MAKE) stop-go-server' INT TERM EXIT; \
+	fi; \
 	sleep 3 && \
 	LANGGRAPH_HTTP='$(HTTP_CONFIG)' \
 	LANGGRAPH_RUNTIME_EDITION=inmem \
@@ -65,6 +90,7 @@ start:
 	LANGGRAPH_STORE='$(STORE_CONFIG)' \
 	LANGGRAPH_CONFIG='{"agent": {"configurable": {"model_name": "openai"}}}' \
 	LANGSMITH_LANGGRAPH_API_VARIANT=test \
+	FF_USE_CORE_API=$(FF_USE_CORE_API) \
 	REDIS_URI=fake \
 	DATABASE_URI=:memory: \
 	MIGRATIONS_PATH=__inmem \
@@ -117,6 +143,20 @@ start-auth-fastapi-jwt:
 		--reload-dir ../runtime_inmem \
 		--no-access-log
 
+start-js-server:
+	@echo "Building and starting Go gRPC server..."
+	$(MAKE) build-go-server
+	$(MAKE) start-go-server
+	@trap '$(MAKE) -C $(CURDIR) stop-go-server' INT TERM EXIT; \
+	echo "Installing JS server dependencies..."; \
+	cd ../public-api-server-js && npm install; \
+	echo "Building JS server..."; \
+	cd ../public-api-server-js && npm run build; \
+	echo "Starting JS server on port 9123..."; \
+	cd ../public-api-server-js && FF_USE_CORE_API=true \
+	LANGSERVE_GRAPHS=$(CORE_API_LANGSERVE_GRAPHS) \
+	PORT=9123 npm start
+
 VERSION_KIND ?= patch
 
 bump-version:

{langgraph_api-0.4.28 → langgraph_api-0.4.29}/PKG-INFO

@@ -1,12 +1,14 @@
 Metadata-Version: 2.4
 Name: langgraph-api
-Version: 0.4.28
+Version: 0.4.29
 Author-email: Nuno Campos <nuno@langchain.dev>, Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0
 License-File: LICENSE
 Requires-Python: >=3.11
 Requires-Dist: cloudpickle>=3.0.0
 Requires-Dist: cryptography<45.0,>=42.0.0
+Requires-Dist: grpcio-tools<2.0.0,>=1.75.0
+Requires-Dist: grpcio<2.0.0,>=1.75.0
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: jsonschema-rs<0.30,>=0.20.0
 Requires-Dist: langchain-core>=0.3.64
@@ -16,6 +18,7 @@ Requires-Dist: langgraph-sdk>=0.2.0
 Requires-Dist: langgraph>=0.4.0
 Requires-Dist: langsmith>=0.3.45
 Requires-Dist: orjson>=3.9.7
+Requires-Dist: protobuf<7.0.0,>=6.32.1
 Requires-Dist: pyjwt>=2.9.0
 Requires-Dist: sse-starlette<2.2.0,>=2.1.0
 Requires-Dist: starlette>=0.38.6

{langgraph_api-0.4.28 → langgraph_api-0.4.29}/constraints.txt

@@ -17,3 +17,6 @@ jsonschema-rs>=0.20.0
 structlog>=24.1.0
 cloudpickle>=3.0.0
 truststore>=0.1
+protobuf>=6.32.1,<7.0.0
+grpcio>=1.75.0,<2.0.0
+grpcio-tools>=1.75.0,<2.0.0

langgraph_api-0.4.29/langgraph_api/__init__.py

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

{langgraph_api-0.4.28 → langgraph_api-0.4.29}/langgraph_api/api/assistants.py

@@ -12,8 +12,9 @@ from starlette.responses import Response
 from starlette.routing import BaseRoute
 
 from langgraph_api import store as api_store
-from langgraph_api.feature_flags import USE_RUNTIME_CONTEXT_API
+from langgraph_api.feature_flags import FF_USE_CORE_API, USE_RUNTIME_CONTEXT_API
 from langgraph_api.graph import get_assistant_id, get_graph
+from langgraph_api.grpc_ops.ops import Assistants as GrpcAssistants
 from langgraph_api.js.base import BaseRemotePregel
 from langgraph_api.route import ApiRequest, ApiResponse, ApiRoute
 from langgraph_api.schema import ASSISTANT_FIELDS
@@ -40,6 +41,7 @@ from langgraph_runtime.retry import retry_db
 
 logger = structlog.stdlib.get_logger(__name__)
 
+CrudAssistants = GrpcAssistants if FF_USE_CORE_API else Assistants
 
 EXCLUDED_CONFIG_SCHEMA = (
     "__pregel_checkpointer",
@@ -164,7 +166,7 @@ async def create_assistant(request: ApiRequest) -> ApiResponse:
     if assistant_id := payload.get("assistant_id"):
         validate_uuid(assistant_id, "Invalid assistant ID: must be a UUID")
     async with connect() as conn:
-        assistant = await Assistants.put(
+        assistant = await CrudAssistants.put(
             conn,
             assistant_id or str(uuid4()),
             config=payload.get("config") or {},
@@ -175,6 +177,7 @@ async def create_assistant(request: ApiRequest) -> ApiResponse:
             name=payload.get("name") or "Untitled",
             description=payload.get("description"),
         )
+
     return ApiResponse(await fetchone(assistant, not_found_code=409))
 
 
@@ -187,7 +190,7 @@ async def search_assistants(
     select = validate_select_columns(payload.get("select") or None, ASSISTANT_FIELDS)
     offset = int(payload.get("offset") or 0)
     async with connect() as conn:
-        assistants_iter, next_offset = await Assistants.search(
+        assistants_iter, next_offset = await CrudAssistants.search(
             conn,
             graph_id=payload.get("graph_id"),
             metadata=payload.get("metadata"),
@@ -210,7 +213,7 @@ async def count_assistants(
     """Count assistants."""
     payload = await request.json(AssistantCountRequest)
     async with connect() as conn:
-        count = await Assistants.count(
+        count = await CrudAssistants.count(
             conn,
             graph_id=payload.get("graph_id"),
             metadata=payload.get("metadata"),
@@ -226,7 +229,7 @@ async def get_assistant(
     """Get an assistant by ID."""
     validate_uuid(assistant_id, "Invalid assistant ID: must be a UUID")
     async with connect() as conn:
-        assistant = await Assistants.get(conn, assistant_id)
+        assistant = await CrudAssistants.get(conn, assistant_id)
     return ApiResponse(await fetchone(assistant))
 
 
@@ -386,7 +389,7 @@ async def patch_assistant(
     validate_uuid(assistant_id, "Invalid assistant ID: must be a UUID")
     payload = await request.json(AssistantPatch)
     async with connect() as conn:
-        assistant = await Assistants.patch(
+        assistant = await CrudAssistants.patch(
             conn,
             assistant_id,
             config=payload.get("config"),
@@ -405,7 +408,7 @@ async def delete_assistant(request: ApiRequest) -> Response:
     assistant_id = request.path_params["assistant_id"]
     validate_uuid(assistant_id, "Invalid assistant ID: must be a UUID")
     async with connect() as conn:
-        aid = await Assistants.delete(conn, assistant_id)
+        aid = await CrudAssistants.delete(conn, assistant_id)
     await fetchone(aid)
     return Response(status_code=204)
 
@@ -417,7 +420,7 @@ async def get_assistant_versions(request: ApiRequest) -> ApiResponse:
     validate_uuid(assistant_id, "Invalid assistant ID: must be a UUID")
     payload = await request.json(AssistantVersionsSearchRequest)
     async with connect() as conn:
-        assistants_iter = await Assistants.get_versions(
+        assistants_iter = await CrudAssistants.get_versions(
             conn,
             assistant_id,
             metadata=payload.get("metadata") or {},
@@ -439,7 +442,7 @@ async def set_latest_assistant_version(request: ApiRequest) -> ApiResponse:
     payload = await request.json(AssistantVersionChange)
     validate_uuid(assistant_id, "Invalid assistant ID: must be a UUID")
     async with connect() as conn:
-        assistant = await Assistants.set_latest(
+        assistant = await CrudAssistants.set_latest(
             conn, assistant_id, payload.get("version")
         )
     return ApiResponse(await fetchone(assistant, not_found_code=404))

{langgraph_api-0.4.28 → langgraph_api-0.4.29}/langgraph_api/feature_flags.py

@@ -1,3 +1,5 @@
+import os
+
 from langgraph.version import __version__
 
 # Only gate features on the major.minor version; Lets you ignore the rc/alpha/etc. releases anyway
@@ -7,3 +9,10 @@ OMIT_PENDING_SENDS = LANGGRAPH_PY_MINOR >= (0, 5)
 USE_RUNTIME_CONTEXT_API = LANGGRAPH_PY_MINOR >= (0, 6)
 USE_NEW_INTERRUPTS = LANGGRAPH_PY_MINOR >= (0, 6)
 USE_DURABILITY = LANGGRAPH_PY_MINOR >= (0, 6)
+
+# Feature flag for new gRPC-based persistence layer
+FF_USE_CORE_API = os.getenv("FF_USE_CORE_API", "false").lower() in (
+    "true",
+    "1",
+    "yes",
+)

{langgraph_api-0.4.28 → langgraph_api-0.4.29}/langgraph_api/graph.py

@@ -21,9 +21,8 @@ from langgraph.pregel import Pregel
 from langgraph.store.base import BaseStore
 from starlette.exceptions import HTTPException
 
-from langgraph_api import asyncio as lg_asyncio
 from langgraph_api import config
-from langgraph_api.feature_flags import USE_RUNTIME_CONTEXT_API
+from langgraph_api.feature_flags import FF_USE_CORE_API, USE_RUNTIME_CONTEXT_API
 from langgraph_api.js.base import BaseRemotePregel, is_js_path
 from langgraph_api.schema import Config
 from langgraph_api.utils.config import run_in_executor, var_child_runnable_config
@@ -51,13 +50,19 @@ async def register_graph(
     description: str | None = None,
 ) -> None:
     """Register a graph."""
+    from langgraph_api.grpc_ops.ops import Assistants as AssistantsGrpc
     from langgraph_runtime.database import connect
-    from langgraph_runtime.ops import Assistants
+    from langgraph_runtime.ops import Assistants as AssistantsRuntime
+
+    Assistants = AssistantsGrpc if FF_USE_CORE_API else AssistantsRuntime
 
     await logger.ainfo(f"Registering graph with id '{graph_id}'", graph_id=graph_id)
-    GRAPHS[graph_id] = graph
-    if callable(graph):
-        FACTORY_ACCEPTS_CONFIG[graph_id] = len(inspect.signature(graph).parameters) > 0
+    if not FF_USE_CORE_API:
+        GRAPHS[graph_id] = graph
+        if callable(graph):
+            FACTORY_ACCEPTS_CONFIG[graph_id] = (
+                len(inspect.signature(graph).parameters) > 0
+            )
 
     from langgraph_runtime.retry import retry_db
 
@@ -87,12 +92,6 @@ async def register_graph(
     await register_graph_db()
 
 
-def register_graph_sync(
-    graph_id: str, graph: GraphValue, config: dict | None = None
-) -> None:
-    lg_asyncio.run_coroutine_threadsafe(register_graph(graph_id, graph, config))
-
-
 @asynccontextmanager
 async def _generate_graph(value: Any) -> AsyncIterator[Any]:
     """Yield a graph object regardless of its type."""

langgraph_api-0.4.29/langgraph_api/grpc_ops/client.py

@@ -0,0 +1,80 @@
+"""gRPC client wrapper for LangGraph persistence services."""
+
+import os
+
+import structlog
+from grpc import aio  # type: ignore[import]
+
+from .generated.core_api_pb2_grpc import AdminStub, AssistantsStub
+
+logger = structlog.stdlib.get_logger(__name__)
+
+
+class GrpcClient:
+    """gRPC client for LangGraph persistence services."""
+
+    def __init__(
+        self,
+        server_address: str | None = None,
+    ):
+        """Initialize the gRPC client.
+
+        Args:
+            server_address: The gRPC server address (default: localhost:50051)
+        """
+        self.server_address = server_address or os.getenv(
+            "GRPC_SERVER_ADDRESS", "localhost:50051"
+        )
+        self._channel: aio.Channel | None = None
+        self._assistants_stub: AssistantsStub | None = None
+        self._admin_stub: AdminStub | None = None
+
+    async def __aenter__(self):
+        """Async context manager entry."""
+        await self.connect()
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        """Async context manager exit."""
+        await self.close()
+
+    async def connect(self):
+        """Connect to the gRPC server."""
+        if self._channel is not None:
+            return
+
+        self._channel = aio.insecure_channel(self.server_address)
+
+        self._assistants_stub = AssistantsStub(self._channel)
+        self._admin_stub = AdminStub(self._channel)
+
+        await logger.adebug(
+            "Connected to gRPC server", server_address=self.server_address
+        )
+
+    async def close(self):
+        """Close the gRPC connection."""
+        if self._channel is not None:
+            await self._channel.close()
+            self._channel = None
+            self._assistants_stub = None
+            self._admin_stub = None
+            await logger.adebug("Closed gRPC connection")
+
+    @property
+    def assistants(self) -> AssistantsStub:
+        """Get the assistants service stub."""
+        if self._assistants_stub is None:
+            raise RuntimeError(
+                "Client not connected. Use async context manager or call connect() first."
+            )
+        return self._assistants_stub
+
+    @property
+    def admin(self) -> AdminStub:
+        """Get the admin service stub."""
+        if self._admin_stub is None:
+            raise RuntimeError(
+                "Client not connected. Use async context manager or call connect() first."
+            )
+        return self._admin_stub

langgraph_api-0.4.29/langgraph_api/grpc_ops/generated/__init__.py

@@ -0,0 +1,5 @@
+# Generated protobuf files
+from . import core_api_pb2
+from . import core_api_pb2_grpc
+
+__all__ = ["core_api_pb2", "core_api_pb2_grpc"]