feat: proactive plugin

examples/chat-sdk/src/state.ts

@@ -33,7 +33,6 @@ export class MemoryStateAdapter implements StateAdapter {
       if (existingTimeout)
         clearTimeout(existingTimeout)
 
-      // eslint-disable-next-line @masknet/prefer-timer-id
       const timeout = setTimeout(() => {
         this.lists.delete(key)
         this.listTimeouts.delete(key)

packages/core/src/types/plugin.ts

@@ -77,6 +77,7 @@ export interface ResponseOptions<T = unknown> extends PluginHookBase<T> {
 export interface SessionInitOptions<T = unknown> {
   agentName: string
   context: AgentContext<T>
+  send?: (input: ItemParam) => string
   sessionId: string
 }
 

packages/core/src/utils/agent.ts

@@ -185,26 +185,20 @@ export const createAgent = <T = unknown>(options: CreateAgentOptions<T>): Agent<
     const resolveContext = (runContext?: Partial<AgentContext<T>>): AgentContext<T> =>
       merge(merge(context, currentSessionContext), runContext)
 
-    const createSessionOptions = (): SessionInitOptions<T> => ({
+    let sessionReady: Promise<void> | undefined
+
+    const sessionCallbacks = {
+      ensureSessionReady: async (): Promise<void> => {},
+    }
+
+    const createSessionOptions = (send: (input: { context?: Partial<AgentContext<T>>, input: ItemParam, signal?: AbortSignal }) => string): SessionInitOptions<T> => ({
       agentName: options.name,
       context: resolveContext(),
+      send: input => send({ input }),
       sessionId: id,
     })
 
-    let sessionReady: Promise<void> | undefined
-
-    const ensureSessionReady = async () => {
-      sessionReady ??= ready.then(async () => {
-        for (const plugin of plugins)
-          await plugin.onSessionInit?.(createSessionOptions())
-      })
-
-      return sessionReady
-    }
-
     const loadSession = async (): Promise<SessionState<T> | undefined> => {
-      await ensureSessionReady()
-
       for (const plugin of plugins) {
         if (plugin.storage == null)
           continue
@@ -242,13 +236,22 @@ export const createAgent = <T = unknown>(options: CreateAgentOptions<T>): Agent<
           await plugin.onTurnDone?.(turnContext)
       },
       plugins,
-      ready: async () => ensureSessionReady(),
+      ready: async () => sessionCallbacks.ensureSessionReady(),
       responseOptions: options.options,
       saveSession,
       sessionContext: initialSessionContext,
       sessionId: id,
     })
 
+    sessionCallbacks.ensureSessionReady = async () => {
+      sessionReady ??= ready.then(async () => {
+        for (const plugin of plugins)
+          await plugin.onSessionInit?.(createSessionOptions(runtime.send))
+      })
+
+      return sessionReady
+    }
+
     const subscribeSession = (channel: string, listener: PluginChannelListener) => {
       const register = () => {
         if (channel === 'apeira') {

packages/plugin-proactive/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "@apeira/plugin-proactive",
+  "type": "module",
+  "version": "0.0.4",
+  "description": "Proactive agent behavior with DMN, scheduler, and autonomous ticks.",
+  "author": "Moeru AI",
+  "license": "MIT",
+  "homepage": "https://github.com/moeru-ai/apeira",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/moeru-ai/apeira.git",
+    "directory": "packages/plugin-proactive"
+  },
+  "bugs": "https://github.com/moeru-ai/apeira/issues",
+  "keywords": [
+    "agent",
+    "ai",
+    "xsai",
+    "proactive"
+  ],
+  "sideEffects": false,
+  "exports": {
+    ".": "./src/index.ts",
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "dist"
+  ],
+  "publishConfig": {
+    "exports": {
+      ".": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      },
+      "./package.json": "./package.json"
+    },
+    "main": "./dist/index.js",
+    "types": "./dist/index.d.ts"
+  },
+  "scripts": {
+    "build": "pkgroll",
+    "test": "vitest run"
+  },
+  "peerDependencies": {
+    "@apeira/core": "workspace:"
+  },
+  "dependencies": {
+    "@xsai/tool": "catalog:xsai",
+    "zod": "catalog:"
+  },
+  "devDependencies": {
+    "@apeira/core": "workspace:"
+  }
+}

packages/plugin-proactive/src/dmn.ts

@@ -0,0 +1,52 @@
+export type DmnState = 'foraging' | 'paused' | 'resting' | 'working'
+
+export const DMN_TICK_INTERVALS: Record<DmnState, null | number> = {
+  foraging: 30000,
+  paused: null,
+  resting: 300000,
+  working: 3000,
+}
+
+export interface DmnContext {
+  lastTickAt: number
+  lastUserInputAt: number
+  state: DmnState
+}
+
+export const createDmnContext = (): DmnContext => ({
+  lastTickAt: 0,
+  lastUserInputAt: 0,
+  state: 'resting',
+})
+
+/**
+ * Compute next DMN state based on turn result
+ * - tool calls made → working
+ * - no tool calls → decays from working → foraging → resting
+ * - paused stays unchanged
+ */
+export const nextDmnState = (
+  current: DmnContext,
+  hasToolCalls: boolean,
+  modelSlept: boolean,
+): DmnState => {
+  if (current.state === 'paused')
+    return 'paused'
+
+  if (modelSlept)
+    return 'resting'
+
+  if (hasToolCalls)
+    return 'working'
+
+  if (current.state === 'working')
+    return 'foraging'
+
+  if (current.state === 'foraging')
+    return 'resting'
+
+  return 'resting'
+}
+
+export const shouldSkipTick = (ctx: DmnContext): boolean =>
+  Date.now() - ctx.lastUserInputAt < 5000