git.iko.soy/opencode-android
on 0bc1eda help sign in
opencode-android/app/src/main/java/soy/iko/opencode/data/model/Provider.kt
66 lines 2.1 kB history raw
1package soy.iko.opencode.data.model
3import kotlinx.serialization.Serializable
5/** Response of `GET /config/providers`. */
6@Serializable
7data class ProvidersResponse(
8 val providers: List<Provider> = emptyList(),
9 val default: Map<String, String> = emptyMap(),
12@Serializable
13data class Provider(
14 val id: String,
15 val name: String? = null,
16 val models: Map<String, ModelInfo> = emptyMap(),
17) {
18 val displayName: String get() = name?.takeIf { it.isNotBlank() } ?: id
21@Serializable
22data class ModelInfo(
23 val id: String? = null,
24 val name: String? = null,
25) {
26 fun displayName(modelKey: String): String =
27 name?.takeIf { it.isNotBlank() } ?: id ?: modelKey
30/** A flattened provider/model selection used by the prompt body and the model picker. */
31@Serializable
32data class ModelRef(
33 val providerID: String,
34 val modelID: String,
37/** A single selectable entry in the model picker, flattened from [ProvidersResponse]. */
38data class ModelOption(
39 val providerID: String,
40 val modelID: String,
41 val providerLabel: String,
42 val modelLabel: String,
43) {
44 val ref: ModelRef get() = ModelRef(providerID, modelID)
47/** Flatten providers into a sorted list of pickable options. */
48fun ProvidersResponse.toOptions(): List<ModelOption> =
49 providers.flatMap { provider ->
50 provider.models.map { (modelKey, info) ->
51 ModelOption(
52 providerID = provider.id,
53 modelID = info.id ?: modelKey,
54 providerLabel = provider.displayName,
55 modelLabel = info.displayName(modelKey),
56 )
57 }
58 }.sortedWith(compareBy({ it.providerLabel.lowercase() }, { it.modelLabel.lowercase() }))
60/** The default option to preselect: the server's default for some provider, else the first model. */
61fun ProvidersResponse.defaultOption(options: List<ModelOption> = toOptions()): ModelOption? {
62 default.entries.firstOrNull()?.let { (providerID, modelID) ->
63 options.firstOrNull { it.providerID == providerID && it.modelID == modelID }?.let { return it }
64 }
65 return options.firstOrNull()