Skip to content

Commit 2a41601

Browse files
committed
OpenAI service wrapper - grader functions added
1 parent 2bbe97c commit 2a41601

File tree

1 file changed

+24
-3
lines changed

1 file changed

+24
-3
lines changed

openai-core/src/main/scala/io/cequence/openaiscala/service/adapter/OpenAIServiceWrapper.scala

Lines changed: 24 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import akka.stream.scaladsl.Source
44
import akka.util.ByteString
55
import io.cequence.openaiscala.domain.Batch._
66
import io.cequence.openaiscala.domain._
7+
import io.cequence.openaiscala.domain.graders.Grader
78
import io.cequence.openaiscala.domain.response._
89
import io.cequence.openaiscala.domain.settings._
910
import io.cequence.openaiscala.service.adapter.ServiceWrapperTypes._
@@ -15,10 +16,10 @@ import io.cequence.openaiscala.service.{
1516
import io.cequence.wsclient.service.adapter.DelegatedCloseableServiceWrapper
1617
import io.cequence.wsclient.service.adapter.ServiceWrapperTypes.CloseableServiceWrapper
1718
import io.cequence.openaiscala.domain.responsesapi.{
18-
Inputs,
19-
Response,
2019
InputItemsResponse,
21-
InputTokensCount
20+
InputTokensCount,
21+
Inputs,
22+
Response
2223
}
2324
import io.cequence.openaiscala.domain.responsesapi.{
2425
CreateModelResponseSettings,
@@ -602,6 +603,10 @@ trait OpenAIServiceWrapper
602603
): Future[Seq[Batch]] =
603604
wrap(_.listBatches(pagination, order))
604605

606+
///////////////////
607+
// Responses API //
608+
///////////////////
609+
605610
override def createModelResponse(
606611
inputs: Inputs,
607612
settings: CreateModelResponseSettings
@@ -645,6 +650,22 @@ trait OpenAIServiceWrapper
645650
): Future[InputItemsResponse] = wrap(
646651
_.listModelResponseInputItems(responseId, after, before, include, limit, order)
647652
)
653+
654+
/////////////
655+
// Graders //
656+
/////////////
657+
658+
override def runGrader(
659+
grader: Grader,
660+
modelSample: String,
661+
item: Map[String, Any]
662+
): Future[String] = wrap(
663+
_.runGrader(grader, modelSample, item)
664+
)
665+
666+
override def validateGrader(grader: Grader): Future[Grader] = wrap(
667+
_.validateGrader(grader)
668+
)
648669
}
649670

650671
private class OpenAICoreServiceWrapperImpl(

0 commit comments

Comments
 (0)