```diff
--- test_files/197-original.txt	2025-03-07 19:06:32
+++ test_files/197-modified.txt	2025-03-07 19:06:32
@@ -2,12 +2,14 @@
 Action for adding memory.
 """
 
+# pylint: disable=import-outside-toplevel
+
 import os
 import typing as t
 
 from pydantic import BaseModel
 
-from composio.tools.local.base.action import Action
+from composio.tools.base.local import LocalAction
 
 
 class AddMemoryRequest(BaseModel):
@@ -23,32 +25,20 @@
     message: t.Optional[str] = None
 
 
-class AddMemory(Action):
+class AddMemory(LocalAction[AddMemoryRequest, AddMemoryResponse]):
     """Add memory to zep session."""
 
-    _display_name = "Create A Zep Session"
-    _request_schema = AddMemoryRequest
-    _response_schema = AddMemoryResponse
     _tags = ["memory", "history"]
-    _tool_name = "zeptool"
 
-    def execute(
-        self,
-        request_data: AddMemoryRequest,
-        authorisation_data: dict,
-    ) -> AddMemoryResponse:
-        """Create session."""
-        from zep_cloud.client import Zep  # pylint: disable=import-outside-toplevel
-        from zep_cloud.types import Message  # pylint: disable=import-outside-toplevel
+    def execute(self, request: AddMemoryRequest, metadata: t.Dict) -> AddMemoryResponse:
+        """Add memory to a session."""
 
-        client = Zep(
-            api_key=os.environ.get(
-                "ZEP_API_KEY",
-                authorisation_data.get("api_key"),
-            ),
-        )
+        from zep_cloud.client import Zep
+        from zep_cloud.types import Message
+
+        client = Zep(api_key=os.environ.get("ZEP_API_KEY", metadata.get("api_key")))
         result = client.memory.add(
-            session_id=request_data.session_id,
+            session_id=request.session_id,
             messages=[
                 Message(
                     role=message["role"],
@@ -56,7 +46,7 @@
                     content=message["content"],
                     metadata=message.get("metadata"),
                 )
-                for message in request_data.messages
+                for message in request.messages
             ],
         )
         return AddMemoryResponse(message=result.message)
```
