kongbrain 0.1.0

package/src/state.ts

@@ -0,0 +1,119 @@
+import type { PluginCompleteParams, PluginCompleteResult } from "openclaw/plugin-sdk";
+import type { KongBrainConfig } from "./config.js";
+import type { SurrealStore } from "./surreal.js";
+import type { EmbeddingService } from "./embeddings.js";
+import type { AdaptiveConfig } from "./orchestrator.js";
+import type { MemoryDaemon } from "./daemon-manager.js";
+
+/** Provider-agnostic LLM completion function. */
+export type CompleteFn = (params: PluginCompleteParams) => Promise<PluginCompleteResult>;
+
+// --- Per-session mutable state ---
+
+const DEFAULT_TOOL_LIMIT = 10;
+
+export class SessionState {
+  readonly sessionId: string;
+  readonly sessionKey: string;
+
+  // Turn tracking
+  lastUserTurnId = "";
+  lastUserText = "";
+  lastAssistantText = "";
+  toolCallCount = 0;
+  toolLimit = DEFAULT_TOOL_LIMIT;
+  turnTextLength = 0;
+  toolCallsSinceLastText = 0;
+  softInterrupted = false;
+  turnStartMs = Date.now();
+  userTurnCount = 0;
+
+  // Thinking capture
+  readonly pendingThinking: string[] = [];
+
+  // Memory daemon
+  daemon: MemoryDaemon | null = null;
+  newContentTokens = 0;
+  readonly DAEMON_TOKEN_THRESHOLD = 12000;
+
+  // Current adaptive config (set by orchestrator preflight each turn)
+  currentConfig: AdaptiveConfig | null = null;
+
+  // Pending tool args for artifact tracking
+  readonly pendingToolArgs = new Map<string, unknown>();
+
+  // 5-pillar IDs (populated at bootstrap)
+  agentId = "";
+  projectId = "";
+  taskId = "";
+
+  constructor(sessionId: string, sessionKey: string) {
+    this.sessionId = sessionId;
+    this.sessionKey = sessionKey;
+  }
+
+  /** Reset per-turn counters at the start of each prompt. */
+  resetTurn(): void {
+    this.toolCallCount = 0;
+    this.toolLimit = DEFAULT_TOOL_LIMIT;
+    this.turnTextLength = 0;
+    this.toolCallsSinceLastText = 0;
+    this.softInterrupted = false;
+    this.turnStartMs = Date.now();
+    this.pendingThinking.length = 0;
+  }
+}
+
+// --- Global plugin state (shared across all sessions) ---
+
+/** Function to enqueue a system event visible to the user. */
+export type EnqueueSystemEventFn = (text: string, options: { sessionKey: string }) => boolean;
+
+export class GlobalPluginState {
+  readonly config: KongBrainConfig;
+  readonly store: SurrealStore;
+  readonly embeddings: EmbeddingService;
+  readonly complete: CompleteFn;
+  workspaceDir?: string;
+  enqueueSystemEvent?: EnqueueSystemEventFn;
+  private sessions = new Map<string, SessionState>();
+
+  constructor(
+    config: KongBrainConfig,
+    store: SurrealStore,
+    embeddings: EmbeddingService,
+    complete: CompleteFn,
+  ) {
+    this.config = config;
+    this.store = store;
+    this.embeddings = embeddings;
+    this.complete = complete;
+  }
+
+  /** Get or create a SessionState for the given session key. */
+  getOrCreateSession(sessionKey: string, sessionId: string): SessionState {
+    let session = this.sessions.get(sessionKey);
+    if (!session) {
+      session = new SessionState(sessionId, sessionKey);
+      this.sessions.set(sessionKey, session);
+    }
+    return session;
+  }
+
+  /** Get an existing session by key. */
+  getSession(sessionKey: string): SessionState | undefined {
+    return this.sessions.get(sessionKey);
+  }
+
+  /** Remove a session from the map (after dispose/cleanup). */
+  removeSession(sessionKey: string): void {
+    this.sessions.delete(sessionKey);
+  }
+
+  /** Shut down all shared resources. */
+  async shutdown(): Promise<void> {
+    this.sessions.clear();
+    await this.embeddings.dispose();
+    await this.store.dispose();
+  }
+}