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

feat: only update job metadata when flag is set (#4824)

Browse Source
This commit is contained in:
cthomas
2025-09-19 14:55:01 -07:00
committed by Caren Thomas
parent f760eaf394
commit 5166ab2f86
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
letta/server/rest_api/routers/v1/agents.py
View File

@@ -1494,7 +1494,7 @@ async def cancel_agent_run(
results = {}
for run_id in run_ids:
run = await server.job_manager.get_job_by_id_async(job_id=run_id, actor=actor)
if run.metadata.get("temporal") and settings.temporal_endpoint:
if run.metadata.get("lettuce") and settings.temporal_endpoint:
client = await Client.connect(
settings.temporal_endpoint,
namespace=settings.temporal_namespace,
@@ -1650,6 +1650,7 @@ async def send_message_async(
MetricRegistry().user_message_counter.add(1, get_ctx_attributes())
actor = await server.user_manager.get_actor_or_default_async(actor_id=headers.actor_id)
# Create a new job
use_lettuce = headers.experimental_params.message_async and settings.temporal_endpoint
run = Run(
user_id=actor.id,
status=JobStatus.created,
@@ -1659,7 +1660,7 @@ async def send_message_async(
metadata={
"job_type": "send_message_async",
"agent_id": agent_id,
"temporal": settings.temporal_endpoint != None,
"lettuce": use_lettuce,
},
request_config=LettaRequestConfig(
use_assistant_message=request.use_assistant_message,
@@ -1670,7 +1671,7 @@ async def send_message_async(
)
run = await server.job_manager.create_job_async(pydantic_job=run, actor=actor)
if headers.experimental_params.message_async and settings.temporal_endpoint:
if use_lettuce:
agent_state = await server.agent_manager.get_agent_by_id_async(
agent_id, actor, include_relationships=["memory", "multi_agent_group", "sources", "tool_exec_environment_variables", "tools"]
)

2
letta/server/rest_api/routers/v1/runs.py
View File

@@ -114,7 +114,7 @@ async def retrieve_run(
try:
job = await server.job_manager.get_job_by_id_async(job_id=run_id, actor=actor)
if job.metadata.get("temporal") and settings.temporal_endpoint:
if job.metadata.get("lettuce") and settings.temporal_endpoint:
client = await Client.connect(
settings.temporal_endpoint,
namespace=settings.temporal_namespace,