ml-dash 0.6.9__py3-none-any.whl → 0.6.10__py3-none-any.whl

ml_dash/__init__.py

@@ -36,14 +36,60 @@ Usage:
         exp.log("Training started")
 """
 
-from .client import RemoteClient
+from .client import RemoteClient, userinfo
 from .experiment import Experiment, OperationMode, ml_dash_experiment
 from .log import LogBuilder, LogLevel
 from .params import ParametersBuilder
 from .run import RUN
 from .storage import LocalStorage
 
-__version__ = "0.6.4"
+__version__ = "0.6.10"
+
+# Minimum version required - blocks older versions
+MINIMUM_REQUIRED_VERSION = "0.6.10"
+
+
+def _check_version_compatibility():
+    """
+    Enforce minimum version requirement.
+
+    Raises ImportError if installed version is below minimum required version.
+    This ensures users have the latest features (userinfo, namespace auto-detection, etc.)
+    """
+    try:
+        from packaging import version
+    except ImportError:
+        # If packaging is not available, skip check
+        # (unlikely since it's a common dependency)
+        return
+
+    current = version.parse(__version__)
+    minimum = version.parse(MINIMUM_REQUIRED_VERSION)
+
+    if current < minimum:
+        raise ImportError(
+            f"\n"
+            f"{'=' * 80}\n"
+            f"ERROR: ml-dash version {__version__} is too old!\n"
+            f"{'=' * 80}\n"
+            f"\n"
+            f"This version of ml-dash ({__version__}) is no longer supported.\n"
+            f"Minimum required version: {MINIMUM_REQUIRED_VERSION}\n"
+            f"\n"
+            f"Please upgrade to the latest version:\n"
+            f"\n"
+            f"  pip install --upgrade ml-dash\n"
+            f"\n"
+            f"Or install specific version:\n"
+            f"\n"
+            f"  pip install ml-dash>={MINIMUM_REQUIRED_VERSION}\n"
+            f"\n"
+            f"{'=' * 80}\n"
+        )
+
+
+# Enforce version check on import
+_check_version_compatibility()
 
 __all__ = [
   "Experiment",
@@ -55,4 +101,5 @@ __all__ = [
   "LogBuilder",
   "ParametersBuilder",
   "RUN",
+  "userinfo",
 ]

ml_dash/auth/token_storage.py

@@ -292,12 +292,3 @@ def decode_jwt_payload(token: str) -> dict:
     return {}
 
 
-def get_jwt_user():
-  # Load token
-  storage = get_token_storage()
-  token = storage.load("ml-dash-token")
-
-  if token:
-    user = decode_jwt_payload(token)
-    return user
-  return None

ml_dash/auto_start.py

@@ -31,19 +31,34 @@ import atexit
 # Token is auto-loaded from storage when first used
 # If not authenticated, operations will fail with AuthenticationError
 # Prefix format: {owner}/{project}/path...
-# Using getpass to get current user as owner for local convenience
 import getpass
 from datetime import datetime
 
-from .auth.token_storage import get_jwt_user
 from .experiment import Experiment
 
-_user = get_jwt_user()
-# Fallback to system username if not authenticated
-_username = _user["username"] if _user else getpass.getuser()
+# Get username for dxp namespace
+# Note: We use userinfo for fresh data (recommended approach)
+# Falls back to system username if not authenticated
+try:
+    from .client import userinfo
+    _username = userinfo.username or getpass.getuser()
+except Exception:
+    # If userinfo fails (e.g., no network), fall back to system user
+    _username = getpass.getuser()
+
 _now = datetime.now()
 
-dxp = Experiment()
+# Create pre-configured singleton experiment in REMOTE mode
+# - dash_url=True: Use default remote server (https://api.dash.ml)
+# - dash_root=None: Remote-only mode (no local storage)
+# - user: Uses authenticated username from userinfo (fresh from server)
+# - Token is auto-loaded from storage when first used
+# - If not authenticated, operations will fail with AuthenticationError
+dxp = Experiment(
+    user=_username,      # Use authenticated username for namespace
+    dash_url=True,       # Use remote API (https://api.dash.ml)
+    dash_root=None,      # Remote-only mode (no local .dash/)
+)
 
 
 # Register cleanup handler to complete experiment on Python exit (if still open)

ml_dash/cli_commands/profile.py

@@ -24,9 +24,9 @@ def add_parser(subparsers):
     help="Output as JSON",
   )
   parser.add_argument(
-    "--refresh",
+    "--cached",
     action="store_true",
-    help="Fetch fresh profile from server (not from cached token)",
+    help="Use cached token data (default: fetch fresh from server)",
   )
 
 
@@ -45,27 +45,17 @@ def _fetch_fresh_profile(remote_url: str, token: str) -> dict:
 
     client = RemoteClient(remote_url, api_key=token)
 
-    # Query for full user profile
-    query = """
-    query GetUserProfile {
-      me {
-        id
-        username
-        name
-        email
-      }
-    }
-    """
-
-    result = client.graphql_query(query)
-    me = result.get("me", {})
+    # Use the new get_current_user() method
+    user_data = client.get_current_user()
 
-    if me:
+    if user_data:
       return {
-        "sub": me.get("id"),
-        "username": me.get("username"),
-        "name": me.get("name"),
-        "email": me.get("email"),
+        "sub": user_data.get("id"),
+        "username": user_data.get("username"),
+        "name": user_data.get("name"),
+        "email": user_data.get("email"),
+        "given_name": user_data.get("given_name"),
+        "family_name": user_data.get("family_name"),
       }
   except Exception as e:
     # If API call fails, return None to fall back to token decoding
@@ -131,8 +121,13 @@ def cmd_profile(args) -> int:
       info["authenticated"] = False
       info["error"] = "Token expired. Please run 'ml-dash login' to re-authenticate."
     else:
-      # Fetch fresh profile from server if requested, or fall back to token
-      if args.refresh:
+      # Fetch fresh profile from server by default, use cached token only if --cached flag is set
+      if args.cached:
+        # Use cached token data
+        info["user"] = token_payload
+        info["source"] = "token"
+      else:
+        # Fetch fresh data from server (default behavior)
         fresh_profile = _fetch_fresh_profile(config.remote_url, token)
         if fresh_profile:
           info["user"] = fresh_profile
@@ -141,9 +136,6 @@ def cmd_profile(args) -> int:
           info["user"] = token_payload
           info["source"] = "token"
           info["warning"] = "Could not fetch fresh profile from server, using cached token data"
-      else:
-        info["user"] = token_payload
-        info["source"] = "token"
 
       if expiry_message:
         info["token_status"] = expiry_message
@@ -199,9 +191,9 @@ def cmd_profile(args) -> int:
   if info.get("warning"):
     warning_text = f"\n[yellow]⚠ {info['warning']}[/yellow]"
 
-  # Show tip for refreshing
-  if source == "token":
-    tip_text = "\n[dim]Tip: Use --refresh to fetch fresh data from server[/dim]"
+  # Show tip for using cached data
+  if source == "server":
+    tip_text = "\n[dim]Tip: Use --cached to use cached token data (faster but may be outdated)[/dim]"
   else:
     tip_text = None
 

ml_dash/client.py

@@ -6,6 +6,95 @@ from typing import Optional, Dict, Any, List
 import httpx
 
 
+class UserInfo:
+    """
+    Singleton user info object that fetches current user from API server.
+
+    Fetches user info from API server on first access (lazy loading).
+    This queries the API for fresh user data, ensuring up-to-date information.
+
+    Usage:
+        >>> from ml_dash import userinfo
+        >>> if userinfo.username:
+        ...     print(f"Namespace: {userinfo.username}")
+        ...     print(f"Email: {userinfo.email}")
+        ...     print(f"Project: {userinfo.username}/my-project")
+    """
+
+    def __init__(self):
+        self._data = None
+        self._fetched = False
+
+    def _fetch(self):
+        """Fetch user info from API server (lazy loading)."""
+        if self._fetched:
+            return
+
+        self._fetched = True
+        try:
+            client = RemoteClient("https://api.dash.ml")
+            self._data = client.get_current_user()
+        except Exception:
+            self._data = None
+
+    @property
+    def username(self) -> Optional[str]:
+        """Username (namespace) - e.g., 'tom_tao_e4c2c9'"""
+        self._fetch()
+        return self._data.get("username") if self._data else None
+
+    @property
+    def email(self) -> Optional[str]:
+        """User email"""
+        self._fetch()
+        return self._data.get("email") if self._data else None
+
+    @property
+    def name(self) -> Optional[str]:
+        """Full name"""
+        self._fetch()
+        return self._data.get("name") if self._data else None
+
+    @property
+    def given_name(self) -> Optional[str]:
+        """First/given name"""
+        self._fetch()
+        return self._data.get("given_name") if self._data else None
+
+    @property
+    def family_name(self) -> Optional[str]:
+        """Last/family name"""
+        self._fetch()
+        return self._data.get("family_name") if self._data else None
+
+    @property
+    def picture(self) -> Optional[str]:
+        """Profile picture URL"""
+        self._fetch()
+        return self._data.get("picture") if self._data else None
+
+    @property
+    def id(self) -> Optional[str]:
+        """User ID"""
+        self._fetch()
+        return self._data.get("id") if self._data else None
+
+    def __bool__(self) -> bool:
+        """Return True if user is authenticated and data was fetched successfully."""
+        self._fetch()
+        return self._data is not None
+
+    def __repr__(self) -> str:
+        self._fetch()
+        if self._data:
+            return f"UserInfo(username='{self.username}', email='{self.email}')"
+        return "UserInfo(not authenticated)"
+
+
+# Create singleton instance
+userinfo = UserInfo()
+
+
 def _serialize_value(value: Any) -> Any:
     """
     Convert value to JSON-serializable format.
@@ -140,6 +229,44 @@ class RemoteClient:
         except Exception:
             return None
 
+    def get_current_user(self) -> Optional[Dict[str, Any]]:
+        """
+        Get current authenticated user's info from server.
+
+        This queries the API server for fresh user data, ensuring up-to-date information.
+
+        Returns:
+            User info dict with keys: username, email, name, given_name, family_name, picture
+            Returns None if not authenticated or if query fails
+
+        Example:
+            >>> client = RemoteClient("https://api.dash.ml")
+            >>> user = client.get_current_user()
+            >>> print(user["username"])  # e.g., "tom_tao_e4c2c9"
+            >>> print(user["email"])     # e.g., "user@example.com"
+        """
+        try:
+            self._ensure_authenticated()
+
+            # Query server for current user's complete profile
+            query = """
+            query GetCurrentUser {
+              me {
+                id
+                username
+                email
+                name
+                given_name
+                family_name
+                picture
+              }
+            }
+            """
+            result = self.graphql_query(query)
+            return result.get("me")
+        except Exception:
+            return None
+
     def _ensure_authenticated(self):
         """Check if authenticated, raise error if not."""
         if not self.api_key:

ml_dash/run.py

@@ -223,7 +223,7 @@ class RUN:
         # experiments/vision/resnet/train.py
         from ml_dash import RUN
 
