[model-gateway] feat: add DAG parallel execution support and workflow optimization (#14999)

This commit is contained in:
Simo Lin
2025-12-12 08:31:12 -08:00
committed by GitHub
parent 10c68f6236
commit 306e5b8d0b
13 changed files with 768 additions and 193 deletions
+18 -6
View File
@@ -770,12 +770,24 @@ pub async fn startup(config: ServerConfig) -> Result<(), Box<dyn std::error::Err
.subscribe(Arc::new(LoggingSubscriber))
.await;
engine.register_workflow(create_worker_registration_workflow(&config.router_config));
engine.register_workflow(create_external_worker_registration_workflow());
engine.register_workflow(create_worker_removal_workflow());
engine.register_workflow(create_mcp_registration_workflow());
engine.register_workflow(create_wasm_module_registration_workflow());
engine.register_workflow(create_wasm_module_removal_workflow());
engine
.register_workflow(create_worker_registration_workflow(&config.router_config))
.expect("worker_registration workflow should be valid");
engine
.register_workflow(create_external_worker_registration_workflow())
.expect("external_worker_registration workflow should be valid");
engine
.register_workflow(create_worker_removal_workflow())
.expect("worker_removal workflow should be valid");
engine
.register_workflow(create_mcp_registration_workflow())
.expect("mcp_registration workflow should be valid");
engine
.register_workflow(create_wasm_module_registration_workflow())
.expect("wasm_module_registration workflow should be valid");
engine
.register_workflow(create_wasm_module_removal_workflow())
.expect("wasm_module_removal workflow should be valid");
app_context
.workflow_engine
.set(engine)