✨ Fix types in published tasks package

Author: coyotte508

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,18 @@
+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/tsconfig.json

@@ -15,6 +15,6 @@
 		"declaration": true,
 		"declarationMap": true
 	},
-	"include": ["src", "scripts"],
+	"include": ["src"],
 	"exclude": ["dist"]
 }

packages/tasks/tsconfig.local.json

@@ -0,0 +1,4 @@
+{
+	"extends": "./tsconfig.json",
+	"include": ["src", "scripts"]
+}