wandb 0.22.0__py3-none-win32.whl → 0.22.2__py3-none-win32.whl

wandb/automations/_generated/__init__.py

@@ -1,99 +1,5 @@
 # Generated by ariadne-codegen
 
-from .create_automation import CreateAutomation
-from .create_generic_webhook_integration import (
-    CreateGenericWebhookIntegration,
-    CreateGenericWebhookIntegrationCreateGenericWebhookIntegration,
-    CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration,
-    CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration,
-)
-from .delete_automation import DeleteAutomation
-from .enums import (
-    AlertSeverity,
-    EventTriggeringConditionType,
-    TriggeredActionType,
-    TriggerScopeType,
-)
-from .fragments import (
-    ArtifactPortfolioScopeFields,
-    ArtifactSequenceScopeFields,
-    FilterEventFields,
-    GenericWebhookActionFields,
-    GenericWebhookActionFieldsIntegrationGenericWebhookIntegration,
-    GenericWebhookActionFieldsIntegrationIntegration,
-    GenericWebhookIntegrationConnectionFields,
-    GenericWebhookIntegrationConnectionFieldsEdges,
-    GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
-    GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration,
-    GenericWebhookIntegrationFields,
-    IntegrationConnectionFields,
-    IntegrationConnectionFieldsEdges,
-    IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
-    IntegrationConnectionFieldsEdgesNodeIntegration,
-    IntegrationConnectionFieldsEdgesNodeSlackIntegration,
-    NoOpActionFields,
-    NotificationActionFields,
-    NotificationActionFieldsIntegrationIntegration,
-    NotificationActionFieldsIntegrationSlackIntegration,
-    PageInfoFields,
-    ProjectConnectionFields,
-    ProjectConnectionFieldsEdges,
-    ProjectConnectionFieldsEdgesNode,
-    ProjectScopeFields,
-    QueueJobActionFields,
-    QueueJobActionFieldsQueue,
-    SlackIntegrationConnectionFields,
-    SlackIntegrationConnectionFieldsEdges,
-    SlackIntegrationConnectionFieldsEdgesNodeIntegration,
-    SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration,
-    SlackIntegrationFields,
-    TriggerFields,
-    TriggerFieldsActionGenericWebhookTriggeredAction,
-    TriggerFieldsActionNoOpTriggeredAction,
-    TriggerFieldsActionNotificationTriggeredAction,
-    TriggerFieldsActionQueueJobTriggeredAction,
-    TriggerFieldsEventFilterEventTriggeringCondition,
-    TriggerFieldsScopeArtifactPortfolio,
-    TriggerFieldsScopeArtifactSequence,
-    TriggerFieldsScopeProject,
-)
-from .generic_webhook_integrations_by_entity import (
-    GenericWebhookIntegrationsByEntity,
-    GenericWebhookIntegrationsByEntityEntity,
-)
-from .get_automations import GetAutomations, GetAutomationsSearchScope
-from .get_automations_by_entity import (
-    GetAutomationsByEntity,
-    GetAutomationsByEntitySearchScope,
-)
-from .input_types import (
-    CreateFilterTriggerInput,
-    CreateGenericWebhookIntegrationInput,
-    GenericWebhookActionInput,
-    NoOpTriggeredActionInput,
-    NotificationActionInput,
-    QueueJobActionInput,
-    TriggeredActionConfig,
-    UpdateFilterTriggerInput,
-)
-from .integrations_by_entity import IntegrationsByEntity, IntegrationsByEntityEntity
-from .operations import (
-    CREATE_AUTOMATION_GQL,
-    CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL,
-    DELETE_AUTOMATION_GQL,
-    GENERIC_WEBHOOK_INTEGRATIONS_BY_ENTITY_GQL,
-    GET_AUTOMATIONS_BY_ENTITY_GQL,
-    GET_AUTOMATIONS_GQL,
-    INTEGRATIONS_BY_ENTITY_GQL,
-    SLACK_INTEGRATIONS_BY_ENTITY_GQL,
-    UPDATE_AUTOMATION_GQL,
-)
-from .slack_integrations_by_entity import (
-    SlackIntegrationsByEntity,
-    SlackIntegrationsByEntityEntity,
-)
-from .update_automation import UpdateAutomation
-
 __all__ = [
     "CREATE_AUTOMATION_GQL",
     "CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL",
@@ -105,22 +11,14 @@ __all__ = [
     "SLACK_INTEGRATIONS_BY_ENTITY_GQL",
     "UPDATE_AUTOMATION_GQL",
     "GetAutomations",
-    "GetAutomationsSearchScope",
     "GetAutomationsByEntity",
-    "GetAutomationsByEntitySearchScope",
     "CreateAutomation",
     "UpdateAutomation",
     "DeleteAutomation",
     "IntegrationsByEntity",
-    "IntegrationsByEntityEntity",
     "SlackIntegrationsByEntity",
-    "SlackIntegrationsByEntityEntity",
     "GenericWebhookIntegrationsByEntity",
-    "GenericWebhookIntegrationsByEntityEntity",
     "CreateGenericWebhookIntegration",
-    "CreateGenericWebhookIntegrationCreateGenericWebhookIntegration",
-    "CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration",
-    "CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration",
     "CreateFilterTriggerInput",
     "CreateGenericWebhookIntegrationInput",
     "GenericWebhookActionInput",
@@ -133,45 +31,74 @@ __all__ = [
     "ArtifactSequenceScopeFields",
     "FilterEventFields",
     "GenericWebhookActionFields",
-    "GenericWebhookActionFieldsIntegrationGenericWebhookIntegration",
-    "GenericWebhookActionFieldsIntegrationIntegration",
     "GenericWebhookIntegrationConnectionFields",
-    "GenericWebhookIntegrationConnectionFieldsEdges",
-    "GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration",
-    "GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration",
     "GenericWebhookIntegrationFields",
     "IntegrationConnectionFields",
-    "IntegrationConnectionFieldsEdges",
-    "IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration",
-    "IntegrationConnectionFieldsEdgesNodeIntegration",
-    "IntegrationConnectionFieldsEdgesNodeSlackIntegration",
     "NoOpActionFields",
     "NotificationActionFields",
-    "NotificationActionFieldsIntegrationIntegration",
-    "NotificationActionFieldsIntegrationSlackIntegration",
     "PageInfoFields",
     "ProjectConnectionFields",
-    "ProjectConnectionFieldsEdges",
-    "ProjectConnectionFieldsEdgesNode",
     "ProjectScopeFields",
     "QueueJobActionFields",
-    "QueueJobActionFieldsQueue",
     "SlackIntegrationConnectionFields",
-    "SlackIntegrationConnectionFieldsEdges",
-    "SlackIntegrationConnectionFieldsEdgesNodeIntegration",
-    "SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration",
     "SlackIntegrationFields",
     "TriggerFields",
-    "TriggerFieldsActionGenericWebhookTriggeredAction",
-    "TriggerFieldsActionNoOpTriggeredAction",
-    "TriggerFieldsActionNotificationTriggeredAction",
-    "TriggerFieldsActionQueueJobTriggeredAction",
-    "TriggerFieldsEventFilterEventTriggeringCondition",
-    "TriggerFieldsScopeArtifactPortfolio",
-    "TriggerFieldsScopeArtifactSequence",
-    "TriggerFieldsScopeProject",
     "AlertSeverity",
     "EventTriggeringConditionType",
     "TriggerScopeType",
     "TriggeredActionType",
 ]
+from .create_automation import CreateAutomation
+from .create_generic_webhook_integration import CreateGenericWebhookIntegration
+from .delete_automation import DeleteAutomation
+from .enums import (
+    AlertSeverity,
+    EventTriggeringConditionType,
+    TriggeredActionType,
+    TriggerScopeType,
+)
+from .fragments import (
+    ArtifactPortfolioScopeFields,
+    ArtifactSequenceScopeFields,
+    FilterEventFields,
+    GenericWebhookActionFields,
+    GenericWebhookIntegrationConnectionFields,
+    GenericWebhookIntegrationFields,
+    IntegrationConnectionFields,
+    NoOpActionFields,
+    NotificationActionFields,
+    PageInfoFields,
+    ProjectConnectionFields,
+    ProjectScopeFields,
+    QueueJobActionFields,
+    SlackIntegrationConnectionFields,
+    SlackIntegrationFields,
+    TriggerFields,
+)
+from .generic_webhook_integrations_by_entity import GenericWebhookIntegrationsByEntity
+from .get_automations import GetAutomations
+from .get_automations_by_entity import GetAutomationsByEntity
+from .input_types import (
+    CreateFilterTriggerInput,
+    CreateGenericWebhookIntegrationInput,
+    GenericWebhookActionInput,
+    NoOpTriggeredActionInput,
+    NotificationActionInput,
+    QueueJobActionInput,
+    TriggeredActionConfig,
+    UpdateFilterTriggerInput,
+)
+from .integrations_by_entity import IntegrationsByEntity
+from .operations import (
+    CREATE_AUTOMATION_GQL,
+    CREATE_GENERIC_WEBHOOK_INTEGRATION_GQL,
+    DELETE_AUTOMATION_GQL,
+    GENERIC_WEBHOOK_INTEGRATIONS_BY_ENTITY_GQL,
+    GET_AUTOMATIONS_BY_ENTITY_GQL,
+    GET_AUTOMATIONS_GQL,
+    INTEGRATIONS_BY_ENTITY_GQL,
+    SLACK_INTEGRATIONS_BY_ENTITY_GQL,
+    UPDATE_AUTOMATION_GQL,
+)
+from .slack_integrations_by_entity import SlackIntegrationsByEntity
+from .update_automation import UpdateAutomation

wandb/automations/_generated/create_generic_webhook_integration.py

@@ -21,7 +21,7 @@ class CreateGenericWebhookIntegration(GQLBase):
 class CreateGenericWebhookIntegrationCreateGenericWebhookIntegration(GQLBase):
     integration: Union[
         CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationIntegration,
-        CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration,
+        GenericWebhookIntegrationFields,
     ] = Field(discriminator="typename__")
 
 
@@ -33,11 +33,5 @@ class CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationI
     ]
 
 
-class CreateGenericWebhookIntegrationCreateGenericWebhookIntegrationIntegrationGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
 CreateGenericWebhookIntegration.model_rebuild()
 CreateGenericWebhookIntegrationCreateGenericWebhookIntegration.model_rebuild()

wandb/automations/_generated/fragments.py

@@ -52,7 +52,7 @@ class GenericWebhookActionFields(GQLBase):
     )
     integration: Union[
         GenericWebhookActionFieldsIntegrationIntegration,
-        GenericWebhookActionFieldsIntegrationGenericWebhookIntegration,
+        GenericWebhookIntegrationFields,
     ] = Field(discriminator="typename__")
     request_payload: Optional[str] = Field(alias="requestPayload")
 
