File size: 1,923 Bytes
e7abd9e 23c96f8 e7abd9e 58582d3 e7abd9e 58582d3 e7abd9e 58582d3 e7abd9e 58582d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
from typing import Optional
from app.config import HF_TOKEN, API
from app.core.cache import cache_config
from app.core.formatting import LogFormatter
import logging
logger = logging.getLogger(__name__)
class HuggingFaceService:
def __init__(self):
self.api = API
self.token = HF_TOKEN
self.cache_dir = cache_config.models_cache
async def check_authentication(self) -> bool:
"""Check if the HF token is valid"""
if not self.token:
return False
try:
logger.info(LogFormatter.info("Checking HF token validity..."))
self.api.get_token_permission()
logger.info(LogFormatter.success("HF token is valid"))
return True
except Exception as e:
logger.error(LogFormatter.error("HF token validation failed", e))
return False
async def get_user_info(self) -> Optional[dict]:
"""Get information about the authenticated user"""
try:
logger.info(LogFormatter.info("Fetching user information..."))
info = self.api.get_token_permission()
logger.info(
LogFormatter.success(
f"User info retrieved for: {info.get('user', 'Unknown')}"
)
)
return info
except Exception as e:
logger.error(LogFormatter.error("Failed to get user info", e))
return None
def _log_repo_operation(self, operation: str, repo: str, details: str = None):
"""Helper to log repository operations"""
logger.info(
LogFormatter.section(f"HF REPOSITORY OPERATION - {operation.upper()}")
)
stats = {
"Operation": operation,
"Repository": repo,
}
if details:
stats["Details"] = details
for line in LogFormatter.tree(stats):
logger.info(line)
|