langgraph-runtime-inmem 0.31.0.dev1__tar.gz → 0.31.0.dev2__tar.gz

{langgraph_runtime_inmem-0.31.0.dev1 → langgraph_runtime_inmem-0.31.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-runtime-inmem
-Version: 0.31.0.dev1
+Version: 0.31.0.dev2
 Summary: Inmem implementation for the LangGraph API server.
 Author-email: Will Fu-Hinthorn <will@langchain.dev>
 License: Elastic-2.0

{langgraph_runtime_inmem-0.31.0.dev1 → langgraph_runtime_inmem-0.31.0.dev2}/langgraph_runtime_inmem/__init__.py

@@ -10,7 +10,7 @@ from langgraph_runtime_inmem import (
     store,
 )
 
-__version__ = "0.31.0.dev1"
+__version__ = "0.31.0.dev2"
 __all__ = [
     "ops",
     "database",

{langgraph_runtime_inmem-0.31.0.dev1 → langgraph_runtime_inmem-0.31.0.dev2}/langgraph_runtime_inmem/lifespan.py

@@ -66,6 +66,16 @@ async def lifespan(
     await api_checkpointer.start_checkpointer()
     await start_ui_bundler()
 
+    if config.DATADOG_METRICS_ENABLED or config.LSD_PROM_METRICS_ENABLED:
+        from langgraph_api.metrics_otlp import (  # noqa: PLC0415
+            COUNTER_SERVER_STARTED,
+            get_otlp_metrics_reporter,
+        )
+
+        reporter = get_otlp_metrics_reporter()
+        reporter.initialize()
+        reporter.inc_counter(COUNTER_SERVER_STARTED)
+
     async def _log_graph_load_failure(err: graph.GraphLoadError) -> None:
         cause = err.__cause__ or err.cause
         log_fields = err.log_fields()
@@ -134,6 +144,17 @@ async def lifespan(
     except asyncio.CancelledError:
         pass
     finally:
+        otlp_reporter = None
+        if config.DATADOG_METRICS_ENABLED or config.LSD_PROM_METRICS_ENABLED:
+            from langgraph_api.metrics_otlp import (  # noqa: PLC0415
+                COUNTER_SERVER_REQUESTED_TO_STOP,
+                get_otlp_metrics_reporter,
+            )
+
+            # Shutdown has been requested; teardown is starting.
+            otlp_reporter = get_otlp_metrics_reporter()
+            otlp_reporter.inc_counter(COUNTER_SERVER_REQUESTED_TO_STOP)
+
         await api_store.exit_store()
         await api_checkpointer.exit_checkpointer()
         await stop_ui_bundler()
@@ -142,6 +163,16 @@ async def lifespan(
         await stop_webhook_http_client()
         await stop_pool()
 
+        if otlp_reporter is not None:
+            from langgraph_api.metrics_otlp import (  # noqa: PLC0415
+                COUNTER_SERVER_STOPPED,
+            )
+
+            # Teardown completed successfully. Emit before shutting down the
+            # reporter so the final OTLP export includes this counter.
+            otlp_reporter.inc_counter(COUNTER_SERVER_STOPPED)
+            otlp_reporter.shutdown()
+
 
 async def queue_with_signal():
     try: