diff --git a/facefusion/apis/endpoints/capabilities.py b/facefusion/apis/endpoints/capabilities.py index 0af0b811..8476a35a 100644 --- a/facefusion/apis/endpoints/capabilities.py +++ b/facefusion/apis/endpoints/capabilities.py @@ -15,6 +15,6 @@ async def get_capabilities(request : Request) -> JSONResponse: 'image': facefusion.choices.image_formats, 'video': facefusion.choices.video_formats }, - 'arguments': args_store.get_capabilities() + 'arguments': args_store.get_api_set() } return JSONResponse(capabilities, status_code = HTTP_200_OK) diff --git a/facefusion/args_store.py b/facefusion/args_store.py index 8d6de6e9..a5ac85f7 100644 --- a/facefusion/args_store.py +++ b/facefusion/args_store.py @@ -12,20 +12,28 @@ ARGS_STORE : ArgsStore =\ } +def get_api_set() -> ArgumentValue: + return ARGS_STORE.get('api') + + +def get_cli_set() -> ArgumentValue: + return ARGS_STORE.get('cli') + + +def get_sys_set() -> ArgumentValue: + return ARGS_STORE.get('sys') + + def get_api_args() -> List[str]: - return list(ARGS_STORE.get('api').keys()) - - -def get_sys_args() -> List[str]: - return list(ARGS_STORE.get('sys').keys()) + return list(get_api_set().keys()) def get_cli_args() -> List[str]: - return list(ARGS_STORE.get('cli').keys()) + return list(get_cli_set().keys()) -def get_capabilities() -> Dict[str, ArgumentValue]: - return ARGS_STORE.get('api') +def get_sys_args() -> List[str]: + return list(get_cli_set().keys()) def register_argument(action : Action, scopes : List[Scope]) -> None: