Коммит заменяет SQLDelight на ksqlite и добавляет journal/context/outbox

2026-09-21 01:12:31 +03:00
parent bd65c29b48
commit acc7237e51
61 changed files with 1623 additions and 1750 deletions
@@ -0,0 +1,24 @@
+package pw.binom.agentik.journal
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+
+/**
+ * Часть контента сообщения на уровне хранилища. Намеренно НЕ зависит от
+ * `pw.binom.agentik.proto.Content` — маппинг `:proto.Content ↔ Content` живёт
+ * в `Mapping.kt` storage impl'ов.
+ */
+@Serializable
+sealed interface Content {
+    @Serializable
+    @SerialName("text")
+    data class Text(val body: String) : Content
+
+    @Serializable
+    @SerialName("image")
+    data class Image(val data: ByteArray, val mime: String) : Content {
+        override fun equals(other: Any?): Boolean =
+            this === other || (other is Image && mime == other.mime && data.contentEquals(other.data))
+        override fun hashCode(): Int = 31 * mime.hashCode() + data.contentHashCode()
+    }
+}
@@ -0,0 +1,41 @@
+package pw.binom.agentik.journal
+
+import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.flow.flow
+import kotlin.time.Instant
+
+/**
+ * Append-only audit log сообщений — read-only представление.
+ *
+ * Producer-операция [MutableJournalStore.append] находится на
+ * [MutableJournalStore] — этот интерфейс только для чтения, чтобы
+ * consumer'ы физически не могли писать в audit log.
+ *
+ * Никаких обновлений, никакого удаления (кроме каскадного вместе
+ * с ConversationStore.delete).
+ */
+interface JournalStore : AutoCloseable {
+
+    suspend fun list(conversationId: String, after: Instant, offset: Int, limit: Int): List<MessageRecord>
+
+    /**
+     * Cold-flow paging через [list]. Default-реализация делает N+1 round-trip
+     * (по странице через `list()` пока не получит короткую страницу). Для
+     * in-memory backend'ов это OK; remote/SQLite impl'ы могут override'нуть
+     * на `Channel` / cursor-батчинг, чтобы избежать per-page round-trip.
+     */
+    fun listFlow(conversationId: String, after: Instant, pageSize: Int = PAGE_SIZE): Flow<MessageRecord> = flow {
+        var offset = 0
+        while (true) {
+            val page = list(conversationId, after, offset, pageSize)
+            if (page.isEmpty()) return@flow
+            for (rec in page) emit(rec)
+            if (page.size < pageSize) return@flow
+            offset += page.size
+        }
+    }
+
+    companion object {
+        const val PAGE_SIZE = 100
+    }
+}
@@ -0,0 +1,12 @@
+package pw.binom.agentik.journal
+
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.json.JsonElement
+
+@Serializable
+data class MessageContext(
+    val origin: MessageOrigin,
+    val sourceId: String? = null,
+    val description: String? = null,
+    val metadata: JsonElement? = null,
+)
@@ -0,0 +1,20 @@
+package pw.binom.agentik.journal
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+
+/**
+ * Контекст инициации хода (кто/что и почему). Дубликат типа из `:proto` —
+ * живёт здесь чтобы не тащить `:proto` в слой хранения данных.
+ */
+@Serializable
+enum class MessageOrigin {
+    @SerialName("user")
+    USER,
+
+    @SerialName("system")
+    SYSTEM,
+
+    @SerialName("event")
+    EVENT,
+}
@@ -0,0 +1,71 @@
+package pw.binom.agentik.journal
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+import kotlin.time.Instant
+
+/**
+ * Запись в таблице `message` (append-only audit).
+ */
+@Serializable
+sealed interface MessageRecord {
+    val id: String
+    val conversationId: String
+    val createdAt: Instant
+
+    @Serializable
+    sealed interface Body : MessageRecord {
+        val content: List<Content>
+    }
+
+    @Serializable
+    @SerialName("user")
+    data class UserMessage(
+        override val id: String,
+        override val conversationId: String,
+        override val content: List<Content>,
+        override val createdAt: Instant,
+        val context: MessageContext? = null,
+    ) : Body
+
+    @Serializable
+    @SerialName("assistant")
+    data class AssistantMessage(
+        override val id: String,
+        override val conversationId: String,
+        override val content: List<Content>,
+        override val createdAt: Instant,
+        val tokens: TurnTokens? = null,
+    ) : Body
+
+    @Serializable
+    @SerialName("tool_call")
+    data class ToolCall(
+        override val id: String,
+        override val conversationId: String,
+        val toolName: String,
+        val toolTitle: String?,
+        val toolArgsJson: String,
+        override val createdAt: Instant,
+    ) : MessageRecord
+
+    @Serializable
+    @SerialName("tool_result")
+    data class ToolResult(
+        override val id: String,
+        override val conversationId: String,
+        val toolCallId: String,
+        val result: String?,
+        override val createdAt: Instant,
+    ) : MessageRecord
+
+    @Serializable
+    @SerialName("error")
+    data class Error(
+        override val id: String,
+        override val conversationId: String,
+        val message: String,
+        val code: String?,
+        override val createdAt: Instant,
+    ) : MessageRecord
+}
@@ -0,0 +1,17 @@
+package pw.binom.agentik.journal
+
+/**
+ * Mutable вариант [JournalStore] — добавляет producer-операцию [append].
+ *
+ * Этот интерфейс предназначен **только для producer'ов** (ChatAgent,
+ * ConversationLoop, ToolDispatcher, sub-agents, A2A-bridge).
+ * Consumer'ы (DebugRoutes, admin dashboards, parent agents) должны
+ * принимать **read-only** [JournalStore] — тогда невозможно случайно
+ * записать в audit log из observer'а.
+ *
+ * **Append семантика**: см. KDoc [MessageStore.append][JournalStore] —
+ * на этом интерфейсе (не дублируем).
+ */
+interface MutableJournalStore : JournalStore {
+    suspend fun append(record: MessageRecord)
+}
@@ -0,0 +1,47 @@
+package pw.binom.agentik.journal
+
+import kotlinx.serialization.SerialName
+import kotlinx.serialization.Serializable
+import kotlinx.serialization.builtins.ListSerializer
+import kotlinx.serialization.json.Json
+
+private val bodyJson = Json {
+    ignoreUnknownKeys = true
+    encodeDefaults = true
+    explicitNulls = false
+}
+
+@Serializable
+data class MessageBodyPayload(
+    val content: List<Content>,
+    @SerialName("context")
+    val context: MessageContext? = null,
+    val tokens: TurnTokens? = null,
+)
+
+fun encodeBodyPayload(
+    content: List<Content>,
+    context: MessageContext? = null,
+    tokens: TurnTokens? = null,
+): String = bodyJson.encodeToString(
+    MessageBodyPayload.serializer(),
+    MessageBodyPayload(content = content, context = context, tokens = tokens),
+)
+
+fun decodeBodyPayload(json: String): BodyDecoded = readPayload(json)
+
+data class BodyDecoded(
+    val content: List<Content>,
+    val context: MessageContext?,
+    val tokens: TurnTokens? = null,
+)
+
+private fun readPayload(json: String): BodyDecoded {
+    return try {
+        val p = bodyJson.decodeFromString(MessageBodyPayload.serializer(), json)
+        BodyDecoded(p.content, p.context, p.tokens)
+    } catch (e: kotlinx.serialization.SerializationException) {
+        val arr = bodyJson.decodeFromString(ListSerializer(Content.serializer()), json)
+        BodyDecoded(arr, null, null)
+    }
+}
@@ -0,0 +1,18 @@
+package pw.binom.agentik.journal
+
+import kotlinx.serialization.Serializable
+
+/**
+ * Token usage одного assistant turn'а.
+ */
+@Serializable
+data class TurnTokens(
+    val input: Int,
+    val output: Int,
+) {
+    val total: Int get() = input + output
+    init {
+        require(input >= 0) { "input tokens must be non-negative, got $input" }
+        require(output >= 0) { "output tokens must be non-negative, got $output" }
+    }
+}