fix state deserialization issue

Author: maxi297

airbyte_cdk/models/airbyte_protocol.py

@@ -2,6 +2,10 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
 
+"""
+This file is necessary because the `AirbyteStateBlob` implementation in the protocol lib is incomplete and given we use the incomplete implementation, we will get `TypeError: AirbyteStateBlob.__init__() takes 1 positional argument but 2 were given`. Hence, we need to redefine all the classes that could serialize AirbyteStateBlob to use the CDK implementation, not the protocol lib one.
+"""
+
 from dataclasses import InitVar, dataclass
 from typing import Annotated, Any, Dict, List, Mapping, Optional, Union
 
@@ -86,3 +90,4 @@ class AirbyteMessage:
     state: Optional[AirbyteStateMessage] = None
     trace: Optional[AirbyteTraceMessage] = None  # type: ignore [name-defined]
     control: Optional[AirbyteControlMessage] = None  # type: ignore [name-defined]
+    destination_catalog: Optional[DestinationCatalog] = None  # type: ignore [name-defined]

airbyte_cdk/models/airbyte_protocol_serializers.py

@@ -1,17 +1,16 @@
 # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
 from typing import Any, Dict
 
-from airbyte_protocol_dataclasses.models import (  # type: ignore[attr-defined] # all classes are imported to airbyte_protocol via *
+from .airbyte_protocol import (  # type: ignore[attr-defined] # all classes are imported to airbyte_protocol via *
     AirbyteMessage,
-    AirbyteStateMessage,
+    AirbyteStateBlob,
     AirbyteStreamState,
     ConfiguredAirbyteCatalog,
     ConfiguredAirbyteStream,
     ConnectorSpecification,
 )
 from serpyco_rs import CustomType, Serializer
 
-from .airbyte_protocol import AirbyteStateBlob
 
 
 class AirbyteStateBlobType(CustomType[AirbyteStateBlob, Dict[str, Any]]):