diff --git a/pyproject.toml b/pyproject.toml
index 1e434a85..5179dfdf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "openpipe-art"
-version = "0.5.0"
+version = "0.5.1"
 description = "The OpenPipe Agent Reinforcement Training (ART) library"
 readme = "README.md"
 requires-python = ">=3.10"
diff --git a/uv.lock b/uv.lock
index 728d3b69..1a20b540 100644
--- a/uv.lock
+++ b/uv.lock
@@ -4145,7 +4145,7 @@ wheels = [
 
 [[package]]
 name = "openpipe-art"
-version = "0.5.0"
+version = "0.5.1"
 source = { editable = "." }
 dependencies = [
     { name = "litellm" },