Merge 'integration_2025-06-19_956583143170' into 'master'

merge branch integration_2025-06-19_956583143170 into master

See merge request: !679

Author: volc-sdk-team

meta.json

@@ -1,4 +1,4 @@
 {
-	"lasted": "3.0.14",
-	"meta_commit": "64d939687708ac4d90f6aeeab91c8c32179c176c"
+	"lasted": "3.0.15",
+	"meta_commit": "70254a5aacc72701cd24a00d3b0b9be1d11b2698"
 }

setup.py

@@ -3,7 +3,7 @@
 from setuptools import setup, find_packages  # noqa: H301
 
 NAME = "volcengine-python-sdk"
-VERSION = "3.0.14"
+VERSION = "3.0.15"
 # To install the library, run the following
 #
 # python setup.py install
@@ -29,7 +29,7 @@
             "httpx>=0.23.0, <1",
             "anyio>=3.5.0, <5",
             "cached-property; python_version < '3.8'",
-            "cryptography>=43.0.3, <43.0.4"
+            "cryptography>=42.0.0"
         ]
     },
 )

volcenginesdkarkruntime/_client.py

@@ -245,6 +245,13 @@ def _get_endpoint_sts_token(self, endpoint_id: str):
             self._sts_token_manager = StsTokenManager(self.ak, self.sk, self.region)
         return self._sts_token_manager.get(endpoint_id)
 
+    def _get_bot_sts_token(self, bot_id: str):
+        if self._sts_token_manager is None:
+            if self.ak is None or self.sk is None:
+                raise ArkAPIError("must set ak and sk before get endpoint token.")
+            self._sts_token_manager = StsTokenManager(self.ak, self.sk, self.region)
+        return self._sts_token_manager.get(bot_id, resource_type="bot")
+
     def _get_endpoint_certificate(self, endpoint_id: str) -> key_agreement_client:
         if self._certificate_manager is None:
             cert_path = os.environ.get("E2E_CERTIFICATE_PATH")

volcenginesdkarkruntime/_utils/_key_agreement.py

@@ -69,13 +69,13 @@ def marshal_cryptography_pub_key(key) -> bytes:
 class key_agreement_client:
     def __init__(self, certificate_pem_string: str) -> None:
         """Load cert and extract public key"""
-        __fixed_version__ = "43.0.3"  # version check
+        __fixed_version__ = "42.0.0"  # version check
         from cryptography import __version__
 