@@ -75,7 +75,7 @@ class GenericWebhookIntegrationConnectionFieldsEdges(GQLBase):
         Annotated[
             Union[
                 GenericWebhookIntegrationConnectionFieldsEdgesNodeIntegration,
-                GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
+                GenericWebhookIntegrationFields,
             ],
             Field(discriminator="typename__"),
         ]
@@ -109,8 +109,8 @@ class IntegrationConnectionFieldsEdges(GQLBase):
         Annotated[
             Union[
                 IntegrationConnectionFieldsEdgesNodeIntegration,
-                IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration,
-                IntegrationConnectionFieldsEdgesNodeSlackIntegration,
+                GenericWebhookIntegrationFields,
+                SlackIntegrationFields,
             ],
             Field(discriminator="typename__"),
         ]
@@ -131,8 +131,7 @@ class NotificationActionFields(GQLBase):
         "NotificationTriggeredAction"
     )
     integration: Union[
-        NotificationActionFieldsIntegrationIntegration,
-        NotificationActionFieldsIntegrationSlackIntegration,
+        NotificationActionFieldsIntegrationIntegration, SlackIntegrationFields
     ] = Field(discriminator="typename__")
     title: Optional[str]
     message: Optional[str]
@@ -194,7 +193,7 @@ class SlackIntegrationConnectionFieldsEdges(GQLBase):
         Annotated[
             Union[
                 SlackIntegrationConnectionFieldsEdgesNodeIntegration,
-                SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration,
+                SlackIntegrationFields,
             ],
             Field(discriminator="typename__"),
         ]
