diff --git a/lang_agent/fs_bkends/__init__.py b/lang_agent/fs_bkends/__init__.py new file mode 100644 index 0000000..988c2fd --- /dev/null +++ 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]] \ No newline at end of file