@letta-ai/letta-code 0.34.12 → 0.34.14

package/dist/mcp-oauth.js

@@ -17392,7 +17392,7 @@ class StreamableHTTPClientTransport {
 // src/mcp-client.ts
 var DEFAULT_CLIENT_INFO = {
   name: "letta-code",
-  version: "0.34.12"
+  version: "0.34.14"
 };
 async function connectMcpServer(config2, options = {}) {
   let client = new Client(options.clientInfo ?? DEFAULT_CLIENT_INFO);
@@ -17583,17 +17583,17 @@ async function startOAuthCallbackServerOnPort(port) {
   let completed = false;
   let settle;
   let reject;
-  const codePromise = new Promise((resolve, rejectPromise) => {
-    settle = (code) => {
+  const callbackPromise = new Promise((resolve, rejectPromise) => {
+    settle = (result) => {
       completed = true;
-      resolve(code);
+      resolve(result);
     };
     reject = (error2) => {
       completed = true;
       rejectPromise(error2);
     };
   });
-  codePromise.catch(() => {
+  callbackPromise.catch(() => {
     return;
   });
   server = createServer((request, response) => {
@@ -17624,6 +17624,7 @@ async function startOAuthCallbackServerOnPort(port) {
     const error2 = url2.searchParams.get("error");
     const code = url2.searchParams.get("code");
     const state = url2.searchParams.get("state");
+    const domain = url2.searchParams.get("domain");
     if (!state || state !== expectedState.value) {
       response.writeHead(400, { "Content-Type": "text/html" });
       response.end(callbackPage("Authorization failed", "Invalid OAuth callback"));
@@ -17643,7 +17644,7 @@ async function startOAuthCallbackServerOnPort(port) {
       return;
     }
     response.writeHead(200, { "Content-Type": "text/html" });
-    settle?.(code);
+    settle?.(domain ? { code, domain } : { code });
     response.end(callbackPage("Authorization complete", "You can close this tab and return to Letta Code."), () => {
       closeServer(true);
     });
@@ -17684,7 +17685,7 @@ Content-Length: 0\r
   return {
     redirectUrl: `http://127.0.0.1:${address.port}/callback`,
     expectedState,
-    waitForCode: () => codePromise.finally(() => clearTimeout(timeout)),
+    waitForCallback: () => callbackPromise.finally(() => clearTimeout(timeout)),
     close: () => {
       clearTimeout(timeout);
       if (!completed)
@@ -17709,6 +17710,21 @@ function escapeHtml(value) {
   });
 }
 
+// src/mcp-oauth-client-name.ts
+var DEFAULT_MCP_OAUTH_CLIENT_NAME = "Letta Code";
+var MAX_MCP_OAUTH_CLIENT_NAME_LENGTH = 64;
+function resolveMcpOAuthClientName(clientName) {
+  if (clientName === undefined)
+    return DEFAULT_MCP_OAUTH_CLIENT_NAME;
+  const trimmed = clientName.trim();
+  if (!trimmed)
+    throw new Error("MCP OAuth client name must not be empty");
+  if (trimmed.length > MAX_MCP_OAUTH_CLIENT_NAME_LENGTH) {
+    throw new Error(`MCP OAuth client name must be at most ${MAX_MCP_OAUTH_CLIENT_NAME_LENGTH} characters`);
+  }
+  return trimmed;
+}
+
 // src/mcp-oauth-identity.ts
 import { createHash } from "node:crypto";
 function mcpOAuthCredentialKey(agentId, serverName, serverUrl) {
@@ -17865,6 +17881,7 @@ var credentialCoordinators = new Map;
 var storageObjectIds = new WeakMap;
 var nextStorageObjectId = 1;
 async function createMcpOAuthSessionWithStorage(options) {
+  const clientName = resolveMcpOAuthClientName(options.clientName);
   const coordinationKey = storageCoordinationKey(options.storage, options.credentialKey, options.storageNamespace);
   retainCredentialCoordinator(coordinationKey);
   let provider;
@@ -17892,13 +17909,20 @@ async function createMcpOAuthSessionWithStorage(options) {
       onStatus: options.onStatus,
       openBrowser: options.openBrowser ?? openSystemBrowser,
       expectedState: callback?.expectedState,
-      generation: loaded.generation
+      generation: loaded.generation,
+      clientName
     });
     const activeProvider = provider;
     const activeCallback = callback;
     return {
       authProvider: activeProvider,
-      ...activeCallback ? { waitForAuthorizationCode: () => activeCallback.waitForCode() } : {},
+      ...activeCallback ? {
+        waitForAuthorizationCode: async () => {
+          const { code, domain } = await activeCallback.waitForCallback();
+          options.onAuthorizationCallback?.({ domain });
+          return code;
+        }
+      } : {},
       exportCredentials: () => activeProvider.exportCredentials(),
       closeCallback: async () => activeCallback?.close(),
       close: async () => {
@@ -17934,6 +17958,7 @@ class PersistentMcpOAuthProvider {
   onStatus;
   openBrowser;
   expectedState;
+  clientName;
   closed = false;
   constructor(options) {
     this.credentialKey = options.credentialKey;
@@ -17943,6 +17968,7 @@ class PersistentMcpOAuthProvider {
     this.onStatus = options.onStatus;
     this.openBrowser = options.openBrowser;
     this.expectedState = options.expectedState;
+    this.clientName = options.clientName;
     this.generation = options.generation;
     const coordinator = getCredentialCoordinator(options.coordinationKey);
     this.coordinatorValue = coordinator;
@@ -17962,7 +17988,7 @@ class PersistentMcpOAuthProvider {
   }
   get clientMetadata() {
     return {
-      client_name: "Letta Code",
+      client_name: this.clientName,
       redirect_uris: [this.redirectUrl],
       grant_types: ["authorization_code", "refresh_token"],
       response_types: ["code"],
@@ -18432,7 +18458,7 @@ async function authorizeMcpServerWithStorage(options) {
     throw new Error("MCP OAuth storage namespace must be non-empty and contain no NUL bytes");
   }
   const credentialKey = mcpOAuthCredentialKey(options.agentId, options.serverName, options.serverUrl);
-  const flightKey = `${options.storageNamespace}\x00${credentialKey}\x00${fetchIdentity(options.fetch)}`;
+  const flightKey = `${options.storageNamespace}\x00${credentialKey}\x00${fetchIdentity(options.fetch)}\x00${options.clientName ?? ""}`;
   let flight = inFlightAuthorizations.get(flightKey);
   if (!flight) {
     const controller = new AbortController;
@@ -18462,7 +18488,9 @@ async function runAuthorization(options, credentialKey, signal) {
     storage,
     interactive: true,
     openBrowser: options.openBrowser,
-    onStatus: options.onStatus
+    onStatus: options.onStatus,
+    onAuthorizationCallback: options.onAuthorizationCallback,
+    clientName: options.clientName
   });
   let oauth;
   try {
@@ -18578,4 +18606,4 @@ export {
   authorizeMcpServerWithStorage
 };
 
-//# debugId=5F827E1066B4B8B864756E2164756E21
+//# debugId=6B58BCFFD3DE56CB64756E2164756E21