chore: Type improvements to account, address_book and consensus (#160)

* chore: type fixes account

Signed-off-by: exploreriii <[email protected]>

* chore: type fixes address_book

Signed-off-by: exploreriii <[email protected]>

* chore: type fixes consensus

Signed-off-by: exploreriii <[email protected]>

* fix: remove duplicate eq

Signed-off-by: exploreriii <[email protected]>

---------

Signed-off-by: exploreriii <[email protected]>

Author: exploreriii

src/hiero_sdk_python/account/account_balance.py

@@ -1,4 +1,4 @@
-from typing import Union, Dict
+from typing import Dict
 
 from hiero_sdk_python.tokens.token_id import TokenId
 from hiero_sdk_python.hbar import Hbar
@@ -42,7 +42,7 @@ def _from_proto(cls, proto: CryptoGetAccountBalanceResponse) -> "AccountBalance"
         if proto.tokenBalances:
             for token_balance in proto.tokenBalances:
                 token_id: TokenId = TokenId._from_proto(token_balance.tokenId)
-                balance: Union[Hbar,int] = token_balance.balance
+                balance: int = token_balance.balance
                 token_balances[token_id] = balance
 
         return cls(hbars=hbars, token_balances=token_balances)

src/hiero_sdk_python/account/account_create_transaction.py

@@ -103,7 +103,7 @@ def set_auto_renew_period(self, seconds: Union[int,Duration]) -> "AccountCreateT
             self.auto_renew_period = seconds
         else:
             raise TypeError("Duration of invalid type")
-
+        return self
 
     def set_account_memo(self, memo: str) -> "AccountCreateTransaction":
         """

src/hiero_sdk_python/account/account_id.py

@@ -72,11 +72,6 @@ def __repr__(self):
         """
         return f"AccountId(shard={self.shard}, realm={self.realm}, num={self.num})"
 
-    def __eq__(self, other):
-            if not isinstance(other, AccountId):
-                return False
-            return (self.shard, self.realm, self.num) == (other.shard, other.realm, other.num)
-
     def __eq__(self, other: object) -> bool:
         """
         Checks equality between two AccountId instances.

src/hiero_sdk_python/address_book/node_address.py

@@ -1,7 +1,7 @@
 from typing import List, TypedDict
 
 from hiero_sdk_python.account.account_id import AccountId
-from hiero_sdk_python.address_book.endpoint import Endpoint
+from hiero_sdk_python.address_book.endpoint import Endpoint, EndpointDict
 from hiero_sdk_python.hapi.services.basic_types_pb2 import NodeAddress as NodeAddressProto
 
 class NodeDict(TypedDict):
@@ -20,7 +20,7 @@ class NodeDict(TypedDict):
     node_account_id: str
     node_id: int
     node_cert_hash: str
-    service_endpoints: List[Endpoint]
+    service_endpoints: List[EndpointDict]
     description: str
 
 class NodeAddress:
@@ -136,7 +136,7 @@ def _from_dict(cls, node: NodeDict) -> 'NodeAddress':
         Create a NodeAddress from a dictionary.
         """
 
-        service_endpoints: List[Endpoint] = node.get('service_endpoints', [])
+        service_endpoints: List[EndpointDict] = node.get('service_endpoints', [])
         public_key: str = node.get('public_key')
         account_id: AccountId = AccountId.from_string(node.get('node_account_id'))
         node_id: int = node.get('node_id')

src/hiero_sdk_python/consensus/topic_info.py

@@ -2,8 +2,9 @@
 from hiero_sdk_python.hapi.services.basic_types_pb2 import Key, AccountID
 from hiero_sdk_python.hapi.services.timestamp_pb2 import Timestamp
 from hiero_sdk_python.hapi.services import consensus_topic_info_pb2
-from hiero_sdk_python import Duration
+from hiero_sdk_python.Duration import Duration
 from hiero_sdk_python.utils.key_format import format_key
+from typing import Optional
 
 class TopicInfo:
     def __init__(
@@ -12,11 +13,12 @@ def __init__(
         running_hash: bytes,
         sequence_number: int,
         expiration_time: Timestamp,
-        admin_key: Key,
-        submit_key: Key,
-        auto_renew_period: Duration,
-        auto_renew_account: AccountID,
-        ledger_id: bytes,
+        admin_key: Optional[Key],
+        submit_key: Optional[Key],
+        auto_renew_period: Optional[Duration],
+        auto_renew_account: Optional[AccountID],
+        ledger_id: Optional[bytes],
+
     ) -> None:
         """
         Initializes a new instance of the TopicInfo class.
@@ -89,6 +91,9 @@ def __str__(self) -> str:
             exp_dt = datetime.fromtimestamp(self.expiration_time.seconds)
 
         running_hash_hex: str = self.running_hash.hex() if self.running_hash else None
+        ledger_id_hex: Optional[str] = (
+            self.ledger_id.hex() if isinstance(self.ledger_id, (bytes, bytearray)) else None
+        )
 
         return (
             "TopicInfo(\n"
@@ -100,6 +105,6 @@ def __str__(self) -> str:
             f"  submit_key={format_key(self.submit_key)},\n"
             f"  auto_renew_period={self.auto_renew_period.seconds},\n"
             f"  auto_renew_account={self.auto_renew_account},\n"
-            f"  ledger_id={self.ledger_id}\n"
+            f"  ledger_id=0x{ledger_id_hex}\n"
             ")"
         )

src/hiero_sdk_python/consensus/topic_message.py

@@ -17,7 +17,7 @@ def __init__(self, response: mirror_proto.ConsensusTopicResponse) -> None:
         Args:
             response: The ConsensusTopicResponse containing chunk data.
         """
-        self.consensus_timestamp: Timestamp = Timestamp._from_protobuf(response.consensusTimestamp).to_date()
+        self.consensus_timestamp: datetime = Timestamp._from_protobuf(response.consensusTimestamp).to_date()
         self.content_size: int = len(response.message)
         self.running_hash: Union[bytes, int] = response.runningHash
         self.sequence_number: Union[bytes, int] = response.sequenceNumber