-        if __version__ != __fixed_version__:
+        if __version__ < __fixed_version__:
             raise Exception(
-                "The cryptography package of Ark SDK only supports version {}, "
-                "please install the cryptography package by using pip install cryptography=={}".format(
+                "The cryptography package of Ark SDK only supports versions after {}, "
+                'please install the cryptography package by using pip install "cryptography>={}"'.format(
                     __fixed_version__, __fixed_version__
                 )
             )

volcenginesdkarkruntime/resources/batch_chat/completions.py

@@ -155,7 +155,10 @@ def _decrypt(
     ) -> ChatCompletion:
         if resp.choices is not None:
             for index, choice in enumerate(resp.choices):
-                if choice.message is not None and choice.message.content is not None:
+                if (
+                    choice.message is not None and choice.finish_reason != 'content_filter'
+                    and choice.message.content is not None
+                ):
                     choice.message.content = aes_gcm_decrypt_base64_string(
                         key, nonce, choice.message.content
                     )
@@ -299,7 +302,10 @@ async def _decrypt(
     ) -> ChatCompletion:
         if resp.choices is not None:
             for index, choice in enumerate(resp.choices):
-                if choice.message is not None and choice.message.content is not None:
+                if (
+                    choice.message is not None and choice.finish_reason != 'content_filter'
+                    and choice.message.content is not None
+                ):
                     choice.message.content = aes_gcm_decrypt_base64_string(
                         key, nonce, choice.message.content
                     )

volcenginesdkarkruntime/resources/chat/completions.py

@@ -107,7 +107,10 @@ def _decrypt_chunk(
         for chunk in resp:
             if chunk.choices is not None:
                 for index, choice in enumerate(chunk.choices):
-                    if choice.delta is not None and choice.delta.content is not None:
+                    if (
+                        choice.delta is not None and choice.delta.content is not None
+                        and choice.finish_reason != 'content_filter'
+                    ):
                         choice.delta.content = aes_gcm_decrypt_base64_string(
                             key, nonce, choice.delta.content
                         )
@@ -124,7 +127,7 @@ def _decrypt(
             if resp.choices is not None:
                 for index, choice in enumerate(resp.choices):
                     if (
-                        choice.message is not None
+                        choice.message is not None and choice.finish_reason != 'content_filter'
                         and choice.message.content is not None
                     ):
                         choice.message.content = aes_gcm_decrypt_base64_string(
@@ -163,6 +166,7 @@ def create(
         tool_choice: ChatCompletionToolChoiceOptionParam | None = None,
         response_format: completion_create_params.ResponseFormat | None = None,
         thinking: completion_create_params.Thinking | None = None,
+        max_completion_tokens: Optional[int] | None = None,
         user: str | None = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
@@ -203,6 +207,7 @@ def create(
                 "tool_choice": tool_choice,
                 "response_format": response_format,
                 "thinking": thinking,
+                "max_completion_tokens": max_completion_tokens,
             },
             options=make_request_options(
                 extra_headers=extra_headers,
@@ -250,7 +255,10 @@ async def _decrypt_chunk(
         async for chunk in resp:
             if chunk.choices is not None:
                 for index, choice in enumerate(chunk.choices):
-                    if choice.delta is not None and choice.delta.content is not None:
+                    if (
+                        choice.delta is not None and choice.delta.content is not None
+                        and choice.finish_reason != 'content_filter'
+                    ):
                         choice.delta.content = aes_gcm_decrypt_base64_string(
                             key, nonce, choice.delta.content
                         )
@@ -267,7 +275,7 @@ async def _decrypt(
             if resp.choices is not None:
                 for index, choice in enumerate(resp.choices):
                     if (
-                        choice.message is not None
+                        choice.message is not None and choice.finish_reason != 'content_filter'
                         and choice.message.content is not None
                     ):
                         choice.message.content = aes_gcm_decrypt_base64_string(
@@ -307,6 +315,7 @@ async def create(
         tool_choice: ChatCompletionToolChoiceOptionParam | None = None,
         response_format: completion_create_params.ResponseFormat | None = None,
         thinking: completion_create_params.Thinking | None = None,
+        max_completion_tokens: Optional[int] | None = None,
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
@@ -346,6 +355,7 @@ async def create(
                 "tool_choice": tool_choice,
                 "response_format": response_format,
                 "thinking": thinking,
+                "max_completion_tokens": max_completion_tokens,
             },
             options=make_request_options(
                 extra_headers=extra_headers,

volcenginesdkcore/api_client.py

@@ -63,7 +63,7 @@ def __init__(self, configuration=None, header_name=None, header_value=None,
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'volcstack-python-sdk/3.0.14'
+        self.user_agent = 'volcstack-python-sdk/3.0.15'
         self.client_side_validation = configuration.client_side_validation
 
         self.interceptor_chain = InterceptorChain()

volcenginesdkcore/configuration.py

@@ -257,5 +257,5 @@ def to_debug_report(self):
                "OS: {env}\n" \
                "Python Version: {pyversion}\n" \
                "Version of the API: 0.1.0\n" \
-               "SDK Package Version: 3.0.14".\
+               "SDK Package Version: 3.0.15".\
             format(env=sys.platform, pyversion=sys.version)

volcenginesdkdbw/__init__.py

@@ -0,0 +1,33 @@
+# coding: utf-8
+
+# flake8: noqa
+
+"""
+    dbw
+
+    No description provided (generated by Swagger Codegen https://github.com/swagger-api/swagger-codegen)  # noqa: E501
+
+    OpenAPI spec version: common-version
+    
+    Generated by: https://github.com/swagger-api/swagger-codegen.git
+"""
+
+
+from __future__ import absolute_import
+
+# import apis into sdk package
+from volcenginesdkdbw.api.dbw_api import DBWApi
+
+# import models into sdk package
+from volcenginesdkdbw.models.describe_audit_log_config_request import DescribeAuditLogConfigRequest
+from volcenginesdkdbw.models.describe_audit_log_config_response import DescribeAuditLogConfigResponse
+from volcenginesdkdbw.models.describe_audit_log_detail_request import DescribeAuditLogDetailRequest
+from volcenginesdkdbw.models.describe_audit_log_detail_response import DescribeAuditLogDetailResponse
+from volcenginesdkdbw.models.describe_audit_log_detail_row_for_describe_audit_log_detail_output import DescribeAuditLogDetailRowForDescribeAuditLogDetailOutput
+from volcenginesdkdbw.models.describe_slow_logs_request import DescribeSlowLogsRequest
+from volcenginesdkdbw.models.describe_slow_logs_response import DescribeSlowLogsResponse
+from volcenginesdkdbw.models.modify_audit_log_config_request import ModifyAuditLogConfigRequest
+from volcenginesdkdbw.models.modify_audit_log_config_response import ModifyAuditLogConfigResponse
+from volcenginesdkdbw.models.search_param_for_describe_audit_log_detail_input import SearchParamForDescribeAuditLogDetailInput
+from volcenginesdkdbw.models.search_param_for_describe_slow_logs_input import SearchParamForDescribeSlowLogsInput
+from volcenginesdkdbw.models.slow_log_for_describe_slow_logs_output import SlowLogForDescribeSlowLogsOutput

volcenginesdkdbw/api/__init__.py

@@ -0,0 +1,6 @@
+from __future__ import absolute_import
+
+# flake8: noqa
+
+# import apis into api package
+from volcenginesdkdbw.api.dbw_api import DBWApi