diff --git a/lang_agent/fs_bkends/base.py b/lang_agent/fs_bkends/base.py index e7ddec8..0ee0f7f 100644 --- a/lang_agent/fs_bkends/base.py +++ b/lang_agent/fs_bkends/base.py @@ -7,6 +7,7 @@ from abc import ABC, abstractmethod class BaseFilesystemBackend(ABC): backend: Any + config: Any @abstractmethod def _build_backend(self): @@ -21,4 +22,7 @@ class BaseFilesystemBackend(ABC): def get_deepagent_params(self): """extra params to pass into the creation of deepagents""" - return {} \ No newline at end of file + if hasattr(self.config, "rt_skills_dir"): + return {"skills" : [self.config.rt_skills_dir]} + else: + return {} \ No newline at end of file diff --git a/lang_agent/fs_bkends/statebk.py b/lang_agent/fs_bkends/statebk.py index d43d5b2..6176980 100644 --- a/lang_agent/fs_bkends/statebk.py +++ b/lang_agent/fs_bkends/statebk.py @@ -54,17 +54,7 @@ class StateBk(BaseFilesystemBackend): def _build_backend(self): self.skills_dict = build_skill_fs_dict(self.config.skills_dir) self.backend = lambda rt : StateBackend(rt) - - def get_backend(self): - return self.backend - def _get_rt_skill_dir(self)->List[str]: - """get runtime skill dir""" - return [self.config.rt_skills_dir] - def get_inf_inp(self): """get inference input for deepagent""" - return {"files":self.skills_dict} - - def get_deepagent_params(self): - return {"skills" : self._get_rt_skill_dir()} \ No newline at end of file + return {"files":self.skills_dict} \ No newline at end of file