fix: restore ai task image analysis config

Author: Desmond-Dong

custom_components/ai_hub/__init__.py

@@ -250,6 +250,7 @@ async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
         # Create AI task subentry with defaults
         ai_task_data = {
             CONF_RECOMMENDED: True,
+            CONF_CHAT_URL: AI_HUB_CHAT_URL,
             CONF_CHAT_MODEL: RECOMMENDED_AI_TASK_MODEL,
             CONF_TEMPERATURE: RECOMMENDED_AI_TASK_TEMPERATURE,
             CONF_TOP_P: RECOMMENDED_AI_TASK_TOP_P,

custom_components/ai_hub/ai_task.py

@@ -14,8 +14,10 @@
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from .consts import (
+    AI_HUB_CHAT_URL,
     AI_HUB_IMAGE_GEN_URL,
     CONF_CHAT_MODEL,
+    CONF_CHAT_URL,
     CONF_IMAGE_MODEL,
     CONF_IMAGE_URL,
     ERROR_GETTING_RESPONSE,
@@ -47,6 +49,16 @@ def _get_conversation_model(config_entry: ConfigEntry) -> str:
     return RECOMMENDED_CHAT_MODEL
 
 
+def _get_conversation_chat_url(config_entry: ConfigEntry) -> str:
+    """Get the chat URL from Conversation Agent subentry."""
+    for subentry in config_entry.subentries.values():
+        if subentry.subentry_type == SUBENTRY_CONVERSATION:
+            chat_url = subentry.data.get(CONF_CHAT_URL, AI_HUB_CHAT_URL)
+            if isinstance(chat_url, str) and chat_url.strip():
+                return chat_url
+    return AI_HUB_CHAT_URL
+
+
 async def async_setup_entry(
     hass: HomeAssistant,
     config_entry: ConfigEntry,
@@ -100,6 +112,15 @@ def __init__(
         ):
             self._attr_supported_features |= ai_task.AITaskEntityFeature.GENERATE_IMAGE
 
+    def _get_model_config(self, chat_log: conversation.ChatLog | None = None) -> dict[str, Any]:
+        """AI Task follows Conversation model and chat URL settings."""
+        model_config = super()._get_model_config(chat_log)
+        chat_url = self.subentry.data.get(CONF_CHAT_URL)
+        if not isinstance(chat_url, str) or not chat_url.strip():
+            chat_url = _get_conversation_chat_url(self.entry)
+        model_config[CONF_CHAT_URL] = chat_url
+        return model_config
+
     async def _async_generate_data(
         self,
         task: ai_task.GenDataTask,

custom_components/ai_hub/consts/defaults.py

@@ -129,6 +129,8 @@
 
 RECOMMENDED_AI_TASK_OPTIONS: Final = {
     CONF_RECOMMENDED: True,
+    CONF_CHAT_URL: AI_HUB_CHAT_URL,
+    CONF_CHAT_MODEL: RECOMMENDED_AI_TASK_MODEL,
     CONF_IMAGE_MODEL: RECOMMENDED_IMAGE_MODEL,
     CONF_IMAGE_URL: AI_HUB_IMAGE_GEN_URL,
     CONF_TEMPERATURE: RECOMMENDED_AI_TASK_TEMPERATURE,

custom_components/ai_hub/entity.py

@@ -287,7 +287,7 @@ async def _async_handle_chat_log(
                         part["text"] = str(part.get("text", ""))
                         _LOGGER.warning("Message %d part %d had non-string text, converted", i, j)
 
-        api_url = options.get(CONF_CHAT_URL)
+        api_url = model_config.get(CONF_CHAT_URL, options.get(CONF_CHAT_URL))
         if not isinstance(api_url, str) or not api_url.strip():
             raise HomeAssistantError(
                 translation_domain=DOMAIN,

custom_components/ai_hub/manifest.json

@@ -9,5 +9,5 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/ha-china/ai_hub/issues",
   "requirements": ["edge-tts==7.2.7", "aiofiles", "aiohttp"],
-  "version": "v2026.04.10"
+  "version": "v2026.04.11"
 }

custom_components/ai_hub/services.py

@@ -226,8 +226,8 @@ async def _handle_analyze_image(call: ServiceCall) -> dict:
         SUBENTRY_CONVERSATION,
         (CONF_CHAT_URL, AI_HUB_CHAT_URL),
         (CONF_CHAT_MODEL, RECOMMENDED_CHAT_MODEL),
-        lambda config: (config[2], config[0]),
-        handle_analyze_image,
+        config_mapper=lambda config: (config[2], config[0]),
+        handler=handle_analyze_image,
     )
 
 
@@ -237,8 +237,8 @@ async def _handle_generate_image(call: ServiceCall) -> dict:
         call,
         SUBENTRY_AI_TASK,
         (CONF_IMAGE_URL, AI_HUB_IMAGE_GEN_URL),
-        lambda config: (config[1], config[0]),
-        handle_generate_image,
+        config_mapper=lambda config: (config[1], config[0]),
+        handler=handle_generate_image,
     )
 
 
@@ -248,8 +248,8 @@ async def _handle_stt_transcribe(call: ServiceCall) -> dict:
         call,
         SUBENTRY_STT,
         (CONF_STT_URL, SILICONFLOW_ASR_URL),
-        lambda config: (config[1], config[0]),
-        handle_stt_transcribe,
+        config_mapper=lambda config: (config[1], config[0]),
+        handler=handle_stt_transcribe,
     )