diff --git a/core/discovery/coap/disc_coap.gni b/core/discovery/coap/disc_coap.gni
index 752787402759f1c2efe1f065ccbf990c2ed0a0a1..1c6ed15c622d30247d054e5a28cc4b40e712e683 100644
--- a/core/discovery/coap/disc_coap.gni
+++ b/core/discovery/coap/disc_coap.gni
@@ -17,7 +17,6 @@ if (dsoftbus_feature_disc_coap == true) {
   disc_coap_src = [
     "$dsoftbus_root_path/core/discovery/coap/common/src/disc_coap_parser.c",
     "$dsoftbus_root_path/core/discovery/coap/nstackx_coap/src/disc_coap.c",
-    "$dsoftbus_root_path/core/discovery/coap/nstackx_coap/src/disc_coap_capability.c",
     "$dsoftbus_root_path/core/discovery/coap/nstackx_coap/src/disc_nstackx_adapter.c",
   ]
 } else {
@@ -36,5 +35,14 @@ disc_coap_inc = [
   "$dsoftbus_root_path/core/common/dfx/hisysevent_adapter/include",
   "//third_party/cJSON",
 ]
+if (dsoftbus_feature_ex_kits) {
+  import(
+      "$dsoftbus_root_path/dsoftbus_enhance/core/discovery/coap/coap_enhance.gni")
+
+  disc_coap_src += coap_enhance_src
+  disc_coap_inc += coap_enhance_inc
+} else {
+  disc_coap_src += [ "$dsoftbus_root_path/core/discovery/coap/nstackx_coap/src/disc_coap_capability.c" ]
+}
 disc_coap_deps =
     [ "$dsoftbus_root_path/components/nstackx/nstackx_ctrl:nstackx_ctrl" ]