feat: 🎸 plugin-nsfw

meetqy · Feb 24, 2023 · 251c083 · 251c083
1 parent c76764a
commit 251c083
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/packages/plugin-nsfw/lib/index.ts b/packages/plugin-nsfw/lib/index.ts
@@ -6,7 +6,7 @@ import { join } from "path";
 let model: nsfw.NSFWJS;
 
 const getModel = async () => {
-  const file = join(__dirname, "./nsfw_model/model/");
+  const file = join(__dirname, "../nsfw_model/model/");
   if (!model) {
     model = await nsfw.load(`file://${file}`, { size: 299 });
   }

diff --git a/packages/plugin-nsfw/package.json b/packages/plugin-nsfw/package.json
@@ -25,11 +25,11 @@
   },
   "files": [
     "dist",
-    "prisma/schema.prisma"
+    "nsfw_model"
   ],
   "main": "./dist/index.js",
   "module": "./dist/index.js",
-  "types": "./dist/lib/index.d.ts",
+  "types": "./dist/index.d.ts",
   "exports": {
     ".": {
       "import": "./dist/index.js",

diff --git a/packages/plugin-nsfw/vite.config.ts b/packages/plugin-nsfw/vite.config.ts
@@ -1,6 +1,13 @@
 import { defineConfig } from "vite";
+import dts from "vite-plugin-dts";
 
 export default defineConfig({
+  plugins: [
+    dts({
+      // 是否跳过类型诊断
+      skipDiagnostics: true,
+    }),
+  ],
   build: {
     lib: {
       entry: "lib/index.ts",