@pikku/kysely 0.13.15 → 0.13.18

package/src/{kysely-ai-agent-run-service.ts → kysely-agent-run-service.ts}

@@ -1,9 +1,9 @@
 import type {
-  AIThread,
-  AIMessage,
+  AgentThread,
+  AgentMessage,
   AgentRunRow,
   AgentRunService,
-} from '@pikku/core/ai-agent'
+} from '@pikku/core/agent'
 import type { Kysely } from 'kysely'
 import { sql } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
@@ -23,7 +23,7 @@ export class KyselyAgentRunService implements AgentRunService {
     owners?: string[]
     limit?: number
     offset?: number
-  }): Promise<AIThread[]> {
+  }): Promise<AgentThread[]> {
     const {
       agentName,
       resourceId,
@@ -37,7 +37,7 @@ export class KyselyAgentRunService implements AgentRunService {
     if (owners && owners.length === 0) return []
 
     let query = this.db
-      .selectFrom('aiThreads as t')
+      .selectFrom('agentThreads as t')
       .select([
         't.id',
         't.resourceId',
@@ -52,7 +52,7 @@ export class KyselyAgentRunService implements AgentRunService {
         't.id',
         'in',
         this.db
-          .selectFrom('aiRun')
+          .selectFrom('agentRun')
           .select('threadId')
           .where('agentName', '=', agentName)
           .distinct()
@@ -87,9 +87,9 @@ export class KyselyAgentRunService implements AgentRunService {
     return result.map((row) => this.mapThreadRow(row))
   }
 
-  async getThread(threadId: string): Promise<AIThread | null> {
+  async getThread(threadId: string): Promise<AgentThread | null> {
     const row = await this.db
-      .selectFrom('aiThreads')
+      .selectFrom('agentThreads')
       .select([
         'id',
         'resourceId',
@@ -105,16 +105,16 @@ export class KyselyAgentRunService implements AgentRunService {
     return this.mapThreadRow(row)
   }
 
-  async getThreadMessages(threadId: string): Promise<AIMessage[]> {
+  async getThreadMessages(threadId: string): Promise<AgentMessage[]> {
     const [msgResult, tcResult] = await Promise.all([
       this.db
-        .selectFrom('aiMessage')
+        .selectFrom('agentMessage')
         .select(['id', 'role', 'content', 'createdAt'])
         .where('threadId', '=', threadId)
         .orderBy('createdAt', 'asc')
         .execute(),
       this.db
-        .selectFrom('aiToolCall')
+        .selectFrom('agentToolCall')
         .select(['id', 'messageId', 'toolName', 'args', 'result'])
         .where('threadId', '=', threadId)
         .orderBy('createdAt', 'asc')
@@ -128,11 +128,11 @@ export class KyselyAgentRunService implements AgentRunService {
       tcByMessage.get(msgId)!.push(tc)
     }
 
-    const messages: AIMessage[] = []
+    const messages: AgentMessage[] = []
     for (const row of msgResult) {
-      const msg: AIMessage = {
+      const msg: AgentMessage = {
         id: row.id,
-        role: row.role as AIMessage['role'],
+        role: row.role as AgentMessage['role'],
         content: row.content ?? undefined,
         createdAt: new Date(row.createdAt as unknown as string),
       }
@@ -170,7 +170,7 @@ export class KyselyAgentRunService implements AgentRunService {
 
   async getThreadRuns(threadId: string): Promise<AgentRunRow[]> {
     const result = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .select([
         'runId',
         'agentName',
@@ -195,7 +195,7 @@ export class KyselyAgentRunService implements AgentRunService {
 
   async deleteThread(threadId: string): Promise<boolean> {
     const result = await this.db
-      .deleteFrom('aiThreads')
+      .deleteFrom('agentThreads')
       .where('id', '=', threadId)
       .executeTakeFirst()
 
@@ -204,7 +204,7 @@ export class KyselyAgentRunService implements AgentRunService {
 
   async getDistinctAgentNames(): Promise<string[]> {
     const result = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .select('agentName')
       .distinct()
       .orderBy('agentName')
@@ -213,7 +213,7 @@ export class KyselyAgentRunService implements AgentRunService {
     return result.map((row) => row.agentName)
   }
 
-  private mapThreadRow(row: any): AIThread {
+  private mapThreadRow(row: any): AgentThread {
     return {
       id: row.id as string,
       resourceId: row.resourceId as string,

package/src/{kysely-ai-run-state-service.ts → kysely-agent-run-state-service.ts}

@@ -1,32 +1,33 @@
 import { sql } from 'kysely'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
+import type { AgentRunState } from '@pikku/core/agent'
+import type { AgentRunScore } from '@pikku/core/agent-scorer'
 import type {
-  AIRunStateService,
+  AgentRunStateService,
   CreateRunInput,
   SaveScoreInput,
 } from '@pikku/core/services'
-import type { AgentRunState, PendingApproval } from '@pikku/core/ai-agent'
-import type { AIRunScore } from '@pikku/core/ai-scorer'
+import type { PendingApproval } from '@pikku/core/agent'
 import { ensurePikkuSchema } from './schema/index.js'
-import { aiSchema } from './schema/ai.schema.js'
-import { getRunScores, saveRunScore } from './kysely-ai-run-scores.js'
+import { agentSchema } from './schema/agent.schema.js'
+import { getRunScores, saveRunScore } from './kysely-agent-run-scores.js'
 
-export class KyselyAIRunStateService implements AIRunStateService {
+export class KyselyAgentRunStateService implements AgentRunStateService {
   private initialized = false
 
   constructor(private db: Kysely<KyselyPikkuDB>) {}
 
   async init(): Promise<void> {
     if (this.initialized) return
-    await ensurePikkuSchema(this.db, aiSchema)
+    await ensurePikkuSchema(this.db, agentSchema)
     this.initialized = true
   }
 
   async createRun(run: CreateRunInput): Promise<string> {
     const runId = `run-${crypto.randomUUID()}`
     await this.db
-      .insertInto('aiRun')
+      .insertInto('agentRun')
       .values({
         runId,
         agentName: run.agentName,
@@ -67,7 +68,7 @@ export class KyselyAIRunStateService implements AIRunStateService {
     }
 
     await this.db
-      .updateTable('aiRun')
+      .updateTable('agentRun')
       .set(values)
       .where('runId', '=', runId)
       .execute()
@@ -75,7 +76,7 @@ export class KyselyAIRunStateService implements AIRunStateService {
 
   async getRun(runId: string): Promise<AgentRunState | null> {
     const row = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .selectAll()
       .where('runId', '=', runId)
       .executeTakeFirst()
@@ -84,7 +85,7 @@ export class KyselyAIRunStateService implements AIRunStateService {
 
   async getRunsByThread(threadId: string): Promise<AgentRunState[]> {
     const rows = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .selectAll()
       .where('threadId', '=', threadId)
       .orderBy('createdAt', 'desc')
@@ -97,7 +98,7 @@ export class KyselyAIRunStateService implements AIRunStateService {
     status: 'approved' | 'denied'
   ): Promise<boolean> {
     const rows = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .select(['runId', 'pendingApprovals'])
       .where('status', '=', 'suspended')
       .execute()
@@ -126,7 +127,7 @@ export class KyselyAIRunStateService implements AIRunStateService {
         // Compare-and-swap on the list we read: a concurrent approver that got
         // there first has already rewritten it, and this update matches no row.
         const result = await this.db
-          .updateTable('aiRun')
+          .updateTable('agentRun')
           .set(updates as any)
           .where('runId', '=', row.runId)
           .where('status', '=', 'suspended')
@@ -142,7 +143,7 @@ export class KyselyAIRunStateService implements AIRunStateService {
     toolCallId: string
   ): Promise<{ run: AgentRunState; approval: PendingApproval } | null> {
     const rows = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .selectAll()
       .where('status', '=', 'suspended')
       .execute()
@@ -170,7 +171,7 @@ export class KyselyAIRunStateService implements AIRunStateService {
     await saveRunScore(this.db, score)
   }
 
-  async getScores(runId: string): Promise<AIRunScore[]> {
+  async getScores(runId: string): Promise<AgentRunScore[]> {
     return getRunScores(this.db, runId)
   }
 

package/src/{kysely-ai-storage-service.ts → kysely-agent-storage-service.ts}

@@ -1,20 +1,24 @@
 import type {
-  AIStorageService,
-  AIRunStateService,
+  AgentMessage,
+  AgentRunState,
+  AgentThread,
+} from '@pikku/core/agent'
+import type { AgentRunScore } from '@pikku/core/agent-scorer'
+import type {
+  AgentRunStateService,
+  AgentStorageService,
   CreateRunInput,
   SaveScoreInput,
 } from '@pikku/core/services'
-import type { AIThread, AIMessage, AgentRunState } from '@pikku/core/ai-agent'
-import type { AIRunScore } from '@pikku/core/ai-scorer'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
 import { parseJson } from './kysely-json.js'
-import { getRunScores, saveRunScore } from './kysely-ai-run-scores.js'
+import { getRunScores, saveRunScore } from './kysely-agent-run-scores.js'
 import { ensurePikkuSchema } from './schema/index.js'
-import { aiSchema } from './schema/ai.schema.js'
+import { agentSchema } from './schema/agent.schema.js'
 
-export class KyselyAIStorageService
-  implements AIStorageService, AIRunStateService
+export class KyselyAgentStorageService
+  implements AgentStorageService, AgentRunStateService
 {
   private initialized = false
 
@@ -22,7 +26,7 @@ export class KyselyAIStorageService
 
   public async init(): Promise<void> {
     if (this.initialized) return
-    await ensurePikkuSchema(this.db, aiSchema)
+    await ensurePikkuSchema(this.db, agentSchema)
     this.initialized = true
   }
 
@@ -33,12 +37,12 @@ export class KyselyAIStorageService
       title?: string
       metadata?: Record<string, unknown>
     }
-  ): Promise<AIThread> {
+  ): Promise<AgentThread> {
     const id = options?.threadId ?? crypto.randomUUID()
     const now = new Date()
 
     await this.db
-      .insertInto('aiThreads')
+      .insertInto('agentThreads')
       .values({
         id,
         resourceId: resourceId,
@@ -59,9 +63,9 @@ export class KyselyAIStorageService
     }
   }
 
-  async getThread(threadId: string): Promise<AIThread> {
+  async getThread(threadId: string): Promise<AgentThread> {
     const row = await this.db
-      .selectFrom('aiThreads')
+      .selectFrom('agentThreads')
       .select([
         'id',
         'resourceId',
@@ -87,9 +91,9 @@ export class KyselyAIStorageService
     }
   }
 
-  async getThreads(resourceId: string): Promise<AIThread[]> {
+  async getThreads(resourceId: string): Promise<AgentThread[]> {
     const result = await this.db
-      .selectFrom('aiThreads')
+      .selectFrom('agentThreads')
       .select([
         'id',
         'resourceId',
@@ -113,21 +117,24 @@ export class KyselyAIStorageService
   }
 
   async deleteThread(threadId: string): Promise<void> {
-    await this.db.deleteFrom('aiThreads').where('id', '=', threadId).execute()
+    await this.db
+      .deleteFrom('agentThreads')
+      .where('id', '=', threadId)
+      .execute()
   }
 
   async getMessages(
     threadId: string,
     options?: { lastN?: number; cursor?: string }
-  ): Promise<AIMessage[]> {
+  ): Promise<AgentMessage[]> {
     let msgQuery = this.db
-      .selectFrom('aiMessage')
+      .selectFrom('agentMessage')
       .select(['id', 'role', 'content', 'createdAt'])
       .where('threadId', '=', threadId)
 
     if (options?.cursor) {
       const cursorRow = await this.db
-        .selectFrom('aiMessage')
+        .selectFrom('agentMessage')
         .select('createdAt')
         .where('id', '=', options.cursor)
         .executeTakeFirst()
@@ -150,7 +157,7 @@ export class KyselyAIStorageService
     }
 
     const tcResult = await this.db
-      .selectFrom('aiToolCall')
+      .selectFrom('agentToolCall')
       .select(['id', 'messageId', 'toolName', 'args', 'result'])
       .where('threadId', '=', threadId)
       .orderBy('createdAt', 'asc')
@@ -163,10 +170,10 @@ export class KyselyAIStorageService
       tcByMessage.get(msgId)!.push(tc)
     }
 
-    const messages: AIMessage[] = []
+    const messages: AgentMessage[] = []
     for (const row of msgResult) {
       const rawContent = row.content as string | undefined
-      let parsedContent: AIMessage['content'] = rawContent ?? undefined
+      let parsedContent: AgentMessage['content'] = rawContent ?? undefined
       if (rawContent) {
         try {
           const parsed = JSON.parse(rawContent)
@@ -179,9 +186,9 @@ export class KyselyAIStorageService
           // Not JSON, use raw string
         }
       }
-      const msg: AIMessage = {
+      const msg: AgentMessage = {
         id: row.id,
-        role: row.role as AIMessage['role'],
+        role: row.role as AgentMessage['role'],
         content: parsedContent,
         createdAt: new Date(row.createdAt),
       }
@@ -217,7 +224,10 @@ export class KyselyAIStorageService
     return messages
   }
 
-  async saveMessages(threadId: string, messages: AIMessage[]): Promise<void> {
+  async saveMessages(
+    threadId: string,
+    messages: AgentMessage[]
+  ): Promise<void> {
     if (messages.length === 0) return
 
     const toolMessages = messages.filter((m) => m.role === 'tool')
@@ -225,7 +235,7 @@ export class KyselyAIStorageService
 
     if (nonToolMessages.length > 0) {
       await this.db
-        .insertInto('aiMessage')
+        .insertInto('agentMessage')
         .values(
           nonToolMessages.map((msg) => ({
             id: msg.id,
@@ -243,7 +253,7 @@ export class KyselyAIStorageService
     )
     if (toolCalls.length > 0) {
       await this.db
-        .insertInto('aiToolCall')
+        .insertInto('agentToolCall')
         .values(
           toolCalls.map((tc) => ({
             id: tc.id,
@@ -260,7 +270,7 @@ export class KyselyAIStorageService
       if (!toolMsg.toolResults) continue
       for (const tr of toolMsg.toolResults) {
         await this.db
-          .updateTable('aiToolCall')
+          .updateTable('agentToolCall')
           .set({ result: tr.result })
           .where('id', '=', tr.id)
           .execute()
@@ -268,7 +278,7 @@ export class KyselyAIStorageService
     }
 
     await this.db
-      .updateTable('aiThreads')
+      .updateTable('agentThreads')
       .set({ updatedAt: new Date() })
       .where('id', '=', threadId)
       .execute()
@@ -279,7 +289,7 @@ export class KyselyAIStorageService
     scope: 'resource' | 'thread'
   ): Promise<Record<string, unknown> | null> {
     const row = await this.db
-      .selectFrom('aiWorkingMemory')
+      .selectFrom('agentWorkingMemory')
       .select('data')
       .where('id', '=', id)
       .where('scope', '=', scope)
@@ -295,7 +305,7 @@ export class KyselyAIStorageService
     data: Record<string, unknown>
   ): Promise<void> {
     await this.db
-      .insertInto('aiWorkingMemory')
+      .insertInto('agentWorkingMemory')
       .values({
         id,
         scope,
@@ -315,7 +325,7 @@ export class KyselyAIStorageService
     const runId = crypto.randomUUID()
 
     await this.db
-      .insertInto('aiRun')
+      .insertInto('agentRun')
       .values({
         runId: runId,
         agentName: run.agentName,
@@ -365,14 +375,14 @@ export class KyselyAIStorageService
     }
 
     await this.db
-      .updateTable('aiRun')
+      .updateTable('agentRun')
       .set(setValues)
       .where('runId', '=', runId)
       .execute()
 
     if (updates.pendingApprovals !== undefined) {
       await this.db
-        .updateTable('aiToolCall')
+        .updateTable('agentToolCall')
         .set({
           approvalStatus: null,
           runId: null,
@@ -393,7 +403,7 @@ export class KyselyAIStorageService
 
   async getRun(runId: string): Promise<AgentRunState | null> {
     const row = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .select([
         'runId',
         'agentName',
@@ -415,7 +425,7 @@ export class KyselyAIStorageService
     if (!row) return null
 
     const approvals = await this.db
-      .selectFrom('aiToolCall')
+      .selectFrom('agentToolCall')
       .select([
         'id',
         'toolName',
@@ -434,7 +444,7 @@ export class KyselyAIStorageService
 
   async getRunsByThread(threadId: string): Promise<AgentRunState[]> {
     const result = await this.db
-      .selectFrom('aiRun')
+      .selectFrom('agentRun')
       .select([
         'runId',
         'agentName',
@@ -457,7 +467,7 @@ export class KyselyAIStorageService
     const runs: AgentRunState[] = []
     for (const row of result) {
       const approvals = await this.db
-        .selectFrom('aiToolCall')
+        .selectFrom('agentToolCall')
         .select([
           'id',
           'toolName',
@@ -482,7 +492,7 @@ export class KyselyAIStorageService
     for (const a of approvals) {
       if (a.type === 'agent-call') {
         await this.db
-          .updateTable('aiToolCall')
+          .updateTable('agentToolCall')
           .set({
             approvalStatus: 'pending',
             runId: runId,
@@ -495,7 +505,7 @@ export class KyselyAIStorageService
           .execute()
       } else {
         await this.db
-          .updateTable('aiToolCall')
+          .updateTable('agentToolCall')
           .set({
             approvalStatus: 'pending',
             runId: runId,
@@ -554,7 +564,7 @@ export class KyselyAIStorageService
     approval: NonNullable<AgentRunState['pendingApprovals']>[number]
   } | null> {
     const tc = await this.db
-      .selectFrom('aiToolCall')
+      .selectFrom('agentToolCall')
       .select([
         'id',
         'toolName',
@@ -602,7 +612,7 @@ export class KyselyAIStorageService
   ): Promise<boolean> {
     // Only the caller that moves the tool call off 'pending' has claimed it.
     const result = await this.db
-      .updateTable('aiToolCall')
+      .updateTable('agentToolCall')
       .set({ approvalStatus: status })
       .where('id', '=', toolCallId)
       .where('approvalStatus', '=', 'pending')
@@ -614,7 +624,7 @@ export class KyselyAIStorageService
     await saveRunScore(this.db, score)
   }
 
-  async getScores(runId: string): Promise<AIRunScore[]> {
+  async getScores(runId: string): Promise<AgentRunScore[]> {
     return getRunScores(this.db, runId)
   }
 

package/src/kysely-audit-service.test.ts

@@ -2,7 +2,7 @@ import { describe, test, beforeEach } from 'node:test'
 import assert from 'node:assert/strict'
 import Database from 'better-sqlite3'
 import { CamelCasePlugin, Kysely, SqliteDialect, sql } from 'kysely'
-import type { AuditEvent } from '../../../core/src/index.js'
+import type { AuditEvent } from '@pikku/core/services'
 
 import { KyselyAuditService } from './kysely-audit-service.js'
 

package/src/kysely-audit-service.ts

@@ -1,11 +1,10 @@
+import type { AuditService } from '@pikku/core/services'
+import type { AuditEvent, AuditEventBatch } from '@pikku/core/services'
 import type {
-  AuditEvent,
-  AuditEventBatch,
   AuditFacets,
   AuditQuery,
   AuditQueryResult,
-  AuditService,
-} from '@pikku/core'
+} from '@pikku/core/services'
 import type { Kysely } from 'kysely'
 import { ensurePikkuSchema } from './schema/index.js'
 import { auditSchema } from './schema/audit.schema.js'
@@ -116,8 +115,7 @@ export class KyselyAuditService implements AuditService {
     if (!batch.length) return
     const rows = batch.map((e) => {
       const metadata = e.metadata as
-        | { tables?: unknown; changedColumns?: unknown }
-        | undefined
+        { tables?: unknown; changedColumns?: unknown } | undefined
       return {
         audit_id: e.eventId ?? e.queryId ?? fallbackId(),
         occurred_at: e.occurredAt ?? new Date().toISOString(),

package/src/kysely-channel-store.ts

@@ -1,5 +1,5 @@
-import type { Channel } from '@pikku/core/channel'
 import { ChannelStore } from '@pikku/core/channel'
+import type { Channel } from '@pikku/core/channel'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
 import { parseJson } from './kysely-json.js'

package/src/kysely-credential-service.ts

@@ -1,13 +1,13 @@
-import type { CredentialService } from '@pikku/core/services'
-import type { Kysely } from 'kysely'
-import type { KyselyPikkuDB } from './kysely-tables.js'
 import {
   deriveKEK,
-  generateKEKSalt,
-  envelopeEncrypt,
   envelopeDecrypt,
+  envelopeEncrypt,
   envelopeRewrap,
+  generateKEKSalt,
 } from '@pikku/core/crypto-utils'
+import type { CredentialService } from '@pikku/core/services'
+import type { Kysely } from 'kysely'
+import type { KyselyPikkuDB } from './kysely-tables.js'
 import { ensurePikkuSchema } from './schema/index.js'
 import { credentialSchema } from './schema/credential.schema.js'
 

package/src/kysely-deployment-service.ts

@@ -4,7 +4,8 @@ import type {
   DeploymentServiceConfig,
   DeploymentConfig,
 } from '@pikku/core/services'
-import type { JWTService, SecretService } from '@pikku/core/services'
+import type { JWTService } from '@pikku/core/services'
+import type { SecretService } from '@pikku/core/services'
 import { getAllFunctionNames } from '@pikku/core/function'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'

package/src/kysely-scope-service.ts

@@ -1,4 +1,5 @@
-import type { Role, ScopeService } from '@pikku/core/services'
+import type { ScopeService } from '@pikku/core/services'
+import type { Role } from '@pikku/core/services'
 import type { FlatScope } from '@pikku/core/scope'
 import type { SystemRole } from '@pikku/core/role'
 import {
@@ -378,7 +379,9 @@ export class KyselyScopeService implements ScopeService {
    * different decisions, and the number is what makes pruning a choice rather
    * than a formality.
    */
-  async findStaleSystemRoles(): Promise<Array<{ role: string; users: number }>> {
+  async findStaleSystemRoles(): Promise<
+    Array<{ role: string; users: number }>
+  > {
     const stale = await this.db
       .selectFrom('pikkuRoles')
       .leftJoin('pikkuUserRole', 'pikkuUserRole.role', 'pikkuRoles.name')

package/src/kysely-secret-service.ts

@@ -1,14 +1,15 @@
-import { createSecretValue, isSecretValue, type SecretValue } from '@pikku/core'
-import type { SecretService, SecretValues } from '@pikku/core/services'
-import type { Kysely } from 'kysely'
-import type { KyselyPikkuDB } from './kysely-tables.js'
 import {
   deriveKEK,
-  generateKEKSalt,
-  envelopeEncrypt,
   envelopeDecrypt,
+  envelopeEncrypt,
   envelopeRewrap,
+  generateKEKSalt,
 } from '@pikku/core/crypto-utils'
+import { createSecretValue, isSecretValue } from '@pikku/core/classification'
+import type { SecretService, SecretValues } from '@pikku/core/services'
+import type { SecretValue } from '@pikku/core/classification'
+import type { Kysely } from 'kysely'
+import type { KyselyPikkuDB } from './kysely-tables.js'
 import { ensurePikkuSchema } from './schema/index.js'
 import { secretSchema } from './schema/secret.schema.js'