Search papers, labs, and topics across Lattice.
This paper introduces LedgerAgent, a novel method for managing task states in policy-adherent tool-calling agents, particularly in customer-service applications. By maintaining a separate ledger of observed task states, LedgerAgent enhances the agent's ability to make informed decisions while adhering to domain policies, mitigating common failure modes associated with implicit state management. The results demonstrate significant improvements in performance across multiple domains, particularly in scenarios requiring strict consistency in decision-making.
LedgerAgent's structured state management reduces policy violations and enhances decision-making accuracy in customer-service agents, outperforming traditional prompt-based methods.
Policy-adherent tool-calling agents in customer-service domains must maintain task states across turns while calling tools and obeying domain policies. Task states consist of relevant facts, identifiers, constraints, and conditions observed through user interaction and tool calls. In standard agents, task states are not represented separately. Observations, tool returns, and policy instructions are placed in the prompt, leaving agents to reconstruct the relevant states from the prompt each time they decide what to do next. This design makes state management implicit, creating two common failure modes. An agent may retrieve the right facts but later ground its decision in stale, missing, or incorrect information; and a syntactically valid tool call may still violate a domain policy that depends on the current task state. We introduce LedgerAgent, an inference-time method for tool-calling agents that maintains observed task states in a separate ledger and renders the states into the prompt. The ledger is also used to check state-dependent policy constraints before environment-changing tool calls are executed, blocking policy violations. Across four customer-service domains and a mixed panel of open- and closed-weight models, LedgerAgent improves average passk over a standard prompt-based tool-calling approach, with the largest gains under stricter multi-trial consistency metrics.