|
1 | 1 | from pydantic import BaseModel, field_validator, Field |
2 | 2 | from typing import List, Optional, Any, Union |
| 3 | +from datetime import datetime, date |
| 4 | +from uuid import UUID |
3 | 5 | import requests |
4 | 6 | import json |
5 | 7 |
|
@@ -194,6 +196,7 @@ class Config: |
194 | 196 | # 定义错误信息结构体 |
195 | 197 | class ErrorInfo(BaseModel): |
196 | 198 | code: str = Field("", alias="Code") |
| 199 | + codeN: int = Field("", alias="CodeN") |
197 | 200 | message: str = Field("", alias="Message") |
198 | 201 |
|
199 | 202 | class Config: |
@@ -320,8 +323,9 @@ def Moderate(self, request: Optional[ModerateV2Request] = None) -> ModerateV2Res |
320 | 323 | except Exception as e: |
321 | 324 | raise Exception(f"处理响应失败: {e}") |
322 | 325 |
|
323 | | - def ModerateStream(self, request: ModerateV2Request, session: ModerateV2StreamSession) -> Optional[ |
324 | | - ModerateV2Response]: |
| 326 | + def ModerateStream( |
| 327 | + self, request: ModerateV2Request, session: ModerateV2StreamSession |
| 328 | + ) -> Optional[ModerateV2Response]: |
325 | 329 | """ |
326 | 330 | 处理流式审核请求 |
327 | 331 | :param request: 当前流式请求片段(ModerateV2Request 类型) |
@@ -438,22 +442,8 @@ def GenerateV2Stream(self, request): |
438 | 442 | # 自定义 JSON 编码器 |
439 | 443 | class CustomJSONEncoder(json.JSONEncoder): |
440 | 444 | 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 | | - } |
455 | 445 | # 处理datetime类型(如2023-10-01T12:00:00) |
456 | | - elif isinstance(obj, datetime): |
| 446 | + if isinstance(obj, datetime): |
457 | 447 | return obj.isoformat() |
458 | 448 | # 处理date类型(如2023-10-01) |
459 | 449 | elif isinstance(obj, date): |
|
0 commit comments