@pikku/kysely 0.13.15 → 0.13.16

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

@@ -1,20 +1,24 @@
 import type {
-  AIStorageService,
-  AIRunStateService,
+  AgentStorageService,
+  AgentRunStateService,
   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 {
+  AgentThread,
+  AgentMessage,
+  AgentRunState,
+} from '@pikku/core/agent'
+import type { AgentRunScore } from '@pikku/core/agent-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.ts

@@ -1,11 +1,13 @@
+import type { AuditService } from '@pikku/core'
 import type {
   AuditEvent,
   AuditEventBatch,
+} from '@pikku/core/ecosystem/services'
+import type {
   AuditFacets,
   AuditQuery,
   AuditQueryResult,
-  AuditService,
-} from '@pikku/core'
+} from '@pikku/core/ecosystem/types'
 import type { Kysely } from 'kysely'
 import { ensurePikkuSchema } from './schema/index.js'
 import { auditSchema } from './schema/audit.schema.js'

package/src/kysely-deployment-service.ts

@@ -1,11 +1,11 @@
-import { buildRemoteHeaders } from '@pikku/core/remote'
+import { buildRemoteHeaders } from '@pikku/core/ecosystem/remote'
 import type {
   DeploymentService,
   DeploymentServiceConfig,
   DeploymentConfig,
-} from '@pikku/core/services'
+} from '@pikku/core/ecosystem/services'
 import type { JWTService, SecretService } from '@pikku/core/services'
-import { getAllFunctionNames } from '@pikku/core/function'
+import { getAllFunctionNames } from '@pikku/core/ecosystem/function'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
 import { ensurePikkuSchema } from './schema/index.js'

package/src/kysely-eventhub-store.ts

@@ -1,4 +1,4 @@
-import { EventHubStore } from '@pikku/core/channel'
+import { EventHubStore } from '@pikku/core/ecosystem/channel'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
 

package/src/kysely-scope-service.ts

