langfun 0.1.2.dev202504180804__tar.gz → 0.1.2.dev202504200803__tar.gz

{langfun-0.1.2.dev202504180804 → langfun-0.1.2.dev202504200803}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202504180804
+Version: 0.1.2.dev202504200803
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202504180804 → langfun-0.1.2.dev202504200803}/langfun/core/agentic/action.py

@@ -199,6 +199,11 @@ class ExecutionTrace(pg.Object, pg.views.html.HtmlTreeView.Extension):
     """Returns all queries from current trace and its child execution items."""
     return self._all_child_items(lf_structured.QueryInvocation)
 
+  @property
+  def all_actions(self) -> list['ActionInvocation']:
+    """Returns all actions from current trace and its child execution items."""
+    return self._all_child_items(ActionInvocation)
+
   @property
   def all_logs(self) -> list[lf.logging.LogEntry]:
     """Returns all logs from current trace and its child execution items."""
@@ -221,7 +226,7 @@ class ExecutionTrace(pg.Object, pg.views.html.HtmlTreeView.Extension):
     for item in self.items:
       if isinstance(item, item_cls):
         child_items.append(item)
-      elif isinstance(item, ActionInvocation):
+      if isinstance(item, ActionInvocation):
         child_items.extend(item.execution._all_child_items(item_cls))  # pylint: disable=protected-access
       elif isinstance(item, ExecutionTrace):
         child_items.extend(item._all_child_items(item_cls))  # pylint: disable=protected-access
@@ -582,6 +587,11 @@ class ActionInvocation(pg.Object, pg.views.html.HtmlTreeView.Extension):
     """Returns all queries made by the action and its child execution items."""
     return self.execution.all_queries
 
+  @property
+  def all_actions(self) -> list['ActionInvocation']:
+    """Returns all actions made by the action and its child execution items."""
+    return self.execution.all_actions
+
   @property
   def all_logs(self) -> list[lf.logging.LogEntry]:
     """Returns all logs made by the action and its child execution items."""

{langfun-0.1.2.dev202504180804 → langfun-0.1.2.dev202504200803}/langfun/core/agentic/action_test.py

@@ -95,6 +95,8 @@ class SessionTest(unittest.TestCase):
     self.assertEqual(len(list(root.logs)), 0)
     # 1 query from Bar, 2 from Foo and 3 from parallel executions.
     self.assertEqual(len(list(root.all_queries)), 6)
+    # 2 actions: Foo and Bar.
+    self.assertEqual(len(list(root.all_actions)), 2)
     # 1 log from Bar and 1 from Foo.
     self.assertEqual(len(list(root.all_logs)), 2)
     self.assertEqual(root.usage_summary.total.num_requests, 6)

{langfun-0.1.2.dev202504180804 → langfun-0.1.2.dev202504200803}/langfun.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202504180804
+Version: 0.1.2.dev202504200803
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors