feat: pipeline model crud

This commit is contained in:
Junyan Qin
2025-03-26 23:19:57 +08:00
parent b85615cece
commit 2c50ab0255
8 changed files with 164 additions and 1 deletions

View File

@@ -26,6 +26,7 @@ from ..persistence import mgr as persistencemgr
from ..api.http.controller import main as http_controller
from ..api.http.service import user as user_service
from ..api.http.service import model as model_service
from ..api.http.service import pipeline as pipeline_service
from ..discover import engine as discover_engine
from ..utils import logcache, ip
from . import taskmgr
@@ -83,6 +84,11 @@ class Application:
instance_secret_meta: config_mgr.ConfigManager = None
pipeline_config_meta_trigger: config_mgr.ConfigManager = None
pipeline_config_meta_safety: config_mgr.ConfigManager = None
pipeline_config_meta_ai: config_mgr.ConfigManager = None
pipeline_config_meta_output: config_mgr.ConfigManager = None
# =========================
ctr_mgr: center_mgr.V2CenterAPI = None
@@ -115,6 +121,8 @@ class Application:
model_service: model_service.ModelsService = None
pipeline_service: pipeline_service.PipelineService = None
def __init__(self):
pass

View File

@@ -8,3 +8,4 @@ from ...config.impls import pymodule
load_python_module_config = config_mgr.load_python_module_config
load_json_config = config_mgr.load_json_config
load_yaml_config = config_mgr.load_yaml_config

View File

@@ -19,6 +19,7 @@ from ...persistence import mgr as persistencemgr
from ...api.http.controller import main as http_controller
from ...api.http.service import user as user_service
from ...api.http.service import model as model_service
from ...api.http.service import pipeline as pipeline_service
from ...discover import engine as discover_engine
from ...utils import logcache
from .. import taskmgr
@@ -127,5 +128,8 @@ class BuildAppStage(stage.BootingStage):
model_service_inst = model_service.ModelsService(ap)
ap.model_service = model_service_inst
pipeline_service_inst = pipeline_service.PipelineService(ap)
ap.pipeline_service = pipeline_service_inst
ctrl = controller.Controller(ap)
ap.ctrl = ctrl

View File

@@ -82,3 +82,8 @@ class LoadConfigStage(stage.BootingStage):
'jwt_secret': secrets.token_hex(16)
})
await ap.instance_secret_meta.dump_config()
ap.pipeline_config_meta_trigger = await config.load_yaml_config("templates/metadata/pipeline/trigger.yaml", "templates/metadata/pipeline/trigger.yaml")
ap.pipeline_config_meta_safety = await config.load_yaml_config("templates/metadata/pipeline/safety.yaml", "templates/metadata/pipeline/safety.yaml")
ap.pipeline_config_meta_ai = await config.load_yaml_config("templates/metadata/pipeline/ai.yaml", "templates/metadata/pipeline/ai.yaml")
ap.pipeline_config_meta_output = await config.load_yaml_config("templates/metadata/pipeline/output.yaml", "templates/metadata/pipeline/output.yaml")