diff --git a/packages/core/vite.config.js b/packages/core/vite.config.js
index 9386a16..5122f40 100644
--- a/packages/core/vite.config.js
+++ b/packages/core/vite.config.js
@@ -12,6 +12,8 @@ export default defineConfig({
 	clearScreen: false,
 	plugins: [
 		dts({
+			include: ['src/**/*.ts'],
+			exclude: ['src/**/*.test.ts'],
 			bundleTypes: true,
 			compilerOptions: {
 				composite: true,
diff --git a/packages/llms/vite.config.js b/packages/llms/vite.config.js
index 8512cc5..c126e3c 100644
--- a/packages/llms/vite.config.js
+++ b/packages/llms/vite.config.js
@@ -13,6 +13,8 @@ export default defineConfig({
 	clearScreen: false,
 	plugins: [
 		dts({
+			include: ['src/**/*.ts'],
+			exclude: ['src/**/*.test.ts'],
 			bundleTypes: true,
 			compilerOptions: {
 				composite: true,
diff --git a/packages/page-agent/vite.config.js b/packages/page-agent/vite.config.js
index 016ddd8..2fa48ff 100644
--- a/packages/page-agent/vite.config.js
+++ b/packages/page-agent/vite.config.js
@@ -12,6 +12,8 @@ export default defineConfig({
 	clearScreen: false,
 	plugins: [
 		dts({
+			include: ['src/**/*.ts'],
+			exclude: ['src/**/*.test.ts'],
 			bundleTypes: true,
 			compilerOptions: {
 				composite: true,
diff --git a/packages/page-controller/vite.config.js b/packages/page-controller/vite.config.js
index aad81b8..f4fa6d1 100644
--- a/packages/page-controller/vite.config.js
+++ b/packages/page-controller/vite.config.js
@@ -14,6 +14,8 @@ export default defineConfig({
 	clearScreen: false,
 	plugins: [
 		dts({
+			include: ['src/**/*.ts'],
+			exclude: ['src/**/*.test.ts'],
 			bundleTypes: true,
 			compilerOptions: {
 				composite: true,
diff --git a/packages/ui/vite.config.js b/packages/ui/vite.config.js
index 8304901..7edc3d8 100644
--- a/packages/ui/vite.config.js
+++ b/packages/ui/vite.config.js
@@ -14,6 +14,8 @@ export default defineConfig({
 	clearScreen: false,
 	plugins: [
 		dts({
+			include: ['src/**/*.ts'],
+			exclude: ['src/**/*.test.ts'],
 			bundleTypes: true,
 			compilerOptions: {
 				composite: true,