Add support for disabling remote enum fetching

python/composio/client/enums/base.py

@@ -2,6 +2,7 @@
 Enum helper base.
 """
 
+import os
 import typing as t
 import warnings
 from pathlib import Path
@@ -25,6 +26,10 @@
 ACTIONS_CACHE = LOCAL_CACHE_DIRECTORY / "actions"
 TRIGGERS_CACHE = LOCAL_CACHE_DIRECTORY / "triggers"
 
+NO_REMOTE_ENUM_FETCHING = (
+    os.environ.get("COMPOSIO_NO_REMOTE_ENUM_FETCHING", "false") == "true"
+)
+
 
 class EnumStringNotFound(ComposioSDKError):
     """Raise when user provides invalid enum string."""
@@ -197,6 +202,16 @@ def _cache_from_local(self) -> t.Optional[EntityType]:
         return None
 
     def _cache_from_remote(self) -> EntityType:
+        if NO_REMOTE_ENUM_FETCHING:
+            raise ComposioSDKError(
+                message=(
+                    f"No metadata found for enum `{self.slug}`, "
+                    "You might be trying to use an app or action "
+                    "that is deprecated, run `composio apps update` "
+                    "and try again"
+                )
+            )
+
         from composio.client import Composio  # pylint: disable=import-outside-toplevel
         from composio.client.endpoints import (  # pylint: disable=import-outside-toplevel
             v2,

python/tests/test_client/test_enum.py

@@ -2,13 +2,16 @@
 Test the auto-generate Enum
 """
 
+from pathlib import Path
 from typing import Dict, List
 from unittest import mock
 
+import pytest
 from pydantic import BaseModel
 
 from composio import action
-from composio.client.enums import Action, App, Tag, Trigger
+from composio.client.enums import Action, App, Tag, Trigger, base
+from composio.exceptions import ComposioSDKError
 from composio.tools.base.local import LocalAction, LocalTool
 
 
@@ -83,6 +86,28 @@ def test_load_remote_trigger(self, _patch) -> None:
         assert enum.slug == Trigger.GITHUB_COMMIT_EVENT.slug
 
 
+class TestDisableRemoteCaching:
+    def setup_method(self) -> None:
+        base.NO_REMOTE_ENUM_FETCHING = True
+
+    def teardown_method(self) -> None:
+        base.NO_REMOTE_ENUM_FETCHING = False
+
+    def test_error(self) -> None:
+        """Test `NO_REMOTE_ENUM_FETCHING` set to True."""
+        enum = Action.GITHUB_META_ROOT
+        enum._path = Path("temp")  # pylint: disable=protected-access
+        with pytest.raises(
+            ComposioSDKError,
+            match=(
+                "No metadata found for enum `GITHUB_META_ROOT`, You might be "
+                "trying to use an app or action that is deprecated, run "
+                "`composio apps update` and try again"
+            ),
+        ):
+            enum.load()
+
+
 def test_tag_enum() -> None:
     """Test `Tag` enum."""
     tag = Tag("ASANA_ALLOCATIONS")

python/tests/test_example.py

@@ -145,7 +145,10 @@
             "values": ["composio_output/CODEINTERPRETER_GET_FILE_CMD_default_", ""],
         },
         "env": {"OPENAI_API_KEY": OPENAI_API_KEY, "COMPOSIO_API_KEY": COMPOSIO_API_KEY},
-        "cwd": EXAMPLES_PATH / "quickstarters" / "sql_agent" / "sql_agent_plotter_crewai",
+        "cwd": EXAMPLES_PATH
+        / "quickstarters"
+        / "sql_agent"
+        / "sql_agent_plotter_crewai",
     },
     "multi_entity_api_key": {
         "plugin": "langchain",