View a markdown version of this page

Getting started with the SageMaker Data Agent for Notebook - Amazon SageMaker Unified Studio

Getting started with the SageMaker Data Agent for Notebook

The SageMaker Data Agent is available directly within your SageMaker AI notebook interface through two interaction modes:

Agent panel

Access comprehensive multi-step workflows by opening the agent panel from the notebook interface. Use this mode for complex analytical tasks requiring planning, data discovery, and multiple operations.

In-line assistance

Generate code directly within notebook cells by using the inline prompt interface. Use this mode for focused tasks like writing specific queries, creating visualizations, or modifying existing code.

Notebook scheduling and runs

Use the agent to create schedules and start background notebook runs from natural language prompts. For example, you can ask the agent to schedule a notebook to run daily or trigger an on-demand background run. For more information about notebook scheduling, see Schedule and automate notebook runs.