Merge branch 'feat/queue-based-graph-engine' into chore/merge-graph-engine

api/extensions/ext_database.py

@@ -13,7 +13,7 @@ logger = logging.getLogger(__name__)
 _GEVENT_COMPATIBILITY_SETUP: bool = False
 
 
-def _safe_rollback(connection) -> None:
+def _safe_rollback(connection):
     """Safely rollback database connection.
 
     Args:
@@ -25,7 +25,7 @@ def _safe_rollback(connection) -> None:
         logger.exception("Failed to rollback connection")
 
 
-def _setup_gevent_compatibility() -> None:
+def _setup_gevent_compatibility():
     global _GEVENT_COMPATIBILITY_SETUP  # pylint: disable=global-statement
 
     # Avoid duplicate registration
@@ -33,7 +33,7 @@ def _setup_gevent_compatibility() -> None:
         return
 
     @event.listens_for(Pool, "reset")
-    def _safe_reset(dbapi_connection, connection_record, reset_state) -> None:  # pylint: disable=unused-argument
+    def _safe_reset(dbapi_connection, connection_record, reset_state):  # pylint: disable=unused-argument
         if reset_state.terminate_only:
             return
 

api/extensions/ext_orjson.py

@@ -3,6 +3,6 @@ from flask_orjson import OrjsonProvider
 from dify_app import DifyApp
 
 
-def init_app(app: DifyApp) -> None:
+def init_app(app: DifyApp):
     """Initialize Flask-Orjson extension for faster JSON serialization"""
     app.json = OrjsonProvider(app)

api/extensions/storage/clickzetta_volume/clickzetta_volume_storage.py

@@ -40,7 +40,7 @@ class ClickZettaVolumeConfig(BaseModel):
 
     @model_validator(mode="before")
     @classmethod
-    def validate_config(cls, values: dict) -> dict:
+    def validate_config(cls, values: dict):
         """Validate the configuration values.
 
         This method will first try to use CLICKZETTA_VOLUME_* environment variables,
@@ -217,7 +217,7 @@ class ClickZettaVolumeStorage(BaseStorage):
             logger.exception("SQL execution failed: %s", sql)
             raise
 
-    def _ensure_table_volume_exists(self, dataset_id: str) -> None:
+    def _ensure_table_volume_exists(self, dataset_id: str):
         """Ensure table volume exists for the given dataset_id."""
         if self._config.volume_type != "table" or not dataset_id:
             return
@@ -252,7 +252,7 @@ class ClickZettaVolumeStorage(BaseStorage):
             # Don't raise exception, let the operation continue
             # The table might exist but not be visible due to permissions
 
-    def save(self, filename: str, data: bytes) -> None:
+    def save(self, filename: str, data: bytes):
         """Save data to ClickZetta Volume.
 
         Args:

api/extensions/storage/clickzetta_volume/file_lifecycle.py

@@ -38,7 +38,7 @@ class FileMetadata:
     tags: Optional[dict[str, str]] = None
     parent_version: Optional[int] = None
 
-    def to_dict(self) -> dict:
+    def to_dict(self):
         """Convert to dictionary format"""
         data = asdict(self)
         data["created_at"] = self.created_at.isoformat()

api/extensions/storage/clickzetta_volume/volume_permissions.py

@@ -623,7 +623,7 @@ class VolumePermissionError(Exception):
 
 def check_volume_permission(
     permission_manager: VolumePermissionManager, operation: str, dataset_id: Optional[str] = None
-) -> None:
+):
     """Permission check decorator function
 
     Args:

api/extensions/storage/opendal_storage.py

@@ -40,7 +40,7 @@ class OpenDALStorage(BaseStorage):
         self.op = self.op.layer(retry_layer)
         logger.debug("added retry layer to opendal operator")
 
-    def save(self, filename: str, data: bytes) -> None:
+    def save(self, filename: str, data: bytes):
         self.op.write(path=filename, bs=data)
         logger.debug("file %s saved", filename)