use annotated sandbox
This commit is contained in:
@@ -17,7 +17,8 @@ from lang_agent.graphs.graph_states import State
|
|||||||
from lang_agent.config import LLMNodeConfig
|
from lang_agent.config import LLMNodeConfig
|
||||||
from lang_agent.base import GraphBase
|
from lang_agent.base import GraphBase
|
||||||
|
|
||||||
from lang_agent.fs_bkends import StateBk, StateBkConfig, LocalShell, LocalShellConfig, DaytonaSandboxBk, DaytonaSandboxConfig
|
# from lang_agent.fs_bkends import StateBk, StateBkConfig, LocalShell, LocalShellConfig, DaytonaSandboxBk, DaytonaSandboxConfig
|
||||||
|
from lang_agent.fs_bkends import BaseFilesystemBackend, StateBkConfig, AnnotatedStateBk
|
||||||
|
|
||||||
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
@tyro.conf.configure(tyro.conf.SuppressFixed)
|
||||||
@dataclass
|
@dataclass
|
||||||
@@ -31,7 +32,7 @@ class DeepAgentConfig(LLMNodeConfig):
|
|||||||
|
|
||||||
# file_backend_config: StateBkConfig = field(default_factory=StateBkConfig)
|
# file_backend_config: StateBkConfig = field(default_factory=StateBkConfig)
|
||||||
# file_backend_config: LocalShellConfig = field(default_factory=LocalShellConfig)
|
# file_backend_config: LocalShellConfig = field(default_factory=LocalShellConfig)
|
||||||
file_backend_config: DaytonaSandboxConfig = field(default_factory=DaytonaSandboxConfig)
|
file_backend_config: AnnotatedStateBk = field(default_factory=StateBkConfig)
|
||||||
|
|
||||||
def __post_init__(self):
|
def __post_init__(self):
|
||||||
super().__post_init__()
|
super().__post_init__()
|
||||||
@@ -50,7 +51,7 @@ class DeepAgent(GraphBase):
|
|||||||
tags=["main_llm"])
|
tags=["main_llm"])
|
||||||
|
|
||||||
self.tool_man: ToolManager = self.config.tool_manager_config.setup()
|
self.tool_man: ToolManager = self.config.tool_manager_config.setup()
|
||||||
self.file_backend: StateBk = self.config.file_backend_config.setup()
|
self.file_backend: BaseFilesystemBackend = self.config.file_backend_config.setup()
|
||||||
bkend_agent_params = self.file_backend.get_deepagent_params()
|
bkend_agent_params = self.file_backend.get_deepagent_params()
|
||||||
|
|
||||||
self.mem = MemorySaver()
|
self.mem = MemorySaver()
|
||||||
|
|||||||
Reference in New Issue
Block a user