Skip to content

Commit 9e40768

Browse files
committed
[FineTuning] Support cancel job
1 parent b1a3e62 commit 9e40768

5 files changed

Lines changed: 26 additions & 4 deletions

File tree

src/main/java/org/devlive/sdk/openai/DefaultApi.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,4 +225,11 @@ Single<FineTuningResponse> fetchCreateFineTuningJob(@Url String url,
225225
*/
226226
@GET
227227
Single<FineTuningEntity> fetchFineTuningJobContent(@Url String url);
228+
229+
/**
230+
* Immediately cancel a fine-tune job.
231+
* 立即取消微调作业。
232+
*/
233+
@POST
234+
Single<FineTuningEntity> fetchCancelFineTuningJob(@Url String url);
228235
}

src/main/java/org/devlive/sdk/openai/DefaultClient.java

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,13 +233,20 @@ public FineTuningResponse fineTuningJobEvents(String jobId)
233233
.blockingGet();
234234
}
235235

236-
public FineTuningEntity fineTuningJobContent(String jobId)
236+
public FineTuningEntity retrieveFineTuningJob(String jobId)
237237
{
238238
String url = String.format(ProviderUtils.getUrl(provider, UrlModel.FETCH_FINE_TUNING_JOBS_CONTENT), jobId);
239239
return this.api.fetchFineTuningJobContent(url)
240240
.blockingGet();
241241
}
242242

243+
public FineTuningEntity cancelFineTuningJob(String jobId)
244+
{
245+
String url = String.format(ProviderUtils.getUrl(provider, UrlModel.FETCH_FINE_TUNING_JOBS_CANCEL), jobId);
246+
return this.api.fetchCancelFineTuningJob(url)
247+
.blockingGet();
248+
}
249+
243250
private ObjectMapper createObjectMapper()
244251
{
245252
ObjectMapper objectMapper = new ObjectMapper();

src/main/java/org/devlive/sdk/openai/model/UrlModel.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,5 +18,6 @@ public enum UrlModel
1818
FETCH_FILES,
1919
FETCH_FINE_TUNING_JOBS,
2020
FETCH_FINE_TUNING_JOBS_EVENTS,
21-
FETCH_FINE_TUNING_JOBS_CONTENT
21+
FETCH_FINE_TUNING_JOBS_CONTENT,
22+
FETCH_FINE_TUNING_JOBS_CANCEL,
2223
}

src/main/java/org/devlive/sdk/openai/utils/ProviderUtils.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ public class ProviderUtils
3030
DEFAULT_PROVIDER.put(UrlModel.FETCH_FINE_TUNING_JOBS, "v1/fine_tuning/jobs");
3131
DEFAULT_PROVIDER.put(UrlModel.FETCH_FINE_TUNING_JOBS_EVENTS, "v1/fine_tuning/jobs/%s/events");
3232
DEFAULT_PROVIDER.put(UrlModel.FETCH_FINE_TUNING_JOBS_CONTENT, "v1/fine_tuning/jobs/%s");
33+
DEFAULT_PROVIDER.put(UrlModel.FETCH_FINE_TUNING_JOBS_CANCEL, "v1/fine_tuning/jobs/%s/cancel");
3334

3435
AZURE_PROVIDER.put(UrlModel.FETCH_COMPLETIONS, "completions");
3536
AZURE_PROVIDER.put(UrlModel.FETCH_CHAT_COMPLETIONS, "chat/completions");

src/test/java/org/devlive/sdk/openai/FineTuningTest.java

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,14 @@ public void testFineTuningJobEvents()
4040
}
4141

4242
@Test
43-
public void testFineTuningJobContent()
43+
public void testRetrieveFineTuningJob()
4444
{
45-
Assert.assertThrows(RequestException.class, () -> client.fineTuningJobContent("ftjob-abc123"));
45+
Assert.assertThrows(RequestException.class, () -> client.retrieveFineTuningJob("ftjob-abc123"));
46+
}
47+
48+
@Test
49+
public void testCancelFineTuningJob()
50+
{
51+
Assert.assertThrows(RequestException.class, () -> client.cancelFineTuningJob("ftjob-abc123"));
4652
}
4753
}

0 commit comments

Comments
 (0)