kiku-agent 1.0.0

package/server/agent/agentState.js

@@ -0,0 +1,140 @@
+export const AgentStates = Object.freeze({
+  IDLE: 'IDLE',
+  FETCHING_TOOL: 'FETCHING_TOOL',
+  STREAMING_THOUGHTS: 'STREAMING_THOUGHTS',
+  STREAMING_ANSWER: 'STREAMING_ANSWER',
+  COMPLETE: 'COMPLETE',
+});
+
+const TRANSITIONS = Object.freeze({
+  [AgentStates.IDLE]: new Set([
+    AgentStates.FETCHING_TOOL,
+    AgentStates.STREAMING_THOUGHTS,
+    AgentStates.STREAMING_ANSWER,
+  ]),
+  [AgentStates.FETCHING_TOOL]: new Set([
+    AgentStates.FETCHING_TOOL,
+    AgentStates.STREAMING_THOUGHTS,
+    AgentStates.STREAMING_ANSWER,
+    AgentStates.COMPLETE,
+  ]),
+  [AgentStates.STREAMING_THOUGHTS]: new Set([
+    AgentStates.FETCHING_TOOL,
+    AgentStates.STREAMING_ANSWER,
+    AgentStates.COMPLETE,
+  ]),
+  [AgentStates.STREAMING_ANSWER]: new Set([
+    AgentStates.COMPLETE,
+    AgentStates.FETCHING_TOOL,
+  ]),
+  [AgentStates.COMPLETE]: new Set([
+    AgentStates.IDLE,
+  ]),
+});
+
+class MiniEmitter {
+  constructor() {
+    this._listeners = new Map();
+  }
+
+  on(event, fn) {
+    if (!this._listeners.has(event)) this._listeners.set(event, new Set());
+    this._listeners.get(event).add(fn);
+    return () => this.off(event, fn);
+  }
+
+  off(event, fn) {
+    const set = this._listeners.get(event);
+    if (set) set.delete(fn);
+  }
+
+  emit(event, ...args) {
+    const set = this._listeners.get(event);
+    if (set) {
+      for (const fn of set) {
+        try { fn(...args); } catch {}
+      }
+    }
+  }
+}
+
+export class AgentStateMachine extends MiniEmitter {
+  constructor() {
+    super();
+    this._state = AgentStates.IDLE;
+    this._enteredAt = Date.now();
+    this._toolName = null;
+    this._queryId = null;
+  }
+
+  get state() { return this._state; }
+
+  get elapsedMs() { return Date.now() - this._enteredAt; }
+
+  get toolName() { return this._toolName; }
+
+  get queryId() { return this._queryId; }
+
+  transition(to, meta = {}) {
+    const from = this._state;
+
+    if (from === to) {
+      if (to !== AgentStates.FETCHING_TOOL) return false;
+    }
+
+    const allowed = TRANSITIONS[from];
+    if (!allowed || !allowed.has(to)) {
+      return false;
+    }
+
+    this.emit('leave:' + from, { to, ...meta });
+    this._state = to;
+    this._enteredAt = Date.now();
+
+    if (to === AgentStates.FETCHING_TOOL) {
+      this._toolName = meta.toolName || null;
+    } else {
+      this._toolName = null;
+    }
+
+    this.emit('transition', { from, to, ...meta });
+    this.emit('enter:' + to, { from, ...meta });
+    return true;
+  }
+
+  startQuery(queryId) {
+    this._queryId = queryId;
+    return this.transition(AgentStates.IDLE);
+  }
+
+  reset() {
+    this._queryId = null;
+    this._toolName = null;
+    if (this._state !== AgentStates.IDLE) {
+      this.transition(AgentStates.IDLE);
+    }
+    if (this._state !== AgentStates.IDLE) {
+      const from = this._state;
+      this.emit('leave:' + from, { to: AgentStates.IDLE, forced: true });
+      this._state = AgentStates.IDLE;
+      this._enteredAt = Date.now();
+      this.emit('transition', { from, to: AgentStates.IDLE, forced: true });
+      this.emit('enter:' + AgentStates.IDLE, { from, forced: true });
+    }
+  }
+
+  get isProcessing() {
+    return this._state !== AgentStates.IDLE && this._state !== AgentStates.COMPLETE;
+  }
+
+  get label() {
+    switch (this._state) {
+      case AgentStates.IDLE: return 'Idle';
+      case AgentStates.FETCHING_TOOL: return `Tool: ${this._toolName || '?'}`;
+      case AgentStates.STREAMING_THOUGHTS: return 'Thinking…';
+      case AgentStates.STREAMING_ANSWER: return 'Streaming…';
+      case AgentStates.COMPLETE: return 'Done';
+      default: return this._state;
+    }
+  }
+}