chore: add ast-grep rule to convert Optional[T] to T | None (#25560)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
This commit is contained in:
-LAN-
2025-09-15 13:06:33 +08:00
committed by GitHub
parent 2e44ebe98d
commit bab4975809
394 changed files with 2555 additions and 2792 deletions

View File

@ -1,5 +1,3 @@
from typing import Optional
from pydantic import BaseModel, Field
from sqlalchemy import select
@ -87,7 +85,7 @@ class ApiModeration(Moderation):
return result
@staticmethod
def _get_api_based_extension(tenant_id: str, api_based_extension_id: str) -> Optional[APIBasedExtension]:
def _get_api_based_extension(tenant_id: str, api_based_extension_id: str) -> APIBasedExtension | None:
stmt = select(APIBasedExtension).where(
APIBasedExtension.tenant_id == tenant_id, APIBasedExtension.id == api_based_extension_id
)

View File

@ -1,6 +1,5 @@
from abc import ABC, abstractmethod
from enum import StrEnum, auto
from typing import Optional
from pydantic import BaseModel, Field
@ -34,7 +33,7 @@ class Moderation(Extensible, ABC):
module: ExtensionModule = ExtensionModule.MODERATION
def __init__(self, app_id: str, tenant_id: str, config: Optional[dict] = None):
def __init__(self, app_id: str, tenant_id: str, config: dict | None = None):
super().__init__(tenant_id, config)
self.app_id = app_id

View File

@ -1,6 +1,6 @@
import logging
from collections.abc import Mapping
from typing import Any, Optional
from typing import Any
from core.app.app_config.entities import AppConfig
from core.moderation.base import ModerationAction, ModerationError
@ -21,7 +21,7 @@ class InputModeration:
inputs: Mapping[str, Any],
query: str,
message_id: str,
trace_manager: Optional[TraceQueueManager] = None,
trace_manager: TraceQueueManager | None = None,
) -> tuple[bool, Mapping[str, Any], str]:
"""
Process sensitive_word_avoidance.

View File

@ -1,7 +1,7 @@
import logging
import threading
import time
from typing import Any, Optional
from typing import Any
from flask import Flask, current_app
from pydantic import BaseModel, ConfigDict
@ -27,11 +27,11 @@ class OutputModeration(BaseModel):
rule: ModerationRule
queue_manager: AppQueueManager
thread: Optional[threading.Thread] = None
thread: threading.Thread | None = None
thread_running: bool = True
buffer: str = ""
is_final_chunk: bool = False
final_output: Optional[str] = None
final_output: str | None = None
model_config = ConfigDict(arbitrary_types_allowed=True)
def should_direct_output(self) -> bool:
@ -127,7 +127,7 @@ class OutputModeration(BaseModel):
if result.action == ModerationAction.DIRECT_OUTPUT:
break
def moderation(self, tenant_id: str, app_id: str, moderation_buffer: str) -> Optional[ModerationOutputsResult]:
def moderation(self, tenant_id: str, app_id: str, moderation_buffer: str) -> ModerationOutputsResult | None:
try:
moderation_factory = ModerationFactory(
name=self.rule.type, app_id=app_id, tenant_id=tenant_id, config=self.rule.config