chore: fix message count to increase on total chunck message

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

Author: manishdait

src/hiero_sdk_python/query/topic_message_query.py

@@ -184,10 +184,11 @@ def _build_query_request(self, state: SubscriptionState) -> mirror_proto.Consens
 
     def _handle_response(self, response, state: SubscriptionState, on_message: Callable[[TopicMessage], None]) -> None:
         """Handles single or chunked messages."""
-        state.count += 1
         state.last_message = response
 
         if not self._chunking_enabled or not response.HasField("chunkInfo") or response.chunkInfo.total <= 1:
+            state.count += 1
+
             message = TopicMessage.of_single(response)
             on_message(message)
             return
@@ -201,8 +202,9 @@ def _handle_response(self, response, state: SubscriptionState, on_message: Calla
         chunks.append(response)
 
         if len(chunks) == response.chunkInfo.total:
-            del state.pending_messages[initial_tx_id]
+            state.count += 1
 
+            del state.pending_messages[initial_tx_id]
             message = TopicMessage.of_many(chunks)
             on_message(message)