-        RUN.__post_init__(entry=__file__)
+        RUN(entry=__file__)
         # Result: RUN.prefix = "vision/resnet", RUN.name = "resnet"
     """
     # Use provided entry or try to auto-detect from caller

{ml_dash-0.6.9.dist-info → ml_dash-0.6.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ml-dash
-Version: 0.6.9
+Version: 0.6.10
 Summary: ML experiment tracking and data storage
 Keywords: machine-learning,experiment-tracking,mlops,data-storage
 Author: Ge Yang, Tom Tao

{ml_dash-0.6.9.dist-info → ml_dash-0.6.10.dist-info}/RECORD

@@ -1,11 +1,11 @@
-ml_dash/__init__.py,sha256=B1oYUXt2DnCctz5nWEqkg1M_XtbJNxaOPxbg54C0Vdc,1555
+ml_dash/__init__.py,sha256=bsn9dwFrkXyZNcm6A4pAKS3ow1Q-2C5Z1jpak-at0Wo,2995
 ml_dash/auth/__init__.py,sha256=3lwM-Y8UBHPU1gFW2JNpmXlPVTnkGudWLKNFFKulQfo,1200
 ml_dash/auth/constants.py,sha256=ku4QzQUMNjvyJwjy7AUdywMAZd59jXSxNHZxDiagUWU,280
 ml_dash/auth/device_flow.py,sha256=DQOdPNlZCuU1umZOA_A6WXdRM3zWphnyo9IntToBl_A,7921
 ml_dash/auth/device_secret.py,sha256=qUsz6M9S1GEIukvmz57eJEp57srSx74O4MU9mZEeDlE,1158
 ml_dash/auth/exceptions.py,sha256=IeBwUzoaTyFtPwd4quFOIel49inIzuabe_ChEeEXEWI,725
-ml_dash/auth/token_storage.py,sha256=L18W8J7D1LlCDlY3Q32l0RXeNh0o7YVDQeeGYm64Dgw,8163
-ml_dash/auto_start.py,sha256=FHTTui33bxfk5rHf-VY2RFcd2JQ_LsNpkUu7Qtpc-GY,1721
+ml_dash/auth/token_storage.py,sha256=9YQXGrn41UVyc1wUvZYbTYLzxSt5NGOyNFNjeX28bjA,7976
+ml_dash/auto_start.py,sha256=mYNjLGI2kyylIfOX5wGOR74gb9UlXg1n5OUQu7aw5SE,2412
 ml_dash/buffer.py,sha256=i4-PZ703_yuKJPAXpmWBGm8jHAAdIBqiA0NIZQEc3wo,26201
 ml_dash/cli.py,sha256=X8LsQA8Wfa1XuXsbvePaGo6NYer7f8CNzy33VT3jrqg,2740
 ml_dash/cli_commands/__init__.py,sha256=bjAmV7MsW-bhtW_4SnLJ0Cfkt9h82vMDC8ebW1Ke8KE,38
@@ -15,9 +15,9 @@ ml_dash/cli_commands/download.py,sha256=Jw-ZeVH8SL9t2yRNCwSmQ0qUOzI4_iWGWt7SwAfM
 ml_dash/cli_commands/list.py,sha256=H442wOAcWYtDwq6BS7lpZbkKhqfTXBCHGctbw8zT1Zw,20841
 ml_dash/cli_commands/login.py,sha256=zX-urtUrfzg2qOGtKNYQgj6UloN9kzj4zEO6h_xwuNs,6782
 ml_dash/cli_commands/logout.py,sha256=lTUUNyRXqvo61qNkCd4KBrPUujDAHnNqsHkU6bHie0U,1332
-ml_dash/cli_commands/profile.py,sha256=AsdwAA2bmSTVnYyJ31iddkSwYJfYqJ60V1UGjFpa3YI,5869
+ml_dash/cli_commands/profile.py,sha256=PoRO1XA4bnOINptj4AO0SyNDBADeryPJBfgC74327e4,5997
 ml_dash/cli_commands/upload.py,sha256=SSUfXC3qoNpoFvPM_ia-ing_N50LNiAvMy9op6FM9Ew,49664
-ml_dash/client.py,sha256=L-FZysVAuIovJJZ1MlDMPhOPJFyL77pMuBxtucG88x8,62005
+ml_dash/client.py,sha256=WgdQRwI9OzEB9dtBtkjFOWNP1t1jj7wr4gvcDUYzZBM,65806
 ml_dash/config.py,sha256=oz2xvoBh2X_xUXWr92cPD5nFxXMT5LxVNypv5B5O0fA,3116
 ml_dash/experiment.py,sha256=sq5Bu-vcKhzTQlUsI6SzIHUZyUkxn7ZSMp_AbhFe4Bw,43462
 ml_dash/files.py,sha256=tGJCTxPfd9vmfvIEqstZjzLvqmHzMZffPXHz0jU9bYU,54441
@@ -26,11 +26,11 @@ ml_dash/metric.py,sha256=ghD1jnuv6dbjV1Jlo7q0mx9UEzpdto2Y1-oDWrSfg04,25809
 ml_dash/params.py,sha256=pPFvknJAJX5uhckzjO1r-HNnKbQFFKDISFmOXNET5eY,9046
 ml_dash/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ml_dash/remote_auto_start.py,sha256=5fvQDHv1CWEKFb6WAa5_uyEInwV_SvotXjOO_6i6ZKE,1687
-ml_dash/run.py,sha256=_a1l5qswwMgrhvajx_lI-I5IXhfHDeH3hfLAJ7Uk_Gs,8847
+ml_dash/run.py,sha256=Hlt_YHaN95TC3TDejzLjFmW9EWyKWi6ruibw-eiPm2U,8833
 ml_dash/snowflake.py,sha256=14rEpRU5YltsmmmZW0EMUy_hdv5S5ME9gWVtmdmwfiU,4917
 ml_dash/storage.py,sha256=x1W-dK6wQY36-LVOJ4kA8Dn07ObNQuIErQWJ3b0PoGY,44910
 ml_dash/track.py,sha256=Dfg1ZnmKZ_FlE5ZfG8Qld_wN4RIMs3nrOxrxwf3thiY,8164
-ml_dash-0.6.9.dist-info/WHEEL,sha256=z-mOpxbJHqy3cq6SvUThBZdaLGFZzdZPtgWLcP2NKjQ,79
-ml_dash-0.6.9.dist-info/entry_points.txt,sha256=dYs2EHX1uRNO7AQGNnVaJJpgiy0Z9q7tiy4fHSyaf3Q,46
-ml_dash-0.6.9.dist-info/METADATA,sha256=f2YU7ukCO9VmIQ1zrctXaRBOA5D6t5caT89JLMD6Sxw,9535
-ml_dash-0.6.9.dist-info/RECORD,,
+ml_dash-0.6.10.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
+ml_dash-0.6.10.dist-info/entry_points.txt,sha256=dYs2EHX1uRNO7AQGNnVaJJpgiy0Z9q7tiy4fHSyaf3Q,46
+ml_dash-0.6.10.dist-info/METADATA,sha256=z9L0TSRTGJUSZ1gDYNR3Pj5IW3oh0AYeMFjEplnhFYA,9536
+ml_dash-0.6.10.dist-info/RECORD,,

{ml_dash-0.6.9.dist-info → ml_dash-0.6.10.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.15
+Generator: uv 0.9.26
 Root-Is-Purelib: true
-Tag: py3-none-any
+Tag: py3-none-any