langfun 0.1.2.dev202504180804__py3-none-any.whl → 0.1.2.dev202504190803__py3-none-any.whl

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/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.dist-info → langfun-0.1.2.dev202504190803.dist-info}/METADATA

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

{langfun-0.1.2.dev202504180804.dist-info → langfun-0.1.2.dev202504190803.dist-info}/RECORD

@@ -26,10 +26,10 @@ langfun/core/subscription_test.py,sha256=Y4ZdbZEwm83YNZBxHff0QR4QUa4rdaNXA3_jfIc
 langfun/core/template.py,sha256=jNhYSrbLIn9kZOa03w5QZbyjgfnzJzE_ZrrMvvWY4t4,24929
 langfun/core/template_test.py,sha256=g7x4mgNIAXEEj-4W1D5whGfl5YikLEQoylKPzaeDomk,17069
 langfun/core/agentic/__init__.py,sha256=qR3jlfUO4rhIoYdRDLz-d22YZf3FvU4FW88vsjiGDQQ,1224
-langfun/core/agentic/action.py,sha256=8o8UUPKZO2TdUGWU3QrJ4fmm4JdV-kH7G12JdrrYHZE,32744
+langfun/core/agentic/action.py,sha256=AVc2aSZEvyblG_PRsri4kg5fnPrrnqjHU_sYB9EdR-o,33123
 langfun/core/agentic/action_eval.py,sha256=-ZcWt_eYqnTyOe9HFGyrLJkQRd4iG3hfN4IZ-NYAWwA,4534
 langfun/core/agentic/action_eval_test.py,sha256=tRUkWmOE9p0rpNOq19xAY2oDEnYsEEykjg6sUpAwJk0,2832
-langfun/core/agentic/action_test.py,sha256=e2INg-E7MaRNuN1qaSXLSnrjrsyVY3uSICiZpMuJr8c,5521
+langfun/core/agentic/action_test.py,sha256=-QDP_U8hIZA2IYmaSDIWEvVjtyXMTq3S49onjLrsqJ8,5604
 langfun/core/coding/__init__.py,sha256=5utju_fwEsImaiftx4oXKl9FAM8p281k8-Esdh_-m1w,835
 langfun/core/coding/python/__init__.py,sha256=4ByknuoNU-mOIHwHKnTtmo6oD64oMFtlqPlYWmA5Wic,1736
 langfun/core/coding/python/correction.py,sha256=7zBedlhQKMPA4cfchUMxAOFl6Zl5RqCyllRHGWys40s,7092
@@ -156,8 +156,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
 langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
 langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
 langfun/core/templates/selfplay_test.py,sha256=Ot__1P1M8oJfoTp-M9-PQ6HUXqZKyMwvZ5f7yQ3yfyM,2326
-langfun-0.1.2.dev202504180804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202504180804.dist-info/METADATA,sha256=3olh4UsWwT8_aleiF-94tKmiiUkfTaSry6xnFBw-sQw,7692
-langfun-0.1.2.dev202504180804.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-langfun-0.1.2.dev202504180804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202504180804.dist-info/RECORD,,
+langfun-0.1.2.dev202504190803.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202504190803.dist-info/METADATA,sha256=Y9E3a9r77DSfmn698f83r8ElM9OmeeAcL-WSkUT2sTw,7692
+langfun-0.1.2.dev202504190803.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+langfun-0.1.2.dev202504190803.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202504190803.dist-info/RECORD,,