@@ -223,16 +222,14 @@ class TriggerFields(GQLBase):
     description: Optional[str]
     enabled: bool
     scope: Union[
-        TriggerFieldsScopeProject,
-        TriggerFieldsScopeArtifactSequence,
-        TriggerFieldsScopeArtifactPortfolio,
+        ProjectScopeFields, ArtifactSequenceScopeFields, ArtifactPortfolioScopeFields
     ] = Field(discriminator="typename__")
-    event: TriggerFieldsEventFilterEventTriggeringCondition
+    event: FilterEventFields
     action: Union[
-        TriggerFieldsActionQueueJobTriggeredAction,
-        TriggerFieldsActionNotificationTriggeredAction,
-        TriggerFieldsActionGenericWebhookTriggeredAction,
-        TriggerFieldsActionNoOpTriggeredAction,
+        QueueJobActionFields,
+        NotificationActionFields,
+        GenericWebhookActionFields,
+        NoOpActionFields,
     ] = Field(discriminator="typename__")
 
 
@@ -243,68 +240,6 @@ class UpdateAutomationResult(GQLBase):
     trigger: Optional[TriggerFields]
 
 
-class TriggerFieldsScopeArtifactPortfolio(ArtifactPortfolioScopeFields):
-    typename__: Typename[Literal["ArtifactPortfolio"]]
-
-
-class TriggerFieldsScopeArtifactSequence(ArtifactSequenceScopeFields):
-    typename__: Typename[Literal["ArtifactSequence"]]
-
-
-class TriggerFieldsEventFilterEventTriggeringCondition(FilterEventFields):
-    typename__: Typename[Literal["FilterEventTriggeringCondition"]]
-
-
-class TriggerFieldsActionGenericWebhookTriggeredAction(GenericWebhookActionFields):
-    typename__: Typename[Literal["GenericWebhookTriggeredAction"]]
-
-
-class GenericWebhookActionFieldsIntegrationGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
-class GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
-class IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration(
-    GenericWebhookIntegrationFields
-):
-    typename__: Typename[Literal["GenericWebhookIntegration"]]
-
-
-class TriggerFieldsActionNoOpTriggeredAction(NoOpActionFields):
-    typename__: Typename[Literal["NoOpTriggeredAction"]]
-
-
-class TriggerFieldsActionNotificationTriggeredAction(NotificationActionFields):
-    typename__: Typename[Literal["NotificationTriggeredAction"]]
-
-
-class TriggerFieldsScopeProject(ProjectScopeFields):
-    typename__: Typename[Literal["Project"]]
-
-
-class TriggerFieldsActionQueueJobTriggeredAction(QueueJobActionFields):
-    typename__: Typename[Literal["QueueJobTriggeredAction"]]
-
-
-class IntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
-    typename__: Typename[Literal["SlackIntegration"]]
-
-
-class NotificationActionFieldsIntegrationSlackIntegration(SlackIntegrationFields):
-    typename__: Typename[Literal["SlackIntegration"]]
-
-
-class SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration(SlackIntegrationFields):
-    typename__: Typename[Literal["SlackIntegration"]]
-
-
 ArtifactPortfolioScopeFields.model_rebuild()
 ArtifactSequenceScopeFields.model_rebuild()
 CreateAutomationResult.model_rebuild()
