Add get_action_from_memory

Author: lalalune

agentaction/__init__.py

@@ -6,6 +6,7 @@
     get_last_action,
     get_available_actions,
     get_formatted_actions,
+    get_action_from_memory,
     search_actions,
     use_action,
     add_action,
@@ -23,6 +24,7 @@
     "get_last_action",
     "get_available_actions",
     "get_formatted_actions",
+    "get_action_from_memory",
     "search_actions",
     "use_action",
     "add_action",

agentaction/main.py

@@ -124,6 +124,21 @@ def get_available_actions(search_text, n_results=5):
 
     return available_actions
 
+def get_action_from_memory(action_name):
+    """
+    Retrieve an action from memory based on the action's name.
+
+    Args:
+        action_name: The name of the action to retrieve.
+
+    Returns:
+        A dictionary representing the action.
+    """
+    action = get_memories("actions", filter_metadata={"name": action_name}, n_results=1)
+    if len(action) == 0:
+        return None
+    return action[0]
+
 
 def search_actions(search_text, n_results=5):
     """

setup.py

@@ -12,7 +12,7 @@
 
 setup(
     name="agentaction",
-    version="0.1.2",
+    version="0.1.3",
     description="Action chaining and history for agents",
     long_description=readme,  # added this line
     long_description_content_type="text/markdown",  # and this line

test.py

@@ -6,6 +6,7 @@
     get_action_history,
     get_last_action,
     get_available_actions,
+    get_action_from_memory,
     search_actions,
     use_action,
     add_action,
@@ -112,6 +113,14 @@ def test_get_available_actions():
     assert len(available_actions) > 0  # Should be at least one action
     cleanup()  # Cleanup after the test
 
+def get_get_action_from_memory():
+    cleanup()  # Ensure clean state before test
+    test_action = setup_test_action()
+    add_action("test", test_action)
+    
+    memory = get_action_from_memory("test")
+    assert memory is not None
+    cleanup()  # Cleanup after the test
 
 # Define a directory for testing import_actions
 TEST_DIR = "test_actions_dir"