__init__ for fs_backend

2026-02-28 15:49:25 +08:00
parent a832178810
commit 7be4aa1283
1 changed files with 11 additions and 0 deletions
--- a/lang_agent/fs_bkends/init.py
+++ b/lang_agent/fs_bkends/init.py
@@ -0,0 +1,11 @@
+import tyro
+
+from lang_agent.fs_bkends.base import BaseFilesystemBackend
+from lang_agent.fs_bkends.statebk import StateBk, StateBkConfig
+
+statebk_dict = {
+    "statebk": StateBkConfig(),
+}
+
+statebk_union = tyro.extras.subcommand_type_from_defaults(statebk_dict, prefix_names=False)
+AnnotatedStateBk = tyro.conf.OmitSubcommandPrefixes[tyro.conf.SuppressFixed[statebk_union]]