generate description for cached action (#3603)

This commit is contained in:
Shuchang Zheng
2025-10-02 18:53:08 -07:00
committed by GitHub
parent dd235e6ce4
commit 1cfa23ae38
4 changed files with 110 additions and 5 deletions

View File

@@ -2673,6 +2673,25 @@ class AgentDB:
await session.refresh(new_action)
return Action.model_validate(new_action)
async def update_action_reasoning(
self,
organization_id: str,
action_id: str,
reasoning: str,
) -> Action:
async with self.Session() as session:
action = (
await session.scalars(
select(ActionModel).filter_by(action_id=action_id).filter_by(organization_id=organization_id)
)
).first()
if action:
action.reasoning = reasoning
await session.commit()
await session.refresh(action)
return Action.model_validate(action)
raise NotFoundError(f"Action {action_id}")
async def retrieve_action_plan(self, task: Task) -> list[Action]:
async with self.Session() as session:
subquery = (