diff --git a/update.sh b/update.sh
new file mode 100755
index 0000000..f656fc6
--- /dev/null
+++ b/update.sh
@@ -0,0 +1,22 @@
+#!/bin/bash
+
+if [[ -z "$1" ]]; then
+    echo "Usage: bash update.sh version" 1>&2
+    echo ""
+    echo "Example: bash update.sh v0.34.0" 1>&2
+    exit 1
+fi
+
+VERSION=$1
+
+pushd third_party/substrait/
+git checkout $VERSION
+SUBSTRAIT_HASH=$(git rev-parse --short HEAD)
+popd
+
+VERSION=$(echo $VERSION | sed 's/v//')
+
+sed -i "s#__substrait_hash__.*#__substrait_hash__ = \"$SUBSTRAIT_HASH\"#g" src/substrait/__init__.py
+sed -i "s#__substrait_version__.*#__substrait_version__ = \"$VERSION\"#g" src/substrait/__init__.py
+
+./gen_proto.sh