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>

api/tests/unit_tests/core/workflow/nodes/llm/test_node.py

@@ -1,7 +1,6 @@
 import base64
 import uuid
 from collections.abc import Sequence
-from typing import Optional
 from unittest import mock
 
 import pytest
@@ -47,7 +46,7 @@ class MockTokenBufferMemory:
         self.history_messages = history_messages or []
 
     def get_history_prompt_messages(
-        self, max_token_limit: int = 2000, message_limit: Optional[int] = None
+        self, max_token_limit: int = 2000, message_limit: int | None = None
     ) -> Sequence[PromptMessage]:
         if message_limit is not None:
             return self.history_messages[-message_limit * 2 :]

api/tests/unit_tests/libs/test_flask_utils.py

@@ -1,6 +1,5 @@
 import contextvars
 import threading
-from typing import Optional
 
 import pytest
 from flask import Flask
@@ -29,7 +28,7 @@ def login_app(app: Flask) -> Flask:
     login_manager.init_app(app)
 
     @login_manager.user_loader
-    def load_user(user_id: str) -> Optional[User]:
+    def load_user(user_id: str) -> User | None:
         if user_id == "test_user":
             return User("test_user")
         return None

api/tests/unit_tests/services/test_dataset_service_batch_update_document_status.py

@@ -1,5 +1,4 @@
 import datetime
-from typing import Optional
 
 # Mock redis_client before importing dataset_service
 from unittest.mock import Mock, call, patch
@@ -37,7 +36,7 @@ class DocumentBatchUpdateTestDataFactory:
         enabled: bool = True,
         archived: bool = False,
         indexing_status: str = "completed",
-        completed_at: Optional[datetime.datetime] = None,
+        completed_at: datetime.datetime | None = None,
         **kwargs,
     ) -> Mock:
         """Create a mock document with specified attributes."""

api/tests/unit_tests/services/test_dataset_service_update_dataset.py

@@ -1,5 +1,5 @@
 import datetime
-from typing import Any, Optional
+from typing import Any
 
 # Mock redis_client before importing dataset_service
 from unittest.mock import Mock, create_autospec, patch
@@ -24,9 +24,9 @@ class DatasetUpdateTestDataFactory:
         description: str = "old_description",
         indexing_technique: str = "high_quality",
         retrieval_model: str = "old_model",
-        embedding_model_provider: Optional[str] = None,
-        embedding_model: Optional[str] = None,
-        collection_binding_id: Optional[str] = None,
+        embedding_model_provider: str | None = None,
+        embedding_model: str | None = None,
+        collection_binding_id: str | None = None,
         **kwargs,
     ) -> Mock:
         """Create a mock dataset with specified attributes."""