@vellumai/cli 0.12.3 → 0.12.4-staging.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vellumai/cli",
-  "version": "0.12.3",
+  "version": "0.12.4-staging.2",
   "description": "CLI tools for vellum-assistant",
   "type": "module",
   "exports": {

package/src/commands/restore.ts

@@ -334,7 +334,12 @@ async function restorePlatform(
     );
 
     try {
-      await rollbackPlatformAssistant(token, opts.version, entry.runtimeUrl);
+      await rollbackPlatformAssistant(
+        token,
+        entry.assistantId,
+        opts.version,
+        entry.runtimeUrl,
+      );
     } catch (err) {
       const msg = err instanceof Error ? err.message : String(err);
       if (msg.includes("401") || msg.includes("403")) {
@@ -628,14 +633,7 @@ async function runLocalRestore(opts: {
   stagedRelativePath: string | undefined;
   bundleData: Buffer | undefined;
 }): Promise<void> {
-  const {
-    entry,
-    name,
-    version,
-    dryRun,
-    stagedRelativePath,
-    bundleData,
-  } = opts;
+  const { entry, name, version, dryRun, stagedRelativePath, bundleData } = opts;
   let accessToken = opts.accessToken;
 
   if (dryRun) {
@@ -762,18 +760,15 @@ async function runLocalRestore(opts: {
             accessToken,
             stagedRelativePath,
           )
-        : await loopbackSafeFetch(
-            `${entry.runtimeUrl}/v1/migrations/import`,
-            {
-              method: "POST",
-              headers: {
-                Authorization: `Bearer ${accessToken}`,
-                "Content-Type": "application/octet-stream",
-              },
-              body: bundleData ? new Uint8Array(bundleData) : undefined,
-              signal: AbortSignal.timeout(120_000),
+        : await loopbackSafeFetch(`${entry.runtimeUrl}/v1/migrations/import`, {
+            method: "POST",
+            headers: {
+              Authorization: `Bearer ${accessToken}`,
+              "Content-Type": "application/octet-stream",
             },
-          );
+            body: bundleData ? new Uint8Array(bundleData) : undefined,
+            signal: AbortSignal.timeout(120_000),
+          });
       if (!response.ok) {
         const body = await response.text();
         console.error(`Error: Import failed (${response.status}): ${body}`);

package/src/commands/rollback.ts

@@ -156,7 +156,12 @@ async function rollbackPlatformViaEndpoint(
 
   let result: { detail: string; version: string | null };
   try {
-    result = await rollbackPlatformAssistant(token, version, entry.runtimeUrl);
+    result = await rollbackPlatformAssistant(
+      token,
+      entry.assistantId,
+      version,
+      entry.runtimeUrl,
+    );
   } catch (err) {
     const detail = err instanceof Error ? err.message : String(err);
 

package/src/lib/platform-client.ts

@@ -829,12 +829,13 @@ export async function fetchCurrentUser(
 
 export async function rollbackPlatformAssistant(
   token: string,
+  assistantId: string,
   version?: string,
   platformUrl?: string,
 ): Promise<{ detail: string; version: string | null }> {
   const resolvedUrl = platformUrl || getPlatformUrl();
   const response = await loopbackSafeFetch(
-    `${resolvedUrl}/v1/assistants/rollback/`,
+    `${resolvedUrl}/v1/assistants/${encodeURIComponent(assistantId)}/rollback/`,
     {
       method: "POST",
       headers: await authHeaders(token, platformUrl),