@truefoundry/assistant-ui-runtime 0.1.2 → 0.1.3-rc.1

package/dist/index.js

@@ -2326,19 +2326,25 @@ function repositoryItemsFromMessages(messages) {
   return items;
 }
 
+// src/private/getGatewayFromPrivateClient.ts
+function getGatewayFromPrivateClient(privateClient) {
+  const internal = privateClient;
+  if (internal.client == null) {
+    throw new Error("PrivateAgentSessionClient is missing an internal gateway client.");
+  }
+  return internal.client;
+}
+
 // src/private/draftSessionBridge.ts
 var DRAFT_SESSION_LAST_UPDATED_AT_HEADER = "x-tfy-session-last-updated-at";
-function createDraftSessionBridge(gateway) {
-  async function getDraft(draftSessionId) {
-    const response = await gateway.agents.private.draftSessions.get(draftSessionId);
-    return response.data;
-  }
+function createDraftSessionBridge(privateClient) {
   return {
     async getDraftAgentSpec(draftSessionId) {
-      const draft = await getDraft(draftSessionId);
+      const draft = await privateClient.getDraftSession({ draftSessionId });
       return draft.agentSpec;
     },
     async syncAgentSpec(draftSessionId, agentSpec) {
+      const gateway = getGatewayFromPrivateClient(privateClient);
       const response = await gateway.agents.private.draftSessions.update(
         draftSessionId,
         { agentSpec }
@@ -2379,10 +2385,10 @@ function sessionListStartTimestamp() {
 // src/private/truefoundryDraftThreadListAdapter.ts
 var THREAD_LIST_PAGE_SIZE = 20;
 function createTrueFoundryDraftThreadListAdapter(options) {
-  const { gateway, defaultAgentSpec, getAgentSpec } = options;
+  const { privateClient, defaultAgentSpec, getAgentSpec } = options;
   return {
     async list({ after } = {}) {
-      const page = await gateway.agents.private.draftSessions.list({
+      const page = await privateClient.listDraftSessions({
         limit: THREAD_LIST_PAGE_SIZE,
         pageToken: after,
         startTimestamp: sessionListStartTimestamp()
@@ -2399,15 +2405,15 @@ function createTrueFoundryDraftThreadListAdapter(options) {
       };
     },
     async initialize(_threadId) {
-      const response = await gateway.agents.private.draftSessions.create({
+      const draft = await privateClient.createDraftSession({
         agentSpec: getAgentSpec?.() ?? defaultAgentSpec
       });
-      const draft = response.data;
       return { remoteId: draft.id, externalId: void 0 };
     },
     async fetch(remoteId) {
-      const response = await gateway.agents.private.draftSessions.get(remoteId);
-      const draft = response.data;
+      const draft = await privateClient.getDraftSession({
+        draftSessionId: remoteId
+      });
       return {
         status: "regular",
         remoteId: draft.id,
@@ -2444,1035 +2450,12 @@ var EMPTY_DRAFT_EXTRAS = {
   }
 };
 
-// ../../node_modules/.pnpm/truefoundry-gateway-sdk@0.3.1/node_modules/truefoundry-gateway-sdk/dist/esm/core/fetcher/HttpResponsePromise.mjs
-var __awaiter = function(thisArg, _arguments, P, generator) {
-  function adopt(value) {
-    return value instanceof P ? value : new P(function(resolve) {
-      resolve(value);
-    });
-  }
-  return new (P || (P = Promise))(function(resolve, reject) {
-    function fulfilled(value) {
-      try {
-        step(generator.next(value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function rejected(value) {
-      try {
-        step(generator["throw"](value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function step(result) {
-      result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected);
-    }
-    step((generator = generator.apply(thisArg, _arguments || [])).next());
-  });
-};
-var HttpResponsePromise = class _HttpResponsePromise extends Promise {
-  constructor(promise) {
-    super((resolve) => {
-      resolve(void 0);
-    });
-    this.innerPromise = promise;
-  }
-  /**
-   * Creates an `HttpResponsePromise` from a function that returns a promise.
-   *
-   * @param fn - A function that returns a promise resolving to a `WithRawResponse` object.
-   * @param args - Arguments to pass to the function.
-   * @returns An `HttpResponsePromise` instance.
-   */
-  static fromFunction(fn, ...args) {
-    return new _HttpResponsePromise(fn(...args));
-  }
-  /**
-   * Creates a function that returns an `HttpResponsePromise` from a function that returns a promise.
-   *
-   * @param fn - A function that returns a promise resolving to a `WithRawResponse` object.
-   * @returns A function that returns an `HttpResponsePromise` instance.
-   */
-  static interceptFunction(fn) {
-    return (...args) => {
-      return _HttpResponsePromise.fromPromise(fn(...args));
-    };
-  }
-  /**
-   * Creates an `HttpResponsePromise` from an existing promise.
-   *
-   * @param promise - A promise resolving to a `WithRawResponse` object.
-   * @returns An `HttpResponsePromise` instance.
-   */
-  static fromPromise(promise) {
-    return new _HttpResponsePromise(promise);
-  }
-  /**
-   * Creates an `HttpResponsePromise` from an executor function.
-   *
-   * @param executor - A function that takes resolve and reject callbacks to create a promise.
-   * @returns An `HttpResponsePromise` instance.
-   */
-  static fromExecutor(executor) {
-    const promise = new Promise(executor);
-    return new _HttpResponsePromise(promise);
-  }
-  /**
-   * Creates an `HttpResponsePromise` from a resolved result.
-   *
-   * @param result - A `WithRawResponse` object to resolve immediately.
-   * @returns An `HttpResponsePromise` instance.
-   */
-  static fromResult(result) {
-    const promise = Promise.resolve(result);
-    return new _HttpResponsePromise(promise);
-  }
-  unwrap() {
-    if (!this.unwrappedPromise) {
-      this.unwrappedPromise = this.innerPromise.then(({ data }) => data);
-    }
-    return this.unwrappedPromise;
-  }
-  /** @inheritdoc */
-  then(onfulfilled, onrejected) {
-    return this.unwrap().then(onfulfilled, onrejected);
-  }
-  /** @inheritdoc */
-  catch(onrejected) {
-    return this.unwrap().catch(onrejected);
-  }
-  /** @inheritdoc */
-  finally(onfinally) {
-    return this.unwrap().finally(onfinally);
-  }
-  /**
-   * Retrieves the data and raw response.
-   *
-   * @returns A promise resolving to a `WithRawResponse` object.
-   */
-  withRawResponse() {
-    return __awaiter(this, void 0, void 0, function* () {
-      return yield this.innerPromise;
-    });
-  }
-};
-
-// ../../node_modules/.pnpm/truefoundry-gateway-sdk@0.3.1/node_modules/truefoundry-gateway-sdk/dist/esm/core/pagination/Page.mjs
-var __awaiter2 = function(thisArg, _arguments, P, generator) {
-  function adopt(value) {
-    return value instanceof P ? value : new P(function(resolve) {
-      resolve(value);
-    });
-  }
-  return new (P || (P = Promise))(function(resolve, reject) {
-    function fulfilled(value) {
-      try {
-        step(generator.next(value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function rejected(value) {
-      try {
-        step(generator["throw"](value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function step(result) {
-      result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected);
-    }
-    step((generator = generator.apply(thisArg, _arguments || [])).next());
-  });
-};
-var __await = function(v) {
-  return this instanceof __await ? (this.v = v, this) : new __await(v);
-};
-var __asyncGenerator = function(thisArg, _arguments, generator) {
-  if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-  var g = generator.apply(thisArg, _arguments || []), i, q = [];
-  return i = Object.create((typeof AsyncIterator === "function" ? AsyncIterator : Object).prototype), verb("next"), verb("throw"), verb("return", awaitReturn), i[Symbol.asyncIterator] = function() {
-    return this;
-  }, i;
-  function awaitReturn(f) {
-    return function(v) {
-      return Promise.resolve(v).then(f, reject);
-    };
-  }
-  function verb(n, f) {
-    if (g[n]) {
-      i[n] = function(v) {
-        return new Promise(function(a, b) {
-          q.push([n, v, a, b]) > 1 || resume(n, v);
-        });
-      };
-      if (f) i[n] = f(i[n]);
-    }
-  }
-  function resume(n, v) {
-    try {
-      step(g[n](v));
-    } catch (e) {
-      settle(q[0][3], e);
-    }
-  }
-  function step(r) {
-    r.value instanceof __await ? Promise.resolve(r.value.v).then(fulfill, reject) : settle(q[0][2], r);
-  }
-  function fulfill(value) {
-    resume("next", value);
-  }
-  function reject(value) {
-    resume("throw", value);
-  }
-  function settle(f, v) {
-    if (f(v), q.shift(), q.length) resume(q[0][0], q[0][1]);
-  }
-};
-var __asyncValues = function(o) {
-  if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-  var m = o[Symbol.asyncIterator], i;
-  return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function() {
-    return this;
-  }, i);
-  function verb(n) {
-    i[n] = o[n] && function(v) {
-      return new Promise(function(resolve, reject) {
-        v = o[n](v), settle(resolve, reject, v.done, v.value);
-      });
-    };
-  }
-  function settle(resolve, reject, d, v) {
-    Promise.resolve(v).then(function(v2) {
-      resolve({ value: v2, done: d });
-    }, reject);
-  }
-};
-var Page = class {
-  constructor({ response, rawResponse, hasNextPage, getItems, loadPage }) {
-    this.response = response;
-    this.rawResponse = rawResponse;
-    this.data = getItems(response);
-    this._hasNextPage = hasNextPage;
-    this.getItems = getItems;
-    this.loadNextPage = loadPage;
-  }
-  /**
-   * Retrieves the next page
-   * @returns this
-   */
-  getNextPage() {
-    return __awaiter2(this, void 0, void 0, function* () {
-      const { data, rawResponse } = yield this.loadNextPage(this.response).withRawResponse();
-      this.response = data;
-      this.rawResponse = rawResponse;
-      this.data = this.getItems(this.response);
-      return this;
-    });
-  }
-  /**
-   * @returns whether there is a next page to load
-   */
-  hasNextPage() {
-    return this._hasNextPage(this.response);
-  }
-  iterMessages() {
-    return __asyncGenerator(this, arguments, function* iterMessages_1() {
-      for (const item of this.data) {
-        yield yield __await(item);
-      }
-      while (this.hasNextPage()) {
-        yield __await(this.getNextPage());
-        for (const item of this.data) {
-          yield yield __await(item);
-        }
-      }
-    });
-  }
-  [Symbol.asyncIterator]() {
-    return __asyncGenerator(this, arguments, function* _a() {
-      var _b, e_1, _c, _d;
-      try {
-        for (var _e = true, _f = __asyncValues(this.iterMessages()), _g; _g = yield __await(_f.next()), _b = _g.done, !_b; _e = true) {
-          _d = _g.value;
-          _e = false;
-          const message = _d;
-          yield yield __await(message);
-        }
-      } catch (e_1_1) {
-        e_1 = { error: e_1_1 };
-      } finally {
-        try {
-          if (!_e && !_b && (_c = _f.return)) yield __await(_c.call(_f));
-        } finally {
-          if (e_1) throw e_1.error;
-        }
-      }
-    });
-  }
-};
-
-// ../../node_modules/.pnpm/truefoundry-gateway-sdk@0.3.1/node_modules/truefoundry-gateway-sdk/dist/esm/agents/TurnStreamData.mjs
-function parseSequenceNumber(id) {
-  if (!id) {
-    throw new Error("Missing SSE sequence number id.");
-  }
-  const parsed = Number(id);
-  if (!Number.isFinite(parsed)) {
-    throw new Error(`Invalid SSE sequence number id: "${id}".`);
-  }
-  return parsed;
-}
-
-// ../../node_modules/.pnpm/truefoundry-gateway-sdk@0.3.1/node_modules/truefoundry-gateway-sdk/dist/esm/agents/Turn.mjs
-var __awaiter3 = function(thisArg, _arguments, P, generator) {
-  function adopt(value) {
-    return value instanceof P ? value : new P(function(resolve) {
-      resolve(value);
-    });
-  }
-  return new (P || (P = Promise))(function(resolve, reject) {
-    function fulfilled(value) {
-      try {
-        step(generator.next(value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function rejected(value) {
-      try {
-        step(generator["throw"](value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function step(result) {
-      result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected);
-    }
-    step((generator = generator.apply(thisArg, _arguments || [])).next());
-  });
-};
-var __classPrivateFieldSet = function(receiver, state, value, kind, f) {
-  if (kind === "m") throw new TypeError("Private method is not writable");
-  if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
-  if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
-  return kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value), value;
-};
-var __classPrivateFieldGet = function(receiver, state, kind, f) {
-  if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-  if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-  return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var __await2 = function(v) {
-  return this instanceof __await2 ? (this.v = v, this) : new __await2(v);
-};
-var __asyncValues2 = function(o) {
-  if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-  var m = o[Symbol.asyncIterator], i;
-  return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function() {
-    return this;
-  }, i);
-  function verb(n) {
-    i[n] = o[n] && function(v) {
-      return new Promise(function(resolve, reject) {
-        v = o[n](v), settle(resolve, reject, v.done, v.value);
-      });
-    };
-  }
-  function settle(resolve, reject, d, v) {
-    Promise.resolve(v).then(function(v2) {
-      resolve({ value: v2, done: d });
-    }, reject);
-  }
-};
-var __asyncGenerator2 = function(thisArg, _arguments, generator) {
-  if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-  var g = generator.apply(thisArg, _arguments || []), i, q = [];
-  return i = Object.create((typeof AsyncIterator === "function" ? AsyncIterator : Object).prototype), verb("next"), verb("throw"), verb("return", awaitReturn), i[Symbol.asyncIterator] = function() {
-    return this;
-  }, i;
-  function awaitReturn(f) {
-    return function(v) {
-      return Promise.resolve(v).then(f, reject);
-    };
-  }
-  function verb(n, f) {
-    if (g[n]) {
-      i[n] = function(v) {
-        return new Promise(function(a, b) {
-          q.push([n, v, a, b]) > 1 || resume(n, v);
-        });
-      };
-      if (f) i[n] = f(i[n]);
-    }
-  }
-  function resume(n, v) {
-    try {
-      step(g[n](v));
-    } catch (e) {
-      settle(q[0][3], e);
-    }
-  }
-  function step(r) {
-    r.value instanceof __await2 ? Promise.resolve(r.value.v).then(fulfill, reject) : settle(q[0][2], r);
-  }
-  function fulfill(value) {
-    resume("next", value);
-  }
-  function reject(value) {
-    resume("throw", value);
-  }
-  function settle(f, v) {
-    if (f(v), q.shift(), q.length) resume(q[0][0], q[0][1]);
-  }
-};
-var _Turn_instances;
-var _Turn_client;
-var _Turn_state;
-var _Turn_applyEvent;
-var DEFAULT_POLL_INTERVAL_MS = 3e3;
-var MIN_POLL_INTERVAL_MS = 3e3;
-var Turn = class {
-  constructor(turn, session, client) {
-    _Turn_instances.add(this);
-    _Turn_client.set(this, void 0);
-    _Turn_state.set(this, void 0);
-    this.id = turn.id;
-    this.sessionId = turn.sessionId;
-    this.previousTurnId = turn.previousTurnId;
-    this.input = turn.input;
-    this.createdBySubject = turn.createdBySubject;
-    this.createdAt = turn.createdAt;
-    __classPrivateFieldSet(this, _Turn_state, turn.state, "f");
-    this.session = session;
-    __classPrivateFieldSet(this, _Turn_client, client, "f");
-  }
-  /**
-   * @returns {TrueFoundryGatewayApi.TurnState} Updated by `refresh()`, `stream()`, and `waitForCompletion()`.
-   */
-  get state() {
-    return __classPrivateFieldGet(this, _Turn_state, "f");
-  }
-  // Terminal states are listed explicitly (not `status !== "running"`) so a newly added
-  // non-terminal status keeps polling by default; new terminal states must be added here.
-  isTerminal(state) {
-    switch (state.status) {
-      case "done":
-      case "cancelled":
-      case "error":
-        return true;
-      default:
-        return false;
-    }
-  }
-  // Refetch from the server, update #state in place, and return self.
-  /**
-   * Refetch from the server, update state in-place and return self.
-   *
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {this} This turn with updated state.
-   */
-  refresh(requestOptions) {
-    return __awaiter3(this, void 0, void 0, function* () {
-      const response = yield __classPrivateFieldGet(this, _Turn_client, "f").agents.sessions.getTurn(this.sessionId, this.id, requestOptions);
-      __classPrivateFieldSet(this, _Turn_state, response.data.state, "f");
-      return this;
-    });
-  }
-  /**
-   * Poll getTurn until terminal.
-   *
-   * @param opts.pollIntervalMs - Poll interval ms while waiting. Minimum 3000.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {TrueFoundryGatewayApi.TurnState} Terminal turn state.
-   */
-  waitForCompletion(opts, requestOptions) {
-    return __awaiter3(this, void 0, void 0, function* () {
-      var _a;
-      const pollIntervalMs = (_a = opts === null || opts === void 0 ? void 0 : opts.pollIntervalMs) !== null && _a !== void 0 ? _a : DEFAULT_POLL_INTERVAL_MS;
-      if (pollIntervalMs < MIN_POLL_INTERVAL_MS) {
-        throw new Error(`pollIntervalMs must be at least ${MIN_POLL_INTERVAL_MS}ms`);
-      }
-      while (!this.isTerminal(__classPrivateFieldGet(this, _Turn_state, "f"))) {
-        yield new Promise((r) => setTimeout(r, pollIntervalMs));
-        yield this.refresh(requestOptions);
-      }
-      return __classPrivateFieldGet(this, _Turn_state, "f");
-    });
-  }
-  // Reconnect to the turn's SSE; let the server decide what to stream (no client-side guard).
-  // Updates #state in place from any event that carries a state snapshot.
-  /**
-   * Reconnect to the turn's SSE. Updates state in-place from lifecycle events.
-   *
-   * @param opts.afterSequenceNumber - Sequence number to resume SSE subscription after.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @yields {TurnStreamData} SSE stream items.
-   */
-  stream(opts, requestOptions) {
-    return __asyncGenerator2(this, arguments, function* stream_1() {
-      var _a, e_1, _b, _c;
-      const sse = yield __await2(__classPrivateFieldGet(this, _Turn_client, "f").agents.sessions.subscribeToTurn(this.sessionId, this.id, { afterSequenceNumber: opts === null || opts === void 0 ? void 0 : opts.afterSequenceNumber }, requestOptions));
-      try {
-        for (var _d = true, _e = __asyncValues2(sse.withMetadata()), _f; _f = yield __await2(_e.next()), _a = _f.done, !_a; _d = true) {
-          _c = _f.value;
-          _d = false;
-          const { data: event, id } = _c;
-          __classPrivateFieldGet(this, _Turn_instances, "m", _Turn_applyEvent).call(this, event);
-          yield yield __await2({ sequenceNumber: parseSequenceNumber(id), event });
-        }
-      } catch (e_1_1) {
-        e_1 = { error: e_1_1 };
-      } finally {
-        try {
-          if (!_d && !_a && (_b = _e.return)) yield __await2(_b.call(_e));
-        } finally {
-          if (e_1) throw e_1.error;
-        }
-      }
-    });
-  }
-  /**
-   * Cancel the running last turn for the session.
-   *
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {void}
-   */
-  cancel(requestOptions) {
-    return __awaiter3(this, void 0, void 0, function* () {
-      yield __classPrivateFieldGet(this, _Turn_client, "f").agents.sessions.cancel(this.sessionId, {}, requestOptions);
-    });
-  }
-  // Expose the autogen Fern Page as-is (it is already async-iterable); no re-wrapping.
-  /**
-   * Paginated turn events; use `stream()` for live SSE.
-   *
-   * @param opts.pageToken - Token from the previous response nextPageToken.
-   * @param opts.limit - Page size. Default 25.
-   * @param opts.order - Sort by creation time. Default `asc`.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {Promise<core.Page<TrueFoundryGatewayApi.TurnEvent, TrueFoundryGatewayApi.ListEventsResponse>>} Paginated turn events.
-   */
-  listEvents(opts, requestOptions) {
-    return __classPrivateFieldGet(this, _Turn_client, "f").agents.sessions.listTurnEvents(this.sessionId, this.id, opts, requestOptions);
-  }
-};
-_Turn_client = /* @__PURE__ */ new WeakMap(), _Turn_state = /* @__PURE__ */ new WeakMap(), _Turn_instances = /* @__PURE__ */ new WeakSet(), _Turn_applyEvent = function _Turn_applyEvent2(event) {
-  if (event.type === "turn.created" || event.type === "turn.done")
-    __classPrivateFieldSet(this, _Turn_state, event.state, "f");
-};
-
-// ../../node_modules/.pnpm/truefoundry-gateway-sdk@0.3.1/node_modules/truefoundry-gateway-sdk/dist/esm/agents/PreparedTurn.mjs
-var __awaiter4 = function(thisArg, _arguments, P, generator) {
-  function adopt(value) {
-    return value instanceof P ? value : new P(function(resolve) {
-      resolve(value);
-    });
-  }
-  return new (P || (P = Promise))(function(resolve, reject) {
-    function fulfilled(value) {
-      try {
-        step(generator.next(value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function rejected(value) {
-      try {
-        step(generator["throw"](value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function step(result) {
-      result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected);
-    }
-    step((generator = generator.apply(thisArg, _arguments || [])).next());
-  });
-};
-var __classPrivateFieldSet2 = function(receiver, state, value, kind, f) {
-  if (kind === "m") throw new TypeError("Private method is not writable");
-  if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
-  if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
-  return kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value), value;
-};
-var __classPrivateFieldGet2 = function(receiver, state, kind, f) {
-  if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-  if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-  return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var __asyncValues3 = function(o) {
-  if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-  var m = o[Symbol.asyncIterator], i;
-  return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function() {
-    return this;
-  }, i);
-  function verb(n) {
-    i[n] = o[n] && function(v) {
-      return new Promise(function(resolve, reject) {
-        v = o[n](v), settle(resolve, reject, v.done, v.value);
-      });
-    };
-  }
-  function settle(resolve, reject, d, v) {
-    Promise.resolve(v).then(function(v2) {
-      resolve({ value: v2, done: d });
-    }, reject);
-  }
-};
-var __await3 = function(v) {
-  return this instanceof __await3 ? (this.v = v, this) : new __await3(v);
-};
-var __asyncDelegator = function(o) {
-  var i, p;
-  return i = {}, verb("next"), verb("throw", function(e) {
-    throw e;
-  }), verb("return"), i[Symbol.iterator] = function() {
-    return this;
-  }, i;
-  function verb(n, f) {
-    i[n] = o[n] ? function(v) {
-      return (p = !p) ? { value: __await3(o[n](v)), done: false } : f ? f(v) : v;
-    } : f;
-  }
-};
-var __asyncGenerator3 = function(thisArg, _arguments, generator) {
-  if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-  var g = generator.apply(thisArg, _arguments || []), i, q = [];
-  return i = Object.create((typeof AsyncIterator === "function" ? AsyncIterator : Object).prototype), verb("next"), verb("throw"), verb("return", awaitReturn), i[Symbol.asyncIterator] = function() {
-    return this;
-  }, i;
-  function awaitReturn(f) {
-    return function(v) {
-      return Promise.resolve(v).then(f, reject);
-    };
-  }
-  function verb(n, f) {
-    if (g[n]) {
-      i[n] = function(v) {
-        return new Promise(function(a, b) {
-          q.push([n, v, a, b]) > 1 || resume(n, v);
-        });
-      };
-      if (f) i[n] = f(i[n]);
-    }
-  }
-  function resume(n, v) {
-    try {
-      step(g[n](v));
-    } catch (e) {
-      settle(q[0][3], e);
-    }
-  }
-  function step(r) {
-    r.value instanceof __await3 ? Promise.resolve(r.value.v).then(fulfill, reject) : settle(q[0][2], r);
-  }
-  function fulfill(value) {
-    resume("next", value);
-  }
-  function reject(value) {
-    resume("throw", value);
-  }
-  function settle(f, v) {
-    if (f(v), q.shift(), q.length) resume(q[0][0], q[0][1]);
-  }
-};
-var _PreparedTurn_client;
-var _PreparedTurn_input;
-var _PreparedTurn_previousTurnIdInput;
-var _PreparedTurn_start;
-var _PreparedTurn_turn;
-var PreparedTurn = class {
-  constructor(init, session, client) {
-    _PreparedTurn_client.set(this, void 0);
-    _PreparedTurn_input.set(this, void 0);
-    _PreparedTurn_previousTurnIdInput.set(this, void 0);
-    _PreparedTurn_start.set(this, void 0);
-    _PreparedTurn_turn.set(this, void 0);
-    this.session = session;
-    this.sessionId = session.id;
-    __classPrivateFieldSet2(this, _PreparedTurn_client, client, "f");
-    __classPrivateFieldSet2(this, _PreparedTurn_input, init.input, "f");
-    __classPrivateFieldSet2(this, _PreparedTurn_previousTurnIdInput, init.previousTurnId, "f");
-  }
-  // Remaining data getters delegate to the inner Turn (undefined until started).
-  /**
-   * @returns {string | undefined} Undefined until `execute()` starts the turn.
-   */
-  get id() {
-    var _a;
-    return (_a = __classPrivateFieldGet2(this, _PreparedTurn_turn, "f")) === null || _a === void 0 ? void 0 : _a.id;
-  }
-  /**
-   * @returns {string | undefined} Undefined until `execute()` starts the turn.
-   */
-  get previousTurnId() {
-    var _a;
-    return (_a = __classPrivateFieldGet2(this, _PreparedTurn_turn, "f")) === null || _a === void 0 ? void 0 : _a.previousTurnId;
-  }
-  /**
-   * @returns {TrueFoundryGatewayApi.TurnState | undefined} Undefined until `execute()` starts the turn.
-   */
-  get state() {
-    var _a;
-    return (_a = __classPrivateFieldGet2(this, _PreparedTurn_turn, "f")) === null || _a === void 0 ? void 0 : _a.state;
-  }
-  /**
-   * @returns {TrueFoundryGatewayApi.Subject | undefined} Undefined until `execute()` starts the turn.
-   */
-  get createdBySubject() {
-    var _a;
-    return (_a = __classPrivateFieldGet2(this, _PreparedTurn_turn, "f")) === null || _a === void 0 ? void 0 : _a.createdBySubject;
-  }
-  /**
-   * @returns {string | undefined} Undefined until `execute()` starts the turn.
-   */
-  get createdAt() {
-    var _a;
-    return (_a = __classPrivateFieldGet2(this, _PreparedTurn_turn, "f")) === null || _a === void 0 ? void 0 : _a.createdAt;
-  }
-  /**
-   * @returns {TrueFoundryGatewayApi.TurnInputItem[] | undefined} Staged input before execute; inner turn input after.
-   */
-  get input() {
-    var _a, _b;
-    return (_b = (_a = __classPrivateFieldGet2(this, _PreparedTurn_turn, "f")) === null || _a === void 0 ? void 0 : _a.input) !== null && _b !== void 0 ? _b : __classPrivateFieldGet2(this, _PreparedTurn_input, "f");
-  }
-  execute(opts, requestOptions) {
-    if (__classPrivateFieldGet2(this, _PreparedTurn_start, "f") != null)
-      throw new Error("Turn already started; use stream() / waitForCompletion().");
-    __classPrivateFieldSet2(this, _PreparedTurn_start, this.openCreateTurn(requestOptions), "f");
-    const { stream = true, pollIntervalMs } = opts !== null && opts !== void 0 ? opts : {};
-    return stream === false ? this.startAndWait(pollIntervalMs, requestOptions) : this.runStreaming();
-  }
-  // Post-execution behaviors. Each throws via mustGetTurn() until execute() has started the turn,
-  // then delegates to the inner Turn. stream() here is the re-subscribe path (subscribeToTurn).
-  /**
-   * Resubscribe via subscribeToTurn after `execute()`.
-   *
-   * @param opts.afterSequenceNumber - Sequence number to resume SSE subscription after.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @yields {TurnStreamData} SSE stream items.
-   */
-  stream(opts, requestOptions) {
-    return __asyncGenerator3(this, arguments, function* stream_1() {
-      yield __await3(yield* __asyncDelegator(__asyncValues3(this.mustGetTurn().stream(opts, requestOptions))));
-    });
-  }
-  /**
-   * Refetch from the server, update state in-place and return self.
-   *
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {this} This prepared turn with updated state.
-   */
-  refresh(requestOptions) {
-    return __awaiter4(this, void 0, void 0, function* () {
-      yield this.mustGetTurn().refresh(requestOptions);
-      return this;
-    });
-  }
-  /**
-   * Poll getTurn until terminal.
-   *
-   * @param opts.pollIntervalMs - Poll interval ms while waiting. Minimum 3000.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {TrueFoundryGatewayApi.TurnState} Terminal turn state.
-   */
-  waitForCompletion(opts, requestOptions) {
-    return __awaiter4(this, void 0, void 0, function* () {
-      return this.mustGetTurn().waitForCompletion(opts, requestOptions);
-    });
-  }
-  /**
-   * Cancel the running last turn for the session.
-   *
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {void}
-   */
-  cancel(requestOptions) {
-    return __awaiter4(this, void 0, void 0, function* () {
-      return this.mustGetTurn().cancel(requestOptions);
-    });
-  }
-  /**
-   * Paginated turn events; use `stream()` for live SSE.
-   *
-   * @param opts.pageToken - Token from the previous response nextPageToken.
-   * @param opts.limit - Page size. Default 25.
-   * @param opts.order - Sort by creation time. Default `asc`.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {Promise<core.Page<TrueFoundryGatewayApi.TurnEvent, TrueFoundryGatewayApi.ListEventsResponse>>} Paginated turn events.
-   */
-  listEvents(opts, requestOptions) {
-    return this.mustGetTurn().listEvents(opts, requestOptions);
-  }
-  // execute(stream:true) path: consume the already-open createTurn SSE (#start), adopting the inner Turn on turn.created.
-  runStreaming() {
-    return __asyncGenerator3(this, arguments, function* runStreaming_1() {
-      yield __await3(yield* __asyncDelegator(__asyncValues3(this.consumeStream(yield __await3(__classPrivateFieldGet2(this, _PreparedTurn_start, "f"))))));
-    });
-  }
-  // execute(stream:false) path: drive the open createTurn SSE until turn.created mints the inner Turn, then poll to terminal.
-  startAndWait(pollIntervalMs, requestOptions) {
-    return __awaiter4(this, void 0, void 0, function* () {
-      const turn = yield this.createTurnIfNotExist();
-      return turn.waitForCompletion({ pollIntervalMs }, requestOptions);
-    });
-  }
-  // Consume an SSE stream, adopting the inner Turn from the first turn.created and yielding all events.
-  // Isolated so a future reconnect wrapper can re-open the stream (subscribeToTurn) on disconnect.
-  // NOTE: reconnect/resume is out of current scope.
-  consumeStream(sse) {
-    return __asyncGenerator3(this, arguments, function* consumeStream_1() {
-      var _a, e_1, _b, _c;
-      try {
-        for (var _d = true, _e = __asyncValues3(sse.withMetadata()), _f; _f = yield __await3(_e.next()), _a = _f.done, !_a; _d = true) {
-          _c = _f.value;
-          _d = false;
-          const { data: event, id } = _c;
-          if (event.type === "turn.created" && __classPrivateFieldGet2(this, _PreparedTurn_turn, "f") == null)
-            this.adoptTurn(event);
-          else if (__classPrivateFieldGet2(this, _PreparedTurn_turn, "f") != null && event.type === "turn.done")
-            this.replaceTurnState(event.state);
-          yield yield __await3({ sequenceNumber: parseSequenceNumber(id), event });
-        }
-      } catch (e_1_1) {
-        e_1 = { error: e_1_1 };
-      } finally {
-        try {
-          if (!_d && !_a && (_b = _e.return)) yield __await3(_b.call(_e));
-        } finally {
-          if (e_1) throw e_1.error;
-        }
-      }
-    });
-  }
-  // Returns the inner Turn or throws if execute() has not started it yet.
-  mustGetTurn() {
-    if (__classPrivateFieldGet2(this, _PreparedTurn_turn, "f") == null) {
-      throw new Error("Turn not started yet; call execute() first.");
-    }
-    return __classPrivateFieldGet2(this, _PreparedTurn_turn, "f");
-  }
-  // Drive the in-flight createTurn SSE (#start) only until the first turn.created has built the inner
-  // Turn, then break (which closes the underlying SSE via the generator's return()). Returns the Turn.
-  createTurnIfNotExist() {
-    return __awaiter4(this, void 0, void 0, function* () {
-      var _a, e_2, _b, _c;
-      if (__classPrivateFieldGet2(this, _PreparedTurn_turn, "f") == null) {
-        try {
-          for (var _d = true, _e = __asyncValues3(this.consumeStream(yield __classPrivateFieldGet2(this, _PreparedTurn_start, "f"))), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
-            _c = _f.value;
-            _d = false;
-            const _event = _c;
-            if (__classPrivateFieldGet2(this, _PreparedTurn_turn, "f") != null)
-              break;
-          }
-        } catch (e_2_1) {
-          e_2 = { error: e_2_1 };
-        } finally {
-          try {
-            if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
-          } finally {
-            if (e_2) throw e_2.error;
-          }
-        }
-      }
-      return this.mustGetTurn();
-    });
-  }
-  // Open the createTurn SSE with the pending input/previousTurnId. Called synchronously by execute()
-  // so the POST is in flight (and #start latched) before execute returns.
-  openCreateTurn(requestOptions) {
-    return __classPrivateFieldGet2(this, _PreparedTurn_client, "f").agents.sessions.createTurn(this.sessionId, { input: __classPrivateFieldGet2(this, _PreparedTurn_input, "f"), previousTurnId: __classPrivateFieldGet2(this, _PreparedTurn_previousTurnIdInput, "f") }, requestOptions);
-  }
-  // Build the inner Turn directly from the turn.created event (no extra getTurn round trip).
-  // The TurnCreatedEvent member of the TurnStreamingEvent union carries everything Turn needs.
-  // input comes from the request we sent (this.#input), not the event.
-  adoptTurn(event) {
-    __classPrivateFieldSet2(this, _PreparedTurn_turn, new Turn({
-      id: event.turnId,
-      sessionId: this.sessionId,
-      previousTurnId: event.previousTurnId,
-      input: __classPrivateFieldGet2(this, _PreparedTurn_input, "f"),
-      state: event.state,
-      createdBySubject: event.createdBy,
-      createdAt: event.createdAt
-    }, this.session, __classPrivateFieldGet2(this, _PreparedTurn_client, "f")), "f");
-  }
-  // Replace the inner Turn with a fresh one carrying the new state. State changes are rare, so
-  // rebuilding via Turn's own constructor (rather than mutating it) keeps Turn the source of truth.
-  replaceTurnState(state) {
-    const turn = this.mustGetTurn();
-    __classPrivateFieldSet2(this, _PreparedTurn_turn, new Turn({
-      id: turn.id,
-      sessionId: turn.sessionId,
-      previousTurnId: turn.previousTurnId,
-      input: turn.input,
-      state,
-      createdBySubject: turn.createdBySubject,
-      createdAt: turn.createdAt
-    }, this.session, __classPrivateFieldGet2(this, _PreparedTurn_client, "f")), "f");
-  }
-};
-_PreparedTurn_client = /* @__PURE__ */ new WeakMap(), _PreparedTurn_input = /* @__PURE__ */ new WeakMap(), _PreparedTurn_previousTurnIdInput = /* @__PURE__ */ new WeakMap(), _PreparedTurn_start = /* @__PURE__ */ new WeakMap(), _PreparedTurn_turn = /* @__PURE__ */ new WeakMap();
-
-// ../../node_modules/.pnpm/truefoundry-gateway-sdk@0.3.1/node_modules/truefoundry-gateway-sdk/dist/esm/agents/AgentSession.mjs
-var __awaiter5 = function(thisArg, _arguments, P, generator) {
-  function adopt(value) {
-    return value instanceof P ? value : new P(function(resolve) {
-      resolve(value);
-    });
-  }
-  return new (P || (P = Promise))(function(resolve, reject) {
-    function fulfilled(value) {
-      try {
-        step(generator.next(value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function rejected(value) {
-      try {
-        step(generator["throw"](value));
-      } catch (e) {
-        reject(e);
-      }
-    }
-    function step(result) {
-      result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected);
-    }
-    step((generator = generator.apply(thisArg, _arguments || [])).next());
-  });
-};
-var __classPrivateFieldSet3 = function(receiver, state, value, kind, f) {
-  if (kind === "m") throw new TypeError("Private method is not writable");
-  if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
-  if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
-  return kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value), value;
-};
-var __classPrivateFieldGet3 = function(receiver, state, kind, f) {
-  if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
-  if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
-  return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
-};
-var _AgentSession_client;
-var AgentSession = class {
-  constructor(session, client) {
-    _AgentSession_client.set(this, void 0);
-    this.id = session.id;
-    this.agentName = session.agentName;
-    this.title = session.title;
-    this.createdBySubject = session.createdBySubject;
-    this.createdAt = session.createdAt;
-    this.updatedAt = session.updatedAt;
-    __classPrivateFieldSet3(this, _AgentSession_client, client, "f");
-  }
-  /**
-   * Stage a turn locally; call `execute()` to start `createTurn`.
-   *
-   * @param opts.input - Turn input items passed to createTurn.
-   * @param opts.previousTurnId - Previous turn to chain from. Default `auto`.
-   * @returns {PreparedTurn} Staged turn.
-   */
-  prepareTurn(opts) {
-    return new PreparedTurn({ input: opts === null || opts === void 0 ? void 0 : opts.input, previousTurnId: opts === null || opts === void 0 ? void 0 : opts.previousTurnId }, this, __classPrivateFieldGet3(this, _AgentSession_client, "f"));
-  }
-  /**
-   * List turns in this session.
-   *
-   * @param opts.pageToken - Token from the previous response nextPageToken.
-   * @param opts.limit - Page size. Default 10.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {core.Page<Turn, TrueFoundryGatewayApi.ListTurnsResponse>} Paginated turns.
-   */
-  listTurns(opts, requestOptions) {
-    return __awaiter5(this, void 0, void 0, function* () {
-      const client = __classPrivateFieldGet3(this, _AgentSession_client, "f");
-      const sessionId = this.id;
-      const page = yield client.agents.sessions.listTurns(sessionId, opts, requestOptions);
-      return new Page({
-        response: page.response,
-        rawResponse: page.rawResponse,
-        hasNextPage: (response) => !!(response === null || response === void 0 ? void 0 : response.pagination.nextPageToken),
-        getItems: (response) => {
-          var _a;
-          return ((_a = response === null || response === void 0 ? void 0 : response.data) !== null && _a !== void 0 ? _a : []).map((turn) => new Turn(turn, this, client));
-        },
-        loadPage: (response) => HttpResponsePromise.fromPromise(client.agents.sessions.listTurns(sessionId, Object.assign(Object.assign({}, opts), { pageToken: response === null || response === void 0 ? void 0 : response.pagination.nextPageToken }), requestOptions).then((nextPage) => ({ data: nextPage.response, rawResponse: nextPage.rawResponse })))
-      });
-    });
-  }
-  /**
-   * Fetch a turn by ID.
-   *
-   * @param opts.turnId - Unique identifier of the turn to fetch.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {Turn} Turn data.
-   */
-  getTurn(opts, requestOptions) {
-    return __awaiter5(this, void 0, void 0, function* () {
-      const response = yield __classPrivateFieldGet3(this, _AgentSession_client, "f").agents.sessions.getTurn(this.id, opts.turnId, requestOptions);
-      return new Turn(response.data, this, __classPrivateFieldGet3(this, _AgentSession_client, "f"));
-    });
-  }
-  /**
-   * Cancel the running last turn for the session.
-   *
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {void}
-   */
-  cancel(requestOptions) {
-    return __awaiter5(this, void 0, void 0, function* () {
-      yield __classPrivateFieldGet3(this, _AgentSession_client, "f").agents.sessions.cancel(this.id, {}, requestOptions);
-    });
-  }
-  /**
-   * Paginated session events across turns (newest first); subscribe to a running turn for live events.
-   *
-   * @param opts.pageToken - Token from the previous response nextPageToken.
-   * @param opts.lastTurnId - Newest turn in the listing window (initial load only). Omit to use the session last turn.
-   * @param opts.limit - Page size. Default 100.
-   * @param requestOptions - Overrides client timeout, retries, abortSignal, headers, queryParams.
-   * @returns {Promise<core.Page<TrueFoundryGatewayApi.SessionEventItem, TrueFoundryGatewayApi.ListSessionEventsResponse>>} Paginated session events.
-   */
-  listEvents(opts, requestOptions) {
-    return __classPrivateFieldGet3(this, _AgentSession_client, "f").agents.sessions.listEvents(this.id, opts, requestOptions);
-  }
-};
-_AgentSession_client = /* @__PURE__ */ new WeakMap();
-
 // src/private/bindDraftAgentSession.ts
 var inflightByDraftId = /* @__PURE__ */ new Map();
-function getGatewayFromSessionClient(client) {
-  const internal = client;
-  if (internal.client == null) {
-    throw new Error("AgentSessionClient is missing an internal gateway client.");
-  }
-  return internal.client;
-}
-function draftToSessionRecord(draft) {
-  return {
-    id: draft.id,
-    agentName: draft.agentName ?? "",
-    title: draft.title,
-    createdBySubject: draft.createdBySubject,
-    createdAt: draft.createdAt,
-    updatedAt: draft.updatedAt
-  };
-}
-async function bindDraftAgentSession(client, gateway, draftSessionId) {
+async function bindDraftAgentSession(privateClient, draftSessionId) {
   let inflight = inflightByDraftId.get(draftSessionId);
   if (inflight == null) {
-    inflight = (async () => {
-      const response = await gateway.agents.private.draftSessions.get(draftSessionId);
-      return new AgentSession(
-        draftToSessionRecord(response.data),
-        getGatewayFromSessionClient(client)
-      );
-    })().finally(() => {
+    inflight = privateClient.getDraftSession({ draftSessionId }).then((draft) => draft).finally(() => {
       if (inflightByDraftId.get(draftSessionId) === inflight) {
         inflightByDraftId.delete(draftSessionId);
       }
@@ -3485,8 +2468,8 @@ async function bindDraftAgentSession(client, gateway, draftSessionId) {
 // src/sessions.ts
 var inflightBySessionId = /* @__PURE__ */ new Map();
 function getSession(client, sessionId, options) {
-  if (options?.draftGateway != null) {
-    return bindDraftAgentSession(client, options.draftGateway, sessionId);
+  if (options?.privateClient != null) {
+    return bindDraftAgentSession(options.privateClient, sessionId);
   }
   let inflight = inflightBySessionId.get(sessionId);
   if (inflight == null) {
@@ -3567,15 +2550,15 @@ function resolveTrueFoundryAgentConfig(options) {
 }
 function resolveTrueFoundryAgentRuntimeOptions(options) {
   const agent = resolveTrueFoundryAgentConfig(options);
-  if (agent.mode === "draft" && options.gateway == null) {
+  if (agent.mode === "draft" && options.privateClient == null) {
     throw new Error(
-      "Draft agent mode requires a `gateway` TrueFoundryGateway client."
+      "Draft agent mode requires a `privateClient` PrivateAgentSessionClient."
     );
   }
   return {
     ...options,
     agent,
-    gateway: options.gateway
+    privateClient: options.privateClient
   };
 }
 
@@ -3787,7 +2770,7 @@ import { useCallback as useCallback2, useEffect as useEffect2, useMemo as useMem
 // src/loadSessionSnapshot.ts
 var inflightBySessionId2 = /* @__PURE__ */ new Map();
 function loadSessionSnapshot(client, sessionId, sessionOptions, onProgress) {
-  const cacheKey = sessionOptions?.draftGateway != null ? `draft:${sessionId}` : sessionId;
+  const cacheKey = sessionOptions?.privateClient != null ? `draft:${sessionId}` : sessionId;
   let inflight = inflightBySessionId2.get(cacheKey);
   if (inflight == null) {
     inflight = getSession(client, sessionId, sessionOptions).then(
@@ -4035,12 +3018,12 @@ function useTrueFoundryAgentMessages({
   onError,
   initializeSession,
   resolveConversationSessionId,
-  draftGateway,
+  privateClient,
   getTurnHeaders
 }) {
   const sessionOptions = useMemo2(
-    () => draftGateway != null ? { draftGateway } : void 0,
-    [draftGateway]
+    () => privateClient != null ? { privateClient } : void 0,
+    [privateClient]
   );
   const [snapshot, setSnapshot] = useState2(createEmptySessionSnapshot);
   const [isRunning, setIsRunning] = useState2(false);
@@ -4581,14 +3564,14 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
   const {
     client,
     agent,
-    gateway,
+    privateClient,
     adapters,
     onError,
     listEventsConcurrency,
     ...sharedOptions
   } = options;
   const draftBridgeRef = useRef3(
-    agent.mode === "draft" && gateway != null ? createDraftSessionBridge(gateway) : null
+    agent.mode === "draft" && privateClient != null ? createDraftSessionBridge(privateClient) : null
   );
   const draftSessionId = useAuiState(
     (state) => agent.mode === "draft" ? state.threadListItem.remoteId ?? void 0 : void 0
@@ -4645,7 +3628,7 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     listEventsConcurrency,
     onError,
     initializeSession,
-    draftGateway: agent.mode === "draft" ? gateway : void 0,
+    privateClient: agent.mode === "draft" ? privateClient : void 0,
     getTurnHeaders: agent.mode === "draft" ? getTurnHeaders : void 0
   });
   if (agent.mode === "draft" && draftSpec.agentSpec != null) {
@@ -4667,9 +3650,9 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
   );
   const downloadSandboxFile = useCallback3(
     async (path) => {
-      if (gateway == null) {
+      if (privateClient == null) {
         const error = new Error(
-          "Downloading a sandbox file requires a `gateway` TrueFoundryGateway client."
+          "Downloading a sandbox file requires a `privateClient` PrivateAgentSessionClient."
         );
         onError?.(error);
         throw error;
@@ -4679,10 +3662,11 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
         onError?.(error);
         throw error;
       }
+      const gateway = getGatewayFromPrivateClient(privateClient);
       const response = await gateway.agents.downloadSandboxFile(sandboxId, { path });
       return await response.blob();
     },
-    [gateway, sandboxId, onError]
+    [privateClient, sandboxId, onError]
   );
   const draftExtras = useMemo3(() => {
     if (agent.mode !== "draft") {
@@ -4767,7 +3751,7 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
 }
 function useTrueFoundryAgentRuntime(options) {
   const resolved = resolveTrueFoundryAgentRuntimeOptions(options);
-  const { client, agent, gateway } = resolved;
+  const { client, agent, privateClient } = resolved;
   const pendingAgentSpecRef = useRef3(
     agent.mode === "draft" ? agent.defaultAgentSpec : void 0
   );
@@ -4775,14 +3759,14 @@ function useTrueFoundryAgentRuntime(options) {
   const namedAgentName = agent.mode === "named" ? agent.agentName : void 0;
   const threadListAdapter = useMemo3(() => {
     if (agentMode === "draft") {
-      if (gateway == null) {
+      if (privateClient == null) {
         throw new Error(
-          "Draft agent mode requires a `gateway` TrueFoundryGateway client."
+          "Draft agent mode requires a `privateClient` PrivateAgentSessionClient."
         );
       }
       const draftAgent = agent;
       return createTrueFoundryDraftThreadListAdapter({
-        gateway,
+        privateClient,
         defaultAgentSpec: draftAgent.defaultAgentSpec,
         getAgentSpec: () => pendingAgentSpecRef.current ?? draftAgent.defaultAgentSpec
       });
@@ -4791,7 +3775,7 @@ function useTrueFoundryAgentRuntime(options) {
       client,
       agentName: namedAgentName
     });
-  }, [agentMode, namedAgentName, client, gateway]);
+  }, [agentMode, namedAgentName, client, privateClient]);
   return useRemoteThreadListRuntime({
     allowNesting: true,
     adapter: threadListAdapter,
@@ -4802,6 +3786,67 @@ function useTrueFoundryAgentRuntime(options) {
   });
 }
 
+// src/truefoundryOwnedSessionsThreadListAdapter.ts
+var THREAD_LIST_PAGE_SIZE3 = 20;
+function ownedSessionTitle(session) {
+  if (session.type === "session/draft") {
+    return draftSessionTitle(session);
+  }
+  return session.title ?? session.agentName;
+}
+function createTrueFoundryOwnedSessionsThreadListAdapter(options) {
+  const { privateClient } = options;
+  return {
+    async list({ after } = {}) {
+      const page = await privateClient.listOwnedSessions({
+        limit: THREAD_LIST_PAGE_SIZE3,
+        pageToken: after,
+        startTimestamp: sessionListStartTimestamp()
+      });
+      const threads = page.data.map((session) => ({
+        status: "regular",
+        remoteId: session.id,
+        title: ownedSessionTitle(session),
+        lastMessageAt: new Date(session.updatedAt)
+      }));
+      return {
+        threads,
+        nextCursor: page.response.pagination.nextPageToken ?? void 0
+      };
+    },
+    async initialize() {
+      throw new Error(
+        "Owned sessions history adapter is read-only; create sessions via a named or draft runtime."
+      );
+    },
+    async fetch(remoteId) {
+      const draft = await privateClient.getDraftSession({
+        draftSessionId: remoteId
+      });
+      return {
+        status: "regular",
+        remoteId: draft.id,
+        title: draftSessionTitle(draft),
+        lastMessageAt: new Date(draft.updatedAt)
+      };
+    },
+    async rename() {
+    },
+    async archive() {
+    },
+    async unarchive() {
+    },
+    async delete() {
+    },
+    async generateTitle() {
+      return new ReadableStream();
+    }
+  };
+}
+
+// src/index.ts
+import { PrivateAgentSessionClient } from "truefoundry-gateway-sdk/agents/private";
+
 // src/hooks.ts
 import { useMemo as useMemo4 } from "react";
 import { useAui as useAui2 } from "@assistant-ui/store";
@@ -4946,6 +3991,7 @@ var trueFoundryAttachmentAdapter = {
   }
 };
 export {
+  PrivateAgentSessionClient,
   ROOT_THREAD_ID,
   TOOL_RESPONSE_THREAD_ID_CUSTOM_KEY,
   buildEditedUserMessageContent,
@@ -4957,6 +4003,7 @@ export {
   convertTurnsToThreadMessages,
   createDraftSessionBridge,
   createTrueFoundryDraftThreadListAdapter,
+  createTrueFoundryOwnedSessionsThreadListAdapter,
   createTrueFoundryThreadListAdapter,
   draftSessionTitle,
   findPausedAssistantMessage,