@posthog/agent 1.24.2 → 1.25.0

package/dist/src/schemas.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.d.ts","sourceRoot":"","sources":["../../src/schemas.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAQxB,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;EAI3B,CAAC;AAEH,eAAO,MAAM,4BAA4B;;;;;;;;;;;;;;;;;;;;;;EAMvC,CAAC;AAEH,eAAO,MAAM,2BAA2B;;;;;;;;;;;;;EAGtC,CAAC;AAGH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAQ9B,CAAC;AAEH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAShC,CAAC;AAGH,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;EAIlC,CAAC;AAEH,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EASlC,CAAC;AAEH,eAAO,MAAM,sBAAsB;;;;;;;;;;EAEjC,CAAC;AAGH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;EAIjC,CAAC;AAGH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCASd,CAAC;AAEjB,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAa1B,CAAC;AAGH,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;;EAI7B,CAAC;AAEH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;EAIrC,CAAC;AAGH,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA+B1B,CAAC;AAEH,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;EAO3B,CAAC;AAGH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;EAK5B,CAAC;AAEH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;EAI9B,CAAC;AAEH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;EAG5B,CAAC;AAEH,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IAmB3B,CAAC;AAEH,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAC1D,MAAM,MAAM,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAC1C,OAAO,4BAA4B,CACpC,CAAC;AACF,MAAM,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAChF,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAChE,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AACpE,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AACxE,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AACxE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AACtE,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAC5D,MAAM,MAAM,SAAS,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAAC;AACxD,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAC9D,MAAM,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,0BAA0B,CAAC,CAAC;AAC9E,MAAM,MAAM,SAAS,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,CAAC;AACxD,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAC1D,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAC5D,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAChE,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAC5D,MAAM,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAE1D;;;GAGG;AACH,wBAAgB,eAAe,CAAC,KAAK,EAAE,OAAO,GAAG,UAAU,GAAG,IAAI,CAGjE;AAED;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,GAAG,UAAU,EAAE,CAIhE"}

package/dist/src/schemas.js

@@ -0,0 +1,196 @@
+import { z } from 'zod';
+
+// Base event schema with timestamp
+const BaseEventSchema = z.object({
+    ts: z.number(),
+});
+// Streaming content events
+const TokenEventSchema = BaseEventSchema.extend({
+    type: z.literal("token"),
+    content: z.string(),
+    contentType: z.enum(["text", "thinking", "tool_input"]).optional(),
+});
+const ContentBlockStartEventSchema = BaseEventSchema.extend({
+    type: z.literal("content_block_start"),
+    index: z.number(),
+    contentType: z.enum(["text", "tool_use", "thinking"]),
+    toolName: z.string().optional(),
+    toolId: z.string().optional(),
+});
+const ContentBlockStopEventSchema = BaseEventSchema.extend({
+    type: z.literal("content_block_stop"),
+    index: z.number(),
+});
+// Tool events
+const ToolCallEventSchema = BaseEventSchema.extend({
+    type: z.literal("tool_call"),
+    toolName: z.string(),
+    callId: z.string(),
+    args: z.record(z.string(), z.unknown()),
+    parentToolUseId: z.string().nullable().optional(),
+    tool: z.unknown().optional(),
+    category: z.unknown().optional(),
+});
+const ToolResultEventSchema = BaseEventSchema.extend({
+    type: z.literal("tool_result"),
+    toolName: z.string(),
+    callId: z.string(),
+    result: z.unknown(),
+    isError: z.boolean().optional(),
+    parentToolUseId: z.string().nullable().optional(),
+    tool: z.unknown().optional(),
+    category: z.unknown().optional(),
+});
+// Message lifecycle events
+const MessageStartEventSchema = BaseEventSchema.extend({
+    type: z.literal("message_start"),
+    messageId: z.string().optional(),
+    model: z.string().optional(),
+});
+const MessageDeltaEventSchema = BaseEventSchema.extend({
+    type: z.literal("message_delta"),
+    stopReason: z.string().optional(),
+    stopSequence: z.string().optional(),
+    usage: z
+        .object({
+        outputTokens: z.number(),
+    })
+        .optional(),
+});
+const MessageStopEventSchema = BaseEventSchema.extend({
+    type: z.literal("message_stop"),
+});
+// User message events
+const UserMessageEventSchema = BaseEventSchema.extend({
+    type: z.literal("user_message"),
+    content: z.string(),
+    isSynthetic: z.boolean().optional(),
+});
+// System events
+const StatusEventSchema = BaseEventSchema.extend({
+    type: z.literal("status"),
+    phase: z.string(),
+    kind: z.string().optional(),
+    branch: z.string().optional(),
+    prUrl: z.string().optional(),
+    taskId: z.string().optional(),
+    messageId: z.string().optional(),
+    model: z.string().optional(),
+}).passthrough(); // Allow additional fields
+const InitEventSchema = BaseEventSchema.extend({
+    type: z.literal("init"),
+    model: z.string(),
+    tools: z.array(z.string()),
+    permissionMode: z.string(),
+    cwd: z.string(),
+    apiKeySource: z.string(),
+    agents: z.array(z.string()).optional(),
+    slashCommands: z.array(z.string()).optional(),
+    outputStyle: z.string().optional(),
+    mcpServers: z
+        .array(z.object({ name: z.string(), status: z.string() }))
+        .optional(),
+});
+// Console event for log-style output
+const ConsoleEventSchema = BaseEventSchema.extend({
+    type: z.literal("console"),
+    level: z.enum(["debug", "info", "warn", "error"]),
+    message: z.string(),
+});
+const CompactBoundaryEventSchema = BaseEventSchema.extend({
+    type: z.literal("compact_boundary"),
+    trigger: z.enum(["manual", "auto"]),
+    preTokens: z.number(),
+});
+// Result events
+const DoneEventSchema = BaseEventSchema.extend({
+    type: z.literal("done"),
+    result: z.string().optional(),
+    durationMs: z.number().optional(),
+    durationApiMs: z.number().optional(),
+    numTurns: z.number().optional(),
+    totalCostUsd: z.number().optional(),
+    usage: z.unknown().optional(),
+    modelUsage: z
+        .record(z.string(), z.object({
+        inputTokens: z.number(),
+        outputTokens: z.number(),
+        cacheReadInputTokens: z.number(),
+        cacheCreationInputTokens: z.number(),
+        webSearchRequests: z.number(),
+        costUSD: z.number(),
+        contextWindow: z.number(),
+    }))
+        .optional(),
+    permissionDenials: z
+        .array(z.object({
+        tool_name: z.string(),
+        tool_use_id: z.string(),
+        tool_input: z.record(z.string(), z.unknown()),
+    }))
+        .optional(),
+});
+const ErrorEventSchema = BaseEventSchema.extend({
+    type: z.literal("error"),
+    message: z.string(),
+    error: z.unknown().optional(),
+    errorType: z.string().optional(),
+    context: z.record(z.string(), z.unknown()).optional(),
+    sdkError: z.unknown().optional(),
+});
+// Metric and artifact events
+const MetricEventSchema = BaseEventSchema.extend({
+    type: z.literal("metric"),
+    key: z.string(),
+    value: z.number(),
+    unit: z.string().optional(),
+});
+const ArtifactEventSchema = BaseEventSchema.extend({
+    type: z.literal("artifact"),
+    kind: z.string(),
+    content: z.unknown(),
+});
+const RawSDKEventSchema = BaseEventSchema.extend({
+    type: z.literal("raw_sdk_event"),
+    sdkMessage: z.unknown(),
+});
+const AgentEventSchema = z.discriminatedUnion("type", [
+    TokenEventSchema,
+    ContentBlockStartEventSchema,
+    ContentBlockStopEventSchema,
+    ToolCallEventSchema,
+    ToolResultEventSchema,
+    MessageStartEventSchema,
+    MessageDeltaEventSchema,
+    MessageStopEventSchema,
+    UserMessageEventSchema,
+    StatusEventSchema,
+    InitEventSchema,
+    ConsoleEventSchema,
+    CompactBoundaryEventSchema,
+    DoneEventSchema,
+    ErrorEventSchema,
+    MetricEventSchema,
+    ArtifactEventSchema,
+    RawSDKEventSchema,
+]);
+/**
+ * Parse and validate an AgentEvent from unknown input.
+ * Returns the parsed event if valid, or null if invalid.
+ */
+function parseAgentEvent(input) {
+    const result = AgentEventSchema.safeParse(input);
+    return result.success ? result.data : null;
+}
+/**
+ * Parse and validate multiple AgentEvents from an array of unknown inputs.
+ * Invalid entries are discarded.
+ */
+function parseAgentEvents(inputs) {
+    return inputs
+        .map((input) => parseAgentEvent(input))
+        .filter((event) => event !== null);
+}
+
+export { AgentEventSchema, ArtifactEventSchema, CompactBoundaryEventSchema, ConsoleEventSchema, ContentBlockStartEventSchema, ContentBlockStopEventSchema, DoneEventSchema, ErrorEventSchema, InitEventSchema, MessageDeltaEventSchema, MessageStartEventSchema, MessageStopEventSchema, MetricEventSchema, RawSDKEventSchema, StatusEventSchema, TokenEventSchema, ToolCallEventSchema, ToolResultEventSchema, UserMessageEventSchema, parseAgentEvent, parseAgentEvents };
+//# sourceMappingURL=schemas.js.map

