diff --git a/clients/ts-sdk/src/functions/datasets/datasets.test.ts b/clients/ts-sdk/src/functions/datasets/datasets.test.ts
index f9e2d7d213..bdb67eb443 100644
--- a/clients/ts-sdk/src/functions/datasets/datasets.test.ts
+++ b/clients/ts-sdk/src/functions/datasets/datasets.test.ts
@@ -1,17 +1,50 @@
 import { beforeAll, describe, expectTypeOf } from "vitest";
 import { TrieveSDK } from "../../sdk";
-import { EventReturn } from "../../types.gen";
+import {
+  Dataset,
+  DatasetAndUsage,
+  DatasetQueueLengthsResponse,
+  DatasetUsageCount,
+  GetAllTagsResponse,
+} from "../../types.gen";
 import { TRIEVE } from "../../__tests__/constants";
 import { test } from "../../__tests__/utils";
 
-describe("Events Tests", async () => {
+describe("Dataset Tests", async () => {
   let trieve: TrieveSDK;
   beforeAll(() => {
     trieve = TRIEVE;
   });
 
-  test("getEventsForDataset", async () => {
-    const data = await trieve.getEventsForDataset({});
-    expectTypeOf(data).toEqualTypeOf<EventReturn>();
+  test("getDatasetById", async () => {
+    const data = await trieve.getDatasetById(trieve.datasetId!);
+    expectTypeOf(data).toEqualTypeOf<Dataset>();
+  });
+
+  test("getDatasetUsageById", async () => {
+    const data = await trieve.getDatasetUsageById(trieve.datasetId!);
+    expectTypeOf(data).toEqualTypeOf<DatasetUsageCount>();
+  });
+
+  test("getDatasetsFromOrganization", async () => {
+    const data = await trieve.getDatasetsFromOrganization(
+      trieve.organizationId!,
+    );
+    expectTypeOf(data).toEqualTypeOf<DatasetAndUsage[]>();
+  });
+
+  test("getAllDatasetTags", async () => {
+    const data = await trieve.getAllDatasetTags(
+      {
+        page: 1,
+      },
+      trieve.datasetId!,
+    );
+    expectTypeOf(data).toEqualTypeOf<GetAllTagsResponse>();
+  });
+
+  test("getDatasetQueueLengths", async () => {
+    const data = await trieve.getDatasetQueueLengths(trieve.datasetId!);
+    expectTypeOf(data).toEqualTypeOf<DatasetQueueLengthsResponse>();
   });
 });