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

fix: add tool exec env vars to create agent return (#3898)

Browse Source
This commit is contained in:
cthomas
2025-08-13 14:58:58 -07:00
committed by GitHub
parent c72eced19e
commit e967f6ed6b
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
letta/services/agent_manager.py
View File

@@ -630,6 +630,7 @@ class AgentManager:
[{"agent_id": aid, "identity_id": iid} for iid in identity_ids],
)
env_rows = []
if agent_create.tool_exec_environment_variables:
env_rows = [
{
@@ -640,7 +641,8 @@ class AgentManager:
}
for key, val in agent_create.tool_exec_environment_variables.items()
]
await session.execute(insert(AgentEnvironmentVariable).values(env_rows))
result = await session.execute(insert(AgentEnvironmentVariable).values(env_rows).returning(AgentEnvironmentVariable.id))
env_rows = [{**row, "id": env_var_id} for row, env_var_id in zip(env_rows, result.scalars().all())]
include_relationships = []
if tool_ids:
@@ -656,6 +658,9 @@ class AgentManager:
result = await new_agent.to_pydantic_async(include_relationships=include_relationships)
if agent_create.tool_exec_environment_variables and env_rows:
result.tool_exec_environment_variables = [AgentEnvironmentVariable(**row) for row in env_rows]
# initial message sequence (skip if _init_with_no_messages is True)
if not _init_with_no_messages:
init_messages = await self._generate_initial_message_sequence_async(