package/dist/src/schemas.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.js","sources":["../../src/schemas.ts"],"sourcesContent":["import { z } from \"zod\";\n\n// Base event schema with timestamp\nconst BaseEventSchema = z.object({\n  ts: z.number(),\n});\n\n// Streaming content events\nexport const TokenEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"token\"),\n  content: z.string(),\n  contentType: z.enum([\"text\", \"thinking\", \"tool_input\"]).optional(),\n});\n\nexport const ContentBlockStartEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"content_block_start\"),\n  index: z.number(),\n  contentType: z.enum([\"text\", \"tool_use\", \"thinking\"]),\n  toolName: z.string().optional(),\n  toolId: z.string().optional(),\n});\n\nexport const ContentBlockStopEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"content_block_stop\"),\n  index: z.number(),\n});\n\n// Tool events\nexport const ToolCallEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"tool_call\"),\n  toolName: z.string(),\n  callId: z.string(),\n  args: z.record(z.string(), z.unknown()),\n  parentToolUseId: z.string().nullable().optional(),\n  tool: z.unknown().optional(),\n  category: z.unknown().optional(),\n});\n\nexport const ToolResultEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"tool_result\"),\n  toolName: z.string(),\n  callId: z.string(),\n  result: z.unknown(),\n  isError: z.boolean().optional(),\n  parentToolUseId: z.string().nullable().optional(),\n  tool: z.unknown().optional(),\n  category: z.unknown().optional(),\n});\n\n// Message lifecycle events\nexport const MessageStartEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"message_start\"),\n  messageId: z.string().optional(),\n  model: z.string().optional(),\n});\n\nexport const MessageDeltaEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"message_delta\"),\n  stopReason: z.string().optional(),\n  stopSequence: z.string().optional(),\n  usage: z\n    .object({\n      outputTokens: z.number(),\n    })\n    .optional(),\n});\n\nexport const MessageStopEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"message_stop\"),\n});\n\n// User message events\nexport const UserMessageEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"user_message\"),\n  content: z.string(),\n  isSynthetic: z.boolean().optional(),\n});\n\n// System events\nexport const StatusEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"status\"),\n  phase: z.string(),\n  kind: z.string().optional(),\n  branch: z.string().optional(),\n  prUrl: z.string().optional(),\n  taskId: z.string().optional(),\n  messageId: z.string().optional(),\n  model: z.string().optional(),\n}).passthrough(); // Allow additional fields\n\nexport const InitEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"init\"),\n  model: z.string(),\n  tools: z.array(z.string()),\n  permissionMode: z.string(),\n  cwd: z.string(),\n  apiKeySource: z.string(),\n  agents: z.array(z.string()).optional(),\n  slashCommands: z.array(z.string()).optional(),\n  outputStyle: z.string().optional(),\n  mcpServers: z\n    .array(z.object({ name: z.string(), status: z.string() }))\n    .optional(),\n});\n\n// Console event for log-style output\nexport const ConsoleEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"console\"),\n  level: z.enum([\"debug\", \"info\", \"warn\", \"error\"]),\n  message: z.string(),\n});\n\nexport const CompactBoundaryEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"compact_boundary\"),\n  trigger: z.enum([\"manual\", \"auto\"]),\n  preTokens: z.number(),\n});\n\n// Result events\nexport const DoneEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"done\"),\n  result: z.string().optional(),\n  durationMs: z.number().optional(),\n  durationApiMs: z.number().optional(),\n  numTurns: z.number().optional(),\n  totalCostUsd: z.number().optional(),\n  usage: z.unknown().optional(),\n  modelUsage: z\n    .record(\n      z.string(),\n      z.object({\n        inputTokens: z.number(),\n        outputTokens: z.number(),\n        cacheReadInputTokens: z.number(),\n        cacheCreationInputTokens: z.number(),\n        webSearchRequests: z.number(),\n        costUSD: z.number(),\n        contextWindow: z.number(),\n      }),\n    )\n    .optional(),\n  permissionDenials: z\n    .array(\n      z.object({\n        tool_name: z.string(),\n        tool_use_id: z.string(),\n        tool_input: z.record(z.string(), z.unknown()),\n      }),\n    )\n    .optional(),\n});\n\nexport const ErrorEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"error\"),\n  message: z.string(),\n  error: z.unknown().optional(),\n  errorType: z.string().optional(),\n  context: z.record(z.string(), z.unknown()).optional(),\n  sdkError: z.unknown().optional(),\n});\n\n// Metric and artifact events\nexport const MetricEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"metric\"),\n  key: z.string(),\n  value: z.number(),\n  unit: z.string().optional(),\n});\n\nexport const ArtifactEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"artifact\"),\n  kind: z.string(),\n  content: z.unknown(),\n});\n\nexport const RawSDKEventSchema = BaseEventSchema.extend({\n  type: z.literal(\"raw_sdk_event\"),\n  sdkMessage: z.unknown(),\n});\n\nexport const AgentEventSchema = z.discriminatedUnion(\"type\", [\n  TokenEventSchema,\n  ContentBlockStartEventSchema,\n  ContentBlockStopEventSchema,\n  ToolCallEventSchema,\n  ToolResultEventSchema,\n  MessageStartEventSchema,\n  MessageDeltaEventSchema,\n  MessageStopEventSchema,\n  UserMessageEventSchema,\n  StatusEventSchema,\n  InitEventSchema,\n  ConsoleEventSchema,\n  CompactBoundaryEventSchema,\n  DoneEventSchema,\n  ErrorEventSchema,\n  MetricEventSchema,\n  ArtifactEventSchema,\n  RawSDKEventSchema,\n]);\n\nexport type TokenEvent = z.infer<typeof TokenEventSchema>;\nexport type ContentBlockStartEvent = z.infer<\n  typeof ContentBlockStartEventSchema\n>;\nexport type ContentBlockStopEvent = z.infer<typeof ContentBlockStopEventSchema>;\nexport type ToolCallEvent = z.infer<typeof ToolCallEventSchema>;\nexport type ToolResultEvent = z.infer<typeof ToolResultEventSchema>;\nexport type MessageStartEvent = z.infer<typeof MessageStartEventSchema>;\nexport type MessageDeltaEvent = z.infer<typeof MessageDeltaEventSchema>;\nexport type MessageStopEvent = z.infer<typeof MessageStopEventSchema>;\nexport type UserMessageEvent = z.infer<typeof UserMessageEventSchema>;\nexport type StatusEvent = z.infer<typeof StatusEventSchema>;\nexport type InitEvent = z.infer<typeof InitEventSchema>;\nexport type ConsoleEvent = z.infer<typeof ConsoleEventSchema>;\nexport type CompactBoundaryEvent = z.infer<typeof CompactBoundaryEventSchema>;\nexport type DoneEvent = z.infer<typeof DoneEventSchema>;\nexport type ErrorEvent = z.infer<typeof ErrorEventSchema>;\nexport type MetricEvent = z.infer<typeof MetricEventSchema>;\nexport type ArtifactEvent = z.infer<typeof ArtifactEventSchema>;\nexport type RawSDKEvent = z.infer<typeof RawSDKEventSchema>;\nexport type AgentEvent = z.infer<typeof AgentEventSchema>;\n\n/**\n * Parse and validate an AgentEvent from unknown input.\n * Returns the parsed event if valid, or null if invalid.\n */\nexport function parseAgentEvent(input: unknown): AgentEvent | null {\n  const result = AgentEventSchema.safeParse(input);\n  return result.success ? result.data : null;\n}\n\n/**\n * Parse and validate multiple AgentEvents from an array of unknown inputs.\n * Invalid entries are discarded.\n */\nexport function parseAgentEvents(inputs: unknown[]): AgentEvent[] {\n  return inputs\n    .map((input) => parseAgentEvent(input))\n    .filter((event): event is AgentEvent => event !== null);\n}\n"],"names":[],"mappings":";;AAEA;AACA,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;AAC/B,IAAA,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;AACf,CAAA,CAAC;AAEF;AACO,MAAM,gBAAgB,GAAG,eAAe,CAAC,MAAM,CAAC;AACrD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;AACxB,IAAA,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;AACnB,IAAA,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC,QAAQ,EAAE;AACnE,CAAA;AAEM,MAAM,4BAA4B,GAAG,eAAe,CAAC,MAAM,CAAC;AACjE,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC;AACtC,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;AACjB,IAAA,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;AACrD,IAAA,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC/B,IAAA,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC9B,CAAA;AAEM,MAAM,2BAA2B,GAAG,eAAe,CAAC,MAAM,CAAC;AAChE,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC;AACrC,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;AAClB,CAAA;AAED;AACO,MAAM,mBAAmB,GAAG,eAAe,CAAC,MAAM,CAAC;AACxD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC;AAC5B,IAAA,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;AACpB,IAAA,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;AAClB,IAAA,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IACvC,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;AACjD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AAC5B,IAAA,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AACjC,CAAA;AAEM,MAAM,qBAAqB,GAAG,eAAe,CAAC,MAAM,CAAC;AAC1D,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC;AAC9B,IAAA,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;AACpB,IAAA,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;AAClB,IAAA,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;AACnB,IAAA,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IAC/B,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;AACjD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AAC5B,IAAA,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AACjC,CAAA;AAED;AACO,MAAM,uBAAuB,GAAG,eAAe,CAAC,MAAM,CAAC;AAC5D,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC;AAChC,IAAA,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAChC,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC7B,CAAA;AAEM,MAAM,uBAAuB,GAAG,eAAe,CAAC,MAAM,CAAC;AAC5D,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC;AAChC,IAAA,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AACjC,IAAA,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AACnC,IAAA,KAAK,EAAE;AACJ,SAAA,MAAM,CAAC;AACN,QAAA,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;KACzB;AACA,SAAA,QAAQ,EAAE;AACd,CAAA;AAEM,MAAM,sBAAsB,GAAG,eAAe,CAAC,MAAM,CAAC;AAC3D,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;AAChC,CAAA;AAED;AACO,MAAM,sBAAsB,GAAG,eAAe,CAAC,MAAM,CAAC;AAC3D,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;AAC/B,IAAA,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;AACnB,IAAA,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AACpC,CAAA;AAED;AACO,MAAM,iBAAiB,GAAG,eAAe,CAAC,MAAM,CAAC;AACtD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;AACzB,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;AACjB,IAAA,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC3B,IAAA,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC7B,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC5B,IAAA,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC7B,IAAA,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAChC,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC7B,CAAA,CAAC,CAAC,WAAW,GAAG;AAEV,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC;AACpD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;AACvB,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;IACjB,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;AAC1B,IAAA,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE;AAC1B,IAAA,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;AACf,IAAA,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;AACxB,IAAA,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;AACtC,IAAA,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;AAC7C,IAAA,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAClC,IAAA,UAAU,EAAE;SACT,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;AACxD,SAAA,QAAQ,EAAE;AACd,CAAA;AAED;AACO,MAAM,kBAAkB,GAAG,eAAe,CAAC,MAAM,CAAC;AACvD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC;AAC1B,IAAA,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;AACjD,IAAA,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;AACpB,CAAA;AAEM,MAAM,0BAA0B,GAAG,eAAe,CAAC,MAAM,CAAC;AAC/D,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC;IACnC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;AACnC,IAAA,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;AACtB,CAAA;AAED;AACO,MAAM,eAAe,GAAG,eAAe,CAAC,MAAM,CAAC;AACpD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;AACvB,IAAA,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC7B,IAAA,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AACjC,IAAA,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AACpC,IAAA,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC/B,IAAA,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AACnC,IAAA,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AAC7B,IAAA,UAAU,EAAE;SACT,MAAM,CACL,CAAC,CAAC,MAAM,EAAE,EACV,CAAC,CAAC,MAAM,CAAC;AACP,QAAA,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;AACvB,QAAA,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;AACxB,QAAA,oBAAoB,EAAE,CAAC,CAAC,MAAM,EAAE;AAChC,QAAA,wBAAwB,EAAE,CAAC,CAAC,MAAM,EAAE;AACpC,QAAA,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE;AAC7B,QAAA,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;AACnB,QAAA,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE;AAC1B,KAAA,CAAC;AAEH,SAAA,QAAQ,EAAE;AACb,IAAA,iBAAiB,EAAE;AAChB,SAAA,KAAK,CACJ,CAAC,CAAC,MAAM,CAAC;AACP,QAAA,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;AACrB,QAAA,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;AACvB,QAAA,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;AAC9C,KAAA,CAAC;AAEH,SAAA,QAAQ,EAAE;AACd,CAAA;AAEM,MAAM,gBAAgB,GAAG,eAAe,CAAC,MAAM,CAAC;AACrD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;AACxB,IAAA,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;AACnB,IAAA,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AAC7B,IAAA,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAChC,IAAA,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;AACrD,IAAA,QAAQ,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;AACjC,CAAA;AAED;AACO,MAAM,iBAAiB,GAAG,eAAe,CAAC,MAAM,CAAC;AACtD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;AACzB,IAAA,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;AACf,IAAA,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;AACjB,IAAA,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;AAC5B,CAAA;AAEM,MAAM,mBAAmB,GAAG,eAAe,CAAC,MAAM,CAAC;AACxD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC;AAC3B,IAAA,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;AAChB,IAAA,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;AACrB,CAAA;AAEM,MAAM,iBAAiB,GAAG,eAAe,CAAC,MAAM,CAAC;AACtD,IAAA,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC;AAChC,IAAA,UAAU,EAAE,CAAC,CAAC,OAAO,EAAE;AACxB,CAAA;MAEY,gBAAgB,GAAG,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE;IAC3D,gBAAgB;IAChB,4BAA4B;IAC5B,2BAA2B;IAC3B,mBAAmB;IACnB,qBAAqB;IACrB,uBAAuB;IACvB,uBAAuB;IACvB,sBAAsB;IACtB,sBAAsB;IACtB,iBAAiB;IACjB,eAAe;IACf,kBAAkB;IAClB,0BAA0B;IAC1B,eAAe;IACf,gBAAgB;IAChB,iBAAiB;IACjB,mBAAmB;IACnB,iBAAiB;AAClB,CAAA;AAwBD;;;AAGG;AACG,SAAU,eAAe,CAAC,KAAc,EAAA;IAC5C,MAAM,MAAM,GAAG,gBAAgB,CAAC,SAAS,CAAC,KAAK,CAAC;AAChD,IAAA,OAAO,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,IAAI,GAAG,IAAI;AAC5C;AAEA;;;AAGG;AACG,SAAU,gBAAgB,CAAC,MAAiB,EAAA;AAChD,IAAA,OAAO;SACJ,GAAG,CAAC,CAAC,KAAK,KAAK,eAAe,CAAC,KAAK,CAAC;SACrC,MAAM,CAAC,CAAC,KAAK,KAA0B,KAAK,KAAK,IAAI,CAAC;AAC3D;;;;"}

