Feat: Replace ULID raw record ID with UUID v7 (#300)

airbytehq · Jul 18, 2024 · 0616d8a · 0616d8a
1 parent 4249bd4
commit 0616d8a
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 17 deletions.
diff --git a/airbyte/records.py b/airbyte/records.py
@@ -72,7 +72,7 @@
 from typing import TYPE_CHECKING, Any
 
 import pytz
-import ulid
+from uuid_extensions import uuid7str
 
 from airbyte._util.name_normalizers import LowerCaseNormalizer, NameNormalizerBase
 from airbyte.constants import (
@@ -142,7 +142,7 @@ def from_record_message(
     ) -> StreamRecord:
         """Return a StreamRecord from a RecordMessage."""
         data_dict: dict[str, Any] = record_message.data.copy()
-        data_dict[AB_RAW_ID_COLUMN] = str(ulid.ULID())
+        data_dict[AB_RAW_ID_COLUMN] = uuid7str()
         data_dict[AB_EXTRACTED_AT_COLUMN] = datetime.fromtimestamp(
             record_message.emitted_at / 1000, tz=pytz.utc
         )

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -40,12 +40,12 @@ snowflake-connector-python = "^3.10.0"
 snowflake-sqlalchemy = "^1.5.1"
 sqlalchemy = "1.4.51"
 types-pyyaml = "^6.0.12.12"
-ulid = "^1.1"
 
 # TODO: Remove this arbitrary python constraint once `sqlalchemy-bigquery` has done so.
 sqlalchemy-bigquery = { version = "1.9.0", python = "<3.13" }
 airbyte-api = "^0.49.2"
 google-cloud-bigquery-storage = "^2.25.0"
+uuid7 = "^0.1.0"
 
 [tool.poetry.group.dev.dependencies]
 docker = "^7.0.0"