refactor: Make makeJsonAgent bundleable with @endo/bundle-source

packages/kernel-agents/src/agent.ts

@@ -1,4 +1,4 @@
-import { mergeDisjointRecords } from '@metamask/kernel-utils';
+import { mergeDisjointRecords } from '@metamask/kernel-utils/merge-disjoint-records';
 import type { Logger } from '@metamask/logger';
 import type { LanguageModel } from '@ocap/kernel-language-model-service';
 

packages/kernel-agents/src/attempt.ts

@@ -1,4 +1,4 @@
-import { SampleGenerationError } from '@metamask/kernel-errors';
+import { SampleGenerationError } from '@metamask/kernel-errors/bundleable';
 import type { Logger } from '@metamask/logger';
 import type { LanguageModel } from '@ocap/kernel-language-model-service';
 

packages/kernel-agents/src/capabilities/discover.ts

@@ -1,5 +1,6 @@
 import { E } from '@endo/eventual-send';
-import type { DiscoverableExo, MethodSchema } from '@metamask/kernel-utils';
+import type { DiscoverableExo } from '@metamask/kernel-utils/discoverable';
+import type { MethodSchema } from '@metamask/kernel-utils/schema';
 
 import type { CapabilityRecord, CapabilitySpec } from '../types.ts';
 

packages/kernel-agents/src/strategies/json/prepare-attempt.ts

@@ -1,4 +1,4 @@
-import { mergeDisjointRecords } from '@metamask/kernel-utils';
+import { mergeDisjointRecords } from '@metamask/kernel-utils/merge-disjoint-records';
 import type { Logger } from '@metamask/logger';
 
 import { makeEvaluator } from './evaluator.ts';

packages/kernel-agents/src/strategies/json/sample-collector.ts

@@ -1,4 +1,4 @@
-import { SampleGenerationError } from '@metamask/kernel-errors';
+import { SampleGenerationError } from '@metamask/kernel-errors/bundleable';
 import type { Logger } from '@metamask/logger';
 
 import type { SampleCollector } from '../../types.ts';

packages/kernel-agents/src/task.ts

@@ -1,4 +1,4 @@
-import { makeCounter } from '@metamask/kernel-utils';
+import { makeCounter } from '@metamask/kernel-utils/counter';
 
 import type { Task, CapabilityRecord } from './types.ts';
 

packages/kernel-agents/src/types/capability.ts

@@ -1,4 +1,4 @@
-import type { JsonSchema } from '@metamask/kernel-utils';
+import type { JsonSchema } from '@metamask/kernel-utils/schema';
 
 export type Capability<Args extends Record<string, unknown>, Return = null> = (
   args: Args,

packages/kernel-errors/package.json

@@ -29,6 +29,16 @@
         "default": "./dist/index.cjs"
       }
     },
+    "./bundleable": {
+      "import": {
+        "types": "./dist/bundleable/index.d.mts",
+        "default": "./dist/bundleable/index.mjs"
+      },
+      "require": {
+        "types": "./dist/bundleable/index.d.cts",
+        "default": "./dist/bundleable/index.cjs"
+      }
+    },
     "./package.json": "./package.json"
   },
   "main": "./dist/index.cjs",
@@ -56,7 +66,6 @@
     "test:watch": "vitest --config vitest.config.ts"
   },
   "dependencies": {
-    "@libp2p/interface": "2.11.0",
     "@metamask/superstruct": "^3.2.1",
     "@metamask/utils": "^11.9.0"
   },

packages/kernel-errors/src/BaseError.ts

@@ -1,6 +1,6 @@
 import type { Json } from '@metamask/utils';
 
