From b36ac37b6e96e32bcf82fbc04dce78c117225357 Mon Sep 17 00:00:00 2001
From: Garvit Gupta <garvit@cloudflare.com>
Date: Wed, 21 Aug 2024 11:48:52 -0500
Subject: [PATCH] VS-271: Fix Vectorize getVcetors, deleteVectors payload in
 Wrangler Client

---
 packages/wrangler/src/vectorize/client.ts      | 5 +++--
 packages/wrangler/src/vectorize/deleteByIds.ts | 8 +++++++-
 packages/wrangler/src/vectorize/getByIds.ts    | 8 +++++++-
 packages/wrangler/src/vectorize/types.ts       | 8 ++++++++
 4 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/packages/wrangler/src/vectorize/client.ts b/packages/wrangler/src/vectorize/client.ts
index f61713538ed0..63d0af18c481 100644
--- a/packages/wrangler/src/vectorize/client.ts
+++ b/packages/wrangler/src/vectorize/client.ts
@@ -12,6 +12,7 @@ import type {
 	VectorizeMetadataIndexPropertyName,
 	VectorizeQueryOptions,
 	VectorizeVector,
+	VectorizeVectorIds,
 	VectorizeVectorMutation,
 } from "./types";
 import type { FormData } from "undici";
@@ -155,7 +156,7 @@ export async function queryIndex(
 export async function getByIds(
 	config: Config,
 	indexName: string,
-	ids: Array<string>
+	ids: VectorizeVectorIds
 ): Promise<VectorizeVector[]> {
 	const accountId = await requireAuth(config);
 
@@ -174,7 +175,7 @@ export async function getByIds(
 export async function deleteByIds(
 	config: Config,
 	indexName: string,
-	ids: Array<string>
+	ids: VectorizeVectorIds
 ): Promise<VectorizeAsyncMutation> {
 	const accountId = await requireAuth(config);
 
diff --git a/packages/wrangler/src/vectorize/deleteByIds.ts b/packages/wrangler/src/vectorize/deleteByIds.ts
index 4beee68c2160..ee8a3415f2c2 100644
--- a/packages/wrangler/src/vectorize/deleteByIds.ts
+++ b/packages/wrangler/src/vectorize/deleteByIds.ts
@@ -6,6 +6,7 @@ import type {
 	CommonYargsArgv,
 	StrictYargsOptionsToInterface,
 } from "../yargs-types";
+import type { VectorizeVectorIds } from "./types";
 
 export function options(yargs: CommonYargsArgv) {
 	return yargs
@@ -37,7 +38,12 @@ export async function handler(
 	}
 
 	logger.log(`📋 Deleting vectors...`);
-	const mutation = await deleteByIds(config, args.name, args.ids);
+
+	const ids: VectorizeVectorIds = {
+		ids: args.ids,
+	};
+
+	const mutation = await deleteByIds(config, args.name, ids);
 
 	logger.log(
 		`✅ Successfully enqueued ${args.ids.length} vectors into index '${args.name}' for deletion. Mutation changeset identifier: ${mutation.mutationId}.`
diff --git a/packages/wrangler/src/vectorize/getByIds.ts b/packages/wrangler/src/vectorize/getByIds.ts
index 7422b2089d25..8f27fa015de7 100644
--- a/packages/wrangler/src/vectorize/getByIds.ts
+++ b/packages/wrangler/src/vectorize/getByIds.ts
@@ -6,6 +6,7 @@ import type {
 	CommonYargsArgv,
 	StrictYargsOptionsToInterface,
 } from "../yargs-types";
+import type { VectorizeVectorIds } from "./types";
 
 export function options(yargs: CommonYargsArgv) {
 	return yargs
@@ -37,7 +38,12 @@ export async function handler(
 	}
 
 	logger.log(`📋 Fetching vectors...`);
-	const vectors = await getByIds(config, args.name, args.ids);
+
+	const ids: VectorizeVectorIds = {
+		ids: args.ids,
+	};
+
+	const vectors = await getByIds(config, args.name, ids);
 
 	if (vectors.length === 0) {
 		logger.warn(
diff --git a/packages/wrangler/src/vectorize/types.ts b/packages/wrangler/src/vectorize/types.ts
index 778b0a26e044..7fd7f6ffa62f 100644
--- a/packages/wrangler/src/vectorize/types.ts
+++ b/packages/wrangler/src/vectorize/types.ts
@@ -138,6 +138,14 @@ export interface VectorizeVectorMutation {
 	count: number;
 }
 
+/**
+ * Request type used to pass vector ids to fetch or delete.
+ */
+export interface VectorizeVectorIds {
+	/* List of vector ids that are fetched or deleted. */
+	ids: string[];
+}
+
 /**
  * Result type indicating a mutation on the Vectorize Index.
  * Actual mutations are processed async where the `mutationId` is the unique identifier for the operation.