package/dist/src/session-store.d.ts

@@ -0,0 +1,53 @@
+import type { PostHogAPIClient, TaskRunUpdate } from "./posthog-api.js";
+import type { TaskRun } from "./types.js";
+import { Logger } from "./utils/logger.js";
+export interface SessionPersistenceConfig {
+    taskId: string;
+    runId: string;
+    logUrl: string;
+    sdkSessionId?: string;
+}
+export declare class SessionStore {
+    private posthogAPI?;
+    private pendingEntries;
+    private flushTimeouts;
+    private configs;
+    private logger;
+    constructor(posthogAPI?: PostHogAPIClient, logger?: Logger);
+    /** Register a session for persistence */
+    register(sessionId: string, config: SessionPersistenceConfig): void;
+    /** Unregister and flush pending */
+    unregister(sessionId: string): Promise<void>;
+    /** Check if a session is registered for persistence */
+    isRegistered(sessionId: string): boolean;
+    /**
+     * Append a raw JSON-RPC line for persistence.
+     * Parses and wraps as StoredNotification for the API.
+     */
+    appendRawLine(sessionId: string, line: string): void;
+    /** Load raw JSON-RPC messages from S3 */
+    load(logUrl: string): Promise<unknown[]>;
+    /** Force flush pending entries */
+    flush(sessionId: string): Promise<void>;
+    private scheduleFlush;
+    /** Get the persistence config for a session */
+    getConfig(sessionId: string): SessionPersistenceConfig | undefined;
+    /**
+     * Start a session for persistence.
+     * Loads the task run and updates status to "in_progress".
+     */
+    start(sessionId: string, taskId: string, runId: string): Promise<TaskRun | undefined>;
+    /**
+     * Mark a session as completed.
+     */
+    complete(sessionId: string): Promise<void>;
+    /**
+     * Mark a session as failed.
+     */
+    fail(sessionId: string, error: Error | string): Promise<void>;
+    /**
+     * Update the task run associated with a session.
+     */
+    updateTaskRun(sessionId: string, update: TaskRunUpdate): Promise<TaskRun | undefined>;
+}
+//# sourceMappingURL=session-store.d.ts.map

