From 881727ad272b95b30f5aa5db1e1e146790f8ebaf Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Fri, 11 Jul 2025 13:53:52 +0000
Subject: [PATCH] Show AI Task pref but disabled if not done loading

---
 src/panels/config/core/ai-task-pref.ts             | 14 ++++++++------
 .../config/core/ha-config-section-general.ts       | 11 ++++-------
 2 files changed, 12 insertions(+), 13 deletions(-)

diff --git a/src/panels/config/core/ai-task-pref.ts b/src/panels/config/core/ai-task-pref.ts
index 2547f8a4ad..ec99cfa907 100644
--- a/src/panels/config/core/ai-task-pref.ts
+++ b/src/panels/config/core/ai-task-pref.ts
@@ -1,6 +1,6 @@
 import "@material/mwc-button";
 import { mdiHelpCircle, mdiStarFourPoints } from "@mdi/js";
-import { css, html, LitElement, nothing } from "lit";
+import { css, html, LitElement } from "lit";
 import { customElement, property, state } from "lit/decorators";
 import "../../../components/ha-card";
 import "../../../components/ha-settings-row";
@@ -14,6 +14,7 @@ import {
   type AITaskPreferences,
 } from "../../../data/ai_task";
 import { documentationUrl } from "../../../util/documentation-url";
+import { isComponentLoaded } from "../../../common/config/is_component_loaded";
 
 @customElement("ai-task-pref")
 export class AITaskPref extends LitElement {
@@ -25,16 +26,15 @@ export class AITaskPref extends LitElement {
 
   protected firstUpdated(changedProps) {
     super.firstUpdated(changedProps);
+    if (!this.hass || !isComponentLoaded(this.hass, "ai_task")) {
+      return;
+    }
     fetchAITaskPreferences(this.hass).then((prefs) => {
       this._prefs = prefs;
     });
   }
 
   protected render() {
-    if (!this._prefs) {
-      return nothing;
-    }
-
     return html`
       <ha-card outlined>
         <h1 class="card-header">
@@ -84,7 +84,9 @@ export class AITaskPref extends LitElement {
             <ha-entity-picker
               data-name="gen_data_entity_id"
               .hass=${this.hass}
-              .value=${this._prefs.gen_data_entity_id}
+              .disabled=${this._prefs === undefined &&
+              isComponentLoaded(this.hass, "ai_task")}
+              .value=${this._prefs?.gen_data_entity_id}
               .includeDomains=${["ai_task"]}
               @value-changed=${this._handlePrefChange}
             ></ha-entity-picker>
diff --git a/src/panels/config/core/ha-config-section-general.ts b/src/panels/config/core/ha-config-section-general.ts
index 1d014dfd2b..7f0491af5b 100644
--- a/src/panels/config/core/ha-config-section-general.ts
+++ b/src/panels/config/core/ha-config-section-general.ts
@@ -28,7 +28,6 @@ import "../../../layouts/hass-subpage";
 import "./ai-task-pref";
 import { haStyle } from "../../../resources/styles";
 import type { HomeAssistant, ValueChangedEvent } from "../../../types";
-import { isComponentLoaded } from "../../../common/config/is_component_loaded";
 
 @customElement("ha-config-section-general")
 class HaConfigSectionGeneral extends LitElement {
@@ -267,12 +266,10 @@ class HaConfigSectionGeneral extends LitElement {
               </ha-progress-button>
             </div>
           </ha-card>
-          ${isComponentLoaded(this.hass, "ai_task")
-            ? html`<ai-task-pref
-                .hass=${this.hass}
-                .narrow=${this.narrow}
-              ></ai-task-pref>`
-            : nothing}
+          <ai-task-pref
+            .hass=${this.hass}
+            .narrow=${this.narrow}
+          ></ai-task-pref>
         </div>
       </hass-subpage>
     `;