Spaces:
Sleeping
Sleeping
baseline08_beta01.4_28Sept25: fixing oauth_token, llm_login: LocalTokenNotFoundError
Browse files- 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 |
|