@steventsao/agent-session-core 0.1.11 → 0.1.13

package/dist/index.js

@@ -1,10 +1,226 @@
-/**
- * @agent-session/core
- *
- * Shared types and protocol for agent-session.
- * Used by both client and server packages.
- */
-export * from './socket-events';
-export * from './types';
-export * from './session-config';
-//# sourceMappingURL=index.js.map
+// src/socket-events.ts
+import { z } from "zod";
+var ClientEventType = {
+  // Session
+  INIT: "INIT",
+  PING: "PING",
+  JOIN_SESSION: "JOIN_SESSION",
+  // Sandbox lifecycle
+  START_SANDBOX: "START_SANDBOX",
+  STOP_SANDBOX: "STOP_SANDBOX",
+  // Command execution
+  EXEC: "EXEC",
+  // Agent
+  AGENT_MESSAGE: "AGENT_MESSAGE",
+  AGENT_STOP: "AGENT_STOP",
+  AGENT_RESET: "AGENT_RESET"
+};
+var ServerEventType = {
+  // Connection
+  CONNECTED: "CONNECTED",
+  READY: "READY",
+  PONG: "PONG",
+  MESSAGE_ACK: "MESSAGE_ACK",
+  MESSAGE_ERROR: "MESSAGE_ERROR",
+  // Event resumption (Manus-style)
+  EVENTS_BATCH: "EVENTS_BATCH",
+  // Sandbox status
+  SANDBOX_STATUS: "SANDBOX_STATUS",
+  // Lifecycle events (granular phases)
+  LIFECYCLE: "LIFECYCLE",
+  // Terminal
+  TERM_DATA: "TERM_DATA",
+  EXEC_COMPLETE: "EXEC_COMPLETE",
+  // System messages
+  SYSTEM: "SYSTEM",
+  ERROR: "ERROR",
+  // Agent events
+  AGENT_STARTED: "AGENT_STARTED",
+  AGENT_MESSAGE: "AGENT_MESSAGE",
+  AGENT_ACTION: "AGENT_ACTION",
+  AGENT_STEP_COMPLETE: "AGENT_STEP_COMPLETE",
+  AGENT_DONE: "AGENT_DONE",
+  AGENT_ERROR: "AGENT_ERROR",
+  AGENT_EVENT: "AGENT_EVENT",
+  FILE_UPLOADED: "FILE_UPLOADED",
+  FILES_SYNC: "FILES_SYNC",
+  // TOC extraction events
+  TOC_REQUEST_RECEIVED: "TOC_REQUEST_RECEIVED",
+  TOC_AUTH_COMPLETED: "TOC_AUTH_COMPLETED",
+  TOC_DB_QUERY_STARTED: "TOC_DB_QUERY_STARTED",
+  TOC_DB_QUERY_COMPLETED: "TOC_DB_QUERY_COMPLETED",
+  TOC_GCS_DOWNLOAD_STARTED: "TOC_GCS_DOWNLOAD_STARTED",
+  TOC_GCS_DOWNLOAD_COMPLETED: "TOC_GCS_DOWNLOAD_COMPLETED",
+  TOC_SANDBOX_CREATED: "TOC_SANDBOX_CREATED",
+  TOC_EXTRACTION_STARTED: "TOC_EXTRACTION_STARTED",
+  TOC_EXTRACTION_COMPLETED: "TOC_EXTRACTION_COMPLETED",
+  TOC_RESPONSE_READY: "TOC_RESPONSE_READY"
+};
+var SandboxStatus = {
+  IDLE: "idle",
+  BOOTING: "booting",
+  READY: "ready",
+  ERROR: "error"
+};
+var SandboxLifecycle = {
+  IDLE: "idle",
+  BOOTING: "booting",
+  ENVIRONMENT_READY: "environment_ready",
+  FILES_READY: "files_ready",
+  INSTALLING: "installing",
+  STARTING: "starting",
+  INTERACTION_READY: "interaction_ready",
+  READY: "ready",
+  // alias for backwards compat
+  ERROR: "error"
+};
+function parseSystemMessageToLifecycle(msg) {
+  const lowerMsg = msg.toLowerCase();
+  if (lowerMsg.includes("booting")) return SandboxLifecycle.BOOTING;
+  if (lowerMsg.includes("environment ready")) return SandboxLifecycle.ENVIRONMENT_READY;
+  if (lowerMsg.includes("files ready") || lowerMsg.includes("files loaded") || lowerMsg.includes("restored")) return SandboxLifecycle.FILES_READY;
+  if (lowerMsg.includes("installing")) return SandboxLifecycle.INSTALLING;
+  if (lowerMsg.includes("starting")) return SandboxLifecycle.STARTING;
+  if (lowerMsg.includes("interaction ready") || lowerMsg.includes("agent ready")) return SandboxLifecycle.INTERACTION_READY;
+  if (lowerMsg.includes("ready")) return SandboxLifecycle.READY;
+  if (lowerMsg.includes("error")) return SandboxLifecycle.ERROR;
+  return null;
+}
+var EPHEMERAL_EVENT_TYPES = /* @__PURE__ */ new Set([
+  "CONNECTED",
+  "READY",
+  "PONG",
+  "MESSAGE_ACK",
+  "MESSAGE_ERROR",
+  "TERM_DATA",
+  "EXEC_COMPLETE",
+  "EVENTS_BATCH",
+  "AGENT_STEP_COMPLETE",
+  "AGENT_EVENT",
+  "ERROR"
+]);
+function isPersistableEvent(event) {
+  return !EPHEMERAL_EVENT_TYPES.has(event.type);
+}
+var ToolUseBlockSchema = z.object({
+  type: z.literal("tool_use"),
+  name: z.string(),
+  input: z.record(z.unknown()).optional()
+});
+var TextBlockSchema = z.object({
+  type: z.literal("text"),
+  text: z.string()
+});
+var ContentBlockSchema = z.discriminatedUnion("type", [
+  ToolUseBlockSchema,
+  TextBlockSchema
+]);
+var AssistantMessageSchema = z.object({
+  type: z.literal("assistant"),
+  message: z.object({
+    content: z.array(ContentBlockSchema)
+  })
+});
+var ResultMessageSchema = z.object({
+  type: z.literal("result"),
+  session_id: z.string().optional(),
+  total_cost_usd: z.number().optional(),
+  duration_ms: z.number().optional()
+});
+var ErrorMessageSchema = z.object({
+  type: z.literal("error"),
+  error: z.string()
+});
+var SystemMessageSchema = z.object({
+  type: z.literal("system"),
+  message: z.string()
+});
+var AgentOutputMessageSchema = z.discriminatedUnion("type", [
+  AssistantMessageSchema,
+  ResultMessageSchema,
+  ErrorMessageSchema,
+  SystemMessageSchema
+]);
+function parseAgentOutputMessage(line) {
+  if (!line.trim()) return null;
+  try {
+    const json = JSON.parse(line);
+    const result = AgentOutputMessageSchema.safeParse(json);
+    return result.success ? result.data : null;
+  } catch {
+    return null;
+  }
+}
+var CLIENT_EVENT_TYPES = new Set(Object.values(ClientEventType));
+var SERVER_EVENT_TYPES = new Set(Object.values(ServerEventType));
+function isClientEvent(data) {
+  if (!data || typeof data !== "object") return false;
+  const maybeEvent = data;
+  return typeof maybeEvent.type === "string" && CLIENT_EVENT_TYPES.has(maybeEvent.type);
+}
+function isServerEvent(data) {
+  if (!data || typeof data !== "object") return false;
+  const maybeEvent = data;
+  return typeof maybeEvent.type === "string" && SERVER_EVENT_TYPES.has(maybeEvent.type);
+}
+function parseServerEvent(json) {
+  try {
+    const parsed = JSON.parse(json);
+    if (isServerEvent(parsed)) return parsed;
+    return null;
+  } catch {
+    return null;
+  }
+}
+function parseClientEvent(json) {
+  try {
+    const parsed = JSON.parse(json);
+    if (isClientEvent(parsed)) return parsed;
+    return null;
+  } catch {
+    return null;
+  }
+}
+
+// src/session-config.ts
+var DEFAULT_SESSION_REFRESH_CONFIG = {
+  onSessionChange: "reset",
+  clearMessagesOnReset: true,
+  clearAgentSessionOnReset: true
+};
+var PRESERVE_SESSION_CONFIG = {
+  onSessionChange: "preserve",
+  clearMessagesOnReset: false,
+  clearAgentSessionOnReset: false
+};
+function shouldResetOnSessionChange(config, oldSessionId, newSessionId) {
+  if (config.onSessionChange === "preserve") {
+    return false;
+  }
+  return oldSessionId !== newSessionId && oldSessionId !== null && newSessionId !== null;
+}
+export {
+  AgentOutputMessageSchema,
+  AssistantMessageSchema,
+  ClientEventType,
+  ContentBlockSchema,
+  DEFAULT_SESSION_REFRESH_CONFIG,
+  EPHEMERAL_EVENT_TYPES,
+  ErrorMessageSchema,
+  PRESERVE_SESSION_CONFIG,
+  ResultMessageSchema,
+  SandboxLifecycle,
+  SandboxStatus,
+  ServerEventType,
+  SystemMessageSchema,
+  TextBlockSchema,
+  ToolUseBlockSchema,
+  isClientEvent,
+  isPersistableEvent,
+  isServerEvent,
+  parseAgentOutputMessage,
+  parseClientEvent,
+  parseServerEvent,
+  parseSystemMessageToLifecycle,
+  shouldResetOnSessionChange
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@steventsao/agent-session-core",
-  "version": "0.1.11",
+  "version": "0.1.13",
   "description": "Shared types and protocol for agent-session",
   "type": "module",
   "main": "dist/index.js",
@@ -15,8 +15,8 @@
     "dist"
   ],
   "scripts": {
-    "build": "tsc",
-    "dev": "tsc --watch"
+    "build": "tsup",
+    "dev": "tsup --watch"
   },
   "dependencies": {
     "zod": "^3.23.0"

package/dist/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,cAAc,iBAAiB,CAAC;AAChC,cAAc,SAAS,CAAC;AACxB,cAAc,kBAAkB,CAAC"}

package/dist/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,cAAc,iBAAiB,CAAC;AAChC,cAAc,SAAS,CAAC;AACxB,cAAc,kBAAkB,CAAC"}

package/dist/session-config.d.ts

@@ -1,55 +0,0 @@
-/**
- * Session Refresh Configuration
- *
- * Defines how the client should behave when the sessionId changes.
- * This is critical for document isolation in multi-document applications.
- */
-/**
- * Refresh strategies for session changes
- *
- * - 'reset': Clear all client state and connect fresh (recommended for document isolation)
- * - 'preserve': Attempt to reconnect to existing sandbox (for tab persistence scenarios)
- */
-export type SessionRefreshStrategy = 'reset' | 'preserve';
-export interface SessionRefreshConfig {
-    /**
-     * What to do when sessionId changes:
-     *
-     * 'reset' (default, recommended):
-     *   - Clear sandboxId, sandboxUrl, messages, agentSessionId
-     *   - Disconnect from old socket
-     *   - Connect to new session with clean slate
-     *   - Use this when sessionId represents a document UUID
-     *
-     * 'preserve':
-     *   - Keep existing state during reconnection
-     *   - Useful for browser tab persistence where same user returns
-     *   - NOT recommended when switching between documents
-     */
-    onSessionChange: SessionRefreshStrategy;
-    /**
-     * Whether to clear chat messages when session changes (only applies to 'reset' strategy)
-     * Default: true
-     */
-    clearMessagesOnReset?: boolean;
-    /**
-     * Whether to clear agent session ID when session changes (only applies to 'reset' strategy)
-     * Default: true
-     */
-    clearAgentSessionOnReset?: boolean;
-}
-/**
- * Default configuration - full reset on session change
- * This is the safest option for document isolation
- */
-export declare const DEFAULT_SESSION_REFRESH_CONFIG: SessionRefreshConfig;
-/**
- * Preserve configuration - for tab persistence scenarios
- * Use with caution - only when you know the user is returning to the same document
- */
-export declare const PRESERVE_SESSION_CONFIG: SessionRefreshConfig;
-/**
- * Helper to determine if state should be reset based on config
- */
-export declare function shouldResetOnSessionChange(config: SessionRefreshConfig, oldSessionId: string | null, newSessionId: string | null): boolean;
-//# sourceMappingURL=session-config.d.ts.map

package/dist/session-config.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"session-config.d.ts","sourceRoot":"","sources":["../src/session-config.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH;;;;;GAKG;AACH,MAAM,MAAM,sBAAsB,GAAG,OAAO,GAAG,UAAU,CAAC;AAE1D,MAAM,WAAW,oBAAoB;IACnC;;;;;;;;;;;;;OAaG;IACH,eAAe,EAAE,sBAAsB,CAAC;IAExC;;;OAGG;IACH,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAE/B;;;OAGG;IACH,wBAAwB,CAAC,EAAE,OAAO,CAAC;CACpC;AAED;;;GAGG;AACH,eAAO,MAAM,8BAA8B,EAAE,oBAI5C,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,uBAAuB,EAAE,oBAIrC,CAAC;AAEF;;GAEG;AACH,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,oBAAoB,EAC5B,YAAY,EAAE,MAAM,GAAG,IAAI,EAC3B,YAAY,EAAE,MAAM,GAAG,IAAI,GAC1B,OAAO,CAOT"}

package/dist/session-config.js

@@ -1,35 +0,0 @@
-/**
- * Session Refresh Configuration
- *
- * Defines how the client should behave when the sessionId changes.
- * This is critical for document isolation in multi-document applications.
- */
-/**
- * Default configuration - full reset on session change
- * This is the safest option for document isolation
- */
-export const DEFAULT_SESSION_REFRESH_CONFIG = {
-    onSessionChange: 'reset',
-    clearMessagesOnReset: true,
-    clearAgentSessionOnReset: true,
-};
-/**
- * Preserve configuration - for tab persistence scenarios
- * Use with caution - only when you know the user is returning to the same document
- */
-export const PRESERVE_SESSION_CONFIG = {
-    onSessionChange: 'preserve',
-    clearMessagesOnReset: false,
-    clearAgentSessionOnReset: false,
-};
-/**
- * Helper to determine if state should be reset based on config
- */
-export function shouldResetOnSessionChange(config, oldSessionId, newSessionId) {
-    if (config.onSessionChange === 'preserve') {
-        return false;
-    }
-    // Reset if sessions are different
-    return oldSessionId !== newSessionId && oldSessionId !== null && newSessionId !== null;
-}
-//# sourceMappingURL=session-config.js.map

package/dist/session-config.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"session-config.js","sourceRoot":"","sources":["../src/session-config.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAwCH;;;GAGG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAyB;IAClE,eAAe,EAAE,OAAO;IACxB,oBAAoB,EAAE,IAAI;IAC1B,wBAAwB,EAAE,IAAI;CAC/B,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAyB;IAC3D,eAAe,EAAE,UAAU;IAC3B,oBAAoB,EAAE,KAAK;IAC3B,wBAAwB,EAAE,KAAK;CAChC,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,0BAA0B,CACxC,MAA4B,EAC5B,YAA2B,EAC3B,YAA2B;IAE3B,IAAI,MAAM,CAAC,eAAe,KAAK,UAAU,EAAE,CAAC;QAC1C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,kCAAkC;IAClC,OAAO,YAAY,KAAK,YAAY,IAAI,YAAY,KAAK,IAAI,IAAI,YAAY,KAAK,IAAI,CAAC;AACzF,CAAC"}