airbyte-cdk 6.60.0.post29.dev16508987644__py3-none-any.whl → 6.60.0.post32.dev16509172620__py3-none-any.whl

airbyte_cdk/models/__init__.py

@@ -56,6 +56,7 @@ from .airbyte_protocol_serializers import (
     ab_connector_spec_to_string,
     ab_message_from_string,
     ab_message_to_string,
+    ab_state_message_to_string,
 )
 from .well_known_types import (
     BinaryData,

airbyte_cdk/models/airbyte_protocol_serializers.py

@@ -9,8 +9,6 @@ import dacite
 import orjson
 from pydantic import ValidationError
 
-from airbyte_cdk.connector_builder.test_reader.helpers import airbyte_message_to_json
-
 from .airbyte_protocol import (  # type: ignore[attr-defined] # all classes are imported to airbyte_protocol via *
     AirbyteCatalog,
     AirbyteMessage,
@@ -161,21 +159,16 @@ def ab_state_message_from_string(
         raise ValueError(f"Failed to decode JSON: {e}") from e
 
 
-def ab_state_blob_from_string(
-    state_blob_json: str,
-) -> AirbyteStateBlob:
+def ab_state_message_to_string(
+    state: AirbyteStateMessage,
+) -> str:
     """
-    Convert a JSON string to an AirbyteStateBlob.
+    Convert an AirbyteStateMessage to a JSON string.
 
     Args:
-        state_blob_json (str): The JSON string to convert.
+        state (AirbyteStateMessage): The AirbyteStateMessage to convert.
 
     Returns:
-        AirbyteStateBlob: The deserialized AirbyteStateBlob.
+        str: JSON string representation of the AirbyteStateMessage.
     """
-    try:
-        return AirbyteStateBlob.model_validate_json(state_blob_json)
-    except ValidationError as e:
-        raise ValueError(f"Invalid AirbyteStateBlob format: {e}") from e
-    except orjson.JSONDecodeError as e:
-        raise ValueError(f"Failed to decode JSON: {e}") from e
+    return state.model_dump_json()

airbyte_cdk/models/well_known_types.py

@@ -2,4 +2,4 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
 
-from airbyte_protocol_dataclasses.models.well_known_types import *  # noqa: F403  # Allow '*'
+from airbyte_protocol.models.well_known_types import *  # noqa: F403  # Allow '*'

airbyte_cdk/sources/streams/permissions/identities_stream.py

@@ -6,7 +6,7 @@ import traceback
 from abc import ABC, abstractmethod
 from typing import Any, Dict, Iterable, List, Mapping, MutableMapping, Optional
 
-from airbyte_protocol_dataclasses.models import SyncMode
+from airbyte_protocol.models import SyncMode
 
 from airbyte_cdk.models import AirbyteLogMessage, AirbyteMessage, Level
 from airbyte_cdk.models import Type as MessageType

airbyte_cdk/sql/shared/sql_processor.py

@@ -13,7 +13,7 @@ from typing import TYPE_CHECKING, Any, final
 import pandas as pd
 import sqlalchemy
 import ulid
-from airbyte_protocol_dataclasses.models import AirbyteStateMessage
+from airbyte_protocol.models import AirbyteStateMessage
 from pandas import Index
 from pydantic import BaseModel, Field
 from sqlalchemy import Column, Table, and_, create_engine, insert, null, select, text, update

{airbyte_cdk-6.60.0.post29.dev16508987644.dist-info → airbyte_cdk-6.60.0.post32.dev16509172620.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: airbyte-cdk
-Version: 6.60.0.post29.dev16508987644
+Version: 6.60.0.post32.dev16509172620
 Summary: A framework for writing Airbyte Connectors.
 Home-page: https://airbyte.com
 License: MIT

{airbyte_cdk-6.60.0.post29.dev16508987644.dist-info → airbyte_cdk-6.60.0.post32.dev16509172620.dist-info}/RECORD

@@ -48,11 +48,11 @@ airbyte_cdk/manifest_migrations/migrations/http_requester_request_body_json_data
 airbyte_cdk/manifest_migrations/migrations/http_requester_url_base_to_url.py,sha256=EX1MVYVpoWypA28qoH48wA0SYZjGdlR8bcSixTDzfgo,1346
 airbyte_cdk/manifest_migrations/migrations/registry.yaml,sha256=F-hdapvl_vZnsI7CQsV00Rb7g7j4Nt2zaM83-Tbwgbg,956
 airbyte_cdk/manifest_migrations/migrations_registry.py,sha256=zly2fwaOxDukqC7eowzrDlvhA2v71FjW74kDzvRXhSY,2619
-airbyte_cdk/models/__init__.py,sha256=NjSFgkzPZd5miSliYLQP-OynA7tUrYEybdMXYmzJOmQ,2096
+airbyte_cdk/models/__init__.py,sha256=S0_uDwV-S9U0DZyUKyITa_eLz1gxhy9ScgH1Aql8CR8,2128
 airbyte_cdk/models/airbyte_protocol.py,sha256=-JLm0bZtzk5Gn_e0TAin2qTUVGqppJhDzeQ-j3z2Pso,254
-airbyte_cdk/models/airbyte_protocol_serializers.py,sha256=zEx6USlAJ8SycqsHTfvg2bCZe4-ybcljdqyzcIF0MR4,4986
+airbyte_cdk/models/airbyte_protocol_serializers.py,sha256=qdaVBJtuSL_xm8oq_v2fs69KgCEW5x47IWAjVkBBuw4,4665
 airbyte_cdk/models/connector_metadata.py,sha256=BD6CO8c3mHavxRJAcwP29sHtNNVLVSNFNQLgHOVxrwA,3229
-airbyte_cdk/models/well_known_types.py,sha256=EquepbisGPuCSrs_D7YVVnMR9-ShhUr21wnFz3COiJs,156
+airbyte_cdk/models/well_known_types.py,sha256=idL7n1qrW1VSLQbi1ZU2ITxTgiAZyEBN7KfO8Hpx3C4,144
 airbyte_cdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_cdk/sources/__init__.py,sha256=45J83QsFH3Wky3sVapZWg4C58R_i1thm61M06t2c1AQ,1156
 airbyte_cdk/sources/abstract_source.py,sha256=50vxEBRByiNhT4WJkiFvgM-C6PWqKSJgvuNC_aeg2cw,15547
@@ -356,7 +356,7 @@ airbyte_cdk/sources/streams/http/requests_native_auth/abstract_oauth.py,sha256=0
 airbyte_cdk/sources/streams/http/requests_native_auth/abstract_token.py,sha256=Y3n7J-sk5yGjv_OxtY6Z6k0PEsFZmtIRi-x0KCbaHdA,1010
 airbyte_cdk/sources/streams/http/requests_native_auth/oauth.py,sha256=gVLo7nU-ORJd413TZHMJQV4m_vaGnRqhoXGGWehFjDA,19253
 airbyte_cdk/sources/streams/http/requests_native_auth/token.py,sha256=h5PTzcdH-RQLeCg7xZ45w_484OPUDSwNWl_iMJQmZoI,2526
-airbyte_cdk/sources/streams/permissions/identities_stream.py,sha256=9O9k6k18Xm3Zsiw_vnI_jsHXfMCQiek6V-jMkJJLxn8,2621
+airbyte_cdk/sources/streams/permissions/identities_stream.py,sha256=eAT4XsQ5zLuj8KWT6c5ITw6zZatxtvS7J48xcxgwPuM,2609
 airbyte_cdk/sources/streams/utils/__init__.py,sha256=4Hw-PX1-VgESLF16cDdvuYCzGJtHntThLF4qIiULWeo,61
 airbyte_cdk/sources/types.py,sha256=1oerHQpidFrIluUhWsM3-4Xst4wwUDqzkgtgZY0jjZU,5485
 airbyte_cdk/sources/utils/__init__.py,sha256=TTN6VUxVy6Is8BhYQZR5pxJGQh8yH4duXh4O1TiMiEY,118
@@ -376,7 +376,7 @@ airbyte_cdk/sql/exceptions.py,sha256=7_-K2c_trPy6kM89I2pwsrnVEtXqOspd9Eqrzf2KD2A
 airbyte_cdk/sql/secrets.py,sha256=FRIafU5YbWzoK8jtAcfExwzMGdswMbs0OOo1O7Y5i-g,4345
 airbyte_cdk/sql/shared/__init__.py,sha256=-BU9zpzwx7JxSlS7EmFuGmdB9jK_QhhEJUe5dxErrDw,334
 airbyte_cdk/sql/shared/catalog_providers.py,sha256=lUYtJkkpkIcWUdp4rcL15GUrFy2b8Mf4sQjJu_YjC3M,5772
-airbyte_cdk/sql/shared/sql_processor.py,sha256=jR-hdLZsPf2sNBa_wvWKLvys8ZJ-SQCIiJS71Rszc-w,28215
+airbyte_cdk/sql/shared/sql_processor.py,sha256=C4HqaiTE8xAh7HN_nQtQ0d2Iy02GmHGw70Ko-PheEIk,28203
 airbyte_cdk/sql/types.py,sha256=XEIhRAo_ASd0kVLBkdLf5bHiRhNple-IJrC9TibcDdY,5880
 airbyte_cdk/test/__init__.py,sha256=f_XdkOg4_63QT2k3BbKY34209lppwgw-svzfZstQEq4,199
 airbyte_cdk/test/catalog_builder.py,sha256=2_tdbk5Jk8mUXtxAN3_vhGIzQawYOZ9kg39xgoJbtl4,2963
@@ -424,9 +424,9 @@ airbyte_cdk/utils/slice_hasher.py,sha256=EDxgROHDbfG-QKQb59m7h_7crN1tRiawdf5uU7G
 airbyte_cdk/utils/spec_schema_transformations.py,sha256=-5HTuNsnDBAhj-oLeQXwpTGA0HdcjFOf2zTEMUTTg_Y,816
 airbyte_cdk/utils/stream_status_utils.py,sha256=ZmBoiy5HVbUEHAMrUONxZvxnvfV9CesmQJLDTAIWnWw,1171
 airbyte_cdk/utils/traced_exception.py,sha256=bc5jMk8Z3AnSL-vqsgPCNgHzWqGTKZODHSg7VHiUyj0,6256
-airbyte_cdk-6.60.0.post29.dev16508987644.dist-info/LICENSE.txt,sha256=Wfe61S4BaGPj404v8lrAbvhjYR68SHlkzeYrg3_bbuM,1051
-airbyte_cdk-6.60.0.post29.dev16508987644.dist-info/LICENSE_SHORT,sha256=aqF6D1NcESmpn-cqsxBtszTEnHKnlsp8L4x9wAh3Nxg,55
-airbyte_cdk-6.60.0.post29.dev16508987644.dist-info/METADATA,sha256=jGMcrWLcyl7LPj9lvlJq0ck7t3kl3JY0ylkv599q7Ns,6544
-airbyte_cdk-6.60.0.post29.dev16508987644.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-airbyte_cdk-6.60.0.post29.dev16508987644.dist-info/entry_points.txt,sha256=AKWbEkHfpzzk9nF9tqBUaw1MbvTM4mGtEzmZQm0ZWvM,139
-airbyte_cdk-6.60.0.post29.dev16508987644.dist-info/RECORD,,
+airbyte_cdk-6.60.0.post32.dev16509172620.dist-info/LICENSE.txt,sha256=Wfe61S4BaGPj404v8lrAbvhjYR68SHlkzeYrg3_bbuM,1051
+airbyte_cdk-6.60.0.post32.dev16509172620.dist-info/LICENSE_SHORT,sha256=aqF6D1NcESmpn-cqsxBtszTEnHKnlsp8L4x9wAh3Nxg,55
+airbyte_cdk-6.60.0.post32.dev16509172620.dist-info/METADATA,sha256=uCfGPS4V3KK_Ki1pzfHZ1mln2hNeBWrGJx45cL7PcFc,6544
+airbyte_cdk-6.60.0.post32.dev16509172620.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+airbyte_cdk-6.60.0.post32.dev16509172620.dist-info/entry_points.txt,sha256=AKWbEkHfpzzk9nF9tqBUaw1MbvTM4mGtEzmZQm0ZWvM,139
+airbyte_cdk-6.60.0.post32.dev16509172620.dist-info/RECORD,,