@tianshu-ai/tianshu 0.3.0

package/packages/server/dist/chat/active-harnesses.d.ts

@@ -0,0 +1,34 @@
+import type { AgentHarness } from "@earendil-works/pi-agent-core";
+import type { ServerMsg } from "./ws-protocol.js";
+/**
+ * Register an active harness for a session. Returns a thunk the
+ * caller MUST run when the harness is no longer reachable (turn
+ * complete, abort, exception path).
+ */
+export declare function registerActiveHarness(sessionId: string, harness: AgentHarness): () => void;
+/**
+ * Look up the active harness for a session, if any. Callers must
+ * be defensive: even when this returns a value, the harness may
+ * be in the process of shutting down. Use `harness.followUp(...)`
+ * inside try/catch.
+ */
+export declare function getActiveHarness(sessionId: string): AgentHarness | undefined;
+/**
+ * Register a `send` thunk for `userId`. handler.ts calls this
+ * once per inbound chat WebSocket. Returns an unregister thunk
+ * the caller MUST run on socket close.
+ */
+export declare function registerUserSendChannel(userId: string, send: (msg: ServerMsg) => void): () => void;
+/**
+ * Best-effort fan-out a server event to every chat WebSocket
+ * the user has open. Used by the session-inbox idle runner to
+ * stream tool calls / deltas of a background turn so any open
+ * tab still feels live.
+ *
+ * A throwing send is logged + skipped — we never want a single
+ * bad socket to break the broadcast loop.
+ */
+export declare function broadcastToUser(userId: string, msg: ServerMsg): void;
+/** Test-only: drop everything. Not exported through the package barrel. */
+export declare function _resetActiveHarnesses(): void;
+//# sourceMappingURL=active-harnesses.d.ts.map

package/packages/server/dist/chat/active-harnesses.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"active-harnesses.d.ts","sourceRoot":"","sources":["../../src/chat/active-harnesses.ts"],"names":[],"mappings":"AA+BA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAiBlD;;;;GAIG;AACH,wBAAgB,qBAAqB,CACnC,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,YAAY,GACpB,MAAM,IAAI,CAUZ;AAED;;;;;GAKG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS,CAE5E;AAED;;;;GAIG;AACH,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,CAAC,GAAG,EAAE,SAAS,KAAK,IAAI,GAC7B,MAAM,IAAI,CAaZ;AAED;;;;;;;;GAQG;AACH,wBAAgB,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,IAAI,CAcpE;AAED,2EAA2E;AAC3E,wBAAgB,qBAAqB,IAAI,IAAI,CAG5C"}

package/packages/server/dist/chat/active-harnesses.js

