Correct ServiceCall into SupportsResponse

mdeweerd · Jul 23, 2023 · f3acbd7 · f3acbd7
1 parent ba0c79d
commit f3acbd7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/zha_toolkit/__init__.py b/custom_components/zha_toolkit/__init__.py
@@ -644,7 +644,7 @@ def register_services(hass):  # noqa: C901
     is_response_data_supported = u.is_ha_ge("2023.7.0")
 
     if is_response_data_supported:
-        from home_assistant.core import ServiceCall
+        from homeassistant.core import ServiceCall, SupportsResponse
 
     async def toolkit_service(service):
         """Run command from toolkit module."""
@@ -811,7 +811,7 @@ async def toolkit_service(service):
                 key,
                 toolkit_service,
                 schema=value,
-                supports_response=ServiceCall.OPTIONAL,  # type:ignore[undefined-variable]
+                supports_response=SupportsResponse.OPTIONAL,  # type:ignore[undefined-variable]
             )
         else:
             hass.services.async_register(