@@ -38,7 +38,6 @@ import java.{util => ju}
38
38
import scala .collection .convert .ImplicitConversions .`iterable asJava`
39
39
import scala .collection .convert .ImplicitConversions .`map AsJavaMap`
40
40
import scala .collection .convert .ImplicitConversions .`list asScalaBuffer`
41
- import io .cequence .openaiscala .domain .response .PromptTokensDetails
42
41
43
42
package object impl {
44
43
@@ -101,7 +100,10 @@ package object impl {
101
100
.addParts(0 , Part .newBuilder().setText(content).build())
102
101
.build()
103
102
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
+ )
105
107
}
106
108
107
109
def toSystemVertexAI (
@@ -267,8 +269,7 @@ package object impl {
267
269
ChatCompletionChoiceInfo (
268
270
index = candidate.getIndex,
269
271
message = toOpenAIAssistantMessage(candidate.getContent),
270
- finish_reason =
271
- Some (candidate.getFinishReason.toString()),
272
+ finish_reason = Some (candidate.getFinishReason.toString()),
272
273
logprobs = None
273
274
)
274
275
},
@@ -289,7 +290,7 @@ package object impl {
289
290
OpenAIUsageInfo (
290
291
prompt_tokens = usageInfo.getPromptTokenCount,
291
292
total_tokens = usageInfo.getTotalTokenCount,
292
- completion_tokens = Some (usageInfo.getCandidatesTokenCount),
293
+ completion_tokens = Some (usageInfo.getCandidatesTokenCount)
293
294
// prompt_tokens_details = Some(
294
295
// PromptTokensDetails(
295
296
// cached_tokens = usageInfo.getCachedContentTokenCount.getOrElse(0),
0 commit comments