refactor: 移除旧的comfyui_client模块,重构API路由以使用新的comfy_workflow和queue_manager,提升代码结构和可维护性
This commit is contained in:
@@ -4,7 +4,7 @@ from typing import Optional, List
|
||||
from fastapi import APIRouter, Body, HTTPException, Path
|
||||
from fastapi.responses import JSONResponse
|
||||
|
||||
from workflow_service import comfyui_client
|
||||
from workflow_service.comfy import comfy_workflow
|
||||
from workflow_service.database.api import (
|
||||
save_workflow,
|
||||
get_all_workflows,
|
||||
@@ -94,8 +94,8 @@ async def get_one_workflow_endpoint(base_name: str, version: Optional[str] = Non
|
||||
workflow = json.loads(workflow_data["workflow_json"])
|
||||
return {
|
||||
"workflow": workflow,
|
||||
"api_spec": comfyui_client.parse_api_spec(workflow),
|
||||
"inputs_json_schema": comfyui_client.parse_inputs_json_schema(workflow),
|
||||
"api_spec": comfy_workflow.parse_api_spec(workflow),
|
||||
"inputs_json_schema": comfy_workflow.parse_inputs_json_schema(workflow),
|
||||
}
|
||||
except Exception as e:
|
||||
raise HTTPException(
|
||||
|
||||
Reference in New Issue
Block a user