diff --git a/clients/client-python/build.gradle.kts b/clients/client-python/build.gradle.kts
index af6cfcd2d9f..c5ec7790432 100644
--- a/clients/client-python/build.gradle.kts
+++ b/clients/client-python/build.gradle.kts
@@ -31,6 +31,8 @@ pythonPlugin {
   pythonVersion.set(project.rootProject.extra["pythonVersion"].toString())
 }
 
+val gravitinoVersion: String = project.properties["version"] as String
+
 fun deleteCacheDir(targetDir: String) {
   project.fileTree(project.projectDir).matching {
     include("**/$targetDir/**")
@@ -122,7 +124,7 @@ fun generatePypiProjectHomePage() {
     // relative path of the images in the how-to-use-python-client.md file is incorrect. We need
     // to fix the relative path of the images/markdown to the absolute path.
     val content = outputFile.readText()
-    val docsUrl = "https://datastrato.ai/docs/latest"
+    val docsUrl = "https://gravitino.apache.org/docs/$gravitinoVersion"
 
     // Use regular expression to match the `[](./a/b/c.md?language=python)` or `[](./a/b/c.md#arg1)` link in the content
     // Convert `[](./a/b/c.md?language=python)` to `[](https://datastrato.ai/docs/latest/a/b/c/language=python)`