```diff
--- test_files/017-original.txt	2025-03-07 19:06:13
+++ test_files/017-modified.txt	2025-03-07 19:06:13
@@ -1,12 +1,14 @@
+from composio import Action, App, Tag, Trigger, WorkspaceType, action
+
 from composio_langchain.toolset import ComposioToolSet
 
-from composio.client.enums import Action, App, Tag, Trigger
 
-
 __all__ = (
     "Action",
     "App",
     "Tag",
     "Trigger",
+    "WorkspaceType",
+    "action",
     "ComposioToolSet",
 )
```
