update types

autoblocksai · Sep 26, 2024 · 5ae8d2e · 5ae8d2e
1 parent b541dc0
commit 5ae8d2e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 15 deletions.
diff --git a/src/client.ts b/src/client.ts
@@ -10,7 +10,6 @@ import {
   AutoblocksEnvVar,
   AUTOBLOCKS_HEADERS,
   API_ENDPOINT,
-  RevisionSpecialVersionsEnum,
 } from './util';
 
 export interface View {
@@ -281,7 +280,6 @@ export class AutoblocksAPIClient {
   }> {
     const encodedName = encodeURIComponent(args.name);
     const encodedSchemaVersion = encodeURIComponent(args.schemaVersion);
-
     if (args.revisionId) {
       return this.get(
         `/datasets/${encodedName}/schema-versions/${encodedSchemaVersion}/revisions/${encodeURIComponent(
@@ -290,10 +288,6 @@ export class AutoblocksAPIClient {
       );
     }
 
-    if (args.schemaVersion === RevisionSpecialVersionsEnum.LATEST) {
-      return this.get(`/datasets/${encodedName}/revisions/latest`);
-    }
-
     return this.get(
       `/datasets/${encodedName}/schema-versions/${encodedSchemaVersion}`,
     );

diff --git a/src/datasets-cli/index.ts b/src/datasets-cli/index.ts
@@ -4,7 +4,6 @@ import {
   AutoblocksEnvVar,
   AUTOBLOCKS_HEADERS,
   API_ENDPOINT,
-  RevisionSpecialVersionsEnum,
 } from '../util';
 import { ParsedDataset, PropertySchema, PropertyTypesEnum } from './types';
 
@@ -27,13 +26,11 @@ export const autogenerationConfigs: AutogenerationConfig[] = [
     symbolType: 'interface',
     filesToModify: ['../datasets/index.d.ts', '../datasets/index.d.mts'],
     generate: (args) => {
-      let generated = `interface ${args.symbolName} {`;
+      let generated = `interface ${args.symbolName} {`; // start of definition
 
       args.datasets.forEach((dataset) => {
         generated += `\n  '${dataset.name}': {`; // start of dataset definition
 
-        generated += `\n    '${RevisionSpecialVersionsEnum.LATEST}': Record<string, unknown>;`;
-
         dataset.schemaVersions.forEach((schemaVersion) => {
           generated += `\n    '${schemaVersion.version}': {`; // start of schema version definition
           schemaVersion.schema.forEach((property) => {
@@ -45,7 +42,7 @@ export const autogenerationConfigs: AutogenerationConfig[] = [
         generated += '\n  };'; // end of dataset definition
       });
 
-      generated += '\n}';
+      generated += '\n}'; // end of definition
       return generated;
     },
   },
@@ -130,9 +127,9 @@ function determineStartAndEndIdx(args: {
   );
 }
 
-async function getAllDatasetSchemasFromAPI(args: {
-  apiKey: string;
-}): Promise<ParsedDataset[]> {
+async function getAllDatasetSchemasFromAPI(args: { apiKey: string }): Promise<{
+  datasets: ParsedDataset[];
+}> {
   const resp = await fetch(`${API_ENDPOINT}/datasets/schemas`, {
     method: 'GET',
     headers: {
@@ -183,7 +180,7 @@ export async function run(): Promise<void> {
     );
   }
 
-  const datasets = await getAllDatasetSchemasFromAPI({ apiKey });
+  const { datasets } = await getAllDatasetSchemasFromAPI({ apiKey });
 
   if (datasets.length === 0) {
     console.log('No dataset schemas found.');