From 05e8a39c0ab693e3ac76ae822a6f510a3c4ebca1 Mon Sep 17 00:00:00 2001 From: tushar-composio Date: Mon, 16 Dec 2024 01:58:59 -0800 Subject: [PATCH] fix: don't delete non-action enum data if it doesn't have `replaced_by` (#1014) Co-authored-by: Viraj <35092918+angrybayblade@users.noreply.github.com> --- python/composio/client/enums/enum.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/python/composio/client/enums/enum.py b/python/composio/client/enums/enum.py index 3e155ec13e9..ac13f11fc99 100644 --- a/python/composio/client/enums/enum.py +++ b/python/composio/client/enums/enum.py @@ -8,7 +8,7 @@ from composio.exceptions import ComposioSDKError from composio.storage.base import LocalStorage -from .base import EnumStringNotFound, SentinalObject +from .base import ActionData, EnumStringNotFound, SentinalObject DataT = t.TypeVar("DataT", bound=LocalStorage) @@ -119,11 +119,15 @@ def load(self) -> DataT: data = self.storage.load(self.storage_path) # HACK: if 'replaced_by' field is not present, delete this cached file # as it is outdated. - if hasattr(data, "replaced_by"): - self._data = data - return self._data + if isinstance(data, ActionData): + if hasattr(data, "replaced_by"): + self._data = data # type: ignore + return self._data # type: ignore + + self.storage_path.unlink() - self.storage_path.unlink() + self._data = data + return self._data # Try to fetch from runtime runtime_data = self.load_from_runtime()