package/dist/src/session-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-store.d.ts","sourceRoot":"","sources":["../../src/session-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACxE,OAAO,KAAK,EAAsB,OAAO,EAAE,MAAM,YAAY,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAE3C,MAAM,WAAW,wBAAwB;IACvC,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,UAAU,CAAC,CAAmB;IACtC,OAAO,CAAC,cAAc,CAAgD;IACtE,OAAO,CAAC,aAAa,CAA0C;IAC/D,OAAO,CAAC,OAAO,CAAoD;IACnE,OAAO,CAAC,MAAM,CAAS;gBAEX,UAAU,CAAC,EAAE,gBAAgB,EAAE,MAAM,CAAC,EAAE,MAAM;IA0B1D,yCAAyC;IACzC,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,wBAAwB,GAAG,IAAI;IAInE,mCAAmC;IAC7B,UAAU,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKlD,uDAAuD;IACvD,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAIxC;;;OAGG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;IA2BpD,yCAAyC;IACnC,IAAI,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAuB9C,kCAAkC;IAC5B,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IA6B7C,OAAO,CAAC,aAAa;IAOrB,+CAA+C;IAC/C,SAAS,CAAC,SAAS,EAAE,MAAM,GAAG,wBAAwB,GAAG,SAAS;IAIlE;;;OAGG;IACG,KAAK,CACT,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,OAAO,GAAG,SAAS,CAAC;IA0B/B;;OAEG;IACG,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKhD;;OAEG;IACG,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAUnE;;OAEG;IACG,aAAa,CACjB,SAAS,EAAE,MAAM,EACjB,MAAM,EAAE,aAAa,GACpB,OAAO,CAAC,OAAO,GAAG,SAAS,CAAC;CAyBhC"}

