bug fix

lang_agent/fs_bkends/statebk.py

@@ -1,16 +1,14 @@
-from dataclasses import dataclass, field, is_dataclass
-from typing import Type, TypedDict, Literal, Dict, List, Tuple, Optional
+from dataclasses import dataclass, field
+from typing import Type
 import tyro
 import os.path as osp
-from abc import ABC, abstractmethod
 import glob
 from loguru import logger
 
 from deepagents.backends.utils import create_file_data
 from deepagents.backends import StateBackend
 
-from lang_agent.config import InstantiateConfig
-from lang_agent.fs_bkends import BaseFilesystemBackend
+from lang_agent.fs_bkends.base import BaseFilesystemBackend, FilesystemBackendConfig
 
 def read_as_utf8(file_path:str):
     with open(file_path, "r", encoding="utf-8") as f:
@@ -31,7 +29,7 @@ def build_skill_fs_dict(skill_dir:str, virt_path:str="/skills"):
 
 @tyro.conf.configure(tyro.conf.SuppressFixed)
 @dataclass
-class StateBkConfig(InstantiateConfig):
+class StateBkConfig(FilesystemBackendConfig):
     _target:Type = field(default_factory=lambda:StateBk)
 
     skills_dir:str = "./assets/skills"
@@ -40,10 +38,6 @@ class StateBkConfig(InstantiateConfig):
     rt_skills_dir:str = "/skills"
     """path to directory with skills in runtime directory"""
 
-    def __post_init__(self):
-        err_msg = f"{self.skills_dir} does not exist"
-        assert osp.exists(self.skills_dir), err_msg
-
 
 class StateBk(BaseFilesystemBackend):
     def __init__(self, config:StateBkConfig):