96 lines
2.1 KiB
Nix

{
lib,
buildPythonPackage,
fetchFromGitHub,
# build-system
hatchling,
# dependencies
click,
langgraph,
langgraph-runtime-inmem,
langgraph-sdk,
python-dotenv,
# testing
pytest-asyncio,
pytestCheckHook,
docker-compose,
# passthru
gitUpdater,
}:
buildPythonPackage rec {
pname = "langgraph-cli";
version = "0.3.6";
pyproject = true;
src = fetchFromGitHub {
owner = "langchain-ai";
repo = "langgraph";
tag = "cli==${version}";
hash = "sha256-tBMdFOHSRjw0PtE19XytLU4MmjR3NBLJxUqWoG4L2F8=";
};
sourceRoot = "${src.name}/libs/cli";
build-system = [ hatchling ];
dependencies = [
click
langgraph-sdk
];
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" ];
pythonImportsCheck = [ "langgraph_cli" ];
disabledTests = [
# Flaky tests that generate a Docker configuration then compare to exact text
"test_config_to_docker_simple"
"test_config_to_docker_pipconfig"
"test_config_to_compose_env_vars"
"test_config_to_compose_env_file"
"test_config_to_compose_end_to_end"
"test_config_to_compose_simple_config"
"test_config_to_compose_watch"
# Tests that require docker
"test_dockerfile_command_with_docker_compose"
"test_build_command_with_api_version_and_base_image"
"test_build_command_with_api_version"
"test_build_generate_proper_build_context"
"test_build_command_shows_wolfi_warning"
];
passthru.updateScript = gitUpdater {
rev-prefix = "cli==";
};
meta = {
description = "Official CLI for LangGraph API";
homepage = "https://github.com/langchain-ai/langgraph/tree/main/libs/cli";
changelog = "https://github.com/langchain-ai/langgraph/releases/tag/${version}";
mainProgram = "langgraph";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ sarahec ];
};
}