Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: Sandboxing for tool execution (#2040)

Browse Source 
Co-authored-by: Caren Thomas <carenthomas@Jeffs-MacBook-Pro-2.local>
Co-authored-by: Caren Thomas <carenthomas@jeffs-mbp-2.lan>
Co-authored-by: Caren Thomas <carenthomas@Jeffs-MBP-2.hsd1.ca.comcast.net>
Co-authored-by: Sarah Wooders <sarahwooders@gmail.com>
This commit is contained in:
Matthew Zhou
2024-11-22 10:34:08 -08:00
committed by GitHub
parent 355fe1b4cf
commit 69730988ce
39 changed files with 2845 additions and 864 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
letta/server/server.py
View File

@@ -78,6 +78,7 @@ from letta.schemas.user import User
from letta.services.agents_tags_manager import AgentsTagsManager
from letta.services.block_manager import BlockManager
from letta.services.organization_manager import OrganizationManager
from letta.services.sandbox_config_manager import SandboxConfigManager
from letta.services.source_manager import SourceManager
from letta.services.tool_manager import ToolManager
from letta.services.user_manager import UserManager
@@ -247,6 +248,7 @@ class SyncServer(Server):
self.block_manager = BlockManager()
self.source_manager = SourceManager()
self.agents_tags_manager = AgentsTagsManager()
self.sandbox_config_manager = SandboxConfigManager(tool_settings)
# Make default user and org
if init_with_default_org_and_user:
@@ -381,10 +383,11 @@ class SyncServer(Server):
tool_objs = []
for name in agent_state.tools:
# TODO: This should be a hard failure, but for migration reasons, we patch it for now
try:
tool_obj = self.tool_manager.get_tool_by_name(tool_name=name, actor=actor)
if tool_obj:
tool_obj = self.tool_manager.get_tool_by_name(tool_name=name, actor=actor)
tool_objs.append(tool_obj)
except NoResultFound:
else:
warnings.warn(f"Tried to retrieve a tool with name {name} from the agent_state, but does not exist in tool db.")
# set agent_state tools to only the names of the available tools
@@ -837,10 +840,10 @@ class SyncServer(Server):
tool_objs = []
if request.tools:
for tool_name in request.tools:
try:
tool_obj = self.tool_manager.get_tool_by_name(tool_name=tool_name, actor=actor)
tool_obj = self.tool_manager.get_tool_by_name(tool_name=tool_name, actor=actor)
if tool_obj:
tool_objs.append(tool_obj)
except NoResultFound:
else:
warnings.warn(f"Attempted to add a nonexistent tool {tool_name} to agent {request.name}, skipping.")
# reset the request.tools to only valid tools
request.tools = [t.name for t in tool_objs]