@@ -0,0 +1,118 @@
+// Process-local registry of active AgentHarness instances by
+// session id.
+//
+// Why we need this:
+//   The session inbox (see session-inbox.ts + plugin-sdk's
+//   SessionInboxCapability) needs to know "is the target session
+//   currently mid-turn? if so, route the message through
+//   `harness.followUp(...)` so the agent picks it up". Without a
+//   registry, the inbox can only persist; agents would never see
+//   messages until the user typed something else.
+//
+// Why this is process-local and not durable:
+//   Harness instances are JS objects; they live exactly as long
+//   as the in-flight WebSocket request that created them. There's
+//   nothing to persist. The DB-backed inbox row is the durable
+//   layer; this map is just the "fast path".
+//
+// Concurrency:
+//   `register` returns an `unregister()` thunk. The chat handler
+//   calls register before `harness.prompt(...)` and the
+//   unregister inside its `finally`. There's no cross-tenant
+//   concern because session ids are globally unique within a
+//   tenant DB; same-id collisions across tenants would be a
+//   schema bug.
+//
+// Multi-process / cluster:
+//   Out of scope. If we run multiple host processes one day, the
+//   inbox dispatcher will need a cross-process bus (NATS, Redis,
+//   …) to find the live harness. The DB persistence path keeps
+//   working unchanged.
+const active = new Map();
+/**
+ * userId → set of `send` thunks for every open chat WebSocket
+ * for that user. Keyed by user (not session) because we want
+ * inbox-driven background turns to push their stream events to
+ * any open tab the user has — they may be looking at a
+ * different session card while the inbox runs.
+ *
+ * Each `send` is a closure created by handler.ts that owns its
+ * own ws.send + JSON.stringify; we don't ref the raw ws here so
+ * tests don't need a real WebSocket.
+ */
+const userSendChannels = new Map();
+/**
+ * Register an active harness for a session. Returns a thunk the
+ * caller MUST run when the harness is no longer reachable (turn
+ * complete, abort, exception path).
+ */
+export function registerActiveHarness(sessionId, harness) {
+    active.set(sessionId, harness);
+    return () => {
+        // Only delete if still pointing at the same instance — guards
+        // against a stale unregister thunk fired after the slot was
+        // re-claimed by a new harness for the same session.
+        if (active.get(sessionId) === harness) {
+            active.delete(sessionId);
+        }
+    };
+}
+/**
+ * Look up the active harness for a session, if any. Callers must
+ * be defensive: even when this returns a value, the harness may
+ * be in the process of shutting down. Use `harness.followUp(...)`
+ * inside try/catch.
+ */
+export function getActiveHarness(sessionId) {
+    return active.get(sessionId);
+}
+/**
+ * Register a `send` thunk for `userId`. handler.ts calls this
+ * once per inbound chat WebSocket. Returns an unregister thunk
+ * the caller MUST run on socket close.
+ */
+export function registerUserSendChannel(userId, send) {
+    let set = userSendChannels.get(userId);
+    if (!set) {
+        set = new Set();
+        userSendChannels.set(userId, set);
+    }
+    set.add(send);
+    return () => {
+        const cur = userSendChannels.get(userId);
+        if (!cur)
+            return;
+        cur.delete(send);
+        if (cur.size === 0)
+            userSendChannels.delete(userId);
+    };
+}
+/**
+ * Best-effort fan-out a server event to every chat WebSocket
+ * the user has open. Used by the session-inbox idle runner to
+ * stream tool calls / deltas of a background turn so any open
+ * tab still feels live.
+ *
+ * A throwing send is logged + skipped — we never want a single
+ * bad socket to break the broadcast loop.
+ */
+export function broadcastToUser(userId, msg) {
+    const set = userSendChannels.get(userId);
+    if (!set)
+        return;
+    for (const send of set) {
+        try {
+            send(msg);
+        }
+        catch (err) {
+            // eslint-disable-next-line no-console
+            console.warn(`[active-harnesses] broadcast send failed for ${userId}:`, err instanceof Error ? err.message : err);
+        }
+    }
+}
+/** Test-only: drop everything. Not exported through the package barrel. */
+export function _resetActiveHarnesses() {
+    active.clear();
+    userSendChannels.clear();
+}
+//# sourceMappingURL=active-harnesses.js.map

package/packages/server/dist/chat/active-harnesses.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"active-harnesses.js","sourceRoot":"","sources":["../../src/chat/active-harnesses.ts"],"names":[],"mappings":"AAAA,6DAA6D;AAC7D,cAAc;AACd,EAAE;AACF,oBAAoB;AACpB,2DAA2D;AAC3D,iEAAiE;AACjE,yDAAyD;AACzD,iEAAiE;AACjE,iEAAiE;AACjE,kDAAkD;AAClD,EAAE;AACF,6CAA6C;AAC7C,gEAAgE;AAChE,kEAAkE;AAClE,+DAA+D;AAC/D,6CAA6C;AAC7C,EAAE;AACF,eAAe;AACf,iEAAiE;AACjE,wDAAwD;AACxD,6DAA6D;AAC7D,6DAA6D;AAC7D,4DAA4D;AAC5D,gBAAgB;AAChB,EAAE;AACF,2BAA2B;AAC3B,iEAAiE;AACjE,iEAAiE;AACjE,+DAA+D;AAC/D,uBAAuB;AAKvB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;AAE/C;;;;;;;;;;GAUG;AACH,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAyC,CAAC;AAE1E;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CACnC,SAAiB,EACjB,OAAqB;IAErB,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/B,OAAO,GAAG,EAAE;QACV,8DAA8D;QAC9D,4DAA4D;QAC5D,oDAAoD;QACpD,IAAI,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,OAAO,EAAE,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAAC,SAAiB;IAChD,OAAO,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AAC/B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,uBAAuB,CACrC,MAAc,EACd,IAA8B;IAE9B,IAAI,GAAG,GAAG,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACvC,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC;QAChB,gBAAgB,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACpC,CAAC;IACD,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACd,OAAO,GAAG,EAAE;QACV,MAAM,GAAG,GAAG,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACjB,IAAI,GAAG,CAAC,IAAI,KAAK,CAAC;YAAE,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,eAAe,CAAC,MAAc,EAAE,GAAc;IAC5D,MAAM,GAAG,GAAG,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACzC,IAAI,CAAC,GAAG;QAAE,OAAO;IACjB,KAAK,MAAM,IAAI,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,sCAAsC;YACtC,OAAO,CAAC,IAAI,CACV,gDAAgD,MAAM,GAAG,EACzD,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CACzC,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC;AAED,2EAA2E;AAC3E,MAAM,UAAU,qBAAqB;IACnC,MAAM,CAAC,KAAK,EAAE,CAAC;IACf,gBAAgB,CAAC,KAAK,EAAE,CAAC;AAC3B,CAAC"}

