chore: Added missing timeout support for some execute() implementations (hiero-ledger#1736)

Signed-off-by: Manish Dait <daitmanish88@gmail.com>

Author: manishdait

CHANGELOG.md

@@ -296,6 +296,7 @@ This changelog is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.
 - Update `.github/scripts/bot-advanced-check.sh` to unassign unqualified users.
 - Fixed broken project structure link in `CONTRIBUTING.md` (#1664)
 - Refactor spam list update logic and remove unused pull request creation step `.github/scripts/update-spam-list.js` `.github/workflows/cron-update-spam-list.yml`.
+- Ensure all Query sub-class `execute()` function to correctly propagate the optional `timeout` parameter.
 
 ### Removed
 

src/hiero_sdk_python/account/account_records_query.py

@@ -3,7 +3,7 @@
 """
 
 import traceback
-from typing import List, Optional
+from typing import List, Optional, Union
 
 from hiero_sdk_python.account.account_id import AccountId
 from hiero_sdk_python.channels import _Channel
@@ -96,7 +96,7 @@ def _get_method(self, channel: _Channel) -> _Method:
         """
         return _Method(transaction_func=None, query_func=channel.crypto.getAccountRecords)
 
-    def execute(self, client: Client) -> List[TransactionRecord]:
+    def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -> List[TransactionRecord]:
         """
         Executes the account records query.
 
@@ -108,6 +108,7 @@ def execute(self, client: Client) -> List[TransactionRecord]:
 
         Args:
             client (Client): The client instance to use for execution
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             List[TransactionRecord]: The account records from the network
@@ -118,7 +119,7 @@ def execute(self, client: Client) -> List[TransactionRecord]:
             ReceiptStatusError: If the query fails with a receipt status error
         """
         self._before_execute(client)
-        response = self._execute(client)
+        response = self._execute(client, timeout)
 
         records = []
         for record in response.cryptoGetAccountRecords.records:

src/hiero_sdk_python/contract/contract_bytecode_query.py

@@ -3,7 +3,7 @@
 """
 
 import traceback
-from typing import Optional
+from typing import Optional, Union
 
 from hiero_sdk_python.channels import _Channel
 from hiero_sdk_python.client.client import Client
@@ -106,7 +106,7 @@ def _get_method(self, channel: _Channel) -> _Method:
             transaction_func=None, query_func=channel.smart_contract.ContractGetBytecode
         )
 
-    def execute(self, client: Client) -> bytes:
+    def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -> bytes:
         """
         Executes the contract bytecode query.
 
@@ -118,6 +118,7 @@ def execute(self, client: Client) -> bytes:
 
         Args:
             client (Client): The client instance to use for execution
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             bytes: The bytecode of the contract from the network
@@ -128,7 +129,7 @@ def execute(self, client: Client) -> bytes:
             ReceiptStatusError: If the query fails with a receipt status error
         """
         self._before_execute(client)
-        response = self._execute(client)
+        response = self._execute(client, timeout)
 
         return response.contractGetBytecodeResponse.bytecode
 

src/hiero_sdk_python/contract/contract_call_query.py

@@ -199,7 +199,7 @@ def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -
 
         Args:
             client (Client): The client instance to use for execution
-            timeout (Optional[Union[int, float]): The total execution timeout (in seconds) for this execution.
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             ContractFunctionResult: The result of the contract call

src/hiero_sdk_python/contract/contract_info_query.py

@@ -110,7 +110,7 @@ def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -
 
         Args:
             client (Client): The client instance to use for execution
-            timeout (Optional[Union[int, float]): The total execution timeout (in seconds) for this execution.
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             ContractInfo: The contract info from the network

src/hiero_sdk_python/file/file_contents_query.py

@@ -2,7 +2,7 @@
 Query to get the contents of a file on the network.
 """
 
-from typing import Optional
+from typing import Optional, Union
 
 from hiero_sdk_python.channels import _Channel
 from hiero_sdk_python.client.client import Client
@@ -96,7 +96,7 @@ def _get_method(self, channel: _Channel) -> _Method:
         """
         return _Method(transaction_func=None, query_func=channel.file.getFileContent)
 
-    def execute(self, client: Client) -> str:
+    def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -> str:
         """
         Executes the file contents query.
 
@@ -108,6 +108,7 @@ def execute(self, client: Client) -> str:
 
         Args:
             client (Client): The client instance to use for execution
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             str: The contents of the file from the network
@@ -118,7 +119,7 @@ def execute(self, client: Client) -> str:
             ReceiptStatusError: If the query fails with a receipt status error
         """
         self._before_execute(client)
-        response = self._execute(client)
+        response = self._execute(client, timeout)
 
         return response.fileGetContents.fileContents.contents
 

src/hiero_sdk_python/file/file_info_query.py

@@ -3,7 +3,7 @@
 """
 
 import traceback
-from typing import Optional
+from typing import Optional, Union
 
 from hiero_sdk_python.channels import _Channel
 from hiero_sdk_python.client.client import Client
@@ -95,7 +95,7 @@ def _get_method(self, channel: _Channel) -> _Method:
         """
         return _Method(transaction_func=None, query_func=channel.file.getFileInfo)
 
-    def execute(self, client: Client) -> FileInfo:
+    def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -> FileInfo:
         """
         Executes the file info query.
 
@@ -107,6 +107,7 @@ def execute(self, client: Client) -> FileInfo:
 
         Args:
             client (Client): The client instance to use for execution
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             FileInfo: The file info from the network
@@ -117,7 +118,7 @@ def execute(self, client: Client) -> FileInfo:
             ReceiptStatusError: If the query fails with a receipt status error
         """
         self._before_execute(client)
-        response = self._execute(client)
+        response = self._execute(client, timeout)
 
         return FileInfo._from_proto(response.fileGetInfo.fileInfo)
 

src/hiero_sdk_python/query/account_balance_query.py

@@ -145,7 +145,7 @@ def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -
 
         Args:
             client (Client): The client instance to use for execution
-            timeout (Optional[Union[int, float]): The total execution timeout (in seconds) for this execution.
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             AccountBalance: The account balance from the network

src/hiero_sdk_python/query/account_info_query.py

@@ -98,10 +98,10 @@ def execute(self, client, timeout: Optional[Union[int, float]] = None):
         to return an AccountInfo object.
 
         This function delegates the core logic to `_execute()`, and may propagate exceptions raised by it.
-        timeout (Optional[Union[int, float]): The total execution timeout (in seconds) for this execution.
 
         Args:
             client (Client): The client instance to use for execution
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             AccountInfo: The account info from the network

src/hiero_sdk_python/query/token_info_query.py

@@ -101,7 +101,7 @@ def execute(self, client: Client, timeout: Optional[Union[int, float]] = None) -
 
         Args:
             client (Client): The client instance to use for execution
-            timeout (Optional[Union[int, float]): The total execution timeout (in seconds) for this execution.
+            timeout (Optional[Union[int, float]]): The total execution timeout (in seconds) for this execution.
 
         Returns:
             TokenInfo: The token info from the network