fix: only forbid endpoint, not function (#839)

Signed-off-by: Agustín Ramiro Díaz <[email protected]>
yeagerai · Jan 22, 2025 · bbe73df · bbe73df
1 parent e640dd2
commit bbe73df
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/backend/protocol_rpc/endpoints.py b/backend/protocol_rpc/endpoints.py
@@ -148,7 +148,6 @@ def delete_provider(llm_provider_registry: LLMProviderRegistry, id: int) -> None
     llm_provider_registry.delete(id)
 
 
-@check_forbidden_method_in_hosted_studio
 def create_validator(
     validators_registry: ValidatorsRegistry,
     accounts_manager: AccountsManager,
@@ -743,7 +742,11 @@ def register_all_rpc_endpoints(
         method_name="sim_deleteProvider",
     )
     register_rpc_endpoint(
-        partial(create_validator, validators_registry, accounts_manager),
+        partial(
+            check_forbidden_method_in_hosted_studio(create_validator),
+            validators_registry,
+            accounts_manager,
+        ),
         method_name="sim_createValidator",
     )
     register_rpc_endpoint(