atlan-application-sdk 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

application_sdk/services/objectstore.py

@@ -469,9 +469,35 @@ class ObjectStore:
             Exception: If there's an error with the Dapr binding operation.
         """
         try:
-            with DaprClient(
-                max_grpc_message_length=DAPR_MAX_GRPC_MESSAGE_LENGTH
-            ) as client:
+            # Calculate data size (handle both bytes and str)
+            if isinstance(data, bytes):
+                data_size = len(data)
+            elif isinstance(data, str):
+                data_size = len(data.encode("utf-8"))
+            else:
+                data_size = 0
+
+            # Check if data size exceeds DAPR limit and log warning
+            if data_size > DAPR_MAX_GRPC_MESSAGE_LENGTH:
+                # gRPC adds overhead (headers, metadata, etc.) to messages
+                # Add a buffer of 5% or at least 1KB to account for this overhead
+                grpc_overhead_buffer = max(int(data_size * 0.05), 1024)
+                required_max_length = data_size + grpc_overhead_buffer
+                logger.warning(
+                    f"Data size ({data_size:,} bytes / {data_size / (1024 * 1024):.2f}MB) "
+                    f"exceeds DAPR_MAX_GRPC_MESSAGE_LENGTH ({DAPR_MAX_GRPC_MESSAGE_LENGTH:,} bytes / "
+                    f"{DAPR_MAX_GRPC_MESSAGE_LENGTH / (1024 * 1024):.2f}MB). "
+                    f"Increasing max_grpc_message_length to {required_max_length:,} bytes "
+                    f"(data size + {grpc_overhead_buffer:,} bytes overhead buffer). "
+                    f"This may cause issues with Dapr gRPC communication."
+                )
+                # Set max_grpc_message_length to accommodate the data size plus gRPC overhead
+                max_message_length = required_max_length
+            else:
+                # Data is within limit, use default DAPR limit
+                max_message_length = DAPR_MAX_GRPC_MESSAGE_LENGTH
+
+            with DaprClient(max_grpc_message_length=max_message_length) as client:
                 response = client.invoke_binding(
                     binding_name=store_name,
                     operation=operation,

application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "1.0.3"
+__version__ = "1.0.4"

{atlan_application_sdk-1.0.3.dist-info → atlan_application_sdk-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 1.0.3
+Version: 1.0.4
 Summary: Atlan Application SDK is a Python library for developing applications on the Atlan Platform
 Project-URL: Repository, https://github.com/atlanhq/application-sdk
 Project-URL: Documentation, https://github.com/atlanhq/application-sdk/README.md

{atlan_application_sdk-1.0.3.dist-info → atlan_application_sdk-1.0.4.dist-info}/RECORD

@@ -1,6 +1,6 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
 application_sdk/constants.py,sha256=S3I_WUGFbmAPH5GTtoTKD5rxILGevkZ219zhctLQles,11568
-application_sdk/version.py,sha256=CRNCKWz-am2hAz_MLElN2Zj7G_IhrNsZx4EtpAxsavI,84
+application_sdk/version.py,sha256=cCvhI08C_P1mwYdUOoxxgWbEkuguvfFnVYGhRHHF5p8,84
 application_sdk/worker.py,sha256=D3-wtfGv1DLFKi1YSaE3jTcX66eC00N6RwtBu9RkgNc,7555
 application_sdk/activities/__init__.py,sha256=9nkd4FVaAJ5Qj5bF6aeMz0--0U3Ul3uRZsN2c3iDxg4,11339
 application_sdk/activities/lock_management.py,sha256=6Wdf3jMKitoarHQP91PIJOoGFz4aaOLS_40c7n1yAOA,3902
@@ -97,7 +97,7 @@ application_sdk/server/mcp/server.py,sha256=HG8tFmcc-f9Wj3vZzs2oRoNJzN1s5hwjnKyk
 application_sdk/services/__init__.py,sha256=H-5HZEPdr53MUfAggyHqHhRXDRLZFZsxvJgWbr257Ds,465
 application_sdk/services/atlan_storage.py,sha256=TKzXxu0yXeUcmZehwp8PcnQTC4A9w9RlZ0Fl-Xp1bLE,8509
 application_sdk/services/eventstore.py,sha256=X03JzodKByXh8w8nOl658rnnZfMFTj0IkmiLVbd6IN8,6729
-application_sdk/services/objectstore.py,sha256=Xvm4y8SZN5MYIdUL0-vPhW1EUnlVYgwp9XESrnXzgF8,17656
+application_sdk/services/objectstore.py,sha256=2fMrVuXsPK04CmbAnHX-hicBDSgW_zEMJyu2v8XEjLg,19200
 application_sdk/services/secretstore.py,sha256=Jd2gYyBcF31x8Hs8d5J93SWBXBdt6ULGvSk-Gfxb8Dw,19072
 application_sdk/services/statestore.py,sha256=3-afiM3Vsoe1XDYRokdGTB5I5CwOKyieuX5RwIZf77o,9413
 application_sdk/test_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -157,8 +157,8 @@ application_sdk/workflows/metadata_extraction/__init__.py,sha256=jHUe_ZBQ66jx8bg
 application_sdk/workflows/metadata_extraction/sql.py,sha256=6ZaVt84n-8U2ZvR9GR7uIJKv5v8CuyQjhlnoRJvDszc,12435
 application_sdk/workflows/query_extraction/__init__.py,sha256=n066_CX5RpJz6DIxGMkKS3eGSRg03ilaCtsqfJWQb7Q,117
 application_sdk/workflows/query_extraction/sql.py,sha256=kT_JQkLCRZ44ZpaC4QvPL6DxnRIIVh8gYHLqRbMI-hA,4826
-atlan_application_sdk-1.0.3.dist-info/METADATA,sha256=lP5LvhsVgUCwD-AJCDZgGoye3Sk7j4Z1sQuakUDCLQk,5913
-atlan_application_sdk-1.0.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-atlan_application_sdk-1.0.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-1.0.3.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-1.0.3.dist-info/RECORD,,
+atlan_application_sdk-1.0.4.dist-info/METADATA,sha256=aVhhJBCcV4bD4zE-zpKTZtAT4jPZN4I2Zj0EnanKlQY,5913
+atlan_application_sdk-1.0.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+atlan_application_sdk-1.0.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-1.0.4.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-1.0.4.dist-info/RECORD,,