package/dist/src/session-store.js

@@ -0,0 +1,195 @@
+import { Logger } from './utils/logger.js';
+
+class SessionStore {
+    posthogAPI;
+    pendingEntries = new Map();
+    flushTimeouts = new Map();
+    configs = new Map();
+    logger;
+    constructor(posthogAPI, logger) {
+        this.posthogAPI = posthogAPI;
+        this.logger =
+            logger ?? new Logger({ debug: false, prefix: "[SessionStore]" });
+        // Flush all pending on process exit
+        const flushAllAndExit = async () => {
+            const flushPromises = [];
+            for (const sessionId of this.configs.keys()) {
+                flushPromises.push(this.flush(sessionId));
+            }
+            await Promise.all(flushPromises);
+            process.exit(0);
+        };
+        process.on("beforeExit", () => {
+            flushAllAndExit().catch((e) => this.logger.error("Flush failed:", e));
+        });
+        process.on("SIGINT", () => {
+            flushAllAndExit().catch((e) => this.logger.error("Flush failed:", e));
+        });
+        process.on("SIGTERM", () => {
+            flushAllAndExit().catch((e) => this.logger.error("Flush failed:", e));
+        });
+    }
+    /** Register a session for persistence */
+    register(sessionId, config) {
+        this.configs.set(sessionId, config);
+    }
+    /** Unregister and flush pending */
+    async unregister(sessionId) {
+        await this.flush(sessionId);
+        this.configs.delete(sessionId);
+    }
+    /** Check if a session is registered for persistence */
+    isRegistered(sessionId) {
+        return this.configs.has(sessionId);
+    }
+    /**
+     * Append a raw JSON-RPC line for persistence.
+     * Parses and wraps as StoredNotification for the API.
+     */
+    appendRawLine(sessionId, line) {
+        const config = this.configs.get(sessionId);
+        if (!config) {
+            return;
+        }
+        try {
+            const message = JSON.parse(line);
+            const entry = {
+                type: "notification",
+                timestamp: new Date().toISOString(),
+                notification: message,
+            };
+            const pending = this.pendingEntries.get(sessionId) ?? [];
+            pending.push(entry);
+            this.pendingEntries.set(sessionId, pending);
+            this.scheduleFlush(sessionId);
+        }
+        catch {
+            this.logger.warn("Failed to parse raw line for persistence", {
+                sessionId,
+                lineLength: line.length,
+            });
+        }
+    }
+    /** Load raw JSON-RPC messages from S3 */
+    async load(logUrl) {
+        const response = await fetch(logUrl);
+        if (!response.ok) {
+            return [];
+        }
+        const content = await response.text();
+        if (!content.trim())
+            return [];
+        return content
+            .trim()
+            .split("\n")
+            .map((line) => {
+            try {
+                return JSON.parse(line);
+            }
+            catch {
+                return null;
+            }
+        })
+            .filter((entry) => entry !== null);
+    }
+    /** Force flush pending entries */
+    async flush(sessionId) {
+        const config = this.configs.get(sessionId);
+        const pending = this.pendingEntries.get(sessionId);
+        if (!config || !pending?.length)
+            return;
+        this.pendingEntries.delete(sessionId);
+        const timeout = this.flushTimeouts.get(sessionId);
+        if (timeout) {
+            clearTimeout(timeout);
+            this.flushTimeouts.delete(sessionId);
+        }
+        if (!this.posthogAPI) {
+            this.logger.debug("No PostHog API configured, skipping flush");
+            return;
+        }
+        try {
+            await this.posthogAPI.appendTaskRunLog(config.taskId, config.runId, pending);
+        }
+        catch (error) {
+            this.logger.error("Failed to persist session logs:", error);
+        }
+    }
+    scheduleFlush(sessionId) {
+        const existing = this.flushTimeouts.get(sessionId);
+        if (existing)
+            clearTimeout(existing);
+        const timeout = setTimeout(() => this.flush(sessionId), 500);
+        this.flushTimeouts.set(sessionId, timeout);
+    }
+    /** Get the persistence config for a session */
+    getConfig(sessionId) {
+        return this.configs.get(sessionId);
+    }
+    /**
+     * Start a session for persistence.
+     * Loads the task run and updates status to "in_progress".
+     */
+    async start(sessionId, taskId, runId) {
+        if (!this.posthogAPI) {
+            this.logger.debug("No PostHog API configured, registering session without persistence");
+            this.register(sessionId, {
+                taskId,
+                runId,
+                logUrl: "",
+            });
+            return undefined;
+        }
+        const taskRun = await this.posthogAPI.getTaskRun(taskId, runId);
+        this.register(sessionId, {
+            taskId,
+            runId,
+            logUrl: taskRun.log_url,
+        });
+        await this.updateTaskRun(sessionId, { status: "in_progress" });
+        return taskRun;
+    }
+    /**
+     * Mark a session as completed.
+     */
+    async complete(sessionId) {
+        await this.flush(sessionId);
+        await this.updateTaskRun(sessionId, { status: "completed" });
+    }
+    /**
+     * Mark a session as failed.
+     */
+    async fail(sessionId, error) {
+        await this.flush(sessionId);
+        const message = typeof error === "string" ? error : error.message;
+        await this.updateTaskRun(sessionId, {
+            status: "failed",
+            error_message: message,
+        });
+        this.logger.error("Session failed", { sessionId, error: message });
+    }
+    /**
+     * Update the task run associated with a session.
+     */
+    async updateTaskRun(sessionId, update) {
+        const config = this.configs.get(sessionId);
+        if (!config) {
+            this.logger.error(`Cannot update task run: session ${sessionId} not registered`);
+            return undefined;
+        }
+        if (!this.posthogAPI) {
+            this.logger.debug("No PostHog API configured, skipping task run update");
+            return undefined;
+        }
+        try {
+            return await this.posthogAPI.updateTaskRun(config.taskId, config.runId, update);
+        }
+        catch (error) {
+            this.logger.error("Failed to update task run:", error);
+            return undefined;
+        }
+    }
+}
+
+export { SessionStore };
+//# sourceMappingURL=session-store.js.map

package/dist/src/session-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session-store.js","sources":["../../src/session-store.ts"],"sourcesContent":["import type { PostHogAPIClient, TaskRunUpdate } from \"./posthog-api.js\";\nimport type { StoredNotification, TaskRun } from \"./types.js\";\nimport { Logger } from \"./utils/logger.js\";\n\nexport interface SessionPersistenceConfig {\n  taskId: string;\n  runId: string;\n  logUrl: string;\n  sdkSessionId?: string;\n}\n\nexport class SessionStore {\n  private posthogAPI?: PostHogAPIClient;\n  private pendingEntries: Map<string, StoredNotification[]> = new Map();\n  private flushTimeouts: Map<string, NodeJS.Timeout> = new Map();\n  private configs: Map<string, SessionPersistenceConfig> = new Map();\n  private logger: Logger;\n\n  constructor(posthogAPI?: PostHogAPIClient, logger?: Logger) {\n    this.posthogAPI = posthogAPI;\n    this.logger =\n      logger ?? new Logger({ debug: false, prefix: \"[SessionStore]\" });\n\n    // Flush all pending on process exit\n    const flushAllAndExit = async () => {\n      const flushPromises: Promise<void>[] = [];\n      for (const sessionId of this.configs.keys()) {\n        flushPromises.push(this.flush(sessionId));\n      }\n      await Promise.all(flushPromises);\n      process.exit(0);\n    };\n\n    process.on(\"beforeExit\", () => {\n      flushAllAndExit().catch((e) => this.logger.error(\"Flush failed:\", e));\n    });\n    process.on(\"SIGINT\", () => {\n      flushAllAndExit().catch((e) => this.logger.error(\"Flush failed:\", e));\n    });\n    process.on(\"SIGTERM\", () => {\n      flushAllAndExit().catch((e) => this.logger.error(\"Flush failed:\", e));\n    });\n  }\n\n  /** Register a session for persistence */\n  register(sessionId: string, config: SessionPersistenceConfig): void {\n    this.configs.set(sessionId, config);\n  }\n\n  /** Unregister and flush pending */\n  async unregister(sessionId: string): Promise<void> {\n    await this.flush(sessionId);\n    this.configs.delete(sessionId);\n  }\n\n  /** Check if a session is registered for persistence */\n  isRegistered(sessionId: string): boolean {\n    return this.configs.has(sessionId);\n  }\n\n  /**\n   * Append a raw JSON-RPC line for persistence.\n   * Parses and wraps as StoredNotification for the API.\n   */\n  appendRawLine(sessionId: string, line: string): void {\n    const config = this.configs.get(sessionId);\n    if (!config) {\n      return;\n    }\n\n    try {\n      const message = JSON.parse(line);\n      const entry: StoredNotification = {\n        type: \"notification\",\n        timestamp: new Date().toISOString(),\n        notification: message,\n      };\n\n      const pending = this.pendingEntries.get(sessionId) ?? [];\n      pending.push(entry);\n      this.pendingEntries.set(sessionId, pending);\n\n      this.scheduleFlush(sessionId);\n    } catch {\n      this.logger.warn(\"Failed to parse raw line for persistence\", {\n        sessionId,\n        lineLength: line.length,\n      });\n    }\n  }\n\n  /** Load raw JSON-RPC messages from S3 */\n  async load(logUrl: string): Promise<unknown[]> {\n    const response = await fetch(logUrl);\n\n    if (!response.ok) {\n      return [];\n    }\n\n    const content = await response.text();\n    if (!content.trim()) return [];\n\n    return content\n      .trim()\n      .split(\"\\n\")\n      .map((line) => {\n        try {\n          return JSON.parse(line);\n        } catch {\n          return null;\n        }\n      })\n      .filter((entry): entry is unknown => entry !== null);\n  }\n\n  /** Force flush pending entries */\n  async flush(sessionId: string): Promise<void> {\n    const config = this.configs.get(sessionId);\n    const pending = this.pendingEntries.get(sessionId);\n\n    if (!config || !pending?.length) return;\n\n    this.pendingEntries.delete(sessionId);\n    const timeout = this.flushTimeouts.get(sessionId);\n    if (timeout) {\n      clearTimeout(timeout);\n      this.flushTimeouts.delete(sessionId);\n    }\n\n    if (!this.posthogAPI) {\n      this.logger.debug(\"No PostHog API configured, skipping flush\");\n      return;\n    }\n\n    try {\n      await this.posthogAPI.appendTaskRunLog(\n        config.taskId,\n        config.runId,\n        pending,\n      );\n    } catch (error) {\n      this.logger.error(\"Failed to persist session logs:\", error);\n    }\n  }\n\n  private scheduleFlush(sessionId: string): void {\n    const existing = this.flushTimeouts.get(sessionId);\n    if (existing) clearTimeout(existing);\n    const timeout = setTimeout(() => this.flush(sessionId), 500);\n    this.flushTimeouts.set(sessionId, timeout);\n  }\n\n  /** Get the persistence config for a session */\n  getConfig(sessionId: string): SessionPersistenceConfig | undefined {\n    return this.configs.get(sessionId);\n  }\n\n  /**\n   * Start a session for persistence.\n   * Loads the task run and updates status to \"in_progress\".\n   */\n  async start(\n    sessionId: string,\n    taskId: string,\n    runId: string,\n  ): Promise<TaskRun | undefined> {\n    if (!this.posthogAPI) {\n      this.logger.debug(\n        \"No PostHog API configured, registering session without persistence\",\n      );\n      this.register(sessionId, {\n        taskId,\n        runId,\n        logUrl: \"\",\n      });\n      return undefined;\n    }\n\n    const taskRun = await this.posthogAPI.getTaskRun(taskId, runId);\n\n    this.register(sessionId, {\n      taskId,\n      runId,\n      logUrl: taskRun.log_url,\n    });\n\n    await this.updateTaskRun(sessionId, { status: \"in_progress\" });\n\n    return taskRun;\n  }\n\n  /**\n   * Mark a session as completed.\n   */\n  async complete(sessionId: string): Promise<void> {\n    await this.flush(sessionId);\n    await this.updateTaskRun(sessionId, { status: \"completed\" });\n  }\n\n  /**\n   * Mark a session as failed.\n   */\n  async fail(sessionId: string, error: Error | string): Promise<void> {\n    await this.flush(sessionId);\n    const message = typeof error === \"string\" ? error : error.message;\n    await this.updateTaskRun(sessionId, {\n      status: \"failed\",\n      error_message: message,\n    });\n    this.logger.error(\"Session failed\", { sessionId, error: message });\n  }\n\n  /**\n   * Update the task run associated with a session.\n   */\n  async updateTaskRun(\n    sessionId: string,\n    update: TaskRunUpdate,\n  ): Promise<TaskRun | undefined> {\n    const config = this.configs.get(sessionId);\n    if (!config) {\n      this.logger.error(\n        `Cannot update task run: session ${sessionId} not registered`,\n      );\n      return undefined;\n    }\n\n    if (!this.posthogAPI) {\n      this.logger.debug(\"No PostHog API configured, skipping task run update\");\n      return undefined;\n    }\n\n    try {\n      return await this.posthogAPI.updateTaskRun(\n        config.taskId,\n        config.runId,\n        update,\n      );\n    } catch (error) {\n      this.logger.error(\"Failed to update task run:\", error);\n      return undefined;\n    }\n  }\n}\n"],"names":[],"mappings":";;MAWa,YAAY,CAAA;AACf,IAAA,UAAU;AACV,IAAA,cAAc,GAAsC,IAAI,GAAG,EAAE;AAC7D,IAAA,aAAa,GAAgC,IAAI,GAAG,EAAE;AACtD,IAAA,OAAO,GAA0C,IAAI,GAAG,EAAE;AAC1D,IAAA,MAAM;IAEd,WAAA,CAAY,UAA6B,EAAE,MAAe,EAAA;AACxD,QAAA,IAAI,CAAC,UAAU,GAAG,UAAU;AAC5B,QAAA,IAAI,CAAC,MAAM;AACT,YAAA,MAAM,IAAI,IAAI,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC;;AAGlE,QAAA,MAAM,eAAe,GAAG,YAAW;YACjC,MAAM,aAAa,GAAoB,EAAE;YACzC,KAAK,MAAM,SAAS,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE;gBAC3C,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC3C;AACA,YAAA,MAAM,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC;AAChC,YAAA,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;AACjB,QAAA,CAAC;AAED,QAAA,OAAO,CAAC,EAAE,CAAC,YAAY,EAAE,MAAK;YAC5B,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;AACvE,QAAA,CAAC,CAAC;AACF,QAAA,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAK;YACxB,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;AACvE,QAAA,CAAC,CAAC;AACF,QAAA,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,MAAK;YACzB,eAAe,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;AACvE,QAAA,CAAC,CAAC;IACJ;;IAGA,QAAQ,CAAC,SAAiB,EAAE,MAAgC,EAAA;QAC1D,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,MAAM,CAAC;IACrC;;IAGA,MAAM,UAAU,CAAC,SAAiB,EAAA;AAChC,QAAA,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;AAC3B,QAAA,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;IAChC;;AAGA,IAAA,YAAY,CAAC,SAAiB,EAAA;QAC5B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;IACpC;AAEA;;;AAGG;IACH,aAAa,CAAC,SAAiB,EAAE,IAAY,EAAA;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE;YACX;QACF;AAEA,QAAA,IAAI;YACF,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;AAChC,YAAA,MAAM,KAAK,GAAuB;AAChC,gBAAA,IAAI,EAAE,cAAc;AACpB,gBAAA,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;AACnC,gBAAA,YAAY,EAAE,OAAO;aACtB;AAED,YAAA,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE;AACxD,YAAA,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;YACnB,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC;AAE3C,YAAA,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;QAC/B;AAAE,QAAA,MAAM;AACN,YAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0CAA0C,EAAE;gBAC3D,SAAS;gBACT,UAAU,EAAE,IAAI,CAAC,MAAM;AACxB,aAAA,CAAC;QACJ;IACF;;IAGA,MAAM,IAAI,CAAC,MAAc,EAAA;AACvB,QAAA,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC;AAEpC,QAAA,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE;AAChB,YAAA,OAAO,EAAE;QACX;AAEA,QAAA,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE;AACrC,QAAA,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE;AAAE,YAAA,OAAO,EAAE;AAE9B,QAAA,OAAO;AACJ,aAAA,IAAI;aACJ,KAAK,CAAC,IAAI;AACV,aAAA,GAAG,CAAC,CAAC,IAAI,KAAI;AACZ,YAAA,IAAI;AACF,gBAAA,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACzB;AAAE,YAAA,MAAM;AACN,gBAAA,OAAO,IAAI;YACb;AACF,QAAA,CAAC;aACA,MAAM,CAAC,CAAC,KAAK,KAAuB,KAAK,KAAK,IAAI,CAAC;IACxD;;IAGA,MAAM,KAAK,CAAC,SAAiB,EAAA;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC;AAElD,QAAA,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO,EAAE,MAAM;YAAE;AAEjC,QAAA,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,CAAC;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC;QACjD,IAAI,OAAO,EAAE;YACX,YAAY,CAAC,OAAO,CAAC;AACrB,YAAA,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC;QACtC;AAEA,QAAA,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;AACpB,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,CAAC;YAC9D;QACF;AAEA,QAAA,IAAI;AACF,YAAA,MAAM,IAAI,CAAC,UAAU,CAAC,gBAAgB,CACpC,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,KAAK,EACZ,OAAO,CACR;QACH;QAAE,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC;QAC7D;IACF;AAEQ,IAAA,aAAa,CAAC,SAAiB,EAAA;QACrC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC;AAClD,QAAA,IAAI,QAAQ;YAAE,YAAY,CAAC,QAAQ,CAAC;AACpC,QAAA,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC;QAC5D,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC;IAC5C;;AAGA,IAAA,SAAS,CAAC,SAAiB,EAAA;QACzB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;IACpC;AAEA;;;AAGG;AACH,IAAA,MAAM,KAAK,CACT,SAAiB,EACjB,MAAc,EACd,KAAa,EAAA;AAEb,QAAA,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;AACpB,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,oEAAoE,CACrE;AACD,YAAA,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE;gBACvB,MAAM;gBACN,KAAK;AACL,gBAAA,MAAM,EAAE,EAAE;AACX,aAAA,CAAC;AACF,YAAA,OAAO,SAAS;QAClB;AAEA,QAAA,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK,CAAC;AAE/D,QAAA,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE;YACvB,MAAM;YACN,KAAK;YACL,MAAM,EAAE,OAAO,CAAC,OAAO;AACxB,SAAA,CAAC;AAEF,QAAA,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;AAE9D,QAAA,OAAO,OAAO;IAChB;AAEA;;AAEG;IACH,MAAM,QAAQ,CAAC,SAAiB,EAAA;AAC9B,QAAA,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;AAC3B,QAAA,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;IAC9D;AAEA;;AAEG;AACH,IAAA,MAAM,IAAI,CAAC,SAAiB,EAAE,KAAqB,EAAA;AACjD,QAAA,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC;AAC3B,QAAA,MAAM,OAAO,GAAG,OAAO,KAAK,KAAK,QAAQ,GAAG,KAAK,GAAG,KAAK,CAAC,OAAO;AACjE,QAAA,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE;AAClC,YAAA,MAAM,EAAE,QAAQ;AAChB,YAAA,aAAa,EAAE,OAAO;AACvB,SAAA,CAAC;AACF,QAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IACpE;AAEA;;AAEG;AACH,IAAA,MAAM,aAAa,CACjB,SAAiB,EACjB,MAAqB,EAAA;QAErB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE;YACX,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,CAAA,gCAAA,EAAmC,SAAS,CAAA,eAAA,CAAiB,CAC9D;AACD,YAAA,OAAO,SAAS;QAClB;AAEA,QAAA,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE;AACpB,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qDAAqD,CAAC;AACxE,YAAA,OAAO,SAAS;QAClB;AAEA,QAAA,IAAI;AACF,YAAA,OAAO,MAAM,IAAI,CAAC,UAAU,CAAC,aAAa,CACxC,MAAM,CAAC,MAAM,EACb,MAAM,CAAC,KAAK,EACZ,MAAM,CACP;QACH;QAAE,OAAO,KAAK,EAAE;YACd,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC;AACtD,YAAA,OAAO,SAAS;QAClB;IACF;AACD;;;;"}

package/dist/src/task-manager.d.ts

@@ -1,19 +1,19 @@
 export interface TaskExecutionState {
     taskId: string;
-    status: 'running' | 'completed' | 'failed' | 'canceled' | 'timeout';
-    mode: 'plan_only' | 'plan_and_build' | 'build_only';
-    result?: any;
+    status: "running" | "completed" | "failed" | "canceled" | "timeout";
+    mode: "plan_only" | "plan_and_build" | "build_only";
+    result?: unknown;
     startedAt: number;
     completedAt?: number;
     abortController?: AbortController;
 }
 export declare class TaskManager {
-    private executionStates;
+    executionStates: Map<string, TaskExecutionState>;
     private defaultTimeout;
     generateExecutionId(): string;
-    startExecution(taskId: string, mode: 'plan_only' | 'plan_and_build' | 'build_only', executionId?: string): TaskExecutionState;
-    waitForCompletion(executionId: string): Promise<any>;
-    completeExecution(executionId: string, result: any): void;
+    startExecution(taskId: string, mode: "plan_only" | "plan_and_build" | "build_only", executionId?: string): TaskExecutionState;
+    waitForCompletion(executionId: string): Promise<unknown>;
+    completeExecution(executionId: string, result: unknown): void;
     failExecution(executionId: string, error: Error): void;
     cancelExecution(executionId: string): void;
     getExecution(executionId: string): TaskExecutionState | undefined;

package/dist/src/task-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"task-manager.d.ts","sourceRoot":"","sources":["../../src/task-manager.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,UAAU,GAAG,SAAS,CAAC;IACpE,IAAI,EAAE,WAAW,GAAG,gBAAgB,GAAG,YAAY,CAAC;IACpD,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,eAAe,CAAC;CACnC;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,eAAe,CAAyC;IAChE,OAAO,CAAC,cAAc,CAAkB;IAExC,mBAAmB,IAAI,MAAM;IAI7B,cAAc,CACZ,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,WAAW,GAAG,gBAAgB,GAAG,YAAY,EACnD,WAAW,GAAE,MAAmC,GAC/C,kBAAkB;IAef,iBAAiB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IAkC1D,iBAAiB,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,GAAG,IAAI;IAWzD,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,GAAG,IAAI;IActD,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI;IAkB1C,YAAY,CAAC,WAAW,EAAE,MAAM,GAAG,kBAAkB,GAAG,SAAS;IAIjE,cAAc,CAAC,WAAW,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAI5D,kBAAkB,CAAC,WAAW,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAIpE,OAAO,CAAC,eAAe;IAkBvB,OAAO,CAAC,SAAS,GAAE,MAAuB,GAAG,IAAI;CAQlD"}
+{"version":3,"file":"task-manager.d.ts","sourceRoot":"","sources":["../../src/task-manager.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,UAAU,GAAG,SAAS,CAAC;IACpE,IAAI,EAAE,WAAW,GAAG,gBAAgB,GAAG,YAAY,CAAC;IACpD,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,eAAe,CAAC;CACnC;AAED,qBAAa,WAAW;IACf,eAAe,kCAAyC;IAC/D,OAAO,CAAC,cAAc,CAAkB;IAExC,mBAAmB,IAAI,MAAM;IAI7B,cAAc,CACZ,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,WAAW,GAAG,gBAAgB,GAAG,YAAY,EACnD,WAAW,GAAE,MAAmC,GAC/C,kBAAkB;IAef,iBAAiB,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAuC9D,iBAAiB,CAAC,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,GAAG,IAAI;IAW7D,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,GAAG,IAAI;IActD,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI;IAkB1C,YAAY,CAAC,WAAW,EAAE,MAAM,GAAG,kBAAkB,GAAG,SAAS;IAIjE,cAAc,CAAC,WAAW,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAI5D,kBAAkB,CAAC,WAAW,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAIpE,OAAO,CAAC,eAAe;IAqBvB,OAAO,CAAC,SAAS,GAAE,MAAuB,GAAG,IAAI;CAQlD"}

package/dist/src/task-manager.js

@@ -1,15 +1,15 @@
-import { randomBytes } from 'crypto';
+import { randomBytes } from 'node:crypto';
 
 class TaskManager {
     executionStates = new Map();
     defaultTimeout = 10 * 60 * 1000; // 10 minutes
     generateExecutionId() {
-        return randomBytes(16).toString('hex');
+        return randomBytes(16).toString("hex");
     }
     startExecution(taskId, mode, executionId = this.generateExecutionId()) {
         const executionState = {
             taskId,
-            status: 'running',
+            status: "running",
             mode,
             startedAt: Date.now(),
             abortController: new AbortController(),
@@ -23,7 +23,7 @@ class TaskManager {
         if (!execution) {
             throw new Error(`Execution ${executionId} not found`);
         }
-        if (execution.result && execution.status === 'completed') {
+        if (execution.result && execution.status === "completed") {
             return execution.result;
         }
         return new Promise((resolve, reject) => {
@@ -34,13 +34,14 @@ class TaskManager {
                     reject(new Error(`Execution ${executionId} disappeared`));
                     return;
                 }
-                if (currentExecution.status === 'completed' && currentExecution.result) {
+                if (currentExecution.status === "completed" &&
+                    currentExecution.result) {
                     clearInterval(checkInterval);
                     resolve(currentExecution.result);
                 }
-                else if (currentExecution.status === 'failed' ||
-                    currentExecution.status === 'canceled' ||
-                    currentExecution.status === 'timeout') {
+                else if (currentExecution.status === "failed" ||
+                    currentExecution.status === "canceled" ||
+                    currentExecution.status === "timeout") {
                     clearInterval(checkInterval);
                     reject(new Error(`Execution ${executionId} ${currentExecution.status}`));
                 }
@@ -52,7 +53,7 @@ class TaskManager {
         if (!execution) {
             throw new Error(`Execution ${executionId} not found`);
         }
-        execution.status = 'completed';
+        execution.status = "completed";
         execution.result = result;
         execution.completedAt = Date.now();
     }
@@ -61,11 +62,11 @@ class TaskManager {
         if (!execution) {
             throw new Error(`Execution ${executionId} not found`);
         }
-        execution.status = 'failed';
+        execution.status = "failed";
         execution.completedAt = Date.now();
         execution.result = {
             error: error.message,
-            status: 'failed',
+            status: "failed",
         };
     }
     cancelExecution(executionId) {
@@ -73,13 +74,13 @@ class TaskManager {
         if (!execution) {
             throw new Error(`Execution ${executionId} not found`);
         }
-        execution.status = 'canceled';
+        execution.status = "canceled";
         execution.completedAt = Date.now();
         execution.abortController?.abort();
         if (!execution.result) {
             execution.result = {
-                status: 'canceled',
-                message: 'Execution was canceled',
+                status: "canceled",
+                message: "Execution was canceled",
             };
         }
     }
@@ -95,14 +96,14 @@ class TaskManager {
     scheduleTimeout(executionId, timeout = this.defaultTimeout) {
         setTimeout(() => {
             const execution = this.executionStates.get(executionId);
-            if (execution && execution.status === 'running') {
-                execution.status = 'timeout';
+            if (execution && execution.status === "running") {
+                execution.status = "timeout";
                 execution.completedAt = Date.now();
                 execution.abortController?.abort();
                 if (!execution.result) {
                     execution.result = {
-                        status: 'timeout',
-                        message: 'Execution timed out',
+                        status: "timeout",
+                        message: "Execution timed out",
                     };
                 }
             }