lint

Author: ninthreezy

datatorch/agent/__init__.py

@@ -139,12 +139,12 @@ async def stop() -> None:
 
     print(" ")
 
-    logger.warning("Gracefully exiting agent.")
+    logger.warning("Gracefully exiting runner.")
 
-    logger.info("Closing agent jobs.")
+    logger.info("Closing runner jobs.")
     await _exit_jobs()
 
-    logger.info("Closing agent jobs.")
+    logger.info("Closing runner jobs.")
     await _exit_jobs()
 
     logger.info("Closing all other tasks.")

datatorch/cli/agent/__init__.py

@@ -5,11 +5,28 @@
 from .dir import dir
 
 
-@click.group(help="Commands for managing agents.")
-def agent():
+@click.group(
+    name="runner",
+    help=(
+        "Commands for managing runners — self-hosted compute that "
+        "executes pipeline jobs. (Formerly 'agent'.)"
+    ),
+)
+def runner():
     pass
 
 
-agent.add_command(start)
-agent.add_command(create)
-agent.add_command(dir)
+runner.add_command(start)
+runner.add_command(create)
+runner.add_command(dir)
+
+# Deprecated alias: `datatorch agent <cmd>` keeps working for existing
+# installs/scripts, sharing the exact command objects; hidden from
+# --help so new users learn `runner`. The wire protocol (agent token,
+# GraphQL mutations) and the on-disk agent directory are unchanged.
+agent = click.Group(
+    name="agent",
+    commands=runner.commands,
+    hidden=True,
+    help="Deprecated alias for 'runner'.",
+)

datatorch/cli/agent/create.py

@@ -35,8 +35,10 @@ def create(ctx):
     agent_settings = agent_directory.settings
 
     if agent_settings.agent_id:
-        click.echo("An agent is already installed.")
-        confirmed = click.confirm("Would you like to create a new agent?", default=True)
+        click.echo("A runner is already installed.")
+        confirmed = click.confirm(
+            "Would you like to create a new runner?", default=True
+        )
         if not confirmed:
             return
 
@@ -48,17 +50,17 @@ def create(ctx):
     if not user_settings.api_key:
         ctx.invoke(login, host=user_settings.api_url)
 
-    name = click.prompt("Enter agents name", default=platform.node(), show_default=True)
-    spinner = Spinner("Creating agent")
+    name = click.prompt("Enter runner name", default=platform.node(), show_default=True)
+    spinner = Spinner("Creating runner")
 
     try:
         agent = create_agent(name)["createAgent"]
         agent_settings.agent_id = agent["agent"]["id"]
         agent_settings.agent_token = agent["token"]
         agent_settings.api_url = user_settings.api_url
     except Exception as ex:
-        spinner.done(click.style("Failed to new create agent.", fg="red"))
+        spinner.done(click.style("Failed to create runner.", fg="red"))
         click.echo(ex)
         return
 
-    spinner.done("Successfully created agent.")
+    spinner.done("Successfully created runner.")

datatorch/cli/agent/dir.py

@@ -5,7 +5,7 @@
 from datatorch.agent import agent_directory
 
 
-@click.command(help="Opens agent directory where settings and files are stored.")
+@click.command(help="Opens the runner directory where settings and files are stored.")
 def dir():
     path = agent_directory.dir
     if platform.system() == "Windows":

datatorch/cli/agent/start.py

@@ -15,7 +15,7 @@ def add_signal_handlers(loop: AbstractEventLoop):
         pass  # Ignore if not implemented. Means this program is running in windows.
 
 
-@click.command(help="Run an agent")
+@click.command(help="Start the runner (executes pipeline jobs on this machine)")
 def start():
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)

datatorch/cli/groups.py

@@ -9,7 +9,7 @@
 from .main.upgrade import package_upgrade
 from .main.version import version
 
-from .agent import agent
+from .agent import agent, runner
 from .pipeline import pipeline
 from .action import action
 from .import_cmds import import_cmd
@@ -29,6 +29,7 @@ def main():
 main.add_command(package_upgrade)
 
 main.add_command(pipeline)
+main.add_command(runner)
 main.add_command(agent)
 main.add_command(action)
 main.add_command(import_cmd)

setup.py

@@ -33,7 +33,7 @@
 
 setup(
     name="datatorch",
-    version="0.6.4",
+    version="0.7.0",
     description="A CLI and library for interacting with DataTorch.",
     author="DataTorch",
     author_email="support@datatorch.io",