@@ -4,6 +4,7 @@ import akka.stream.scaladsl.Source
44import akka .util .ByteString
55import io .cequence .openaiscala .domain .Batch ._
66import io .cequence .openaiscala .domain ._
7+ import io .cequence .openaiscala .domain .graders .Grader
78import io .cequence .openaiscala .domain .response ._
89import io .cequence .openaiscala .domain .settings ._
910import io .cequence .openaiscala .service .adapter .ServiceWrapperTypes ._
@@ -15,10 +16,10 @@ import io.cequence.openaiscala.service.{
1516import io .cequence .wsclient .service .adapter .DelegatedCloseableServiceWrapper
1617import io .cequence .wsclient .service .adapter .ServiceWrapperTypes .CloseableServiceWrapper
1718import io .cequence .openaiscala .domain .responsesapi .{
18- Inputs ,
19- Response ,
2019 InputItemsResponse ,
21- InputTokensCount
20+ InputTokensCount ,
21+ Inputs ,
22+ Response
2223}
2324import 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
650671private class OpenAICoreServiceWrapperImpl (
0 commit comments