-import { ErrorCode } from './constants.ts';
+import { ErrorCode } from './error-codes.ts';
 import type {
   MarshaledOcapError,
   OcapError,

packages/kernel-errors/src/bundleable/SampleGenerationError.ts

@@ -0,0 +1,46 @@
+import { ErrorCode } from '../error-codes.ts';
+
+/**
+ * Error options with optional stack trace.
+ */
+type ErrorOptionsWithStack = {
+  cause?: Error;
+  stack?: string;
+};
+
+/**
+ * Bundleable version of SampleGenerationError.
+ * This version does not include marshaling logic to avoid dependencies
+ * on @metamask/utils that prevent vat bundling.
+ *
+ * An error indicating that the LLM generated invalid response.
+ * This error should trigger resampling from the LLM.
+ */
+export class SampleGenerationError extends Error {
+  public readonly code: typeof ErrorCode.SampleGenerationError;
+
+  public readonly data: { sample: string };
+
+  /**
+   * Creates a new SampleGenerationError.
+   *
+   * @param sample - The invalid sample text generated by the LLM.
+   * @param cause - The underlying error that caused sample generation to fail.
+   * @param options - Additional error options including stack.
+   * @param options.stack - The stack trace of the error.
+   */
+  constructor(sample: string, cause: Error, options?: ErrorOptionsWithStack) {
+    super('LLM generated invalid response.', { cause });
+
+    this.name = 'SampleGenerationError';
+    this.code = ErrorCode.SampleGenerationError;
+    this.data = { sample };
+
+    if (options?.stack) {
+      this.stack = options.stack;
+    }
+
+    harden(this);
+  }
+}
+harden(SampleGenerationError);

packages/kernel-errors/src/bundleable/index.ts

@@ -0,0 +1,11 @@
+/**
+ * Bundleable versions of kernel-errors.
+ *
+ * These versions do not include marshaling logic to avoid dependencies
+ * on @metamask/utils that prevent vat bundling with @endo/bundle-source.
+ *
+ * Use these exports when bundling vat code:
+ * import { SampleGenerationError } from '@metamask/kernel-errors/bundleable';
+ */
+export { SampleGenerationError } from './SampleGenerationError.ts';
+export { ErrorCode, ErrorSentinel } from '../error-codes.ts';

packages/kernel-errors/src/constants.ts

@@ -10,8 +10,12 @@ import {
 import { JsonStruct, object } from '@metamask/utils';
 import type { NonEmptyArray } from '@metamask/utils';
 
+import { ErrorCode, ErrorSentinel } from './error-codes.ts';
 import type { MarshaledError, MarshaledOcapError } from './types.ts';
 
+// Re-export for backward compatibility
+export { ErrorCode, ErrorSentinel } from './error-codes.ts';
+
 /**
  * Struct to validate plain {@link Error} objects.
  */
@@ -20,29 +24,6 @@ export const ErrorStruct = define<Error>(
   (value) => value instanceof Error,
 );
 
-/**
- * Enum defining all error codes for Ocap errors.
- */
-export const ErrorCode = {
-  AbortError: 'ABORT_ERROR',
-  DuplicateEndowment: 'DUPLICATE_ENDOWMENT',
-  StreamReadError: 'STREAM_READ_ERROR',
-  VatAlreadyExists: 'VAT_ALREADY_EXISTS',
-  VatDeleted: 'VAT_DELETED',
-  VatNotFound: 'VAT_NOT_FOUND',
-  SubclusterNotFound: 'SUBCLUSTER_NOT_FOUND',
-  SampleGenerationError: 'SAMPLE_GENERATION_ERROR',
-  InternalError: 'INTERNAL_ERROR',
-  ResourceLimitError: 'RESOURCE_LIMIT_ERROR',
-} as const;
-
-export type ErrorCode = (typeof ErrorCode)[keyof typeof ErrorCode];
-
-/**
- * A sentinel value used to identify marshaled errors.
- */
-export const ErrorSentinel = '@@MARSHALED_ERROR';
-
 const ErrorCodeStruct = union(
   Object.values(ErrorCode).map((code) => literal(code)) as NonEmptyArray<
     Struct<ErrorCode>

packages/kernel-errors/src/error-codes.ts

@@ -0,0 +1,22 @@
+/**
+ * Enum defining all error codes for Ocap errors.
+ */
+export const ErrorCode = {
+  AbortError: 'ABORT_ERROR',
+  DuplicateEndowment: 'DUPLICATE_ENDOWMENT',
+  StreamReadError: 'STREAM_READ_ERROR',
+  VatAlreadyExists: 'VAT_ALREADY_EXISTS',
+  VatDeleted: 'VAT_DELETED',
+  VatNotFound: 'VAT_NOT_FOUND',
+  SubclusterNotFound: 'SUBCLUSTER_NOT_FOUND',
+  SampleGenerationError: 'SAMPLE_GENERATION_ERROR',
+  InternalError: 'INTERNAL_ERROR',
+  ResourceLimitError: 'RESOURCE_LIMIT_ERROR',
+} as const;
+
+export type ErrorCode = (typeof ErrorCode)[keyof typeof ErrorCode];
+
+/**
+ * A sentinel value used to identify marshaled errors.
+ */
+export const ErrorSentinel = '@@MARSHALED_ERROR';

packages/kernel-errors/src/index.test.ts

@@ -23,7 +23,6 @@ describe('index', () => {
       'isMarshaledError',
       'isMarshaledOcapError',
       'isOcapError',
-      'isRetryableNetworkError',
       'marshalError',
       'toError',
       'unmarshalError',

packages/kernel-errors/src/index.ts

@@ -22,4 +22,3 @@ export { marshalError } from './marshal/marshalError.ts';
 export { unmarshalError } from './marshal/unmarshalError.ts';
 export { isMarshaledError } from './marshal/isMarshaledError.ts';
 export { isMarshaledOcapError } from './marshal/isMarshaledOcapError.ts';
-export { isRetryableNetworkError } from './utils/isRetryableNetworkError.ts';

packages/kernel-utils/package.json

@@ -49,6 +49,36 @@
         "default": "./dist/discoverable.cjs"
       }
     },
+    "./counter": {
+      "import": {
+        "types": "./dist/counter.d.mts",
+        "default": "./dist/counter.mjs"
+      },
+      "require": {
+        "types": "./dist/counter.d.cts",
+        "default": "./dist/counter.cjs"
+      }
+    },
+    "./merge-disjoint-records": {
+      "import": {
+        "types": "./dist/merge-disjoint-records.d.mts",
+        "default": "./dist/merge-disjoint-records.mjs"
+      },
+      "require": {
+        "types": "./dist/merge-disjoint-records.d.cts",
+        "default": "./dist/merge-disjoint-records.cjs"
+      }
+    },
+    "./schema": {
+      "import": {
+        "types": "./dist/schema.d.mts",
+        "default": "./dist/schema.mjs"
+      },
+      "require": {
+        "types": "./dist/schema.d.cts",
+        "default": "./dist/schema.cjs"
+      }
+    },
     "./package.json": "./package.json"
   },
   "main": "./dist/index.cjs",

packages/kernel-utils/src/counter.ts

@@ -0,0 +1,13 @@
+/**
+ * A simple counter which increments and returns when called.
+ *
+ * @param start - One less than the first returned number.
+ * @returns A counter.
+ */
+export const makeCounter = (start: number = 0) => {
+  let counter: number = start;
+  return () => {
+    counter += 1;
+    return counter;
+  };
+};

packages/kernel-utils/src/misc.ts

@@ -1,18 +1,7 @@
 import { AbortError } from '@metamask/kernel-errors';
 
-/**
- * A simple counter which increments and returns when called.
- *
- * @param start - One less than the first returned number.
- * @returns A counter.
- */
-export const makeCounter = (start: number = 0) => {
-  let counter: number = start;
-  return () => {
-    counter += 1;
-    return counter;
-  };
-};
+// Re-export makeCounter for backward compatibility
+export { makeCounter } from './counter.ts';
 
 /**
  * Delay execution by the specified number of milliseconds.

packages/ocap-kernel/src/remotes/ConnectionFactory.ts

@@ -10,14 +10,15 @@ import { ping } from '@libp2p/ping';
 import { webRTC } from '@libp2p/webrtc';
 import { webSockets } from '@libp2p/websockets';
 import { webTransport } from '@libp2p/webtransport';
-import { AbortError, isRetryableNetworkError } from '@metamask/kernel-errors';
+import { AbortError } from '@metamask/kernel-errors';
 import { fromHex, retryWithBackoff } from '@metamask/kernel-utils';
 import { Logger } from '@metamask/logger';
 import { multiaddr } from '@multiformats/multiaddr';
 import { byteStream } from 'it-byte-stream';
 import { createLibp2p } from 'libp2p';
 
 import type { Channel, InboundConnectionHandler } from './types.ts';
+import { isRetryableNetworkError } from '../utils/isRetryableNetworkError.ts';
 
 /**
  * Connection factory for libp2p network operations.

packages/ocap-kernel/src/remotes/network.ts

@@ -1,8 +1,4 @@
-import {
-  AbortError,
-  isRetryableNetworkError,
-  ResourceLimitError,
-} from '@metamask/kernel-errors';
+import { AbortError, ResourceLimitError } from '@metamask/kernel-errors';
 import {
   abortableDelay,
   DEFAULT_MAX_RETRY_ATTEMPTS,
@@ -22,6 +18,7 @@ import type {
   OnRemoteGiveUp,
   RemoteCommsOptions,
 } from './types.ts';
+import { isRetryableNetworkError } from '../utils/isRetryableNetworkError.ts';
 
 /** Default upper bound for queued outbound messages while reconnecting */
 const DEFAULT_MAX_QUEUE = 200;

yarn.lock

@@ -2337,7 +2337,6 @@ __metadata:
   resolution: "@metamask/kernel-errors@workspace:packages/kernel-errors"
   dependencies:
     "@arethetypeswrong/cli": "npm:^0.17.4"
-    "@libp2p/interface": "npm:2.11.0"
     "@metamask/auto-changelog": "npm:^5.3.0"
     "@metamask/eslint-config": "npm:^15.0.0"
     "@metamask/eslint-config-nodejs": "npm:^15.0.0"