semmyk commited on
Commit
74018a1
Β·
1 Parent(s): 331205c

baseline08_beta01.4_28Sept25: fixing oauth_token, llm_login: LocalTokenNotFoundError

Browse files
Files changed (1) hide show
  1. llm/llm_login.py +10 -3
llm/llm_login.py CHANGED
@@ -61,13 +61,19 @@ def login_huggingface(token: Optional[str] = None):
61
  #pass
62
 
63
  #def is_login_huggingface():
64
- def is_loggedin_huggingface():
65
  #from huggingface_hub import HfApi
66
  from huggingface_hub.utils import HfHubHTTPError
 
67
 
68
- try:
69
  HfApi().whoami()
70
- logger.log(level=20, msg=("βœ”οΈ You are logged in."), extra={"is_logged_in": True})
 
 
 
 
 
71
  disable_immplicit_token()
72
  return True
73
  except HfHubHTTPError as exc:
@@ -81,4 +87,5 @@ def is_loggedin_huggingface():
81
  logger.exception(f"βœ— An unexpected HTTP error occurred: β†’ {exc}\n{tb}", exc_info=True)
82
  #raise RuntimeError(f"βœ— An unexpected HTTP error occurred: β†’ {exc}\n{tb}") from exc
83
  return False
 
84
 
 
61
  #pass
62
 
63
  #def is_login_huggingface():
64
+ def is_loggedin_huggingface(token: Optional[str] = None):
65
  #from huggingface_hub import HfApi
66
  from huggingface_hub.utils import HfHubHTTPError
67
+ from huggingface_hub.errors import LocalTokenNotFoundError
68
 
69
+ try: ##SMY: TOOD: investigate if still needed given oauth_token: gr.OAuthToken
70
  HfApi().whoami()
71
+ logger.log(level=20, msg=("βœ”οΈ HfApi().whoami(): You are logged in."), extra={"is_logged_in": True})
72
+ disable_immplicit_token()
73
+ return True
74
+ except LocalTokenNotFoundError as exc: #Token is required (`token=True`), but no token found
75
+ HfApi().whoami(token)
76
+ logger.log(level=20, msg=("βœ”οΈ HfApi().whoami(): You are logged in with token."), extra={"is_logged_in": True})
77
  disable_immplicit_token()
78
  return True
79
  except HfHubHTTPError as exc:
 
87
  logger.exception(f"βœ— An unexpected HTTP error occurred: β†’ {exc}\n{tb}", exc_info=True)
88
  #raise RuntimeError(f"βœ— An unexpected HTTP error occurred: β†’ {exc}\n{tb}") from exc
89
  return False
90
+ finally: return False
91