@@ -335,24 +270,24 @@ SlackIntegrationConnectionFieldsEdgesNodeIntegration.model_rebuild()
 SlackIntegrationFields.model_rebuild()
 TriggerFields.model_rebuild()
 UpdateAutomationResult.model_rebuild()
-TriggerFieldsScopeArtifactPortfolio.model_rebuild()
-TriggerFieldsScopeArtifactSequence.model_rebuild()
-TriggerFieldsEventFilterEventTriggeringCondition.model_rebuild()
-TriggerFieldsActionGenericWebhookTriggeredAction.model_rebuild()
-GenericWebhookActionFieldsIntegrationGenericWebhookIntegration.model_rebuild()
-GenericWebhookIntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration.model_rebuild()
-IntegrationConnectionFieldsEdgesNodeGenericWebhookIntegration.model_rebuild()
-TriggerFieldsActionNoOpTriggeredAction.model_rebuild()
-TriggerFieldsActionNotificationTriggeredAction.model_rebuild()
+ArtifactPortfolioScopeFields.model_rebuild()
+ArtifactSequenceScopeFields.model_rebuild()
+FilterEventFields.model_rebuild()
+GenericWebhookActionFields.model_rebuild()
+GenericWebhookIntegrationFields.model_rebuild()
+GenericWebhookIntegrationFields.model_rebuild()
+GenericWebhookIntegrationFields.model_rebuild()
+NoOpActionFields.model_rebuild()
+NotificationActionFields.model_rebuild()
 PageInfoFields.model_rebuild()
 PageInfoFields.model_rebuild()
 PageInfoFields.model_rebuild()
 PageInfoFields.model_rebuild()
