diff --git a/bindings/python/fluss/__init__.pyi b/bindings/python/fluss/__init__.pyi
index 4c2142d7..c387d734 100644
--- a/bindings/python/fluss/__init__.pyi
+++ b/bindings/python/fluss/__init__.pyi
@@ -169,6 +169,30 @@ class Config:
     def scanner_log_max_poll_records(self) -> int: ...
     @scanner_log_max_poll_records.setter
     def scanner_log_max_poll_records(self, num: int) -> None: ...
+    @property
+    def writer_batch_timeout_ms(self) -> int: ...
+    @writer_batch_timeout_ms.setter
+    def writer_batch_timeout_ms(self, timeout: int) -> None: ...
+    @property
+    def connect_timeout_ms(self) -> int: ...
+    @connect_timeout_ms.setter
+    def connect_timeout_ms(self, timeout: int) -> None: ...
+    @property
+    def security_protocol(self) -> str: ...
+    @security_protocol.setter
+    def security_protocol(self, protocol: str) -> None: ...
+    @property
+    def security_sasl_mechanism(self) -> str: ...
+    @security_sasl_mechanism.setter
+    def security_sasl_mechanism(self, mechanism: str) -> None: ...
+    @property
+    def security_sasl_username(self) -> str: ...
+    @security_sasl_username.setter
+    def security_sasl_username(self, username: str) -> None: ...
+    @property
+    def security_sasl_password(self) -> str: ...
+    @security_sasl_password.setter
+    def security_sasl_password(self, password: str) -> None: ...
 
 class FlussConnection:
     @staticmethod
@@ -837,10 +861,6 @@ class TableBucket:
     def __str__(self) -> str: ...
     def __repr__(self) -> str: ...
 
-class TableDistribution:
-    def bucket_keys(self) -> List[str]: ...
-    def bucket_count(self) -> Optional[int]: ...
-
 class PartitionInfo:
     """Information about a partition."""
 
@@ -864,6 +884,7 @@ class ErrorCode:
     """
 
     CLIENT_ERROR: int
+    NONE: int
     UNKNOWN_SERVER_ERROR: int
     NETWORK_EXCEPTION: int
     UNSUPPORTED_VERSION: int