moved things to base class
This commit is contained in:
@@ -7,6 +7,7 @@ from abc import ABC, abstractmethod
|
|||||||
|
|
||||||
class BaseFilesystemBackend(ABC):
|
class BaseFilesystemBackend(ABC):
|
||||||
backend: Any
|
backend: Any
|
||||||
|
config: Any
|
||||||
|
|
||||||
@abstractmethod
|
@abstractmethod
|
||||||
def _build_backend(self):
|
def _build_backend(self):
|
||||||
@@ -21,4 +22,7 @@ class BaseFilesystemBackend(ABC):
|
|||||||
|
|
||||||
def get_deepagent_params(self):
|
def get_deepagent_params(self):
|
||||||
"""extra params to pass into the creation of deepagents"""
|
"""extra params to pass into the creation of deepagents"""
|
||||||
return {}
|
if hasattr(self.config, "rt_skills_dir"):
|
||||||
|
return {"skills" : [self.config.rt_skills_dir]}
|
||||||
|
else:
|
||||||
|
return {}
|
||||||
@@ -54,17 +54,7 @@ class StateBk(BaseFilesystemBackend):
|
|||||||
def _build_backend(self):
|
def _build_backend(self):
|
||||||
self.skills_dict = build_skill_fs_dict(self.config.skills_dir)
|
self.skills_dict = build_skill_fs_dict(self.config.skills_dir)
|
||||||
self.backend = lambda rt : StateBackend(rt)
|
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):
|
def get_inf_inp(self):
|
||||||
"""get inference input for deepagent"""
|
"""get inference input for deepagent"""
|
||||||
return {"files":self.skills_dict}
|
return {"files":self.skills_dict}
|
||||||
|
|
||||||
def get_deepagent_params(self):
|
|
||||||
return {"skills" : self._get_rt_skill_dir()}
|
|
||||||
Reference in New Issue
Block a user