整理了初始化过程,现在更容易被继承了。
This commit is contained in:
@@ -23,13 +23,25 @@ from .handler import (
|
||||
from .task_result import TaskResult, TaskStatus
|
||||
|
||||
# API DataSource
|
||||
from .api import ApiClient, ApiDataSource, BaseApiHandler
|
||||
from .api import ApiClient, ApiDataSource, BaseApiHandler, register as register_api_datasource
|
||||
|
||||
# DB DataSource
|
||||
# DB base helpers
|
||||
from .db import DbDataSource, BaseDbHandler
|
||||
|
||||
# JSONL DataSource
|
||||
from .jsonl import JsonlDataSource, BaseJsonlHandler
|
||||
from .jsonl import JsonlDataSource, BaseJsonlHandler, register as register_jsonl_datasource
|
||||
|
||||
|
||||
_BUILTIN_DATASOURCE_TYPES_REGISTERED = False
|
||||
|
||||
|
||||
def ensure_builtin_datasource_types_registered(*, replace: bool = False) -> None:
|
||||
global _BUILTIN_DATASOURCE_TYPES_REGISTERED
|
||||
if _BUILTIN_DATASOURCE_TYPES_REGISTERED and not replace:
|
||||
return
|
||||
register_jsonl_datasource(replace=replace)
|
||||
register_api_datasource(replace=replace)
|
||||
_BUILTIN_DATASOURCE_TYPES_REGISTERED = True
|
||||
|
||||
__all__ = [
|
||||
# 基础类
|
||||
@@ -46,13 +58,14 @@ __all__ = [
|
||||
"TaskResult",
|
||||
"TaskStatus",
|
||||
"ValidationResult",
|
||||
"ensure_builtin_datasource_types_registered",
|
||||
|
||||
# API DataSource
|
||||
"ApiClient",
|
||||
"ApiDataSource",
|
||||
"BaseApiHandler",
|
||||
|
||||
# DB DataSource
|
||||
# DB base helpers
|
||||
"DbDataSource",
|
||||
"BaseDbHandler",
|
||||
|
||||
|
||||
Reference in New Issue
Block a user