-TriggerFieldsScopeProject.model_rebuild()
-TriggerFieldsActionQueueJobTriggeredAction.model_rebuild()
-IntegrationConnectionFieldsEdgesNodeSlackIntegration.model_rebuild()
-NotificationActionFieldsIntegrationSlackIntegration.model_rebuild()
-SlackIntegrationConnectionFieldsEdgesNodeSlackIntegration.model_rebuild()
+ProjectScopeFields.model_rebuild()
+QueueJobActionFields.model_rebuild()
+SlackIntegrationFields.model_rebuild()
+SlackIntegrationFields.model_rebuild()
+SlackIntegrationFields.model_rebuild()
 TriggerFields.model_rebuild()
 TriggerFields.model_rebuild()
 TriggerFields.model_rebuild()

wandb/cli/beta.py

@@ -15,8 +15,7 @@ from wandb.util import get_core_path
 
 @click.group()
 def beta():
-    """Beta versions of wandb CLI commands. Requires wandb-core."""
-    # this is the future that requires wandb-core!
+    """Beta versions of wandb CLI commands."""
     import wandb.env
 
     wandb._sentry.configure_scope(process_context="wandb_beta")
@@ -32,6 +31,21 @@ def beta():
         )
 
 
+@beta.command()
+@click.argument("path", nargs=1, type=click.Path(exists=True), required=False)
+def leet(path: str | None = None) -> None:
+    """Launch W&B LEET: the Lightweight Experiment Exploration Tool.
+
+    LEET is a terminal UI for viewing a W&B run specified by an optional PATH.
+
+    PATH can include a .wandb file or a run directory containing a .wandb file.
+    If PATH is not provided, the command will look for the latest run.
+    """
+    from . import beta_leet
+
+    beta_leet.launch(path)
+
+
 @beta.command()
 @click.argument("paths", type=click.Path(exists=True), nargs=-1)
 @click.option(

wandb/cli/beta_leet.py

@@ -0,0 +1,74 @@
+from __future__ import annotations
+
+import os
+import pathlib
+import subprocess
+import sys
+
+import click
+from typing_extensions import Never
+
+import wandb
+from wandb.env import error_reporting_enabled, is_debug
+from wandb.sdk import wandb_setup
+from wandb.util import get_core_path
+
+from .beta_sync import _find_wandb_files
+
+
+def _fatal(message: str) -> Never:
+    """Print an error message and exit with code 1."""
+    click.echo(f"Error: {message}", err=True)
+    sys.exit(1)
+
+
+def _wandb_file_path(path: str | None) -> str:
+    """Returns absolute path to the .wandb file to display with LEET.
+
+    If `path` is not provided, looks for the latest W&B run.
+
+    Prints an error and exits if a valid path is not found.
+    """
+    if not path:
+        wandb_dir = wandb_setup.singleton().settings.wandb_dir
+
+        wandb_run_path = (pathlib.Path(wandb_dir) / "latest-run").resolve()
+    else:
+        wandb_run_path = pathlib.Path(path).resolve()
+
+    wandb_files = list(_find_wandb_files(wandb_run_path, skip_synced=False))
+
+    if len(wandb_files) == 0:
+        _fatal(f"Could not find a .wandb file in {wandb_run_path}.")
+    elif len(wandb_files) > 1:
+        _fatal(f"Found multiple .wandb files in {wandb_run_path}.")
+
+    return wandb_files[0]
+
+
+def launch(path: str | None) -> Never:
+    wandb._sentry.configure_scope(process_context="leet")
+
+    wandb_file = _wandb_file_path(path)
+
+    try:
+        core_path = get_core_path()
+
+        args = [core_path, "leet"]
+        args.append(wandb_file)
+
+        if not error_reporting_enabled():
+            args.append("--no-observability")
+
+        if is_debug(default="False"):
+            args.extend(["--log-level", "-4"])
+
+        result = subprocess.run(
+            args,
+            env=os.environ,
+            close_fds=True,
+        )
+        sys.exit(result.returncode)
+
+    except Exception as e:
+        wandb._sentry.reraise(e)

wandb/cli/beta_sync.py

@@ -14,7 +14,7 @@ import wandb
 from wandb.proto.wandb_sync_pb2 import ServerSyncResponse
 from wandb.sdk import wandb_setup
 from wandb.sdk.lib import asyncio_compat
-from wandb.sdk.lib.printer import ERROR, Printer, new_printer
+from wandb.sdk.lib.printer import Printer, new_printer
 from wandb.sdk.lib.progress import progress_printer
 from wandb.sdk.lib.service.service_connection import ServiceConnection
 from wandb.sdk.mailbox.mailbox_handle import MailboxHandle
@@ -86,12 +86,10 @@ async def _do_sync(
 
     This is factored out to make the progress animation testable.
     """
-    init_result = await service.init_sync(
-        wandb_files,
-        settings,
-    ).wait_async(timeout=5)
+    init_handle = await service.init_sync(wandb_files, settings)
+    init_result = await init_handle.wait_async(timeout=5)
 
-    sync_handle = service.sync(init_result.id, parallelism=parallelism)
+    sync_handle = await service.sync(init_result.id, parallelism=parallelism)
 
     await _SyncStatusLoop(
         init_result.id,
@@ -130,19 +128,19 @@ class _SyncStatusLoop:
         handle: MailboxHandle[ServerSyncResponse],
     ) -> None:
         response = await handle.wait_async(timeout=None)
-        if messages := list(response.errors):
-            self._printer.display(messages, level=ERROR)
+        for msg in response.messages:
+            self._printer.display(msg.content, level=msg.severity)
         self._done.set()
 
     async def _show_progress_until_done(self) -> None:
         """Show rate-limited status updates until _done is set."""
         with progress_printer(self._printer, "Syncing...") as progress:
             while not await self._rate_limit_check_done():
-                handle = self._service.sync_status(self._id)
+                handle = await self._service.sync_status(self._id)
                 response = await handle.wait_async(timeout=None)
 
-                if messages := list(response.new_errors):
-                    self._printer.display(messages, level=ERROR)
+                for msg in response.new_messages:
+                    self._printer.display(msg.content, level=msg.severity)
                 progress.update(response.stats)
 
     async def _rate_limit_check_done(self) -> bool:

wandb/cli/cli.py

@@ -244,7 +244,7 @@ def projects(entity, display=True):
     return projects
 
 
-@cli.command(context_settings=CONTEXT, help="Login to Weights & Biases")
+@cli.command(context_settings=CONTEXT)
 @click.argument("key", nargs=-1)
 @click.option("--cloud", is_flag=True, help="Login to the cloud instead of local")
 @click.option(
@@ -262,6 +262,16 @@ def projects(entity, display=True):
 )
 @display_error
 def login(key, host, cloud, relogin, anonymously, verify, no_offline=False):
+    """Verify and store your API key for authentication with W&B services.
+
+    By default, only store credentials locally without verifying them with W&B.
+    To verify credentials, set `--verify=True`.
+
+    For server deployments (dedicated cloud or customer-managed instances),
+    specify the host URL using the `--host` flag. You can also set environment
+    variables `WANDB_BASE_URL` and `WANDB_API_KEY` instead of running
+    the `login` command with host parameters.
+    """
     # TODO: handle no_offline
     anon_mode = "must" if anonymously else "never"
 
@@ -422,9 +432,7 @@ def init(ctx, project, entity, reset, mode):
     )
 
 
-@cli.command(
-    context_settings=CONTEXT, help="Upload an offline training directory to W&B"
-)
+@cli.command(context_settings=CONTEXT)
 @click.pass_context
 @click.argument("path", nargs=-1, type=click.Path(exists=True))
 @click.option("--view", is_flag=True, default=False, help="View runs", hidden=True)
@@ -518,9 +526,23 @@ def sync(
     skip_console=None,
     replace_tags=None,
 ):
+    """Synchronize W&B run data to the cloud.
+
+    If PATH is provided, sync runs found at the given path. If a path
+    is not specified, search for `./wandb` first, then search for a
+    `wandb/` subdirectory.
+
+    To sync a specific run:
+
+        wandb sync ./wandb/run-20250813_124246-n67z9ude
+
+    Or equivalently:
+
+        wandb sync ./wandb/run-20250813_124246-n67z9ude/run-n67z9ude.wandb
+    """
     api = _get_cling_api()
     if not api.is_authenticated:
-        wandb.termlog("Login to W&B to sync offline runs")
+        wandb.termlog("Login to W&B to sync runs")
         ctx.invoke(login, no_offline=True)
         api = _get_cling_api(reset=True)
 
@@ -2700,9 +2722,10 @@ Run `git clone {repo}` and restore from there or pass the --no-git flag."""
     return commit, json_config, patch_content, repo, metadata
 
 
-@cli.command("online", help="Enable W&B sync")
+@cli.command("online")
 @display_error
 def online():
+    """Undo `wandb offline`."""
     api = InternalApi()
     try:
         api.clear_setting("mode", persist=True)
@@ -2713,9 +2736,13 @@ def online():
     )
 
 
-@cli.command("offline", help="Disable W&B sync")
+@cli.command("offline")
 @display_error
 def offline():
+    """Save data logged to W&B locally without uploading it to the cloud.
+
+    Use `wandb online` or `wandb sync` to upload offline runs.
+    """
     api = InternalApi()
     try:
         api.set_setting("mode", "offline", persist=True)