From bfdd7deb609cf9b7d37f22e9ea651efe6c3713b2 Mon Sep 17 00:00:00 2001
From: Quanzheng Long <long@langchain.dev>
Date: Mon, 16 Mar 2026 16:41:14 -0700
Subject: [PATCH] makefile

---
 saf-python-sdk/Makefile | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/saf-python-sdk/Makefile b/saf-python-sdk/Makefile
index ff0099649..0fd81a54e 100644
--- a/saf-python-sdk/Makefile
+++ b/saf-python-sdk/Makefile
@@ -1,10 +1,31 @@
+
 PYPI_REPOSITORY ?= pypi
 PYPI_TOKEN ?=
+PYTHON_VERSION ?= 3.13
+TEST_BASE = uv run --python $(PYTHON_VERSION) --with pytest --with anyio --with typing_extensions --with pydantic --with langgraph pytest -q -s
 
-.PHONY: publish-to-pypi-saf-python-sdk
+.PHONY: publish-to-pypi-saf-python-sdk all-tests test_primitives test_sub_agents test_update_elision test_run_pool_size test-benchmark
 publish-to-pypi-saf-python-sdk:
 	@test -n "$(PYPI_TOKEN)" || (echo "PYPI_TOKEN is required"; exit 1)
 	cd "$(CURDIR)" && \
 	MATURIN_PYPI_TOKEN="$(PYPI_TOKEN)" \
 	uvx maturin publish --repository $(PYPI_REPOSITORY) --non-interactive --skip-existing --no-sdist
 
+all-tests:
+	$(TEST_BASE) tests/advanced-graph/test_*.py
+
+test_primitives:
+	$(TEST_BASE) tests/advanced-graph/test_primitives.py
+
+test_sub_agents:
+	$(TEST_BASE) tests/advanced-graph/test_sub_agents.py
+
+test_update_elision:
+	$(TEST_BASE) tests/advanced-graph/test_update_elision.py
+
+test_run_pool_size:
+	$(TEST_BASE) tests/advanced-graph/test_run_pool_size.py
+
+test-benchmark:
+	uv run --python $(PYTHON_VERSION) --with anyio --with typing_extensions --with pydantic --with langgraph python tests/advanced-graph/benchmark_stategraph_vs_advancedgraph.py
+