清理代码

sync_state_machine/datasource/api/datasource.py

@@ -12,7 +12,7 @@ import json
 from typing import Optional
 from pathlib import Path
 from ..datasource import BaseDataSource
-from ..handler import NodeHandler, ensure_handler_compat
+from ..handler import NodeHandler
 from ..task_result import TaskResult
 from .client import ApiClient
 
@@ -72,16 +72,15 @@ class ApiDataSource(BaseDataSource):
         Args:
             handler: API Handler 实例
         """
-        compatible_handler = ensure_handler_compat(handler)
-        compatible_handler.set_api_client(self.client)
-        compatible_handler.set_poll_mode(self.poll_mode)
+        handler.set_api_client(self.client)
+        handler.set_poll_mode(self.poll_mode)
         
         # 调用基类注册方法
-        super().register_handler(compatible_handler)
+        super().register_handler(handler)
 
         # 如果 collection 已设置，确保新 handler 拿到引用
         if self._collection is not None:
-            compatible_handler.set_collection(self._collection)  # type: ignore[arg-type]
+            handler.set_collection(self._collection)  # type: ignore[arg-type]
 
     @staticmethod
     def _fmt_trace_value(value, max_len: int = 600) -> str: