commonMain.batch.Batches.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of anthropic-sdk-kotlin-jvm Show documentation

Kotlin multiplatform client for accessing Ahtropic APIs

The newest version!

/*
 * Copyright 2024 Kazimierz Pogoda / Xemantic
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.xemantic.ai.anthropic.batch

import com.xemantic.ai.anthropic.Response
import com.xemantic.ai.anthropic.message.Message
import kotlinx.datetime.Instant
import kotlinx.serialization.SerialName
import kotlinx.serialization.Serializable

@Serializable
data class MessageBatchRequest(
  val requests: List
)

@Serializable
data class Request(
  @SerialName("custom_id")
  val customId: String,
  val params: Params
) {

  @Serializable
  data class Params(
    val model: String,
    val maxTokens: Int,
    val messages: List
  )

}

@Serializable
data class RequestCounts(
  val processing: Int,
  val succeeded: Int,
  val errored: Int,
  val canceled: Int,
  val expired: Int
)

/**
 * Processing status of the Message Batch.
 */
enum class ProcessingStatus {
  @SerialName("in_progress")
  IN_PROGRESS,
  @SerialName("canceling")
  CANCELING,
  @SerialName("ended")
  ENDED
}


@Serializable
@SerialName("message_batch")
data class MessageBatchResponse(
  val id: String,
  @SerialName("processing_status")
  val processingStatus: ProcessingStatus,
  @SerialName("request_counts")
  val requestCounts: RequestCounts,
  @SerialName("ended_at")
  val endedAt: Instant?,
  @SerialName("created_at")
  val createdAt: Instant,
  @SerialName("expires_at")
  val expiresAt: Instant,
  @SerialName("cancel_initiated_at")
  val cancelInitiatedAt: Instant?,
  @SerialName("results_url")
  val resultsUrl: String?
) : Response(type = "message_batch")