diff --git a/homeassistant/helpers/device_registry.py b/homeassistant/helpers/device_registry.py
index b22b1740a4f..f3051e9dfd7 100644
--- a/homeassistant/helpers/device_registry.py
+++ b/homeassistant/helpers/device_registry.py
@@ -332,6 +332,7 @@ class DeviceRegistry:
         new_identifiers: set[tuple[str, str]] | UndefinedType = UNDEFINED,
         sw_version: str | None | UndefinedType = UNDEFINED,
         via_device_id: str | None | UndefinedType = UNDEFINED,
+        add_config_entry_id: str | UndefinedType = UNDEFINED,
         remove_config_entry_id: str | UndefinedType = UNDEFINED,
         disabled_by: str | None | UndefinedType = UNDEFINED,
         suggested_area: str | None | UndefinedType = UNDEFINED,
@@ -347,6 +348,7 @@ class DeviceRegistry:
             new_identifiers=new_identifiers,
             sw_version=sw_version,
             via_device_id=via_device_id,
+            add_config_entry_id=add_config_entry_id,
             remove_config_entry_id=remove_config_entry_id,
             disabled_by=disabled_by,
             suggested_area=suggested_area,
diff --git a/homeassistant/helpers/entity_registry.py b/homeassistant/helpers/entity_registry.py
index 69415030a87..5d1c495904b 100644
--- a/homeassistant/helpers/entity_registry.py
+++ b/homeassistant/helpers/entity_registry.py
@@ -381,6 +381,7 @@ class EntityRegistry:
         *,
         name: str | None | UndefinedType = UNDEFINED,
         icon: str | None | UndefinedType = UNDEFINED,
+        config_entry_id: str | None | UndefinedType = UNDEFINED,
         area_id: str | None | UndefinedType = UNDEFINED,
         new_entity_id: str | UndefinedType = UNDEFINED,
         new_unique_id: str | UndefinedType = UNDEFINED,
@@ -391,6 +392,7 @@ class EntityRegistry:
             entity_id,
             name=name,
             icon=icon,
+            config_entry_id=config_entry_id,
             area_id=area_id,
             new_entity_id=new_entity_id,
             new_unique_id=new_unique_id,