整理代码

2026-04-03 09:18:44 +08:00
parent 543c3ff906
commit f5729d5a18
50 changed files with 305 additions and 296 deletions
@@ -43,11 +43,21 @@ class BaseJsonlHandler(BaseNodeHandler):
    def __init__(
        self,
        datasource: JsonlDataSource,
-        node_type: str,
-        node_class: type,
-        schema: type
+        node_class_or_node_type,
+        node_class: type | None = None,
+        schema: type | None = None,
+        node_type: str | None = None,
    ):
-        super().__init__(node_type, node_class, schema)
+        if isinstance(node_class_or_node_type, str):
+            resolved_node_type = node_class_or_node_type
+            resolved_node_class = node_class
+            resolved_schema = schema
+        else:
+            resolved_node_class = node_class_or_node_type
+            resolved_node_type = node_type or getattr(resolved_node_class, "node_type", None)
+            resolved_schema = schema or getattr(resolved_node_class, "schema", None)
+
+        super().__init__(resolved_node_type, resolved_node_class, resolved_schema)
        self.datasource = datasource

    async def load(self) -> List['SyncNode']: