@moxt-ai/mobius 0.0.181 → 0.0.182

package/dist/index.js

@@ -19791,6 +19791,8 @@ var SESSION_LIST_KIND = "session.list";
 var SESSION_LISTED_KIND = "session.listed";
 var SESSION_QUERY_KIND = "session.query";
 var SESSION_QUERIED_KIND = "session.queried";
+var SESSION_RESTORE_KIND = "session.restore";
+var SESSION_RESTORED_KIND = "session.restored";
 var SESSION_STATUS_CHANGED_KIND = "session.status.changed";
 var SESSION_CONFIGURE_KIND = "session.configure";
 var SESSION_CONFIG_UPDATED_KIND = "session.config.updated";
@@ -19871,6 +19873,11 @@ var sessionsQueriedEventSchema = external_exports.object({
   kind: external_exports.literal(SESSION_QUERIED_KIND),
   requestId: identifierStringSchema
 }).readonly();
+var sessionRestoreCommandSchema = external_exports.object({
+  kind: external_exports.literal(SESSION_RESTORE_KIND),
+  requestId: identifierStringSchema,
+  sessionId: identifierStringSchema
+}).readonly();
 var fileReadCommandSchema = external_exports.object({
   kind: external_exports.literal(FILE_READ_KIND),
   maxBytes: external_exports.number().int().positive().max(128 * 1024 * 1024),
@@ -20185,6 +20192,12 @@ var sessionReadyEventSchema = external_exports.object({
   runtimeTimings: runtimeTimingsSchema.optional(),
   sessionId: identifierStringSchema
 }).readonly();
+var sessionRestoredEventSchema = external_exports.object({
+  entry: sessionMetadataSchema,
+  kind: external_exports.literal(SESSION_RESTORED_KIND),
+  ready: sessionReadyEventSchema,
+  requestId: identifierStringSchema
+}).readonly();
 var sessionConfigureCommandSchema = external_exports.object({
   ...sessionIdentityFields,
   booleanValue: external_exports.boolean(),
@@ -20400,6 +20413,10 @@ var decodeSessionQueryCommand = (encoded) => decode3(encoded, sessionQueryComman
 var encodeSessionQueryCommand = (message) => encode3(message, sessionQueryCommandSchema);
 var decodeSessionsQueriedEvent = (encoded) => decode3(encoded, sessionsQueriedEventSchema);
 var encodeSessionsQueriedEvent = (message) => encode3(message, sessionsQueriedEventSchema);
+var decodeSessionRestoreCommand = (encoded) => decode3(encoded, sessionRestoreCommandSchema);
+var encodeSessionRestoreCommand = (message) => encode3(message, sessionRestoreCommandSchema);
+var decodeSessionRestoredEvent = (encoded) => decode3(encoded, sessionRestoredEventSchema);
+var encodeSessionRestoredEvent = (message) => encode3(message, sessionRestoredEventSchema);
 var decodeSessionStatusChangedEvent = (encoded) => decode3(encoded, sessionStatusChangedEventSchema);
 var encodeSessionStatusChangedEvent = (message) => encode3(message, sessionStatusChangedEventSchema);
 var decodeSessionPrepareCommand = (encoded) => decode3(encoded, sessionPrepareCommandSchema);
@@ -21482,6 +21499,11 @@ var serviceToDaemonSignatures = {
     encodeSessionQueryCommand,
     decodeSessionQueryCommand
   ),
+  [SESSION_RESTORE_KIND]: defineMessageSignature(
+    SESSION_RESTORE_KIND,
+    encodeSessionRestoreCommand,
+    decodeSessionRestoreCommand
+  ),
   [NATIVE_SESSION_CANCEL_KIND]: defineMessageSignature(
     NATIVE_SESSION_CANCEL_KIND,
     encodeNativeSessionCancel,
@@ -21613,6 +21635,11 @@ var daemonToServiceSignatures = {
     encodeSessionsQueriedEvent,
     decodeSessionsQueriedEvent
   ),
+  [SESSION_RESTORED_KIND]: defineMessageSignature(
+    SESSION_RESTORED_KIND,
+    encodeSessionRestoredEvent,
+    decodeSessionRestoredEvent
+  ),
   [SESSION_STATUS_CHANGED_KIND]: defineMessageSignature(
     SESSION_STATUS_CHANGED_KIND,
     encodeSessionStatusChangedEvent,
@@ -31492,6 +31519,17 @@ var embeddedImageSchema = external_exports.looseObject({
   data: external_exports.string(),
   mimeType: external_exports.string()
 });
+var toolImageSchema = external_exports.union([
+  embeddedImageSchema,
+  external_exports.object({
+    type: external_exports.literal("image"),
+    source: external_exports.object({ type: external_exports.literal("base64"), data: external_exports.string(), media_type: external_exports.string() })
+  }).transform(({ source }) => ({ type: "image", data: source.data, mimeType: source.media_type })),
+  external_exports.object({
+    type: external_exports.literal("image"),
+    file: external_exports.object({ base64: external_exports.string(), type: external_exports.string() })
+  }).transform(({ file: file2 }) => ({ type: "image", data: file2.base64, mimeType: file2.type }))
+]);
 function isRecord6(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
 }
@@ -31611,7 +31649,7 @@ var AcpContentExternalizer = class {
     if (!isRecord6(value)) {
       return value;
     }
-    const image = embeddedImageSchema.safeParse(value);
+    const image = toolImageSchema.safeParse(value);
     if (image.success && image.data.data.length > 0) {
       const prepared = await this.#content(sessionId, image.data, signal);
       if (isRecord6(prepared.event) && typeof prepared.event["imageKey"] === "string") {
@@ -31630,6 +31668,7 @@ var AcpContentExternalizer = class {
   async #toolUpdate(sessionId, update, signal) {
     const images = /* @__PURE__ */ new Map();
     const rawOutput = update.rawOutput === void 0 ? void 0 : await this.#rawOutput(sessionId, update.rawOutput, images, signal);
+    const metadata = update._meta === void 0 || update._meta === null ? update._meta : external_exports.record(external_exports.string(), external_exports.unknown()).parse(await this.#rawOutput(sessionId, update._meta, images, signal));
     const originalJson = images.size === 0 ? void 0 : JSON.stringify(update.rawOutput);
     const rewrittenJson = images.size === 0 ? void 0 : JSON.stringify(rawOutput);
     const agentContent = [];
@@ -31653,10 +31692,11 @@ var AcpContentExternalizer = class {
       eventContent.push({ type: "content", content: prepared.event });
     }
     const output2 = update.rawOutput === void 0 ? {} : { rawOutput };
+    const meta3 = metadata === void 0 ? {} : { _meta: metadata };
     const hasContent = update.content !== void 0 && update.content !== null || agentContent.length > 0;
     return {
-      agent: { ...update, ...output2, ...hasContent ? { content: agentContent } : {} },
-      event: { ...update, ...output2, ...hasContent ? { content: eventContent } : {} }
+      agent: { ...update, ...meta3, ...output2, ...hasContent ? { content: agentContent } : {} },
+      event: { ...update, ...meta3, ...output2, ...hasContent ? { content: eventContent } : {} }
     };
   }
   update = async (sessionId, update, signal) => {
@@ -42135,6 +42175,72 @@ async function executePrompt({
 }
 
 // src/daemon-session-commands.ts
+async function prepareSession(command, config2, directories, audit, sessions, metrics, signal) {
+  return await metrics.measure("prepare-total", async () => {
+    const resolved = await metrics.measure(
+      "workspace-resolve",
+      async () => await resolveSessionDirectories(
+        directories,
+        config2.bindingId,
+        command.directoryId,
+        command.additionalDirectoryIds,
+        signal
+      )
+    );
+    const agent = config2.catalog.find(command.agentId);
+    const preparation = {
+      additionalDirectories: resolved.additionalDirectories,
+      agent,
+      cwd: resolved.cwd,
+      sessionId: command.sessionId,
+      signal
+    };
+    const prepared = command.transient === true ? await sessions.inspectSession(preparation, metrics) : await sessions.prepareSession(preparation, metrics);
+    const configuration = command.configuration;
+    const configOptions = configuration === void 0 ? prepared.configOptions : await metrics.measure(
+      "configuration",
+      async () => await sessions.configureSession({
+        ...preparation,
+        booleanValue: configuration.booleanValue,
+        configId: configuration.configId,
+        requestId: command.requestId,
+        value: configuration.value,
+        valueType: configuration.valueType
+      })
+    );
+    const listing = await metrics.measure(
+      "workspace-list",
+      async () => await directories.list(config2.bindingId, command.directoryId, signal)
+    );
+    audit.record({
+      commandId: command.requestId,
+      event: "session_prepared",
+      sessionId: command.sessionId
+    });
+    if (command.configuration !== void 0) {
+      audit.record({
+        commandId: command.requestId,
+        event: "session_configured",
+        sessionId: command.sessionId
+      });
+    }
+    return {
+      additionalDirectories: resolved.additionalDirectoryEntries,
+      agentId: command.agentId,
+      agentName: prepared.agentName,
+      agentVersion: prepared.agentVersion,
+      capabilities: prepared.capabilities,
+      commands: prepared.commands,
+      configOptions,
+      directoryId: command.directoryId,
+      directoryLabel: listing.directoryLabel,
+      kind: "session.ready",
+      requestId: command.requestId,
+      restored: prepared.restored,
+      sessionId: command.sessionId
+    };
+  });
+}
 async function executeWorkspaceList(socket, command, directories, audit, signal) {
   try {
     const listing = await directories.list(command.bindingId, command.directoryId, signal);
@@ -42159,72 +42265,7 @@ async function executeWorkspaceList(socket, command, directories, audit, signal)
 async function executeSessionPrepare(socket, command, config2, directories, audit, sessions, diagnostics, signal) {
   const metrics = new AccumulatingOperationMetrics();
   try {
-    const readyEvent = await metrics.measure("prepare-total", async () => {
-      const resolved = await metrics.measure(
-        "workspace-resolve",
-        async () => await resolveSessionDirectories(
-          directories,
-          config2.bindingId,
-          command.directoryId,
-          command.additionalDirectoryIds,
-          signal
-        )
-      );
-      const agent = config2.catalog.find(command.agentId);
-      const preparation = {
-        additionalDirectories: resolved.additionalDirectories,
-        agent,
-        cwd: resolved.cwd,
-        sessionId: command.sessionId,
-        signal
-      };
-      const prepared = command.transient === true ? await sessions.inspectSession(preparation, metrics) : await sessions.prepareSession(preparation, metrics);
-      const configuration = command.configuration;
-      const configOptions = configuration === void 0 ? prepared.configOptions : await metrics.measure(
-        "configuration",
-        async () => await sessions.configureSession({
-          ...preparation,
-          booleanValue: configuration.booleanValue,
-          configId: configuration.configId,
-          requestId: command.requestId,
-          value: configuration.value,
-          valueType: configuration.valueType
-        })
-      );
-      const listing = await metrics.measure(
-        "workspace-list",
-        async () => await directories.list(config2.bindingId, command.directoryId, signal)
-      );
-      audit.record({
-        commandId: command.requestId,
-        event: "session_prepared",
-        sessionId: command.sessionId
-      });
-      if (command.configuration !== void 0) {
-        audit.record({
-          commandId: command.requestId,
-          event: "session_configured",
-          sessionId: command.sessionId
-        });
-      }
-      const event = {
-        additionalDirectories: resolved.additionalDirectoryEntries,
-        agentId: command.agentId,
-        agentName: prepared.agentName,
-        agentVersion: prepared.agentVersion,
-        capabilities: prepared.capabilities,
-        commands: prepared.commands,
-        configOptions,
-        directoryId: command.directoryId,
-        directoryLabel: listing.directoryLabel,
-        kind: "session.ready",
-        requestId: command.requestId,
-        restored: prepared.restored,
-        sessionId: command.sessionId,
-        runtimeTimings: metrics.snapshot()
-      };
-      return event;
-    });
+    const readyEvent = await prepareSession(command, config2, directories, audit, sessions, metrics, signal);
     send(socket, {
       ...readyEvent,
       runtimeTimings: metrics.snapshot()
@@ -42252,6 +42293,59 @@ async function executeSessionPrepare(socket, command, config2, directories, audi
     reportOperationFailure(socket, command.requestId, command.sessionId, error62, metrics.snapshot());
   }
 }
+async function executeSessionRestore(socket, command, config2, directories, audit, sessions, nativeSessions, diagnostics, signal) {
+  const metrics = new AccumulatingOperationMetrics();
+  try {
+    const entry = await metrics.measure("session-lookup", async () => {
+      const [found] = await nativeSessions.querySessions([command.sessionId], signal);
+      if (found === void 0) {
+        throw new AcpExecutionError("The Session was not found", {}, "session_not_found");
+      }
+      return found;
+    });
+    const ready = await prepareSession(
+      {
+        additionalDirectoryIds: entry.additionalDirectories.map((directory) => directory.directoryId),
+        agentId: entry.agentId,
+        bindingId: config2.bindingId,
+        directoryId: entry.directoryId,
+        kind: "session.prepare",
+        requestId: command.requestId,
+        sessionId: command.sessionId,
+        sessionState: "provisioned"
+      },
+      config2,
+      directories,
+      audit,
+      sessions,
+      metrics,
+      signal
+    );
+    send(socket, {
+      entry,
+      kind: "session.restored",
+      ready: { ...ready, runtimeTimings: metrics.snapshot() },
+      requestId: command.requestId
+    });
+  } catch (error62) {
+    diagnostics.daemonDiagnostic(
+      "session_restore_failed",
+      command.requestId,
+      {
+        code: operationErrorCode(error62),
+        sessionId: command.sessionId
+      },
+      error62
+    );
+    audit.record({
+      commandId: command.requestId,
+      details: { code: operationErrorCode(error62) },
+      event: "session_restore_failed",
+      sessionId: command.sessionId
+    });
+    reportOperationFailure(socket, command.requestId, command.sessionId, error62, metrics.snapshot());
+  }
+}
 async function executeSessionConfigure(socket, command, config2, directories, audit, sessions, signal) {
   try {
     const resolved = await resolveSessionDirectories(
@@ -42565,6 +42659,21 @@ function createDaemonCommandHandler({
         });
         return;
       }
+      if (kind === SESSION_RESTORE_KIND) {
+        const command2 = message;
+        void executeSessionRestore(
+          socket,
+          command2,
+          config2,
+          directories,
+          audit,
+          sessions,
+          nativeSessions,
+          diagnostics,
+          signal
+        );
+        return;
+      }
       if (kind === SESSION_PERMISSION_RESPOND_KIND) {
         const command2 = message;
         void sessions.respondPermission(command2).then(() => {