update forward api

Signed-off-by: Jat <[email protected]>
FederatedAI · Dec 26, 2022 · 996eab5 · 996eab5
1 parent 1c9ab6f
commit 996eab5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/fate_flow/utils/api_utils.py b/python/fate_flow/utils/api_utils.py
@@ -312,6 +312,8 @@ def forward_api(role, request_config):
     role = role.upper()
     if not hasattr(ServerRegistry, role):
         ServerRegistry.load()
+    if not hasattr(ServerRegistry, role):
+        return {'retcode': 404, 'retmsg': f'role "{role.lower()}" not supported'}
     registry = getattr(ServerRegistry, role)
 
     headers = request_config.get('header', {})