✨ Switch to nodenext resolution

packages/gguf/src/gguf.spec.ts

@@ -33,7 +33,7 @@ describe("gguf", () => {
 			const arrayBuf = await res.arrayBuffer();
 			fs.writeFileSync(".cache/model.gguf", Buffer.from(arrayBuf));
 		}
-	});
+	}, 30_000);
 
 	it("should parse a llama2 7b", async () => {
 		const { metadata, tensorInfos } = await gguf(URL_LLAMA);

packages/tasks/package.json

@@ -9,10 +9,9 @@
 	},
 	"main": "./dist/index.cjs",
 	"module": "./dist/index.js",
-	"types": "./dist/src/index.d.ts",
+	"types": "./dist/index.d.ts",
 	"exports": {
 		".": {
-			"types": "./dist/src/index.d.ts",
 			"require": "./dist/index.cjs",
 			"import": "./dist/index.js"
 		}
@@ -24,13 +23,14 @@
 		"format": "prettier --write .",
 		"format:check": "prettier --check .",
 		"prepublishOnly": "pnpm run inference-codegen && git diff --name-only --exit-code src && pnpm run build",
-		"build": "tsup src/index.ts --format cjs,esm --clean && tsc --emitDeclarationOnly --declaration",
+		"build": "tsup src/index.ts --format cjs,esm --clean && tsc --emitDeclarationOnly --declaration && pnpm run generate-cts",
 		"watch:export": "tsup src/index.ts --format cjs,esm --watch",
 		"watch:types": "tsc --emitDeclarationOnly --declaration --watch",
 		"watch": "npm-run-all --parallel watch:export watch:types",
 		"prepare": "pnpm run build",
 		"check": "tsc",
 		"test": "vitest run",
+		"generate-cts": "tsx scripts/generate-cts.ts",
 		"inference-codegen": "tsx scripts/inference-codegen.ts && prettier --write src/tasks/*/inference.ts",
 		"inference-tgi-import": "tsx scripts/inference-tgi-import.ts && prettier --write src/tasks/text-generation/spec/*.json && prettier --write src/tasks/chat-completion/spec/*.json",
 		"inference-tei-import": "tsx scripts/inference-tei-import.ts && prettier --write src/tasks/feature-extraction/spec/*.json"

packages/tasks/scripts/generate-cts.ts

@@ -0,0 +1,19 @@
+// Just copy over the already generated .d.ts and .d.ts.map files to .d.cts and .d.cts.map files
+import { readdirSync, readFileSync, statSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+
+function recursiveCopy(path: string) {
+	for (const item of readdirSync(path)) {
+		if (item.endsWith(".d.ts")) {
+			const content = readFileSync(join(path, item), "utf-8");
+			writeFileSync(join(path, item.replace(".d.ts", ".d.cts")), content.replaceAll(".d.ts.map", ".d.cts.map"));
+		} else if (item.endsWith(".d.ts.map")) {
+			const content = readFileSync(join(path, item), "utf-8");
+			writeFileSync(join(path, item.replace(".d.ts.map", ".d.cts.map")), content.replaceAll(".d.ts", ".d.cts"));
+		} else if (statSync(join(path, item)).isDirectory()) {
+			recursiveCopy(join(path, item));
+		}
+	}
+}
+
+recursiveCopy("dist");

packages/tasks/src/default-widget-inputs.ts

@@ -1,5 +1,5 @@
-import type { WidgetExample } from "./widget-example";
-import type { WidgetType } from "./pipelines";
+import type { WidgetExample } from "./widget-example.js";
+import type { WidgetType } from "./pipelines.js";
 
 type LanguageCode = string;
 

packages/tasks/src/index.ts

@@ -1,7 +1,7 @@
-export { LIBRARY_TASK_MAPPING } from "./library-to-tasks";
-export { MAPPING_DEFAULT_WIDGET } from "./default-widget-inputs";
-export type { TaskData, TaskDemo, TaskDemoEntry, ExampleRepo } from "./tasks";
-export * from "./tasks";
+export { LIBRARY_TASK_MAPPING } from "./library-to-tasks.js";
+export { MAPPING_DEFAULT_WIDGET } from "./default-widget-inputs.js";
+export type { TaskData, TaskDemo, TaskDemoEntry, ExampleRepo } from "./tasks/index.js";
+export * from "./tasks/index.js";
 export {
 	PIPELINE_DATA,
 	PIPELINE_TYPES,
@@ -13,11 +13,15 @@ export {
 	MODALITY_LABELS,
 	SUBTASK_TYPES,
 	PIPELINE_TYPES_SET,
-} from "./pipelines";
-export { ALL_DISPLAY_MODEL_LIBRARY_KEYS, ALL_MODEL_LIBRARY_KEYS, MODEL_LIBRARIES_UI_ELEMENTS } from "./model-libraries";
-export type { LibraryUiElement, ModelLibraryKey } from "./model-libraries";
-export type { ModelData, TransformersInfo } from "./model-data";
-export type { AddedToken, SpecialTokensMap, TokenizerConfig } from "./tokenizer-data";
+} from "./pipelines.js";
+export {
+	ALL_DISPLAY_MODEL_LIBRARY_KEYS,
+	ALL_MODEL_LIBRARY_KEYS,
+	MODEL_LIBRARIES_UI_ELEMENTS,
+} from "./model-libraries.js";
+export type { LibraryUiElement, ModelLibraryKey } from "./model-libraries.js";
+export type { ModelData, TransformersInfo } from "./model-data.js";
+export type { AddedToken, SpecialTokensMap, TokenizerConfig } from "./tokenizer-data.js";
 export type {
 	WidgetExample,
 	WidgetExampleAttribute,
@@ -38,18 +42,18 @@ export type {
 	WidgetExampleOutputLabels,
 	WidgetExampleOutputAnswerScore,
 	WidgetExampleOutputText,
-} from "./widget-example";
-export { SPECIAL_TOKENS_ATTRIBUTES } from "./tokenizer-data";
+} from "./widget-example.js";
+export { SPECIAL_TOKENS_ATTRIBUTES } from "./tokenizer-data.js";
 
-import * as snippets from "./snippets";
-export * from "./gguf";
+import * as snippets from "./snippets/index.js";
+export * from "./gguf.js";
 
 export { snippets };
 
-export { SKUS, DEFAULT_MEMORY_OPTIONS } from "./hardware";
-export type { HardwareSpec, SkuType } from "./hardware";
-export { LOCAL_APPS } from "./local-apps";
-export type { LocalApp, LocalAppKey, LocalAppSnippet } from "./local-apps";
+export { SKUS, DEFAULT_MEMORY_OPTIONS } from "./hardware.js";
+export type { HardwareSpec, SkuType } from "./hardware.js";
+export { LOCAL_APPS } from "./local-apps.js";
+export type { LocalApp, LocalAppKey, LocalAppSnippet } from "./local-apps.js";
 
-export { DATASET_LIBRARIES_UI_ELEMENTS } from "./dataset-libraries";
-export type { DatasetLibraryUiElement, DatasetLibraryKey } from "./dataset-libraries";
+export { DATASET_LIBRARIES_UI_ELEMENTS } from "./dataset-libraries.js";
+export type { DatasetLibraryUiElement, DatasetLibraryKey } from "./dataset-libraries.js";

packages/tasks/src/library-to-tasks.ts

@@ -1,5 +1,5 @@
-import type { ModelLibraryKey } from "./model-libraries";
-import type { PipelineType } from "./pipelines";
+import type { ModelLibraryKey } from "./model-libraries.js";
+import type { PipelineType } from "./pipelines.js";
 
 /**
  * Mapping from library name to its supported tasks.

packages/tasks/src/local-apps.ts

@@ -1,6 +1,6 @@
-import { parseGGUFQuantLabel } from "./gguf";
-import type { ModelData } from "./model-data";
-import type { PipelineType } from "./pipelines";
+import { parseGGUFQuantLabel } from "./gguf.js";
+import type { ModelData } from "./model-data.js";
+import type { PipelineType } from "./pipelines.js";
 
 export interface LocalAppSnippet {
 	/**

packages/tasks/src/model-data.ts

@@ -1,6 +1,6 @@
-import type { PipelineType } from "./pipelines";
-import type { WidgetExample } from "./widget-example";
-import type { TokenizerConfig } from "./tokenizer-data";
+import type { PipelineType } from "./pipelines.js";
+import type { WidgetExample } from "./widget-example.js";
+import type { TokenizerConfig } from "./tokenizer-data.js";
 
 /**
  * Public interface for model metadata

packages/tasks/src/model-libraries-snippets.ts

@@ -1,6 +1,6 @@
-import type { ModelData } from "./model-data";
-import type { WidgetExampleTextInput, WidgetExampleSentenceSimilarityInput } from "./widget-example";
-import { LIBRARY_TASK_MAPPING } from "./library-to-tasks";
+import type { ModelData } from "./model-data.js";
+import type { WidgetExampleTextInput, WidgetExampleSentenceSimilarityInput } from "./widget-example.js";
+import { LIBRARY_TASK_MAPPING } from "./library-to-tasks.js";
 
 const TAG_CUSTOM_CODE = "custom_code";
 

packages/tasks/src/model-libraries.ts

@@ -1,6 +1,6 @@
-import * as snippets from "./model-libraries-snippets";
-import type { ModelData } from "./model-data";
-import type { ElasticSearchQuery } from "./model-libraries-downloads";
+import * as snippets from "./model-libraries-snippets.js";
+import type { ModelData } from "./model-data.js";
+import type { ElasticSearchQuery } from "./model-libraries-downloads.js";
 
 /**
  * Elements configurable by a model library.

packages/tasks/src/snippets/common.ts

@@ -1,4 +1,4 @@
-import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks";
+import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks/index.js";
 
 export function stringifyMessages(
 	messages: ChatCompletionInputMessage[],

packages/tasks/src/snippets/curl.spec.ts

@@ -1,6 +1,6 @@
-import type { ModelDataMinimal } from "./types";
+import type { ModelDataMinimal } from "./types.js";
 import { describe, expect, it } from "vitest";
-import { getCurlInferenceSnippet } from "./curl";
+import { getCurlInferenceSnippet } from "./curl.js";
 
 describe("inference API snippets", () => {
 	it("conversational llm", async () => {

packages/tasks/src/snippets/index.ts

@@ -1,6 +1,7 @@
-import * as inputs from "./inputs";
-import * as curl from "./curl";
-import * as python from "./python";
-import * as js from "./js";
+import * as inputs from "./inputs.js";
+import * as curl from "./curl.js";
+import * as python from "./python.js";
+import * as js from "./js.js";
+export * from "./types.js";
 
 export { inputs, curl, python, js };

packages/tasks/src/snippets/inputs.ts

@@ -1,6 +1,6 @@
-import type { PipelineType } from "../pipelines";
-import type { ChatCompletionInputMessage } from "../tasks";
-import type { ModelDataMinimal } from "./types";
+import type { PipelineType } from "../pipelines.js";
+import type { ChatCompletionInputMessage } from "../tasks/index.js";
+import type { ModelDataMinimal } from "./types.js";
 
 const inputsZeroShotClassification = () =>
 	`"Hi, I recently bought a device from your company but it is not working as advertised and I would like to get reimbursed!"`;

packages/tasks/src/snippets/js.spec.ts

@@ -1,6 +1,6 @@
-import type { InferenceSnippet, ModelDataMinimal } from "./types";
+import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
 import { describe, expect, it } from "vitest";
-import { getJsInferenceSnippet } from "./js";
+import { getJsInferenceSnippet } from "./js.js";
 
 describe("inference API snippets", () => {
 	it("conversational llm", async () => {

packages/tasks/src/snippets/python.spec.ts

@@ -1,6 +1,6 @@
-import type { InferenceSnippet, ModelDataMinimal } from "./types";
+import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
 import { describe, expect, it } from "vitest";
-import { getPythonInferenceSnippet } from "./python";
+import { getPythonInferenceSnippet } from "./python.js";
 
 describe("inference API snippets", () => {
 	it("conversational llm", async () => {

packages/tasks/src/snippets/types.ts

@@ -1,4 +1,4 @@
-import type { ModelData } from "../model-data";
+import type { ModelData } from "../model-data.js";
 
 /**
  * Minimal model data required for snippets.

packages/tasks/src/tasks/audio-classification/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/audio-to-audio/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/automatic-speech-recognition/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/depth-estimation/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/document-question-answering/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/feature-extraction/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/fill-mask/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/image-classification/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/image-feature-extraction/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/image-segmentation/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/image-text-to-text/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/image-to-3d/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/image-to-image/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [

packages/tasks/src/tasks/image-to-text/data.ts

@@ -1,4 +1,4 @@
-import type { TaskDataCustom } from "..";
+import type { TaskDataCustom } from "../index.js";
 
 const taskData: TaskDataCustom = {
 	datasets: [