From a66609b3886ae274c3328c0a793c806238ec2c37 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Thu, 6 Aug 2026 13:55:18 +0200
Subject: [PATCH] rename capability stuff

---
 facefusion/capability_store.py | 12 ++++++------
 facefusion/types.py            | 15 ++++++++-------
 2 files changed, 14 insertions(+), 13 deletions(-)

diff --git a/facefusion/capability_store.py b/facefusion/capability_store.py
index a25eaa27..eeec34fa 100644
--- a/facefusion/capability_store.py
+++ b/facefusion/capability_store.py
@@ -1,7 +1,7 @@
 from argparse import Action
-from typing import Dict, List
+from typing import List
 
-from facefusion.types import CapabilityGroup, CapabilitySet, CapabilityStore, Group, Scope
+from facefusion.types import Capability, CapabilityGroup, CapabilitySet, CapabilityStore, Group, Scope
 
 CAPABILITY_STORE : CapabilityStore =\
 {
@@ -21,15 +21,15 @@ def get_api_capability_group() -> CapabilityGroup:
 	return capability_group
 
 
-def get_api_capability_set() -> Dict[str, CapabilitySet]:
+def get_api_capability_set() -> CapabilitySet:
 	return CAPABILITY_STORE.get('api')
 
 
-def get_cli_capability_set() -> Dict[str, CapabilitySet]:
+def get_cli_capability_set() -> CapabilitySet:
 	return CAPABILITY_STORE.get('cli')
 
 
-def get_sys_capability_set() -> Dict[str, CapabilitySet]:
+def get_sys_capability_set() -> CapabilitySet:
 	return CAPABILITY_STORE.get('sys')
 
 
@@ -47,7 +47,7 @@ def get_sys_arguments() -> List[str]:
 
 def register_capability_set(actions : List[Action], scopes : List[Scope], groups : List[Group]) -> None:
 	for action in actions:
-		value : CapabilitySet =\
+		value : Capability =\
 		{
 			'default': action.default,
 			'groups': groups
diff --git a/facefusion/types.py b/facefusion/types.py
index 35976f00..d9109b7a 100755
--- a/facefusion/types.py
+++ b/facefusion/types.py
@@ -135,19 +135,20 @@ Choice : TypeAlias = Union[int | str]
 Scope : TypeAlias = Literal['api', 'cli', 'sys']
 Group : TypeAlias = str
 
-CapabilitySet = TypedDict('CapabilitySet',
+Capability = TypedDict('Capability',
 {
 	'default' : Any,
-	'choices' : NotRequired[List[Choice]],
-	'groups' : NotRequired[List[Group]]
+	'groups' : List[Group],
+	'choices' : NotRequired[List[Choice]]
 })
+CapabilitySet : TypeAlias = Dict[str, Capability]
 CapabilityStore = TypedDict('CapabilityStore',
 {
-	'api' : Dict[str, CapabilitySet],
-	'cli' : Dict[str, CapabilitySet],
-	'sys' : Dict[str, CapabilitySet]
+	'api' : CapabilitySet,
+	'cli' : CapabilitySet,
+	'sys' : CapabilitySet
 })
-CapabilityGroup : TypeAlias = Dict[Group, Dict[str, CapabilitySet]]
+CapabilityGroup : TypeAlias = Dict[Group, CapabilitySet]
 
 ProcessState = Literal['checking', 'processing', 'stopping', 'pending']
 UpdateProgress : TypeAlias = Callable[[int], None]