Skip to content

Commit 890a293

Browse files
committed
fix: Reformat
1 parent 4ec39a7 commit 890a293

File tree

1 file changed

+7
-17
lines changed

1 file changed

+7
-17
lines changed

volcenginesdkllmshield/api/llm_shield_sdk_v2.py

Lines changed: 7 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
from pydantic import BaseModel, field_validator, Field
22
from typing import List, Optional, Any, Union
3+
from datetime import datetime, date
4+
from uuid import UUID
35
import requests
46
import json
57

@@ -194,6 +196,7 @@ class Config:
194196
# 定义错误信息结构体
195197
class ErrorInfo(BaseModel):
196198
code: str = Field("", alias="Code")
199+
codeN: int = Field("", alias="CodeN")
197200
message: str = Field("", alias="Message")
198201

199202
class Config:
@@ -320,8 +323,9 @@ def Moderate(self, request: Optional[ModerateV2Request] = None) -> ModerateV2Res
320323
except Exception as e:
321324
raise Exception(f"处理响应失败: {e}")
322325

323-
def ModerateStream(self, request: ModerateV2Request, session: ModerateV2StreamSession) -> Optional[
324-
ModerateV2Response]:
326+
def ModerateStream(
327+
self, request: ModerateV2Request, session: ModerateV2StreamSession
328+
) -> Optional[ModerateV2Response]:
325329
"""
326330
处理流式审核请求
327331
:param request: 当前流式请求片段(ModerateV2Request 类型)
@@ -438,22 +442,8 @@ def GenerateV2Stream(self, request):
438442
# 自定义 JSON 编码器
439443
class CustomJSONEncoder(json.JSONEncoder):
440444
def default(self, obj):
441-
if isinstance(obj, Error):
442-
return {
443-
"Code": obj.Code,
444-
"Message": obj.Message
445-
}
446-
elif isinstance(obj, JudgeDecisionInfo):
447-
return {
448-
"ErrCode": obj.ErrCode,
449-
"ErrMsg": obj.ErrMsg,
450-
"Labels": obj.Labels,
451-
"Matches": [{"Label": match.Label, "Word": match.Word} for match in obj.Matches],
452-
"DecisionCategory": obj.DecisionCategory,
453-
"RuleIDs": obj.RuleIDs
454-
}
455445
# 处理datetime类型(如2023-10-01T12:00:00)
456-
elif isinstance(obj, datetime):
446+
if isinstance(obj, datetime):
457447
return obj.isoformat()
458448
# 处理date类型(如2023-10-01)
459449
elif isinstance(obj, date):

0 commit comments

Comments
 (0)