kimaki 0.26.0 → 0.27.0

package/src/subagent-rate-limit-plugin.ts

@@ -1,228 +0,0 @@
-// OpenCode plugin that aborts task-created subagent sessions after rate limits.
-
-import type { Hooks, Plugin } from '@opencode-ai/plugin'
-
-import {
-  appendToastSessionMarker,
-  createPluginLogger,
-  formatPluginErrorWithStack,
-  setPluginLogFilePath,
-} from './plugin-logger.js'
-import { createPluginClient } from './plugin-opencode-client.js'
-import { initSentry, notifyError } from './sentry.js'
-
-const logger = createPluginLogger('SUBMODEL')
-
-const RATE_LIMIT_TEXT_PATTERNS = [
-  'rate_limit',
-  'rate limit',
-  'resource exhausted',
-  'retry after',
-  'too many requests',
-  'quota exceeded',
-  'balance exhausted',
-  'spending-limit',
-  'run out of credits',
-] as const
-
-type PluginEvent = Parameters<NonNullable<Hooks['event']>>[0]['event']
-
-function isRateLimitText(text: string | undefined): boolean {
-  if (!text) {
-    return false
-  }
-
-  const haystack = text.toLowerCase()
-  return RATE_LIMIT_TEXT_PATTERNS.some((pattern) => {
-    return haystack.includes(pattern)
-  })
-}
-
-function getTaskChildSession(event: PluginEvent) {
-  if (event.type !== 'message.part.updated') {
-    return undefined
-  }
-
-  const part = event.properties.part
-  if (part.type !== 'tool' || part.tool !== 'task' || part.state.status === 'pending') {
-    return undefined
-  }
-
-  const childSessionId = part.state.metadata?.sessionId
-  if (typeof childSessionId !== 'string' || childSessionId.length === 0) {
-    return undefined
-  }
-
-  const subagentType = part.state.input?.subagent_type
-  return {
-    childSessionId,
-    subagentType: typeof subagentType === 'string' ? subagentType : undefined,
-  }
-}
-
-function getEventSessionId(event: PluginEvent): string | undefined {
-  if (event.type === 'session.status' || event.type === 'session.idle') {
-    return event.properties.sessionID
-  }
-  if (event.type === 'session.error') {
-    return event.properties.sessionID
-  }
-  if (event.type === 'message.updated') {
-    return event.properties.info.sessionID
-  }
-  if (event.type === 'message.part.updated') {
-    return event.properties.part.sessionID
-  }
-  if (
-    event.type === 'session.created'
-    || event.type === 'session.updated'
-    || event.type === 'session.deleted'
-  ) {
-    return event.properties.info.id
-  }
-  return undefined
-}
-
-function extractRateLimitReason(event: PluginEvent): string | undefined {
-  if (event.type === 'session.status' && event.properties.status.type === 'retry') {
-    return isRateLimitText(event.properties.status.message)
-      ? event.properties.status.message
-      : undefined
-  }
-
-  if (event.type === 'message.part.updated' && event.properties.part.type === 'retry') {
-    const retryError = event.properties.part.error
-    if (retryError.data.statusCode === 429 || retryError.data.statusCode === 402) {
-      return retryError.data.message
-    }
-    if (isRateLimitText(retryError.data.responseBody)) {
-      return retryError.data.responseBody
-    }
-    return isRateLimitText(retryError.data.message)
-      ? retryError.data.message
-      : undefined
-  }
-
-  const apiError = (() => {
-    if (event.type === 'session.error' && event.properties.error?.name === 'APIError') {
-      return event.properties.error.data
-    }
-    if (
-      event.type === 'message.updated'
-      && event.properties.info.role === 'assistant'
-      && event.properties.info.error?.name === 'APIError'
-    ) {
-      return event.properties.info.error.data
-    }
-    return undefined
-  })()
-
-  if (!apiError) {
-    return undefined
-  }
-  if (apiError.statusCode === 429 || apiError.statusCode === 402) {
-    return apiError.message
-  }
-  // xAI returns 403 with spending-limit or run out of credits text.
-  // Only treat 403 as a usage limit if the body matches specific patterns,
-  // not all 403s (which can be auth errors handled elsewhere).
-  if (apiError.statusCode === 403) {
-    const errorText = [apiError.message, apiError.responseBody].filter(Boolean).join(' ')
-    if (isRateLimitText(errorText)) return apiError.message
-  }
-  if (isRateLimitText(apiError.responseBody)) {
-    return apiError.responseBody
-  }
-  return isRateLimitText(apiError.message) ? apiError.message : undefined
-}
-
-export const subagentRateLimitPlugin: Plugin = async ({ serverUrl, directory }) => {
-  initSentry()
-
-  const dataDir = process.env.KIMAKI_DATA_DIR
-  if (dataDir) {
-    setPluginLogFilePath(dataDir)
-  }
-
-  // Build our own v2 client. The plugin-provided ctx.client (v1) does not
-  // reliably make REST calls (session.abort silently no-ops) from inside the
-  // plugin process. See plugin-opencode-client.ts.
-  const client = createPluginClient({ serverUrl, directory })
-  logger.bindClient(client)
-
-  const subagentSessions = new Map<string, {
-    subagentType?: string
-    aborting: boolean
-  }>()
-
-  return {
-    event: async ({ event }) => {
-      const taskChild = getTaskChildSession(event)
-      if (taskChild) {
-        const existing = subagentSessions.get(taskChild.childSessionId)
-        if (existing) {
-          if (taskChild.subagentType) {
-            existing.subagentType = taskChild.subagentType
-          }
-        } else {
-          subagentSessions.set(taskChild.childSessionId, {
-            subagentType: taskChild.subagentType,
-            aborting: false,
-          })
-        }
-      }
-
-      const eventSessionId = getEventSessionId(event)
-      if (!eventSessionId) {
-        return
-      }
-
-      if (event.type === 'session.deleted' || event.type === 'session.idle') {
-        subagentSessions.delete(eventSessionId)
-        return
-      }
-
-      const rateLimitReason = extractRateLimitReason(event)
-      if (!rateLimitReason) {
-        return
-      }
-
-      const subagent = subagentSessions.get(eventSessionId)
-      if (!subagent || subagent.aborting) {
-        return
-      }
-
-      subagent.aborting = true
-      const abortResult = await (async () => {
-          await client.session.abort({ sessionID: eventSessionId, directory })
-
-          await client.tui.showToast({
-            message: appendToastSessionMarker({
-              message: `Aborting ${subagent.subagentType || 'subagent'} after rate limit so the parent task can recover: ${rateLimitReason}`,
-              sessionId: eventSessionId,
-            }),
-            variant: 'info',
-          }).catch(() => {
-            return
-          })
-
-          logger.info(
-            `Aborted subagent ${eventSessionId} after rate limit`,
-          )
-        })()
-        .catch((error) => {
-          return new Error('Subagent rate-limit abort failed', {
-            cause: error,
-          })
-        })
-
-      subagentSessions.delete(eventSessionId)
-      if (!(abortResult instanceof Error)) {
-        return
-      }
-
-      logger.warn(`[subagent-rate-limit-plugin] ${formatPluginErrorWithStack(abortResult)}`)
-      void notifyError(abortResult, 'subagent rate-limit plugin abort failed')
-    },
-  }
-}