feat: Add WASM compatibility fallback from serpyco-rs to serpyco

airbyte_cdk/models/airbyte_protocol.py

@@ -2,11 +2,16 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
 
+import sys
 from dataclasses import InitVar, dataclass
 from typing import Annotated, Any, Dict, List, Mapping, Optional, Union
 
 from airbyte_protocol_dataclasses.models import *  # noqa: F403  # Allow '*'
-from serpyco_rs.metadata import Alias
+
+if sys.platform == "emscripten":
+    from serpyco.metadata import Alias
+else:
+    from serpyco_rs.metadata import Alias
 
 # ruff: noqa: F405  # ignore fuzzy import issues with 'import *'
 

airbyte_cdk/models/airbyte_protocol_serializers.py

@@ -1,7 +1,15 @@
 # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
-from typing import Any, Dict
-
-from serpyco_rs import CustomType, Serializer
+import sys
+from typing import TYPE_CHECKING, Any, Dict
+
+if TYPE_CHECKING:
+    from serpyco_rs import CustomType, Serializer
+else:
+    USE_RUST_BACKEND = sys.platform != "emscripten"
+    if USE_RUST_BACKEND:
+        from serpyco_rs import CustomType, Serializer
+    else:
+        from serpyco import CustomType, Serializer
 
 from .airbyte_protocol import (  # type: ignore[attr-defined] # all classes are imported to airbyte_protocol via *
     AirbyteCatalog,

airbyte_cdk/test/entrypoint_wrapper.py

@@ -17,6 +17,7 @@
 import json
 import logging
 import re
+import sys
 import tempfile
 import traceback
 from collections import deque
@@ -28,7 +29,11 @@
 
 import orjson
 from pydantic import ValidationError as V2ValidationError
-from serpyco_rs import SchemaValidationError
+
+if sys.platform == "emscripten":
+    from serpyco import SchemaValidationError
+else:
+    from serpyco_rs import SchemaValidationError
 
 from airbyte_cdk.entrypoint import AirbyteEntrypoint
 from airbyte_cdk.exception_handler import assemble_uncaught_exception