Fixed auto delete for retention_time==0

custom_components/llmvision/__init__.py

@@ -94,7 +94,7 @@ async def async_setup_entry(hass, entry):
     hass.data[DOMAIN][entry_uid] = filtered_entry_data
 
     # check if the entry is the calendar entry (has entry rentention_time)
-    if filtered_entry_data.get(CONF_RETENTION_TIME):
+    if filtered_entry_data.get(CONF_RETENTION_TIME) is not None:
         # forward the calendar entity to the platform
         await hass.config_entries.async_forward_entry_setups(entry, ["calendar"])
 
@@ -109,6 +109,7 @@ async def async_remove_entry(hass, entry):
     if entry_uid in hass.data[DOMAIN]:
         # Remove the entry from hass.data
         _LOGGER.info(f"Removing {entry.title} from hass.data")
+        async_unload_entry(hass, entry)
         hass.data[DOMAIN].pop(entry_uid)
     else:
         _LOGGER.warning(

custom_components/llmvision/calendar.py

@@ -2,7 +2,7 @@
 import uuid
 import os
 import json
-from .const import CONF_RETENTION_TIME
+from .const import DOMAIN, CONF_RETENTION_TIME
 from homeassistant.util import dt as dt_util
 from homeassistant.core import HomeAssistant
 from homeassistant.components.calendar import (
@@ -31,7 +31,8 @@ def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry):
         self._attr_name = config_entry.title
         self._attr_unique_id = config_entry.entry_id
         self._events = []
-        self._retention_time = config_entry.options.get(CONF_RETENTION_TIME, 7)
+        self._retention_time = self.hass.data.get(DOMAIN).get(
+            self._attr_unique_id).get(CONF_RETENTION_TIME)
         self._current_event = None
         self._attr_supported_features = (CalendarEntityFeature.DELETE_EVENT)
         # Path to the JSON file where events are stored
@@ -150,6 +151,7 @@ async def _save_events(self) -> None:
         # Delete events outside of retention time window
         now = datetime.datetime.now()
         cutoff_date = now - datetime.timedelta(days=self._retention_time)
+        _LOGGER.info(f"Deleting events before {cutoff_date}")
 
         events_data = [
             {
@@ -161,7 +163,7 @@ async def _save_events(self) -> None:
                 "location": event.location,
             }
             for event in self._events
-            if dt_util.as_local(self._ensure_datetime(event.end)) >= self._ensure_datetime(cutoff_date)
+            if dt_util.as_local(self._ensure_datetime(event.end)) >= self._ensure_datetime(cutoff_date) or self._retention_time == 0
         ]
 
         def write_to_file():

custom_components/llmvision/config_flow.py

@@ -177,11 +177,12 @@ async def groq(self):
             _LOGGER.error("Could not connect to Groq server.")
             raise ServiceValidationError("handshake_failed")
 
-    async def semantic_index(self):
+    async def semantic_index(self) -> bool:
         # check if semantic_index is already configured
         for uid in self.hass.data[DOMAIN]:
             if 'retention_time' in self.hass.data[DOMAIN][uid]:
                 return False
+        return True
 
     def get_configured_providers(self):
         providers = []