Skip to content

docs(cli): update truss ssh setup docstring for model support#2393

Merged
pat-baseten merged 1 commit into
mainfrom
pat/ssh-setup-docstring
Apr 23, 2026
Merged

docs(cli): update truss ssh setup docstring for model support#2393
pat-baseten merged 1 commit into
mainfrom
pat/ssh-setup-docstring

Conversation

@pat-baseten

@pat-baseten pat-baseten commented Apr 23, 2026

Copy link
Copy Markdown
Contributor

Summary

  • truss ssh setup's docstring claims training-only scope, but the proxy script has handled model hostnames since Remote SSH support for model containers #2377.
  • The docstring's hostname example (<job-id>-<replica>...) predates the training-job- prefix that proxy_command.py:120 now requires, so copy-pasting it would fail.

Change

Docstring-only update to truss/cli/ssh_commands.py:

  • Broaden "training jobs" → "workloads" in summary and body.
  • Replace the single stale example with current training-job and model-deployment formats.

truss ssh setup's docstring claims training-only scope and shows a
hostname format (<job-id>-<replica>...) that predates the
training-job- prefix that proxy_command.py now requires. Update the
docstring to match current behavior: both training jobs and model
deployments, correct hostname formats.
@CLAassistant

CLAassistant commented Apr 23, 2026

Copy link
Copy Markdown

CLA assistant check
All committers have signed the CLA.

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Updates truss ssh setup CLI help text to reflect that SSH access supports both training jobs and model deployments, and refreshes the hostname examples to current formats.

Changes:

  • Broaden wording from “training jobs” to “workloads”.
  • Replace the legacy training-job hostname example with current training-job and model-deployment examples.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment thread truss/cli/ssh_commands.py
Comment on lines +43 to +53
"""One-time setup: configure SSH access for Baseten workloads.

Generates an SSH keypair, installs a ProxyCommand script, and adds a
wildcard Host entry to ~/.ssh/config. After running this once, connect
to any running training job with:
to any running workload with:

ssh <job-id>-<replica>.<remote>.ssh.baseten.co
# Training job
ssh training-job-<job_id>-<node>.ssh.baseten.co

Example: ssh 5wo5n3y-0.dev.ssh.baseten.co
# Model deployment (requires runtime.remote_ssh.enabled)
ssh model-<model_id>-<deployment_id>.ssh.baseten.co

@William-Gao1 William-Gao1 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@pat-baseten pat-baseten merged commit 661935d into main Apr 23, 2026
40 checks passed
@pat-baseten pat-baseten deleted the pat/ssh-setup-docstring branch April 23, 2026 15:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants