Skip to content

Commit 724a8ea

Browse files
aanaseerRobertCraigie
authored andcommitted
add fine_tuning.jobs.cancel
1 parent d44dd66 commit 724a8ea

File tree

1 file changed

+20
-1
lines changed
  • src/openai/cli/_api/fine_tuning

1 file changed

+20
-1
lines changed

src/openai/cli/_api/fine_tuning/jobs.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,16 @@ def register(subparser: _SubParsersAction[ArgumentParser]) -> None:
4444
sub.set_defaults(func=CLIFineTuningJobs.list, args_model=CLIFineTuningJobsListArgs)
4545

4646
sub = subparser.add_parser("fine_tuning.jobs.cancel")
47+
sub.add_argument(
48+
"-i",
49+
"--id",
50+
help="The ID of the fine-tuning job to cancel.",
51+
required=True,
52+
)
53+
sub.set_defaults(
54+
func=CLIFineTuningJobs.cancel, args_model=CLIFineTuningJobsCancelArgs
55+
)
56+
4757
sub = subparser.add_parser("fine_tuning.jobs.list_events")
4858

4959

@@ -54,6 +64,8 @@ class CLIFineTuningJobsListArgs(BaseModel):
5464
after: NotGivenOr[str] = NOT_GIVEN
5565
limit: NotGivenOr[int] = NOT_GIVEN
5666

67+
class CLIFineTuningJobsCancelArgs(BaseModel):
68+
id: str
5769

5870
class CLIFineTuningJobs:
5971
@staticmethod
@@ -70,4 +82,11 @@ def list(args: CLIFineTuningJobsListArgs) -> None:
7082
] = get_client().fine_tuning.jobs.list(
7183
after=args.after or NOT_GIVEN, limit=args.limit or NOT_GIVEN
7284
)
73-
print_model(fine_tuning_jobs)
85+
print_model(fine_tuning_jobs)
86+
87+
@staticmethod
88+
def cancel(args: CLIFineTuningJobsCancelArgs) -> None:
89+
fine_tuning_job: FineTuningJob = get_client().fine_tuning.jobs.cancel(
90+
fine_tuning_job_id=args.id
91+
)
92+
print_model(fine_tuning_job)

0 commit comments

Comments
 (0)