Amazon SageMaker Data Agent is now available in the Amazon SageMaker Unified Studio Query Editor

Amazon SageMaker Data Agent is now available in the Query Editor in Amazon SageMaker Unified Studio, extending beyond notebook experience. With Data Agent in Query Editor, you can generate SQL queries from natural language, debug failed queries, and explore your data through a conversational, interactive experience.

Data Agent brings the same conversational experience available in notebooks to your SQL analytics workflow. You can ask "calculate quarterly revenue growth rate by product category for 2025," and the agent proposes a step-by-step plan for you to review before generating contextually accurate SQL for Amazon Redshift and Amazon Athena. This helps you build analytics queries faster, going from question to executable SQL without writing complex joins and aggregations manually. When a query fails, you can use Fix with AI to analyze the error and get suggested corrections. Data Agent maintains awareness of your connected data sources and schema information, so follow-up questions and modifications build on your previous context.

To get started, navigate to a project in SageMaker Unified Studio, open the Query Editor, and open the agent panel. Data Agent in Query Editor is available in IAM domains in all AWS Regions where Amazon SageMaker Unified Studio is supported. For more information, see SageMaker Data Agent and Generative SQL in the Amazon SageMaker Unified Studio User Guide.