python3Packages.langgraph-runtime-inmem: init at 0.6.8 (#431562)

pkgs/development/python-modules/langgraph-cli/default.nix

@@ -8,7 +8,10 @@
 
   # dependencies
   click,
+  langgraph,
+  langgraph-runtime-inmem,
   langgraph-sdk,
+  python-dotenv,
 
   # testing
   pytest-asyncio,
@@ -40,21 +43,20 @@ buildPythonPackage rec {
     langgraph-sdk
   ];
 
-  # Not yet. Depemnds on `langgraph-runtime-inmem` which isn't in github yet
-  # https://github.com/langchain-ai/langgraph/issues/5802
-  # optional-dependencies = {
-  #   "inmem" = [
-  #     langgraph-api
-  #     langgraph-runtime-inmem
-  #     python-dotenv
-  #   ]
-  # }
+  optional-dependencies = {
+    "inmem" = [
+      langgraph
+      langgraph-runtime-inmem
+      python-dotenv
+    ];
+  };
 
   nativeCheckInputs = [
     pytest-asyncio
     pytestCheckHook
     docker-compose
-  ];
+  ]
+  ++ lib.flatten (builtins.attrValues optional-dependencies);
 
   enabledTestPaths = [ "tests/unit_tests" ];
 

pkgs/development/python-modules/langgraph-runtime-inmem/default.nix

@@ -0,0 +1,59 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchPypi,
+  hatchling,
+  blockbuster,
+  langgraph,
+  langgraph-checkpoint,
+  sse-starlette,
+  starlette,
+  structlog,
+}:
+
+buildPythonPackage rec {
+  pname = "langgraph-runtime-inmem";
+  version = "0.6.8";
+  pyproject = true;
+
+  # Not available in any repository
+  src = fetchPypi {
+    pname = "langgraph_runtime_inmem";
+    inherit version;
+    hash = "sha256-chPmwJ+tUJoRK5xX9+r6mbYf95ZbX4Z3mP6Ra19nBxM=";
+  };
+
+  build-system = [
+    hatchling
+  ];
+
+  dependencies = [
+    blockbuster
+    langgraph
+    langgraph-checkpoint
+    sse-starlette
+    starlette
+    structlog
+  ];
+
+  # Can remove after blockbuster version bump
+  # https://github.com/NixOS/nixpkgs/pull/431547
+  pythonRelaxDeps = [
+    "blockbuster"
+  ];
+
+  pythonImportsCheck = [
+    "langgraph_runtime_inmem"
+  ];
+
+  # no tests
+  doCheck = false;
+
+  meta = {
+    description = "Inmem implementation for the LangGraph API server";
+    homepage = "https://pypi.org/project/langgraph-runtime-inmem/";
+    # no changelog
+    license = lib.licenses.elastic20;
+    maintainers = with lib.maintainers; [ sarahec ];
+  };
+}

pkgs/top-level/python-packages.nix

@@ -7892,6 +7892,8 @@ self: super: with self; {
 
   langgraph-prebuilt = callPackage ../development/python-modules/langgraph-prebuilt { };
 
+  langgraph-runtime-inmem = callPackage ../development/python-modules/langgraph-runtime-inmem { };
+
   langgraph-sdk = callPackage ../development/python-modules/langgraph-sdk { };
 
   langid = callPackage ../development/python-modules/langid { };