package/packages/server/dist/chat/agent-loop.d.ts

@@ -0,0 +1,64 @@
+import type { TenantContext } from "../core/index.js";
+import type { PluginRegistry } from "../core/plugins/registry.js";
+export interface AgentLoopRequest {
+    ctx: TenantContext;
+    /** Owner of the worker session. */
+    userId: string;
+    /** Initial user message. */
+    initialUserMessage: string;
+    /** Optional system-prompt override. */
+    systemPrompt?: string;
+    /** Optional model override. */
+    modelId?: string;
+    /** Allow-list of tool names. `null` / undefined = all available. */
+    toolsAllow?: string[] | null;
+    /** Deny-list applied AFTER `toolsAllow`; see `AgentLoopRunnerRequest`. */
+    toolsDeny?: string[] | null;
+    /** Allow-list of skill names. */
+    skillsAllow?: string[] | null;
+    /** Friendly title for the worker session row. */
+    sessionTitle?: string | null;
+    /** Worker role to stamp on the session row. */
+    workerRole?: string | null;
+    /** Worker filesystem slug (matches `_tenant/config/workers/<slug>/`).
+     *  Forwarded to `agentScope.slug` so `tenant_config_write` can
+     *  scope writes to this worker's own bundle. */
+    workerSlug?: string | null;
+    /** Parent (user) session id. */
+    parentSessionId?: string | null;
+    /** Workboard task id when this run is driven by a task. Plumbed
+     *  through to `AgentToolContext.taskId` so per-task tools (most
+     *  importantly the microsandbox `exec` route) can scope their
+     *  resources to the task lifecycle. */
+    taskId?: string | null;
+    /** Plugin registry for tool/skill discovery. */
+    pluginRegistry?: PluginRegistry;
+    /** Tenant root dir on the host. */
+    homeDir?: string;
+    /** Soft per-call timeout caps. 0 disables a layer. */
+    timeouts?: {
+        firstResponseMs?: number;
+        idleMs?: number;
+        maxRunMs?: number;
+    };
+    /** External abort signal. */
+    signal?: AbortSignal;
+    /** Fires once after the worker session row has been inserted. */
+    onSessionStart?: (sessionId: string) => void;
+    /**
+     * Resume an existing worker session instead of creating a fresh
+     * one. When set, `initialUserMessage` is treated as a follow-up
+     * prompt within the existing transcript.
+     */
+    resumeSessionId?: string;
+}
+export interface AgentLoopResult {
+    status: "done" | "stalled" | "aborted" | "error";
+    summary: string;
+    files: string[];
+    sessionId: string;
+    turns: number;
+    reason: "task_complete" | "no_completion" | "first_response_timeout" | "idle_timeout" | "max_run_timeout" | "aborted" | "stream_error" | "exception";
+}
+export declare function runAgentLoop(req: AgentLoopRequest): Promise<AgentLoopResult>;
+//# sourceMappingURL=agent-loop.d.ts.map

package/packages/server/dist/chat/agent-loop.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-loop.d.ts","sourceRoot":"","sources":["../../src/chat/agent-loop.ts"],"names":[],"mappings":"AA4BA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAuBtD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAKlE,MAAM,WAAW,gBAAgB;IAC/B,GAAG,EAAE,aAAa,CAAC;IACnB,mCAAmC;IACnC,MAAM,EAAE,MAAM,CAAC;IACf,4BAA4B;IAC5B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,uCAAuC;IACvC,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,+BAA+B;IAC/B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,oEAAoE;IACpE,UAAU,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAC7B,0EAA0E;IAC1E,SAAS,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAC5B,iCAAiC;IACjC,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAC9B,iDAAiD;IACjD,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,+CAA+C;IAC/C,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B;;oDAEgD;IAChD,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,gCAAgC;IAChC,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC;;;2CAGuC;IACvC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,gDAAgD;IAChD,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,mCAAmC;IACnC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,sDAAsD;IACtD,QAAQ,CAAC,EAAE;QACT,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,6BAA6B;IAC7B,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,iEAAiE;IACjE,cAAc,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IAC7C;;;;OAIG;IACH,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,GAAG,OAAO,CAAC;IACjD,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EACF,eAAe,GACf,eAAe,GACf,wBAAwB,GACxB,cAAc,GACd,iBAAiB,GACjB,SAAS,GACT,cAAc,GACd,WAAW,CAAC;CACjB;AAeD,wBAAsB,YAAY,CAChC,GAAG,EAAE,gBAAgB,GACpB,OAAO,CAAC,eAAe,CAAC,CA0iB1B"}