volcengine-python-sdk 4.0.16__py2.py3-none-any.whl → 4.0.17__py2.py3-none-any.whl

{volcengine_python_sdk-4.0.16.dist-info → volcengine_python_sdk-4.0.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: volcengine-python-sdk
-Version: 4.0.16
+Version: 4.0.17
 Summary: Volcengine SDK for Python
 Home-page: https://github.com/volcengine/volcengine-python-sdk
 License: Apache License 2.0

{volcengine_python_sdk-4.0.16.dist-info → volcengine_python_sdk-4.0.17.dist-info}/RECORD

@@ -3213,8 +3213,8 @@ volcenginesdkcloudtrail20180101/models/trail_for_describe_trails_output.py,sha25
 volcenginesdkcloudtrail20180101/models/update_trail_request.py,sha256=B3ekQIluA7_EJS-lDVFfS9opp_Q0AA85o7feAxhRCuQ,12596
 volcenginesdkcloudtrail20180101/models/update_trail_response.py,sha256=Gw7FcSWqHtttRxkuWdFuh9A69ZHcwHWEf2K8S9AfWQQ,10744
 volcenginesdkcore/__init__.py,sha256=hXXJZvkXO1--TZxyZTuFowfJp0BoBo6c3KYbGNuVc9k,239
-volcenginesdkcore/api_client.py,sha256=t9XrPgW0WDXXlXMXMjQCVjK_Q7rfjhhK-TaYoDF74d0,15114
-volcenginesdkcore/configuration.py,sha256=oz7aMZf8pxT6kPVbPcKEOd5Hm7tVSSz4aC0LeE-6nyI,10769
+volcenginesdkcore/api_client.py,sha256=isLSgMUf3919HNcAi2kFmuZO11w7ygoTygukB7STP9c,15114
+volcenginesdkcore/configuration.py,sha256=CnGmr3CkQ_gp1l-m130mQqzd7pRqA1clUmWG8ENl2I0,10769
 volcenginesdkcore/flatten.py,sha256=g3r61JS_AO7WV6ClRDkXgtnVXcw3c7tbZjeLAJxkSLc,3811
 volcenginesdkcore/metadata.py,sha256=uEwumzeIwgfu3_QBUk6BU_MQg_snF92Kke1J2L5TcGc,3419
 volcenginesdkcore/model.py,sha256=f3cvQ6QiQfecClucu-_-l5xVI3W5_vF-EKkjwZu7Vjc,177
@@ -3226,7 +3226,7 @@ volcenginesdkcore/auth/credential.py,sha256=LhD66-T1w_zhCXMht_JpMXO2pX05_r5dwmyq
 volcenginesdkcore/auth/providers/__init__.py,sha256=ezMR8ym3WLs0LdXFioeWaU1gLljgaH9_OCevWDWx89E,102
 volcenginesdkcore/auth/providers/provider.py,sha256=tkcsu66gX3F8KXcKxLLQm27mOhaAcnqnXGUmgp0Mseo,533
 volcenginesdkcore/auth/providers/static_provider.py,sha256=OGbYQ80a8T9FdR7EjpZ6s8h7_xzYO7OstWizFOiFeuA,548
-volcenginesdkcore/auth/providers/sts_provider.py,sha256=d5v0zLc-ZH1AX0f_B4zuqaJYAKgkuND-_nSLOe-v7aI,3229
+volcenginesdkcore/auth/providers/sts_provider.py,sha256=rgGxyBk3W0dbesShpDSKq1xwrRD8aOdDXN9Junc19Nc,3242
 volcenginesdkcore/endpoint/__init__.py,sha256=K7dXFGGh0QYiALOLgtPxboGYzwCJB5IHseGSQPObZdQ,117
 volcenginesdkcore/endpoint/endpoint_provider.py,sha256=9ajUjf2N2NmThuWrMm5QgLtmh8YbPas5TdvkAZYQ9lM,330
 volcenginesdkcore/endpoint/providers/__init__.py,sha256=FmxBi_siUNTZod9GFOk5VG-UVYT7daNt4hZ2zifqYY8,76
@@ -6725,6 +6725,11 @@ volcenginesdklivesaas20230801/models/whitelist_viewing_restriction_for_update_vi
 volcenginesdklivesaas20230801/models/whitelist_viewing_restriction_for_update_viewing_restriction_output.py,sha256=QuKWHvZ-vHo0-MxiPWyhOVW29N7rVpTW-rbsCG3vyGE,6689
 volcenginesdklivesaas20230801/models/zoom_config_for_get_vod_player_config_output.py,sha256=H9oQplsZBAB68cH70Ja84dq0oV8vT6K3zZbqSSyT55M,3655
 volcenginesdklivesaas20230801/models/zoom_config_for_update_vod_player_config_input.py,sha256=-TMwP9NipvXdwgo1JNDDyq_HsVNNEnN3DFGq-v7iwUQ,3673
+volcenginesdkllmshield/__init__.py,sha256=0tXID7HFo_matygEeWBXQUeB-vc1BBuW36ImIiwqceE,1738
+volcenginesdkllmshield/api/__init__.py,sha256=di_9IIKQ0HCuzkGrTyZ6h38iP7y1xelH8_IJbKi4aWA,71
+volcenginesdkllmshield/api/llm_shield_sdk_v2.py,sha256=t3unTS8U8I5J_L09jVX4ESX8wQNinXSZ-XriJ2VFl_Q,16661
+volcenginesdkllmshield/models/__init__.py,sha256=OpEdwHNrJ0rcWEdVt0Km58pboIBWP4WYA8zv2LhzzWc,259
+volcenginesdkllmshield/models/llm_shield_sign.py,sha256=JmmaxTpGJR-5QltTHFffe8CXUv7rrCiWlx8jzt78r80,5640
 volcenginesdkmcdn/__init__.py,sha256=T6n7RhmCZfockyBT_7tbkB5Zb7moirVG3hC_l6lgUbw,19051
 volcenginesdkmcdn/api/__init__.py,sha256=1vA5hPYIN4Hv1_IMCWN9GHHaSWQrgd6tSGDmcUDeyXg,138
 volcenginesdkmcdn/api/mcdn_api.py,sha256=LbNhFv3JE0DcJFGtIx_79JYWyA1p53n7-F2ozW6YUVc,110963
@@ -13663,14 +13668,14 @@ volcenginesdkwaf/models/vul_for_create_domain_output.py,sha256=2iStwOpXqnVreaLjo
 volcenginesdkwaf/models/waf_action_list_for_get_tls_config_output.py,sha256=cdZlna1vQ3QG-MUGyOyjaycO0WHAT-gXpydZ_PJ3MhI,11680
 volcenginesdkwaf/models/waf_action_list_for_modify_tls_config_input.py,sha256=DMZcjsl_w6q9n0laMPBaLojH8jxCQb3V0dd6md4n9Mg,11786
 volcenginesdkwaf/models/web_backdoor_for_get_vulnerability_config_output.py,sha256=QiIjPj4k2ILlHPlP1aXEhv-aGzFngcp6-vhwwHzPiXc,7549
-volcenginesdkwafruntime/__init__.py,sha256=oRCJ_LTUCaMoR7HlJqM6xnXbr4rb5_luzeM0wsjf63k,517
+volcenginesdkwafruntime/__init__.py,sha256=NqvhBz6iQAKGphWWkJVQ4eVwsfeyY7cf9wNrV1hJaOA,793
 volcenginesdkwafruntime/api/__init__.py,sha256=di_9IIKQ0HCuzkGrTyZ6h38iP7y1xelH8_IJbKi4aWA,71
-volcenginesdkwafruntime/api/waf_runtime_api.py,sha256=Ya_maRHEfUUpkIVY4aU09PsVuOyewvAraYGEMPxqUCc,3776
+volcenginesdkwafruntime/api/waf_runtime_api.py,sha256=eWgdzFcwflEkhkNrPzZL1mqSid7O1OWiwHGUyWJvZNA,4035
 volcenginesdkwafruntime/models/__init__.py,sha256=di_9IIKQ0HCuzkGrTyZ6h38iP7y1xelH8_IJbKi4aWA,71
-volcenginesdkwafruntime/models/llm_stream_session.py,sha256=bYCzl2GYXv2hS6J-yiEYjYHQpq4f7Vs-GslyuqL4bW4,1516
-volcengine_python_sdk-4.0.16.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-volcengine_python_sdk-4.0.16.dist-info/METADATA,sha256=wyVRZgGMzgzfrZtFrz5i52GcnxS26rzldOu8RQx6QK4,674
-volcengine_python_sdk-4.0.16.dist-info/NOTICE.md,sha256=dqWX0O4-gFqGLdHJsXAiF6Q8JHlu_3nFaQSmrMHzujM,254
-volcengine_python_sdk-4.0.16.dist-info/WHEEL,sha256=Kh9pAotZVRFj97E15yTA4iADqXdQfIVTHcNaZTjxeGM,110
-volcengine_python_sdk-4.0.16.dist-info/top_level.txt,sha256=ehvVeTUQW6mfpJtv8XAvo_bnmVbptrq2eqL1Iub3TBo,2195
-volcengine_python_sdk-4.0.16.dist-info/RECORD,,
+volcenginesdkwafruntime/models/llm_stream_session.py,sha256=U9cig3fuZRBT4Ov8_cUsWDKXcMLNjBDxKt-7IoewoUw,1589
+volcengine_python_sdk-4.0.17.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+volcengine_python_sdk-4.0.17.dist-info/METADATA,sha256=3LBnEVQcUSuvePfehA-e2yKQmYpb3cf8vYRA3SO707g,674
+volcengine_python_sdk-4.0.17.dist-info/NOTICE.md,sha256=dqWX0O4-gFqGLdHJsXAiF6Q8JHlu_3nFaQSmrMHzujM,254
+volcengine_python_sdk-4.0.17.dist-info/WHEEL,sha256=Kh9pAotZVRFj97E15yTA4iADqXdQfIVTHcNaZTjxeGM,110
+volcengine_python_sdk-4.0.17.dist-info/top_level.txt,sha256=ynjiHmtvNtMbJ9sneg1RSWBhKzwNfX5yGiVKTa6AXYA,2218
+volcengine_python_sdk-4.0.17.dist-info/RECORD,,

{volcengine_python_sdk-4.0.16.dist-info → volcengine_python_sdk-4.0.17.dist-info}/top_level.txt

@@ -50,6 +50,7 @@ volcenginesdkkafka
 volcenginesdkkms
 volcenginesdklivesaas
 volcenginesdklivesaas20230801
+volcenginesdkllmshield
 volcenginesdkmcdn
 volcenginesdkmcs
 volcenginesdkmlplatform20240701

volcenginesdkcore/api_client.py

@@ -67,7 +67,7 @@ class ApiClient(object):
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'volcstack-python-sdk/4.0.16'
+        self.user_agent = 'volcstack-python-sdk/4.0.17'
         self.client_side_validation = configuration.client_side_validation
 
         self.interceptor_chain = InterceptorChain()

volcenginesdkcore/auth/providers/sts_provider.py

@@ -3,7 +3,7 @@ import time
 import uuid
 from datetime import datetime
 
-import dateutil.parser
+import dateutil.parser, dateutil.tz
 
 from volcenginesdkcore import UniversalApi, UniversalInfo, ApiClient, Configuration
 from .provider import Provider, CredentialValue

volcenginesdkcore/configuration.py

@@ -267,7 +267,7 @@ class Configuration(six.with_metaclass(TypeWithDefault, object)):
                "OS: {env}\n" \
                "Python Version: {pyversion}\n" \
                "Version of the API: 0.1.0\n" \
-               "SDK Package Version: 4.0.16".\
+               "SDK Package Version: 4.0.17".\
             format(env=sys.platform, pyversion=sys.version)
 
     @property

volcenginesdkllmshield/__init__.py

@@ -0,0 +1,82 @@
+# coding: utf-8
+
+# flake8: noqa
+
+"""
+    llmshield
+
+    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
+
+# 导入API模块
+from volcenginesdkllmshield.api.llm_shield_sdk_v2 import ClientV2
+from volcenginesdkllmshield.api.llm_shield_sdk_v2 import (
+    ContentTypeV2,
+    DecisionTypeV2,
+    UserAction,
+    MatchSource,
+    MessageV2,
+    ModerateV2Request,
+    RiskMatchV2,
+    PermitMatchV2,
+    RiskV2,
+    RiskInfoV2,
+    PermitV2,
+    PermitInfoV2,
+    BlockDetailV2,
+    ReplaceDetailV2,
+    DecisionDetailV2,
+    DecisionV2,
+    ModerateV2Result,
+    ErrorInfo,
+    ResponseMetadata,
+    ModerateV2Response,
+    ModerateV2StreamSession,
+    GenerateStreamV2Request,
+    GenerateStreamV2Response,
+    GenerateSummarizeV2,
+    GenerateStreamResult,
+    GenerateStreamV2ResponseData,
+    CustomJSONEncoder
+)
+
+
+__all__ = [
+    # API客户端
+    'ClientV2',
+    # 常量类
+    'ContentTypeV2',
+    'DecisionTypeV2',
+    'UserAction',
+    'MatchSource',
+    # 数据模型
+    'MessageV2',
+    'ModerateV2Request',
+    'RiskMatchV2',
+    'PermitMatchV2',
+    'RiskV2',
+    'RiskInfoV2',
+    'PermitV2',
+    'PermitInfoV2',
+    'BlockDetailV2',
+    'ReplaceDetailV2',
+    'DecisionDetailV2',
+    'DecisionV2',
+    'ModerateV2Result',
+    'ErrorInfo',
+    'ResponseMetadata',
+    'ModerateV2Response',
+    'ModerateV2StreamSession',
+    'GenerateStreamV2Request',
+    'GenerateStreamV2Response',
+    'GenerateSummarizeV2',
+    'GenerateStreamResult',
+    'GenerateStreamV2ResponseData',
+    'CustomJSONEncoder',
+]

volcenginesdkllmshield/api/__init__.py

@@ -0,0 +1,4 @@
+# coding: utf-8
+# flake8: noqa
+
+from __future__ import absolute_import

volcenginesdkllmshield/api/llm_shield_sdk_v2.py

@@ -0,0 +1,470 @@
+from pydantic import BaseModel, field_validator, Field
+from typing import List, Optional, Any, Union
+from datetime import datetime, date
+from uuid import UUID
+import requests
+import json
+
+from ..models.llm_shield_sign import request_sign, Version
+
+LLM_STREAM_SEND_BASE_WINDOW_V2 = 10
+LLM_STREAM_SEND_EXPONENT_V2 = 2
+
+
+# 定义内容类型常量
+class ContentTypeV2:
+    TEXT = 1
+    AUDIO = 2
+    IMAGE = 3
+    VIDEO = 4
+
+
+# 定义决策类型常量
+class DecisionTypeV2:
+    PASS = 1
+    BLOCK = 2
+    MARK = 3
+    REPLACE = 4
+    OPTIMIZE = 5
+
+
+# 定义用户操作常量
+class UserAction:
+    PASS = 1
+    BLOCK = 2
+    MARK = 3
+    REPLACE = 4
+
+
+# 定义匹配来源常量
+class MatchSource:
+    UNKNOWN = 0
+    GLOBAL_CONTENTLIB = 1
+    ADMIN_CONTENTLIB = 2
+    USER_CONTENTLIB = 3
+
+
+# 定义消息结构体
+class MessageV2(BaseModel):
+    role: str = Field("", alias="Role")
+    content: str = Field("", alias="Content")
+    content_type: int = Field(ContentTypeV2.TEXT, alias="ContentType")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义审核请求结构体
+class ModerateV2Request(BaseModel):
+    message: MessageV2 = Field(None, alias="Message")
+    msg_id: str = Field("", alias="MsgID")
+    use_stream: int = Field(0, alias="UseStream")
+    scene: str = Field("", alias="Scene")
+    history: List[MessageV2] = Field([], alias="History")
+
+    class Config:
+        populate_by_name = True
+
+    # 深拷贝构造方法：通过 Pydantic 序列化/反序列化实现
+    def __init__(self, other=None, **data):
+        # 如果传入了其他 ModerateV2Request 实例，则进行深拷贝
+        if other is not None and isinstance(other, ModerateV2Request):
+            # 1. 将其他实例序列化为字典（包含嵌套对象）
+            other_dict = other.model_dump(by_alias=True)  # 使用 alias 键名
+            # 2. 用序列化后的字典初始化当前实例（实现深拷贝）
+            super().__init__(**other_dict)
+        else:
+            # 正常初始化逻辑
+            super().__init__(**data)
+
+
+# 定义风险匹配结构体
+class RiskMatchV2(BaseModel):
+    word: str = Field("", alias="Word")
+    action: Optional[int] = Field(None, alias="Action")
+    source: Optional[int] = Field(None, alias="Source")
+    rule_id: Optional[Any] = Field(None, alias="RuleID")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义放行匹配结构体
+class PermitMatchV2(BaseModel):
+    word: str = Field("", alias="Word")
+    action: Optional[int] = Field(None, alias="Action")
+    source: Optional[int] = Field(None, alias="Source")
+    rule_id: Optional[Any] = Field(None, alias="RuleID")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义风险结构体
+class RiskV2(BaseModel):
+    category: str = Field("", alias="Category")
+    label: str = Field("", alias="Label")
+    prob: Optional[float] = Field(None, alias="Prob")
+    matches: List[RiskMatchV2] = Field([], alias="Matches")
+
+    @field_validator('matches', mode="before")
+    def convert_risk_matches_none_to_list(cls, value):
+        return [] if value is None else value
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义风险信息结构体 - 添加 None 转换
+class RiskInfoV2(BaseModel):
+    risks: List[RiskV2] = Field([], alias="Risks")
+
+    @field_validator('risks', mode="before")
+    def convert_none_to_list(cls, value):
+        return [] if value is None else value
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义放行结构体
+class PermitV2(BaseModel):
+    category: str = Field("", alias="Category")
+    label: str = Field("", alias="Label")
+    prob: Optional[float] = Field(None, alias="Prob")
+    matches: List[PermitMatchV2] = Field([], alias="Matches")
+
+    @field_validator('matches', mode="before")
+    def convert_permit_matches_none_to_list(cls, value):
+        return [] if value is None else value
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义放行信息结构体 - 添加 None 转换
+class PermitInfoV2(BaseModel):
+    permits: List[PermitV2] = Field([], alias="Permits")
+
+    @field_validator('permits', mode="before")
+    def convert_none_to_list(cls, value):
+        return [] if value is None else value
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义拦截详情结构体
+class BlockDetailV2(BaseModel):
+    class Config:
+        extra = "forbid"
+
+
+# 定义替换详情结构体
+class ReplaceDetailV2(BaseModel):
+    replacement: Optional[MessageV2] = Field(None, alias="Replacement")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义决策详情结构体
+class DecisionDetailV2(BaseModel):
+    block_detail: Optional[BlockDetailV2] = Field(None, alias="BlockDetail")
+    replace_detail: Optional[ReplaceDetailV2] = Field(None, alias="ReplaceDetail")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义决策结构体
+class DecisionV2(BaseModel):
+    decision_type: int = Field(0, alias="DecisionType")
+    decision_detail: DecisionDetailV2 = Field(default_factory=DecisionDetailV2, alias="DecisionDetail")
+    decision_strategy_id: Optional[str] = Field(None, alias="DecisionStrategyID")
+    hit_strategy_ids: List[str] = Field([], alias="HitStrategyIDs")
+
+    @field_validator('hit_strategy_ids', mode="before")
+    def convert_hit_strategies_none_to_list(cls, value):
+        return [] if value is None else value
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义审核结果结构体
+class ModerateV2Result(BaseModel):
+    msg_id: str = Field("", alias="MsgID")
+    risk_info: RiskInfoV2 = Field(default_factory=RiskInfoV2, alias="RiskInfo")
+    decision: DecisionV2 = Field(default_factory=DecisionV2, alias="Decision")
+    permit_info: PermitInfoV2 = Field(default_factory=PermitInfoV2, alias="PermitInfo")
+    degraded: bool = Field(False, alias="Degraded")
+    degrade_reason: str = Field("", alias="DegradeReason")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义错误信息结构体
+class ErrorInfo(BaseModel):
+    code: str = Field("", alias="Code")
+    codeN: Union[int, str] = Field("", alias="CodeN")
+    message: str = Field("", alias="Message")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义响应元数据结构体
+class ResponseMetadata(BaseModel):
+    error: Union[ErrorInfo, None] = Field(default_factory=ErrorInfo, alias="Error")
+    requestId: str = Field(..., alias="RequestId")  # 添加requestId字段，映射自RequestId
+    service: Union[str, None] = Field(None, alias="Service")
+    action: Union[str, None] = Field(None, alias="Action")
+    version: Union[str, None] = Field(None, alias="Version")
+    region: Union[str, None] = Field(None, alias="Region")
+
+    class Config:
+        populate_by_name = True
+        validate_by_name = True
+
+
+# 定义审核响应结构体
+class ModerateV2Response(BaseModel):
+    response_metadata: ResponseMetadata = Field(default_factory=ResponseMetadata, alias="ResponseMetadata")
+    result: Union[ModerateV2Result, None] = Field(default_factory=ModerateV2Result, alias="Result")
+
+    class Config:
+        populate_by_name = True
+
+
+class ModerateV2StreamSession:
+    """流式会话结构体，用于积累流式请求、存储未发送长度和默认响应体"""
+
+    def __init__(self):
+        # 用于积累流式的请求（初始为 None，对应 Go 中的指针）
+        self.request: Optional[ModerateV2Request] = None
+        # 未发送的长度（对应 Go 中的 StreamSendLen）
+        self.stream_send_len: int = 0
+        self.CurrentSendWindow = LLM_STREAM_SEND_BASE_WINDOW_V2
+        # 存储默认响应体（初始为 None，对应 Go 中的指针）
+        self.default_body: Optional[ModerateV2Response] = None
+
+
+class GenerateStreamV2Request(BaseModel):
+    """生成流V2版本的请求模型"""
+    msg_id: str = Field(..., alias="MsgID", description="消息ID，表示请求的唯一标识")
+
+    class Config:
+        populate_by_name = True
+        json_schema_extra = {
+            "validate": {"required": True}  # 对应Go中的validate:"required"
+        }
+
+
+class GenerateStreamV2Response(BaseModel):
+    def __init__(self, reader=None):
+        self.Reader = reader
+
+
+class GenerateSummarizeV2(BaseModel):
+    """生成过程的总结信息模型"""
+    token_cost: int = Field(0, alias="TokenCost", description="消耗的token数量")
+    time_cost_ms: int = Field(0, alias="TimeCostMS", description="消耗的时长（毫秒）")
+
+    class Config:
+        populate_by_name = True
+
+
+class GenerateStreamResult(BaseModel):
+    """生成流V2版本的结果模型"""
+    message: Optional[MessageV2] = Field(None, alias="Message", description="优化内容，isFinished为true时为空")
+    is_finished: bool = Field(False, alias="IsFinished", description="标识是否结束")
+
+    # summarize: Optional[GenerateSummarizeV2] = Field(None, alias="Summarize", description="总结信息，isFinished为true时有值")
+
+    class Config:
+        populate_by_name = True
+
+
+class GenerateStreamV2ResponseData(BaseModel):
+    """生成流V2版本的响应数据模型"""
+    response_metadata: ResponseMetadata = Field(..., alias="ResponseMetadata", description="响应元数据")
+    result: GenerateStreamResult = Field(..., alias="Result", description="生成流结果")
+
+    class Config:
+        populate_by_name = True
+
+
+# 定义客户端类
+class ClientV2:
+    def __init__(self, url: str, ak: str, sk: str, region: str, timeout: float):
+        self.url = url
+        self.ak = ak
+        self.sk = sk
+        self.region = region
+        self.http_client = requests.Session()
+        self.http_client.timeout = timeout
+
+    def Moderate(self, request: Optional[ModerateV2Request] = None) -> ModerateV2Response:
+        path = "/v2/moderate"
+        action = "Moderate"
+
+        if request is None:
+            request = ModerateV2Request()
+
+        request_body = request.model_dump_json(by_alias=True).encode("utf-8")
+
+        header = {
+        }
+
+        sign_header = request_sign(header, self.ak, self.sk, self.region, self.url, path, action, request_body)
+
+        try:
+            resp = self.http_client.post(
+                url=self.url + path + "?Action=" + action + "&Version=" + Version,
+                data=request_body,
+                headers=sign_header
+            )
+
+            response = ModerateV2Response.model_validate(resp.json())
+            return response
+
+        except requests.RequestException as e:
+            raise Exception(f"请求失败: {e}")
+        except Exception as e:
+            raise Exception(f"处理响应失败: {e}")
+
+    def ModerateStream(
+            self, request: ModerateV2Request, session: ModerateV2StreamSession
+    ) -> Optional[ModerateV2Response]:
+        """
+        处理流式审核请求
+        :param request: 当前流式请求片段（ModerateV2Request 类型）
+        :param session: 流式会话对象（ModerateV2StreamSession 类型）
+        :return: 审核响应（ModerateV2Response 类型）
+        """
+        # 1. 校验参数合法性
+        path = "/v2/moderate"
+        action = "Moderate"
+        if request is None:
+            request = ModerateV2Request()  # 初始化空请求
+
+        # 本接口仅支持流式调用（use_stream 不能为 0，且 session 不能为空）
+        if request.use_stream == 0 or session is None:
+            raise ValueError("use_stream cannot be 0, and session cannot be None")
+
+        is_first_request = (session.request is None)  # 判断是否为首次请求
+        is_last_request = (request.use_stream == 2)  # 判断是否为最后一次请求
+
+        # 2. 初始化或追加会话请求（深拷贝确保隔离）
+        if session.request is None:
+            # 首次请求：深拷贝初始请求到 session
+            session.request = ModerateV2Request(request)
+        else:
+            # 后续请求：追加当前请求内容到 session 积累的请求中
+            # 示例：追加 message.content（根据实际业务调整）
+            if request.message and request.message.content:
+                if session.request.message is None:
+                    session.request.message = MessageV2()
+                session.request.message.content += request.message.content
+                session.request.use_stream = request.use_stream
+        session.stream_send_len += len(request.message.content)
+
+        # 3. 判断是否需要发送请求到后端
+        # 只有当未检测长度 >= 10 或者是第一次或者是最后一次请求时，才发送请求
+        need_send_request = is_last_request or is_first_request or (
+                session.stream_send_len >= session.CurrentSendWindow)
+
+        # 如果不需要发送请求，直接返回上次的默认响应（如果有）
+        if not need_send_request:
+            return session.default_body
+        else:
+            session.CurrentSendWindow = session.CurrentSendWindow * LLM_STREAM_SEND_EXPONENT_V2
+
+        # 3. 序列化请求（使用 Pydantic 的 model_dump 方法）
+        try:
+            request_body = session.request.model_dump_json(by_alias=True).encode("utf-8")
+            # request_str = session.request.model_dump_json(by_alias=True)
+        except Exception as e:
+            raise IOError(f"Failed to serialize request: {str(e)}")
+        headers = {
+            # "Content-Type": "application/json",
+        }
+        sign_header = request_sign(headers, self.ak, self.sk, self.region, self.url, path, action, request_body)
+        try:
+            response = requests.post(
+                url=self.url + path + "?Action=" + action + "&Version=" + Version,
+                data=request_body,
+                headers=sign_header
+            )
+        except requests.exceptions.RequestException as e:
+            raise IOError(f"HTTP request failed: {str(e)}")
+
+        # 5. 解析响应
+        try:
+            response_data = json.loads(response.text)
+            moderate_response = ModerateV2Response(**response_data)
+        except Exception as e:
+            raise IOError(f"Failed to parse response: {str(e)}")
+
+        # 6. 更新会话状态
+        session.default_body = moderate_response  # 存储响应体
+        session.stream_send_len = 0  # 重置未发送长度（根据实际业务调整）
+        session.request.msg_id = moderate_response.result.msg_id
+
+        # 7. 若为最后一次流式请求（use_stream == 2），打印最终内容
+        if session.request.use_stream == 2:
+            final_content = session.request.message.content if (
+                    session.request.message and session.request.message.content) else ""
+            print(f"最终检测内容: {final_content}")
+
+        return moderate_response
+
+    def GenerateV2Stream(self, request):
+        path = "/v2/generate"
+        action = "Generate"
+        if request is None:
+            request = GenerateStreamV2Request()
+
+        # 将请求结构体序列化为 JSON
+        requestBody = request.model_dump_json(by_alias=True).encode("utf-8")
+
+        headers = {
+            # "Content-Type": "application/json",
+        }
+        try:
+            sign_header = request_sign(headers, self.ak, self.sk, self.region, self.url, path, action, requestBody)
+            # 发送 HTTP 请求
+            resp = self.http_client.post(url=self.url + path + "?Action=" + action + "&Version=" + Version,
+                                         data=requestBody, headers=sign_header, stream=True)
+            if resp.status_code != 200:
+                raise Exception("bad response code: %d" % resp.status_code)
+
+            for line in resp.iter_lines():
+                if line:
+                    line = line.decode('utf-8')
+                    if line.lstrip().startswith('data:'):
+                        sse_data = line[line.index(':') + 1:].strip()
+                        yield sse_data
+        except Exception as e:
+            return None, Exception("failed to send request: %s" % str(e))
+
+
+# 自定义 JSON 编码器
+class CustomJSONEncoder(json.JSONEncoder):
+    def default(self, obj):
+        # 处理datetime类型（如2023-10-01T12:00:00）
+        if isinstance(obj, datetime):
+            return obj.isoformat()
+        # 处理date类型（如2023-10-01）
+        elif isinstance(obj, date):
+            return obj.isoformat()
+        # 处理UUID类型
+        elif isinstance(obj, UUID):
+            return str(obj)
+        # 处理其他未知的自定义类型（返回类型信息便于调试）
+        elif hasattr(obj, '__dict__'):
+            return obj.__dict__  # 返回对象的属性字典
+        # 调用默认处理（会抛出TypeError）
+        return super().default(obj)

volcenginesdkllmshield/models/__init__.py

@@ -0,0 +1,20 @@
+# coding:utf-8
+
+from __future__ import absolute_import
+
+from .llm_shield_sign import (
+    request_sign,
+    Service,
+    Version,
+    ContentType,
+    Method
+)
+
+__all__ = [
+    'request_sign',
+    'Service',
+    'Version',
+    'ContentType',
+    'Method'
+]
+

volcenginesdkllmshield/models/llm_shield_sign.py

@@ -0,0 +1,159 @@
+# coding:utf-8
+"""
+Copyright (year) Beijing Volcano Engine Technology Ltd.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+     http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+import datetime
+import hashlib
+import hmac
+from urllib.parse import quote, urlparse
+import requests
+
+# 以下参数视服务不同而不同，一个服务内通常是一致的
+Service = "llmshield"
+Version = "2025-08-31"
+ContentType = "application/json"
+Method = "POST"
+
+
+def norm_query(params):
+    query = ""
+    for key in sorted(params.keys()):
+        if type(params[key]) == list:
+            for k in params[key]:
+                query = (
+                        query + quote(key, safe="-_.~") + "=" + quote(k, safe="-_.~") + "&"
+                )
+        else:
+            query = (query + quote(key, safe="-_.~") + "=" + quote(params[key], safe="-_.~") + "&")
+    query = query[:-1]
+    return query.replace("+", "%20")
+
+
+# 第一步：准备辅助函数。
+# sha256 非对称加密
+def hmac_sha256(key: bytes, content: str):
+    return hmac.new(key, content.encode("utf-8"), hashlib.sha256).digest()
+
+
+# sha256 hash算法
+def hash_sha256(content: bytes):
+    return hashlib.sha256(content).hexdigest()
+
+
+# 第二步：签名请求函数
+def request_sign(header, ak, sk, region, url, path, action, body):
+    host = urlparse(url).netloc
+    date = utc_now()
+    # 第三步：创建身份证明。其中的 Service 和 Region 字段是固定的。ak 和 sk 分别代表
+    # AccessKeyID 和 SecretAccessKey。同时需要初始化签名结构体。一些签名计算时需要的属性也在这里处理。
+    # 初始化身份证明结构体
+    credential = {
+        "access_key_id": ak,
+        "secret_access_key": sk,
+        "service": Service,
+        "region": region,
+    }
+    # 初始化签名结构体
+    request_param = {
+        "body": body,
+        "host": host,
+        "path": path,
+        "method": Method,
+        "content_type": ContentType,
+        "date": date,
+        "query": {"Action": action, "Version": Version},
+    }
+    if body is None:
+        request_param["body"] = ""
+    # 第四步：接下来开始计算签名。在计算签名前，先准备好用于接收签算结果的 signResult 变量，并设置一些参数。
+    # 初始化签名结果的结构体
+    x_date = request_param["date"].strftime("%Y%m%dT%H%M%SZ")
+    short_x_date = x_date[:8]
+    x_content_sha256 = hash_sha256(request_param["body"])
+    sign_result = {
+        "Host": request_param["host"],
+        "X-Content-Sha256": x_content_sha256,
+        "X-Date": x_date,
+        "Content-Type": request_param["content_type"],
+    }
+    # 第五步：计算 Signature 签名。
+    signed_headers_str = ";".join(
+        ["content-type", "host", "x-content-sha256", "x-date"]
+    )
+    # signed_headers_str = signed_headers_str + ";x-security-token"
+    canonical_request_str = "\n".join(
+        [request_param["method"].upper(),
+         request_param["path"],
+         norm_query(request_param["query"]),
+         "\n".join(
+             [
+                 "content-type:" + request_param["content_type"],
+                 "host:" + request_param["host"],
+                 "x-content-sha256:" + x_content_sha256,
+                 "x-date:" + x_date,
+             ]
+         ),
+         "",
+         signed_headers_str,
+         x_content_sha256,
+         ]
+    )
+
+    # 打印正规化的请求用于调试比对
+    # print(canonical_request_str)
+    hashed_canonical_request = hash_sha256(canonical_request_str.encode("utf-8"))
+
+    # 打印hash值用于调试比对
+    # print(hashed_canonical_request)
+    credential_scope = "/".join([short_x_date, credential["region"], credential["service"], "request"])
+    string_to_sign = "\n".join(["HMAC-SHA256", x_date, credential_scope, hashed_canonical_request])
+
+    # 打印最终计算的签名字符串用于调试比对
+    # print(string_to_sign)
+    k_date = hmac_sha256(credential["secret_access_key"].encode("utf-8"), short_x_date)
+    k_region = hmac_sha256(k_date, credential["region"])
+    k_service = hmac_sha256(k_region, credential["service"])
+    k_signing = hmac_sha256(k_service, "request")
+    signature = hmac_sha256(k_signing, string_to_sign).hex()
+
+    sign_result["Authorization"] = "HMAC-SHA256 Credential={}, SignedHeaders={}, Signature={}".format(
+        credential["access_key_id"] + "/" + credential_scope,
+        signed_headers_str,
+        signature,
+    )
+    header = {**header, **sign_result, "X-Top-Service": Service, "X-Top-Region": region}
+    # header = {**header, **{"X-Security-Token": SessionToken}}
+    # 第六步：将 Signature 签名写入 HTTP Header 中，并发送 HTTP 请求。
+    return header
+
+
+# datetime.utcnow() 在 3.12+ 已经过期，使用如下方法兼容
+def utc_now():
+    try:
+        from datetime import timezone
+        return datetime.datetime.now(timezone.utc)
+    except ImportError:
+        class UTC(datetime.tzinfo):
+            def utcoffset(self, dt):
+                return datetime.timedelta(0)
+
+            def tzname(self, dt):
+                return "UTC"
+
+            def dst(self, dt):
+                return datetime.timedelta(0)
+
+        return datetime.datetime.now(UTC())

volcenginesdkwafruntime/__init__.py

@@ -15,5 +15,7 @@
 from volcenginesdkwaf import *
 from volcenginesdkwafruntime.api.waf_runtime_api import  WAFRuntimeApi
 from volcenginesdkwafruntime.models.llm_stream_session import LLMStreamSession
-
-__all__ = ["WAFRuntimeApi", "LLMStreamSession"]
+from volcenginesdkwafruntime.models.llm_stream_session import  LLM_STREAM_SEND_EXPONENT
+from volcenginesdkwafruntime.models.llm_stream_session import  LLM_STREAM_SEND_BASE_WINDOW
+__all__ = ["WAFRuntimeApi", "LLMStreamSession" ,
+           "LLM_STREAM_SEND_EXPONENT" , "LLM_STREAM_SEND_BASE_WINDOW"] # @opensource-lint-ignore

volcenginesdkwafruntime/api/waf_runtime_api.py

@@ -1,11 +1,13 @@
-
 from volcenginesdkwaf import WAFApi, CheckLLMResponseStreamRequest
-from volcenginesdkwafruntime.models.llm_stream_session import LLMStreamSession
+from volcenginesdkwafruntime.models.llm_stream_session import LLMStreamSession, LLM_STREAM_SEND_EXPONENT, \
+    LLM_STREAM_SEND_BASE_WINDOW
 
 global_llm_send_len = 10
 
+
 class WAFRuntimeApi(WAFApi):
     """继承自 WAFApi 并重写 check_llm_response_stream 方法"""
+
     def check_llm_response_stream(
             self,
             body: CheckLLMResponseStreamRequest,
@@ -52,13 +54,15 @@ class WAFRuntimeApi(WAFApi):
 
             # 重置流缓冲区和发送长度
             session.set_stream_send_len(0)
+            session.CurrentSendWindow = session.CurrentSendWindow * LLM_STREAM_SEND_EXPONENT
 
             return response
 
         # 3. 处理 use_stream 为其他值的情况（累计长度，超过阈值才发送）
         else:
             # 如果未发送长度超过 10 个字符，调用 API
-            if session.get_stream_send_len() > global_llm_send_len:
+            if session.get_stream_send_len() >= session.CurrentSendWindow:
+                session.CurrentSendWindow = session.CurrentSendWindow * LLM_STREAM_SEND_EXPONENT
                 # 准备请求体，使用 session 中的完整流内容
                 body.content = session.get_stream_buf()
                 body.msg_id = session.get_msg_id()

volcenginesdkwafruntime/models/llm_stream_session.py

@@ -2,10 +2,11 @@ from typing import Optional
 
 from volcenginesdkwaf import CheckLLMResponseStreamResponse
 
+LLM_STREAM_SEND_BASE_WINDOW = 10
+LLM_STREAM_SEND_EXPONENT = 2
 
-class LLMStreamSession:
-    """对应 Java 中的 LLMStreamSession 类"""
 
+class LLMStreamSession:
     def __init__(
             self,
             stream_buf: str = "",
@@ -17,6 +18,7 @@ class LLMStreamSession:
         self.stream_send_len = stream_send_len
         self.msg_id = msg_id
         self.default_body = default_body
+        self.CurrentSendWindow = LLM_STREAM_SEND_BASE_WINDOW
 
     def get_stream_buf(self) -> str:
         return self.stream_buf