adapt pyrefly

.github/workflows/api-tests.yml

@@ -43,7 +43,7 @@ jobs:
         run: |
           cd api
           uv add --dev pyrefly
-          uv run pyrefly check || true
+          uv run pyrefly check
 
       - name: Run dify config tests
         run: uv run --project api dev/pytest/pytest_config_tests.py

api/extensions/ext_celery.py

@@ -1,6 +1,6 @@
 import ssl
 from datetime import timedelta
-from typing import Any
+from typing import Any, TypedDict
 
 import pytz
 from celery import Celery, Task
@@ -99,8 +99,12 @@ def __call__(self, *args: object, **kwargs: object) -> object:
     imports = []
     day = dify_config.CELERY_BEAT_SCHEDULER_TIME
 
+    class BeatTask(TypedDict):
+        task: str
+        schedule: crontab | timedelta
+
     # if you add a new task, please add the switch to CeleryScheduleTasksConfig
-    beat_schedule = {}
+    beat_schedule: dict[str, BeatTask] = {}
     if dify_config.ENABLE_CLEAN_EMBEDDING_CACHE_TASK:
         imports.append("schedule.clean_embedding_cache_task")
         beat_schedule["clean_embedding_cache_task"] = {

api/pyproject.toml

@@ -109,7 +109,6 @@ dev = [
     "dotenv-linter~=0.5.0",
     "faker~=32.1.0",
     "lxml-stubs~=0.5.1",
-    "ty~=0.0.1a19",
     "basedpyright~=1.31.0",
     "ruff~=0.14.0",
     "pytest~=8.3.2",

api/pyrefly.toml

@@ -0,0 +1,18 @@
+project-includes = ["."]
+project-excludes = [
+    "tests/",
+    ".venv",
+    "migrations/",
+    "extensions/storage/",
+    "services/",
+    "core/",
+    "controllers/",
+    "extensions/ext_mail.py",
+    "extensions/ext_otel.py",
+    "libs/",
+    "models/",
+    "factories/",
+    "schedule/queue_monitor_task.py",
+]
+python-platform = "linux"
+python-version = "3.11.0"