Post-training can replace prompt-time schema injection for enterprise coding agents

CRAFT uses supervised fine-tuning and reinforcement learning to learn schema knowledge and tool use, reducing token costs and improving consistency.

PaperBig Techcs.AIarXiv:2607.22642v1
Aakash Kolekar · Sahika Genc · Shahriar Shariat · Bunyamin Sisman · Tibor Mezi · Barbara Poblete · +7 more

Amazon Advertising Foundations · Amazon Web Services Agentic AI

Research Digest··2 min read
The authors present CRAFT, a two-stage post-training recipe that enables enterprise coding agents to learn schema and tool-use behavior without exhaustive prompt-time schema injection. They show that CRAFT improves composite Agent Score by +9.6 percentage points, reduces input tokens by ~9x, and cuts schema-discovery loops by up to 5x compared to a schema-stuffed baseline.

What they did

CRAFT consists of two stages: first, schema-stripped PLAN supervised fine-tuning learns domain-structured plans and executable behaviors from validated trajectories that deliberately omit schema context; second, execution-shaped reinforcement learning aligns the policy for tool selection, code quality, plan-code consistency, and self-recovery from failed executions. Training data is curated via a Tri-Gate filter that combines execution validation, data-integrity checks, and an LLM-judge reasoning audit. The method is evaluated in an enterprise advertising analytics environment covering 25 schema-linked core entities and 30 agentic workflows, with beta APIs as the agent-facing tool surface.

Key findings

  • CRAFT improves composite Agent Score by +9.6 percentage points over a schema-stuffed baseline.
  • Consistency improves by +4.1 pp, multi-turn coherence by +4.2 pp.
  • Input-token burden is reduced by approximately 9×.
  • Schema-discovery loops (re-reading documentation) are reduced by up to 5×.

Why it matters

This work shows that enterprise coding agents can internalize complex schema knowledge and tool-use behavior through post-training, greatly reducing reliance on prompt engineering and long context windows. The approach addresses a key production bottleneck: high inference cost, brittle schema updates, and inconsistent multi-turn analysis in deployed analytic agents.

Caveats

The evaluation is limited to a single advertising analytics domain with a fixed set of 30 workflows and 25 entities. The training pipeline requires curated multi-turn trajectories with execution validation, which may not be straightforward to obtain in other domains. The paper also acknowledges reward-shaping limitations and infrastructure challenges for scaling multi-turn reinforcement learning in enterprise settings.

§

Analysis

This paper tackles a practical tension in agent deployment: whether to stuff context into prompts or to train the model on the task. By showing that post-training can internalize schema knowledge, the authors offer a path toward cheaper and more reliable enterprise agents. The results are credible and well-measured, though the domain-specific evaluation leaves open questions about generalizability to other API surfaces.

newspaper

Research Digest

Articles published under the Zotpaper byline are synthesized from multiple source publications by our AI editor and reviewed by our editorial process. Each story combines reporting from credible outlets to give readers a balanced, comprehensive view.