From d9b25f114ce5fef3266f26c89366cb4638ed4125 Mon Sep 17 00:00:00 2001 From: Tyler Lee Date: Tue, 27 Jan 2026 20:02:51 +0800 Subject: [PATCH] Update encryption.py Add support to enable VOLC_ARK_ENCRYPTION via extra_headers --- volcenginesdkarkruntime/resources/encryption.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/volcenginesdkarkruntime/resources/encryption.py b/volcenginesdkarkruntime/resources/encryption.py index 314cc554..6780b93e 100644 --- a/volcenginesdkarkruntime/resources/encryption.py +++ b/volcenginesdkarkruntime/resources/encryption.py @@ -162,7 +162,7 @@ def _content_encryption(args, kwargs): lambda x: client.encrypt_string_with_key(_crypto_key, _crypto_nonce, x), ) info = {"ExpireTime": client.get_cert_expiration_time()} - if os.environ.get("VOLC_ARK_ENCRYPTION") == "AICC": + if os.environ.get("VOLC_ARK_ENCRYPTION") == "AICC" or extra_headers.get("VOLC_ARK_ENCRYPTION", None) == "true": ring_id, key_id = client.get_cert_ring_key_id() info["Version"] = "AICCv0.1" info["KeyID"] = key_id