@@ -1,6 +1,7 @@
-import type { Role, ScopeService } from '@pikku/core/services'
-import type { FlatScope } from '@pikku/core/scope'
-import type { SystemRole } from '@pikku/core/role'
+import type { ScopeService } from '@pikku/core/services'
+import type { Role } from '@pikku/core/ecosystem/services'
+import type { FlatScope } from '@pikku/core/ecosystem/scope'
+import type { SystemRole } from '@pikku/core/ecosystem/role'
 import {
   SystemRoleImmutableError,
   SystemRoleShadowedError,
@@ -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-services.test.ts

@@ -5,7 +5,7 @@ import { SerializePlugin } from './serialize-plugin.js'
 import Database from 'better-sqlite3'
 import { PostgresJSDialect } from 'kysely-postgres-js'
 import postgres from 'postgres'
-import { defineServiceTests } from '@pikku/core/testing'
+import { defineServiceTests } from '@pikku/core/ecosystem/testing'
 
 import type { KyselyPikkuDB } from './kysely-tables.js'
 import { KyselyChannelStore } from './kysely-channel-store.js'
@@ -13,8 +13,8 @@ import { KyselyEventHubStore } from './kysely-eventhub-store.js'
 import { KyselyWorkflowService } from './kysely-workflow-service.js'
 import { KyselyWorkflowRunService } from './kysely-workflow-run-service.js'
 import { KyselyDeploymentService } from './kysely-deployment-service.js'
-import { KyselyAIStorageService } from './kysely-ai-storage-service.js'
-import { KyselyAgentRunService } from './kysely-ai-agent-run-service.js'
+import { KyselyAgentStorageService } from './kysely-agent-storage-service.js'
+import { KyselyAgentRunService } from './kysely-agent-run-service.js'
 import { KyselySecretService } from './kysely-secret-service.js'
 import { KyselyCredentialService } from './kysely-credential-service.js'
 import { KyselySessionStore } from './kysely-session-store.js'
@@ -42,12 +42,12 @@ async function dropAllTables(db: Kysely<KyselyPikkuDB>): Promise<void> {
   const tables = [
     'pikku_deployment_functions',
     'pikku_deployments',
-    'ai_tool_call',
-    'ai_message',
-    'ai_run_score',
-    'ai_run',
-    'ai_working_memory',
-    'ai_threads',
+    'agent_tool_call',
+    'agent_message',
+    'agent_run_score',
+    'agent_run',
+    'agent_working_memory',
+    'agent_threads',
     'channel_subscriptions',
     'channels',
     'workflow_step_history',
@@ -97,8 +97,8 @@ function registerTests(
         await s.init()
         return s
       },
-      aiStorageService: async () => {
-        const s = new KyselyAIStorageService(getDb())
+      agentStorageService: async () => {
+        const s = new KyselyAgentStorageService(getDb())
         await s.init()
         return s
       },

package/src/kysely-tables.ts

@@ -3,7 +3,7 @@ import type {
   WorkflowStatus,
   StepStatus,
   WorkflowVersionStatus,
-} from '@pikku/core/workflow'
+} from '@pikku/core/ecosystem/workflow'
 
 export interface ChannelsTable {
   channelId: string
@@ -86,7 +86,7 @@ export interface WorkflowVersionsTable {
   createdAt: Generated<Date>
 }
 
-export interface AIThreadsTable {
+export interface AgentThreadsTable {
   id: string
   resourceId: string
   title: string | null
@@ -95,7 +95,7 @@ export interface AIThreadsTable {
   updatedAt: Generated<Date>
 }
 
-export interface AIMessageTable {
+export interface AgentMessageTable {
   id: string
   threadId: string
   role: 'system' | 'user' | 'assistant' | 'tool'
@@ -103,7 +103,7 @@ export interface AIMessageTable {
   createdAt: Generated<Date>
 }
 
-export interface AIToolCallTable {
+export interface AgentToolCallTable {
   id: string
   threadId: string
   messageId: string
@@ -119,14 +119,14 @@ export interface AIToolCallTable {
   createdAt: Generated<Date>
 }
 
-export interface AIWorkingMemoryTable {
+export interface AgentWorkingMemoryTable {
   id: string
   scope: string
   data: string
   updatedAt: Generated<Date>
 }
 
-export interface AIRunTable {
+export interface AgentRunTable {
   runId: Generated<string>
   agentName: string
   threadId: string
@@ -151,7 +151,7 @@ export interface AIRunTable {
   updatedAt: Generated<Date>
 }
 
-export interface AIRunScoreTable {
+export interface AgentRunScoreTable {
   id: string
   runId: string
   scorerName: string
@@ -336,12 +336,12 @@ export interface KyselyPikkuDB {
   workflowStep: WorkflowStepTable
   workflowStepHistory: WorkflowStepHistoryTable
   workflowVersions: WorkflowVersionsTable
-  aiThreads: AIThreadsTable
-  aiMessage: AIMessageTable
-  aiToolCall: AIToolCallTable
-  aiWorkingMemory: AIWorkingMemoryTable
-  aiRun: AIRunTable
-  aiRunScore: AIRunScoreTable
+  agentThreads: AgentThreadsTable
+  agentMessage: AgentMessageTable
+  agentToolCall: AgentToolCallTable
+  agentWorkingMemory: AgentWorkingMemoryTable
+  agentRun: AgentRunTable
+  agentRunScore: AgentRunScoreTable
   pikkuDeployments: PikkuDeploymentsTable
   pikkuDeploymentFunctions: PikkuDeploymentFunctionsTable
   secrets: SecretsTable

package/src/kysely-virtual-user-run-store.ts

@@ -1,12 +1,12 @@
 import type { Kysely, Selectable } from 'kysely'
 import { randomUUID } from 'node:crypto'
+import type { VirtualUserDisposition } from '@pikku/core/virtual-user'
 import type {
-  VirtualUserDisposition,
   VirtualUserRunOutcome,
   VirtualUserRunRecord,
   VirtualUserRunStart,
   VirtualUserRunStore,
-} from '@pikku/core/virtual-user'
+} from '@pikku/core/ecosystem/virtual-user'
 import type { KyselyPikkuDB, VirtualUserRunTable } from './kysely-tables.js'
 import { parseJson } from './kysely-json.js'
 import { ensurePikkuSchema } from './schema/index.js'

package/src/kysely-webhook-service.ts

@@ -1,12 +1,12 @@
+import { QueueWebhookService } from '@pikku/core/services'
 import {
   PIKKU_OUTGOING_WEBHOOK_QUEUE_NAME,
-  QueueWebhookService,
   type SendWebhookInput,
   type SendWebhookResult,
   type WebhookAttemptResult,
   type WebhookDeliveryRecord,
   type WebhookDeliveryWithAttempts,
-} from '@pikku/core/services'
+} from '@pikku/core/ecosystem/services'
 import type { QueueService } from '@pikku/core/queue'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'

package/src/kysely-workflow-mirror.ts

@@ -1,4 +1,4 @@
-import type { SerializedError } from '@pikku/core'
+import type { SerializedError } from '@pikku/core/ecosystem/types'
 import type {
   StepState,
   WorkflowRunMirror,
@@ -6,7 +6,7 @@ import type {
   WorkflowStatus,
   WorkflowPlannedStep,
   WorkflowVersionStatus,
-} from '@pikku/core/workflow'
+} from '@pikku/core/ecosystem/workflow'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
 import { ensurePikkuSchema } from './schema/index.js'

package/src/kysely-workflow-run-service.ts

@@ -3,7 +3,7 @@ import type {
   StepState,
   WorkflowStatus,
   WorkflowRunService,
-} from '@pikku/core/workflow'
+} from '@pikku/core/ecosystem/workflow'
 import type { Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
 import { parseJson } from './kysely-json.js'

package/src/kysely-workflow-service.ts

@@ -1,15 +1,6 @@
-import type { SerializedError } from '@pikku/core'
-import {
-  PikkuWorkflowService,
-  type WorkflowPlannedStep,
-  type WorkflowQueueOptions,
-  type WorkflowRun,
-  type WorkflowRunWire,
-  type StepState,
-  type StepStatus,
-  type WorkflowStatus,
-  type WorkflowVersionStatus,
-} from '@pikku/core/workflow'
+import type { SerializedError } from '@pikku/core/ecosystem/types'
+import { PikkuWorkflowService } from '@pikku/core/workflow'
+import type { WorkflowPlannedStep, WorkflowQueueOptions, WorkflowRun, WorkflowRunWire, StepState, StepStatus, WorkflowStatus, WorkflowVersionStatus } from '@pikku/core/ecosystem/workflow'
 import { sql, type Kysely } from 'kysely'
 import type { KyselyPikkuDB } from './kysely-tables.js'
 import { KyselyWorkflowRunService } from './kysely-workflow-run-service.js'