File tree Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Original file line number Diff line number Diff line change @@ -893,6 +893,12 @@ def set_token(cls, token: str) -> HFTokenStatus:
893
893
huggingface_hub .login (token = token , add_to_git_credential = False )
894
894
return cls .get_status ()
895
895
896
+ @classmethod
897
+ def reset_token (cls ) -> HFTokenStatus :
898
+ with SuppressOutput (), contextlib .suppress (Exception ):
899
+ huggingface_hub .logout ()
900
+ return cls .get_status ()
901
+
896
902
897
903
@model_manager_router .get ("/hf_login" , operation_id = "get_hf_login_status" , response_model = HFTokenStatus )
898
904
async def get_hf_login_status () -> HFTokenStatus :
@@ -915,3 +921,8 @@ async def do_hf_login(
915
921
ApiDependencies .invoker .services .logger .warning ("Unable to verify HF token" )
916
922
917
923
return token_status
924
+
925
+
926
+ @model_manager_router .delete ("/hf_login" , operation_id = "reset_hf_token" , response_model = HFTokenStatus )
927
+ async def reset_hf_token () -> HFTokenStatus :
928
+ return HFTokenHelper .reset_token ()
You can’t perform that action at this time.
0 commit comments