Skip to content

Conversation

@armishra
Copy link
Contributor

@armishra armishra commented Nov 12, 2025

🚀 What

Add --team parameter to training project creation

Adds support for selecting a team when creating training projects via truss train push. If multiple teams are available, users are prompted to select one. If no teams are available, no team parameter is sent to the API.

💻 How

  • Added get_teams() GraphQL query method to fetch available teams
  • Added inquire_team() function that prompts for team selection when multiple teams exist
  • Added team_name optional field to TrainingProject model
  • Added --team CLI option to truss train push command
  • Updated deployment functions to accept and pass team_name parameter
  • Modified upsert_training_project() API call to exclude team_name when None using exclude_none=True

🔬 Testing

  • CI
  • E2E Tests:
$ truss train push config.py --remote baseten-staging --team team3 --job-name "archit_staging_team3"
ERROR ClickException: Team 'team3' does not exist. Available teams: architstaging, research, t3

Created TrainingProject on staging:
Screenshot 2025-11-12 at 3 05 40 PM

@linear
Copy link

linear bot commented Nov 12, 2025

@armishra armishra force-pushed the archit/cor-422-trainingproject-creation-through-truss branch 3 times, most recently from 97100f0 to ebd8401 Compare November 12, 2025 23:14
@armishra armishra force-pushed the archit/cor-422-trainingproject-creation-through-truss branch from ebd8401 to fdb586e Compare November 12, 2025 23:21
@armishra armishra closed this Nov 12, 2025
@armishra armishra deleted the archit/cor-422-trainingproject-creation-through-truss branch November 12, 2025 23:51
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.

2 participants