From 5b3c5be42034bf1c108b14344580ec38e986710c Mon Sep 17 00:00:00 2001
From: xiaohongbo <xiaohongbo.xhb@alibaba-inc.com>
Date: Wed, 25 Feb 2026 22:01:59 +0800
Subject: [PATCH 1/2] [python] feat: FlussError.is_retriable()

---
 bindings/python/fluss/__init__.pyi |  2 ++
 bindings/python/src/error.rs       | 20 +++++++++++++++
 bindings/python/test/test_admin.py | 39 ++++++++++++++++++++++++++++++
 3 files changed, 61 insertions(+)

diff --git a/bindings/python/fluss/__init__.pyi b/bindings/python/fluss/__init__.pyi
index 6f9ae0b3..813c596b 100644
--- a/bindings/python/fluss/__init__.pyi
+++ b/bindings/python/fluss/__init__.pyi
@@ -771,6 +771,8 @@ class FlussError(Exception):
     message: str
     error_code: int
     def __init__(self, message: str, error_code: int = -2) -> None: ...
+    def is_retriable(self) -> bool:
+        ...
     def __str__(self) -> str: ...
 
 class LakeSnapshot:
diff --git a/bindings/python/src/error.rs b/bindings/python/src/error.rs
index 606b9f4f..cbcbe4da 100644
--- a/bindings/python/src/error.rs
+++ b/bindings/python/src/error.rs
@@ -23,6 +23,22 @@ use pyo3::prelude::*;
 /// collide with current or future API codes. Consistent with the CPP binding.
 const CLIENT_ERROR_CODE: i32 = -2;
 
+const RETRIABLE_ERROR_CODES: &[i32] = &[
+    1,  // NetworkException
+    3,  // CorruptMessage
+    10, // LogStorageException
+    11, // KvStorageException
+    12, // NotLeaderOrFollower
+    14, // CorruptRecordException
+    21, // UnknownTableOrBucketException
+    25, // RequestTimeOut
+    26, // StorageException
+    28, // NotEnoughReplicasAfterAppendException
+    29, // NotEnoughReplicasException
+    44, // LeaderNotAvailableException
+    51, // RetriableAuthenticateException
+];
+
 /// Fluss errors
 #[pyclass(extends=PyException)]
 #[derive(Debug, Clone)]
@@ -44,6 +60,10 @@ impl FlussError {
         }
     }
 
+    fn is_retriable(&self) -> bool {
+        RETRIABLE_ERROR_CODES.contains(&self.error_code)
+    }
+
     fn __str__(&self) -> String {
         if self.error_code != CLIENT_ERROR_CODE {
             format!("FlussError(code={}): {}", self.error_code, self.message)
diff --git a/bindings/python/test/test_admin.py b/bindings/python/test/test_admin.py
index f203400f..f8ed8632 100644
--- a/bindings/python/test/test_admin.py
+++ b/bindings/python/test/test_admin.py
@@ -190,6 +190,7 @@ async def test_fluss_error_response(admin):
         await admin.get_table_info(table_path)
 
     assert exc_info.value.error_code == fluss.ErrorCode.TABLE_NOT_EXIST
+    assert exc_info.value.is_retriable() is False
 
 
 async def test_error_database_not_exist(admin):
@@ -299,3 +300,41 @@ async def test_error_table_not_partitioned(admin):
 
     await admin.drop_table(table_path, ignore_if_not_exists=True)
     await admin.drop_database(db_name, ignore_if_not_exists=True, cascade=True)
+
+
+def test_fluss_error_is_retriable():
+    client_err = fluss.FlussError("connection failed", fluss.ErrorCode.CLIENT_ERROR)
+    assert client_err.is_retriable() is False
+
+    table_not_exist = fluss.FlussError(
+        "table not found", fluss.ErrorCode.TABLE_NOT_EXIST
+    )
+    assert table_not_exist.is_retriable() is False
+
+    db_not_exist = fluss.FlussError(
+        "database not found", fluss.ErrorCode.DATABASE_NOT_EXIST
+    )
+    assert db_not_exist.is_retriable() is False
+
+    assert (
+        fluss.FlussError("network", fluss.ErrorCode.NETWORK_EXCEPTION).is_retriable()
+        is True
+    )
+    assert (
+        fluss.FlussError(
+            "timeout", fluss.ErrorCode.REQUEST_TIME_OUT
+        ).is_retriable()
+        is True
+    )
+    assert (
+        fluss.FlussError(
+            "leader not available", fluss.ErrorCode.LEADER_NOT_AVAILABLE_EXCEPTION
+        ).is_retriable()
+        is True
+    )
+    assert (
+        fluss.FlussError(
+            "retriable auth", fluss.ErrorCode.RETRIABLE_AUTHENTICATE_EXCEPTION
+        ).is_retriable()
+        is True
+    )

From a6d44a6cbfdaf8ecd3234858754b2b9de3343891 Mon Sep 17 00:00:00 2001
From: xiaohongbo <xiaohongbo.xhb@alibaba-inc.com>
Date: Wed, 25 Feb 2026 22:08:05 +0800
Subject: [PATCH 2/2] optimize format

---
 bindings/python/fluss/__init__.pyi | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/bindings/python/fluss/__init__.pyi b/bindings/python/fluss/__init__.pyi
index 813c596b..c55411bd 100644
--- a/bindings/python/fluss/__init__.pyi
+++ b/bindings/python/fluss/__init__.pyi
@@ -771,8 +771,10 @@ class FlussError(Exception):
     message: str
     error_code: int
     def __init__(self, message: str, error_code: int = -2) -> None: ...
+
     def is_retriable(self) -> bool:
         ...
+
     def __str__(self) -> str: ...
 
 class LakeSnapshot: