Skip to content

Commit d8abc69

Browse files
committed
Vertex - finish reason and usage info fix. Closes #95, #96
1 parent 3e9c931 commit d8abc69

File tree

1 file changed

+6
-5
lines changed
  • google-vertexai-client/src/main/scala/io/cequence/openaiscala/vertexai/service/impl

1 file changed

+6
-5
lines changed

google-vertexai-client/src/main/scala/io/cequence/openaiscala/vertexai/service/impl/package.scala

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@ import java.{util => ju}
3838
import scala.collection.convert.ImplicitConversions.`iterable asJava`
3939
import scala.collection.convert.ImplicitConversions.`map AsJavaMap`
4040
import scala.collection.convert.ImplicitConversions.`list asScalaBuffer`
41-
import io.cequence.openaiscala.domain.response.PromptTokensDetails
4241

4342
package object impl {
4443

@@ -101,7 +100,10 @@ package object impl {
101100
.addParts(0, Part.newBuilder().setText(content).build())
102101
.build()
103102

104-
case x: BaseMessage => throw new OpenAIScalaClientException(s"Unsupported message type: ${x.getClass().getName()}")
103+
case x: BaseMessage =>
104+
throw new OpenAIScalaClientException(
105+
s"Unsupported message type: ${x.getClass().getName()}"
106+
)
105107
}
106108

107109
def toSystemVertexAI(
@@ -267,8 +269,7 @@ package object impl {
267269
ChatCompletionChoiceInfo(
268270
index = candidate.getIndex,
269271
message = toOpenAIAssistantMessage(candidate.getContent),
270-
finish_reason =
271-
Some(candidate.getFinishReason.toString()),
272+
finish_reason = Some(candidate.getFinishReason.toString()),
272273
logprobs = None
273274
)
274275
},
@@ -289,7 +290,7 @@ package object impl {
289290
OpenAIUsageInfo(
290291
prompt_tokens = usageInfo.getPromptTokenCount,
291292
total_tokens = usageInfo.getTotalTokenCount,
292-
completion_tokens = Some(usageInfo.getCandidatesTokenCount),
293+
completion_tokens = Some(usageInfo.getCandidatesTokenCount)
293294
// prompt_tokens_details = Some(
294295
// PromptTokensDetails(
295296
// cached_tokens = usageInfo.getCachedContentTokenCount.getOrElse(0),

0 commit comments

Comments
 (0)