Skip to content

Commit 67c192f

Browse files
author
michal.zyga
committed
class renaming & add test for list fine tuning jobs
1 parent 634f6cd commit 67c192f

File tree

7 files changed

+225
-198
lines changed

7 files changed

+225
-198
lines changed

core/src/main/scala/sttp/openai/OpenAI.scala

+4-4
Original file line numberDiff line numberDiff line change
@@ -552,25 +552,25 @@ class OpenAI(authToken: String, baseUri: Uri = OpenAIUris.OpenAIBaseUri) {
552552
* @param fineTuningRequestBody
553553
* Request body that will be used to create a fine-tuning job.
554554
*/
555-
def createFineTuningJob(fineTuningRequestBody: FineTuningRequestBody): Request[Either[OpenAIException, FineTuningResponse]] =
555+
def createFineTuningJob(fineTuningRequestBody: FineTuningJobRequestBody): Request[Either[OpenAIException, FineTuningJobResponse]] =
556556
openAIAuthRequest
557557
.post(openAIUris.FineTuning)
558558
.body(fineTuningRequestBody)
559-
.response(asJson_parseErrors[FineTuningResponse])
559+
.response(asJson_parseErrors[FineTuningJobResponse])
560560

561561
/** List your organization's fine-tuning jobs
562562
*
563563
* [[https://platform.openai.com/docs/api-reference/fine-tuning/list]]
564564
*/
565565
def listFineTuningJobs(
566566
queryParameters: finetuning.QueryParameters = finetuning.QueryParameters.empty
567-
): Request[Either[OpenAIException, ListFineTuningResponse]] = {
567+
): Request[Either[OpenAIException, ListFineTuningJobResponse]] = {
568568
val uri = openAIUris.FineTuning
569569
.withParams(queryParameters.toMap)
570570

571571
openAIAuthRequest
572572
.get(uri)
573-
.response(asJson_parseErrors[ListFineTuningResponse])
573+
.response(asJson_parseErrors[ListFineTuningJobResponse])
574574
}
575575

576576
/** Gets info about the fine-tune job.

core/src/main/scala/sttp/openai/OpenAISyncClient.scala

+3-3
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ import sttp.openai.requests.embeddings.EmbeddingsRequestBody.EmbeddingsBody
1717
import sttp.openai.requests.embeddings.EmbeddingsResponseBody.EmbeddingResponse
1818
import sttp.openai.requests.files.FilesResponseData.{DeletedFileData, FileData, FilesResponse}
1919
import sttp.openai.requests.finetuning
20-
import sttp.openai.requests.finetuning.{FineTuningRequestBody, FineTuningResponse, ListFineTuningResponse}
20+
import sttp.openai.requests.finetuning.{FineTuningJobRequestBody, FineTuningJobResponse, ListFineTuningJobResponse}
2121
import sttp.openai.requests.images.ImageResponseData.ImageResponse
2222
import sttp.openai.requests.images.creation.ImageCreationRequestBody.ImageCreationBody
2323
import sttp.openai.requests.images.edit.ImageEditsConfig
@@ -359,14 +359,14 @@ class OpenAISyncClient private (
359359
* @param fineTuningRequestBody
360360
* Request body that will be used to create a fine-tuning job.
361361
*/
362-
def createFineTuningJob(fineTuningRequestBody: FineTuningRequestBody): FineTuningResponse =
362+
def createFineTuningJob(fineTuningRequestBody: FineTuningJobRequestBody): FineTuningJobResponse =
363363
sendOrThrow(openAI.createFineTuningJob(fineTuningRequestBody))
364364

365365
/** List your organization's fine-tuning jobs
366366
*
367367
* [[https://platform.openai.com/docs/api-reference/fine-tuning/list]]
368368
*/
369-
def listFineTuningJobs(queryParameters: finetuning.QueryParameters = finetuning.QueryParameters.empty): ListFineTuningResponse =
369+
def listFineTuningJobs(queryParameters: finetuning.QueryParameters = finetuning.QueryParameters.empty): ListFineTuningJobResponse =
370370
sendOrThrow(openAI.listFineTuningJobs(queryParameters))
371371

372372
/** Gets info about the fine-tune job.

core/src/main/scala/sttp/openai/requests/finetuning/FineTuningRequestBody.scala core/src/main/scala/sttp/openai/requests/finetuning/FineTuningJobRequestBody.scala

+3-3
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ import ujson.Str
2828
* @param method
2929
* The method used for fine-tuning.
3030
*/
31-
case class FineTuningRequestBody(
31+
case class FineTuningJobRequestBody(
3232
model: FineTuningModel,
3333
trainingFile: String,
3434
suffix: Option[String] = None,
@@ -37,8 +37,8 @@ case class FineTuningRequestBody(
3737
seed: Option[Int] = None,
3838
method: Option[Method] = None
3939
)
40-
object FineTuningRequestBody {
41-
implicit val fineTuningRequestBodyWriter: SnakePickle.Writer[FineTuningRequestBody] = SnakePickle.macroW[FineTuningRequestBody]
40+
object FineTuningJobRequestBody {
41+
implicit val fineTuningRequestBodyWriter: SnakePickle.Writer[FineTuningJobRequestBody] = SnakePickle.macroW[FineTuningJobRequestBody]
4242
}
4343

4444
sealed abstract class FineTuningModel(val value: String)

core/src/main/scala/sttp/openai/requests/finetuning/FineTuningResponse.scala core/src/main/scala/sttp/openai/requests/finetuning/FineTuningJobResponse.scala

+7-7
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ import ujson.Str
4343
* @param method
4444
* The method used for fine-tuning.
4545
*/
46-
case class FineTuningResponse(
46+
case class FineTuningJobResponse(
4747
id: String,
4848
createdAt: Int,
4949
error: Option[Error] = None,
@@ -64,8 +64,8 @@ case class FineTuningResponse(
6464
method: Method
6565
)
6666

67-
object FineTuningResponse {
68-
implicit val fineTuningResponseDataReader: SnakePickle.Reader[FineTuningResponse] = SnakePickle.macroR[FineTuningResponse]
67+
object FineTuningJobResponse {
68+
implicit val fineTuningResponseDataReader: SnakePickle.Reader[FineTuningJobResponse] = SnakePickle.macroR[FineTuningJobResponse]
6969
}
7070

7171
/** @param code
@@ -119,12 +119,12 @@ object Status {
119119

120120
}
121121

122-
case class ListFineTuningResponse(
122+
case class ListFineTuningJobResponse(
123123
`object`: String = "list",
124-
data: Seq[FineTuningResponse],
124+
data: Seq[FineTuningJobResponse],
125125
hasMore: Boolean
126126
)
127127

128-
object ListFineTuningResponse {
129-
implicit val listFineTuningResponseR: SnakePickle.Reader[ListFineTuningResponse] = SnakePickle.macroR[ListFineTuningResponse]
128+
object ListFineTuningJobResponse {
129+
implicit val listFineTuningResponseR: SnakePickle.Reader[ListFineTuningJobResponse] = SnakePickle.macroR[ListFineTuningJobResponse]
130130
}

core/src/test/scala/sttp/openai/fixtures/FineTuningFixture.scala

-104
This file was deleted.

0 commit comments

Comments
 (0)