diff --git a/scripts/prepare-extension-package-boundary-artifacts.mjs b/scripts/prepare-extension-package-boundary-artifacts.mjs
index ef7edee9fe5..50d77f4721d 100644
--- a/scripts/prepare-extension-package-boundary-artifacts.mjs
+++ b/scripts/prepare-extension-package-boundary-artifacts.mjs
@@ -12,8 +12,10 @@ const VALID_MODES = new Set(["all", "package-boundary"]);
 const ROOT_DTS_INPUTS = [
   "tsconfig.json",
   "tsconfig.plugin-sdk.dts.json",
-  "src",
-  "packages/memory-host-sdk/src",
+  "src/plugin-sdk",
+  "src/video-generation/dashscope-compatible.ts",
+  "src/video-generation/types.ts",
+  "src/types",
 ];
 const PACKAGE_DTS_INPUTS = [
   "tsconfig.json",
diff --git a/tsconfig.plugin-sdk.dts.json b/tsconfig.plugin-sdk.dts.json
index 6ade3489171..afccdc5f4c4 100644
--- a/tsconfig.plugin-sdk.dts.json
+++ b/tsconfig.plugin-sdk.dts.json
@@ -11,6 +11,11 @@
     "rootDir": ".",
     "tsBuildInfoFile": "dist/plugin-sdk/.tsbuildinfo"
   },
-  "include": ["src/**/*.ts", "src/types/**/*.d.ts", "packages/memory-host-sdk/src/**/*.ts"],
+  "include": [
+    "src/plugin-sdk/**/*.ts",
+    "src/video-generation/dashscope-compatible.ts",
+    "src/video-generation/types.ts",
+    "src/types/**/*.d.ts"
+  ],
   "exclude": ["node_modules", "dist", "src/**/*.test.ts"]
 }