add fx to get actions for tool use for contexts; add test

Author: jakelorocco

mellea/stdlib/base.py

@@ -161,6 +161,14 @@ def render_for_generation(self) -> list[Component | CBlock] | None:
         """Provides a linear list of context components to use for generation, or None if that is not possible to construct."""
         ...
 
+    @abc.abstractmethod
+    def actions_for_available_tools(self) -> list[Component | CBlock] | None:
+        """Provides a list of actions to extract tools from for use with during generation, or None if that's not possible.
+
+        Can be used to make the available tools differ from the tools of all the actions in the context.
+        """
+        ...
+
     @abc.abstractmethod
     def full_event_log(self) -> list[Component | CBlock]:
         """Provides a list of all events stored in the context."""
@@ -210,6 +218,14 @@ def __init__(self):
         self._ctx = []
         self._log_ctx = []
 
+    def actions_for_available_tools(self) -> list[Component | CBlock] | None:
+        """Provides a list of actions to extract tools from for use with during generation, or None if that's not possible.
+
+        Can be used to make the available tools differ from the tools of all the actions in the context.
+        In most cases, this will just be the same context as `render_for_generation`.
+        """
+        return self.render_for_generation()
+
     def last_output(self):
         """The last output thunk of the context."""
         for c in self._ctx[::-1]:

test/stdlib_basics/test_base.py

@@ -1,3 +1,4 @@
+import pytest
 from mellea.stdlib.base import Component, CBlock
 from mellea.stdlib.base import LinearContext
 
@@ -33,3 +34,21 @@ def test_context():
     ctx.insert(CBlock("b"))
     ctx.insert(CBlock("c"))
     ctx.insert(CBlock("d"))
+
+
+def test_actions_for_available_tools():
+    ctx = LinearContext(window_size=3)
+    ctx.insert(CBlock("a"))
+    ctx.insert(CBlock("b"))
+    for_generation = ctx.render_for_generation()
+    assert for_generation is not None
+
+    actions = ctx.actions_for_available_tools()
+    assert actions is not None
+
+    assert len(for_generation) == len(actions)
+    for i in range(len(actions)):
+        assert actions[i] == for_generation[i]
+
+if __name__ == "__main__":
+    pytest.main([__file__])