diff --git a/packages/codegen/package.json b/packages/codegen/package.json
index 18afcb8c..1decf3a7 100644
--- a/packages/codegen/package.json
+++ b/packages/codegen/package.json
@@ -19,6 +19,11 @@
       "types": "./dist/index.d.ts",
       "require": "./dist/index.cjs",
       "default": "./dist/index.js"
+    },
+    "./generate": {
+      "types": "./dist/generate.d.ts",
+      "require": "./dist/generate.cjs",
+      "default": "./dist/generate.js"
     }
   },
   "scripts": {
diff --git a/packages/codegen/vite.config.ts b/packages/codegen/vite.config.ts
index f154e6ee..220ff529 100644
--- a/packages/codegen/vite.config.ts
+++ b/packages/codegen/vite.config.ts
@@ -20,7 +20,7 @@ export default defineConfig(({ mode }) => ({
     outDir: "dist",
     target: mode === "esm" ? "esnext" : "es2015",
     lib: {
-      entry: ["src/index.ts", "src/cli.ts"],
+      entry: ["src/index.ts", "src/cli.ts", "src/generate.ts"],
       formats: [mode === "esm" ? "es" : "cjs"],
     },
     rollupOptions: {