com.google.ai.edge.aicore

fun content(@Content.Role role: Int = Role.USER, init: Content.Builder.() -> Unit): Content

content(Role.MODEL) {
  text("Example string")
)

fun generationConfig(init: GenerationConfig.Builder.() -> Unit): GenerationConfig

generationConfig {
  context = context // required
  workerExecutor = workerThread // Set the [ExecutorService] on which background tasks should be run. If no value is specified, a background thread pool will be used.
  callbackExecutor = callbackExecutor // Set the [Executor] on which callbacks should be invoked. If no value is specified, callbacks will be invoked on the UI thread.
  temperature = 0.75f
  topK = 30
  candidateCount = 4
  maxOutputTokens = 300
  stopSequences = listOf("in conclusion", "-----", "do you need")
}

fun Part.asTextOrNull(): String?