@@ -47,10 +47,7 @@ def register(subparser: _SubParsersAction[ArgumentParser]) -> None:
47
47
"--validation-file" ,
48
48
help = "The validation file to use for fine-tuning." ,
49
49
)
50
- sub .set_defaults (
51
- func = CLIFineTuningJobs .create , args_model = CLIFineTuningJobsCreateArgs
52
- )
53
-
50
+ sub .set_defaults (func = CLIFineTuningJobs .create , args_model = CLIFineTuningJobsCreateArgs )
54
51
55
52
sub = subparser .add_parser ("fine_tuning.jobs.retrieve" )
56
53
sub .add_argument (
@@ -59,9 +56,7 @@ def register(subparser: _SubParsersAction[ArgumentParser]) -> None:
59
56
help = "The ID of the fine-tuning job to retrieve." ,
60
57
required = True ,
61
58
)
62
- sub .set_defaults (
63
- func = CLIFineTuningJobs .retrieve , args_model = CLIFineTuningJobsRetrieveArgs
64
- )
59
+ sub .set_defaults (func = CLIFineTuningJobs .retrieve , args_model = CLIFineTuningJobsRetrieveArgs )
65
60
66
61
sub = subparser .add_parser ("fine_tuning.jobs.list" )
67
62
sub .add_argument (
@@ -84,9 +79,7 @@ def register(subparser: _SubParsersAction[ArgumentParser]) -> None:
84
79
help = "The ID of the fine-tuning job to cancel." ,
85
80
required = True ,
86
81
)
87
- sub .set_defaults (
88
- func = CLIFineTuningJobs .cancel , args_model = CLIFineTuningJobsCancelArgs
89
- )
82
+ sub .set_defaults (func = CLIFineTuningJobs .cancel , args_model = CLIFineTuningJobsCancelArgs )
90
83
91
84
sub = subparser .add_parser ("fine_tuning.jobs.list_events" )
92
85
sub .add_argument (
@@ -106,9 +99,8 @@ def register(subparser: _SubParsersAction[ArgumentParser]) -> None:
106
99
help = "Number of fine-tuning job events to retrieve." ,
107
100
type = int ,
108
101
)
109
- sub .set_defaults (
110
- func = CLIFineTuningJobs .list_events , args_model = CLIFineTuningJobsListEventsArgs
111
- )
102
+ sub .set_defaults (func = CLIFineTuningJobs .list_events , args_model = CLIFineTuningJobsListEventsArgs )
103
+
112
104
113
105
class CLIFineTuningJobsCreateArgs (BaseModel ):
114
106
model : str
@@ -117,16 +109,20 @@ class CLIFineTuningJobsCreateArgs(BaseModel):
117
109
suffix : NotGivenOr [str ] = NOT_GIVEN
118
110
validation_file : NotGivenOr [str ] = NOT_GIVEN
119
111
112
+
120
113
class CLIFineTuningJobsRetrieveArgs (BaseModel ):
121
114
id : str
122
115
116
+
123
117
class CLIFineTuningJobsListArgs (BaseModel ):
124
118
after : NotGivenOr [str ] = NOT_GIVEN
125
119
limit : NotGivenOr [int ] = NOT_GIVEN
126
120
121
+
127
122
class CLIFineTuningJobsCancelArgs (BaseModel ):
128
123
id : str
129
124
125
+
130
126
class CLIFineTuningJobsListEventsArgs (BaseModel ):
131
127
id : str
132
128
after : NotGivenOr [str ] = NOT_GIVEN
@@ -148,34 +144,26 @@ def create(args: CLIFineTuningJobsCreateArgs) -> None:
148
144
149
145
@staticmethod
150
146
def retrieve (args : CLIFineTuningJobsRetrieveArgs ) -> None :
151
- fine_tuning_job : FineTuningJob = get_client ().fine_tuning .jobs .retrieve (
152
- fine_tuning_job_id = args .id
153
- )
147
+ fine_tuning_job : FineTuningJob = get_client ().fine_tuning .jobs .retrieve (fine_tuning_job_id = args .id )
154
148
print_model (fine_tuning_job )
155
-
149
+
156
150
@staticmethod
157
151
def list (args : CLIFineTuningJobsListArgs ) -> None :
158
- fine_tuning_jobs : SyncCursorPage [
159
- FineTuningJob
160
- ] = get_client ().fine_tuning .jobs .list (
152
+ fine_tuning_jobs : SyncCursorPage [FineTuningJob ] = get_client ().fine_tuning .jobs .list (
161
153
after = args .after or NOT_GIVEN , limit = args .limit or NOT_GIVEN
162
154
)
163
155
print_model (fine_tuning_jobs )
164
-
156
+
165
157
@staticmethod
166
158
def cancel (args : CLIFineTuningJobsCancelArgs ) -> None :
167
- fine_tuning_job : FineTuningJob = get_client ().fine_tuning .jobs .cancel (
168
- fine_tuning_job_id = args .id
169
- )
159
+ fine_tuning_job : FineTuningJob = get_client ().fine_tuning .jobs .cancel (fine_tuning_job_id = args .id )
170
160
print_model (fine_tuning_job )
171
-
161
+
172
162
@staticmethod
173
163
def list_events (args : CLIFineTuningJobsListEventsArgs ) -> None :
174
- fine_tuning_job_events : SyncCursorPage [
175
- FineTuningJobEvent
176
- ] = get_client ().fine_tuning .jobs .list_events (
164
+ fine_tuning_job_events : SyncCursorPage [FineTuningJobEvent ] = get_client ().fine_tuning .jobs .list_events (
177
165
fine_tuning_job_id = args .id ,
178
166
after = args .after or NOT_GIVEN ,
179
167
limit = args .limit or NOT_GIVEN ,
180
168
)
181
- print_model (fine_tuning_job_events )
169
+ print_model (fine_tuning_job_events )
0 commit comments