How OpenAI Built Its Data Agent

· Source: ByteByteGo · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics, Robotics & Autonomous Systems · Depth: Intermediate, short

Summary

OpenAI's internal data agent, serving thousands of employees daily, employs a surprisingly simple architecture to answer natural language queries by generating and executing SQL. This agent, powered by GPT 5.5 and 13 tools, relies heavily on a "context assembly layer" to navigate over 70,000 database tables. This layer creates comprehensive table descriptions from usage metadata, human annotations, and nightly Codex enrichment, alongside company knowledge and past agent corrections. Beyond context assembly, OpenAI leverages Codex internally for critical tasks like migrating 90,000 tables and 600 petabytes in two months, managing patches for a dozen open-source forks, and assisting support engineers with around 100 fixes daily. Key engineering lessons include prioritizing data foundations, limiting tool complexity, trusting high-quality past queries, guiding goals over specific steps, and setting ambitious timelines.

Key takeaway

For AI Engineers developing internal data agents, prioritize establishing a robust data foundation and a sophisticated context assembly layer over complex multi-agent architectures. Your agent's reliability hinges on high-quality table descriptions and curated knowledge, not an abundance of tools or fine-tuning. Consider leveraging large language models like Codex for automating data migrations, patch releases, and support tasks to achieve ambitious operational timelines.

Key insights

OpenAI's data agent achieves reliability through architectural simplicity and robust context assembly, not complex multi-agent pipelines.

Principles

Method

The agent's core method involves a context assembly layer that retrieves relevant table descriptions (from metadata, human notes, Codex enrichment), company knowledge, and memory to build a comprehensive prompt for the LLM.

In practice

Topics

Best for: AI Architect, CTO, VP of Engineering/Data, AI Engineer, Machine Learning Engineer, MLOps Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by ByteByteGo.