AI coding assistants are reworking software program growth, however information engineering presents distinctive challenges: ruled information entry, shared compute environments, and compliance controls which are designed to stay in place. How do you deliver the ability of agentic AI growth right into a ruled information surroundings? With the AWS Toolkit for Visible Studio Code, you possibly can join Kiro, VS Code, or Cursor on to Amazon SageMaker Unified Studio.
If you join your editor to a SageMaker Unified Studio House (a cloud-based compute surroundings inside your challenge), you get AI-assisted growth together with your most well-liked instruments whereas your information governance, challenge permissions, and compute are managed by SageMaker Unified Studio. Moreover, SageMaker Unified Studio robotically generates steering recordsdata (like AGENTS.md) that present your AI assistant with context about your challenge surroundings, so it understands your information and challenge configuration from the primary immediate.
This put up demonstrates the mixing utilizing Kiro. The identical Distant Entry connection works with VS Code and Cursor. The put up begins by displaying what you are able to do with this integration: utilizing pure language to discover and analyze information in a ruled surroundings. We then stroll by the setup so you possibly can strive it your self.
What’s new
With the AWS Toolkit, you possibly can join Kiro, VS Code, and Cursor to your SageMaker House over a safe SSH tunnel. No extra extensions or SSH key administration required. After the connection is established, your IDE has full entry to your House’s file system, compute, and information providers.
Two capabilities make this particularly highly effective for information work:
- Automated AI steering – When connecting Kiro to SageMaker Unified Studio, Kiro generates
AGENTS.mdandsmus-context.mdrecordsdata that present your AI assistant with context about your surroundings, together with challenge configuration, surroundings particulars, and utilities for locating your information catalog and challenge construction. Kiro detects these recordsdata robotically; different editors can use them as context for their very own AI options. - MCP server help – have Kiro uncover and configure itself for the Mannequin Context Protocol servers in your distant SageMaker area ( like
smus_localandaws-dataprocessing) to present your agent direct entry to your AWS Glue Information Catalog, Amazon Athena queries, and SageMaker Unified Studio challenge metadata.
The next diagram reveals how the parts join:
Structure diagram: How the parts join
See it in motion: AI-assisted growth with ruled information
Earlier than strolling by the setup, we clarify what you are able to do with this integration. This walkthrough makes use of Kiro because the editor. With Kiro related to a SageMaker Unified Studio House, MCP servers configured, and steering paperwork in place, we will use pure language to discover information and construct analytics. The AI assistant has all of the context it wants to do that effectively.
Be aware: Agentic AI output is nondeterministic. The precise code, instrument decisions, and responses Kiro produces will differ between classes, even with the identical immediate. The next walkthrough reveals one consultant session. Your expertise will differ within the specifics, however the patterns and capabilities demonstrated listed here are constant.
Step 1: Discover the information
Begin with a easy immediate:
Even with native MCP instruments accessible, Kiro usually prefers the AWS Command Line Interface (AWS CLI) and bash to retrieve data. That is anticipated and usually doesn’t have an effect on the end result. In case you choose MCP instruments for each operation, you possibly can add that choice to a steering doc.
Kiro used the sagemaker_studio SDK to find the catalog:
Then it drilled into the desk schema:
Kiro found the sagemaker_sample_db.churn dataset, a pattern dataset that ships with SageMaker Unified Studio containing 10,000 rows and 21 columns of buyer churn information (state, account size, name minutes, service calls, churn flag, and extra). Discover that we didn’t write any of this code. We requested a query in pure language, and Kiro selected the precise SDK calls, explored the catalog, and surfaced the outcomes.
One other, extra pure approach to get the identical reply is to ask immediately. Prompting “Allow us to pattern the churn desk.” yields the identical catalog paths and schema output, together with extra metrics like row depend and an information pattern, all from a single conversational immediate:

Determine 1 — The sagemaker_sample_db.churn dataset within the catalog

Determine 2 — Churn dataset schema with 21 columns
With the schema and row depend in hand, Kiro sampled the information to spherical out its understanding of the dataset:

Determine 3 — Complete information pattern after Kiro catalog exploration
Step 2: Run analytics with full context
With the information explored, ask Kiro to run an information high quality analysis:
As a result of Kiro had already explored the catalog and sampled the information, it made sensible decisions about the way to run the evaluation. As a substitute of utilizing PySpark for this 10,000-row desk, Kiro used Athena utilizing sqlutils to run the analysis immediately. It produced a radical information high quality report:
- 10,000 rows, 21 columns, zero nulls throughout all columns. Clear on that entrance.
- 5,000 duplicate rows (50 p.c). Important, price investigating earlier than modeling.
- Outliers minimal. Most columns have lower than 1 p.c outlier price by IQR.
- Churn is sort of 50/50 break up (50.04 p.c False, 49.96 p.c True). Unusually balanced, indicating artificial information.
- Clear sign in key options. Churners and non-churners present variations in
day_mins(7.52 vs. 3.52),eve_mins(5.95 vs. 4.11), andvmail_message(175 vs. 278). - State distribution roughly uniform (~2% every),
intl_planandvmail_planclose to 50/50.
The important thing perception here’s what Kiro did not do. It didn’t default to PySpark as a result of the surroundings helps Spark. Having explored the information first, understanding the desk dimension, column varieties, and that churn is a correct Boolean (not a string), Kiro independently selected the precise engine for the workload and produced appropriate analytics on the primary go.
Greatest observe: Discover first, code second
Begin each AI-assisted growth session with information exploration. Ask your AI assistant to find your catalog, pattern your tables, and perceive the schema earlier than asking it to construct something. This single step helps cut back a typical supply of errors in AI-assisted information work: the LLM making assumptions about information it has not seen.
Exploring your information offers the big language mannequin (LLM) the context it must correctly assist together with your challenge. It saves hallucinations and rework, ends in quicker growth time, and reduces token prices.
Able to strive it your self? The next sections stroll by the complete setup: stipulations, connecting your editor to your SageMaker House, configuring MCP servers, and dealing with notebooks.
Conditions
Earlier than you start, be sure you have the next:
- A SageMaker Unified Studio area and challenge with at the very least one challenge that has a compute surroundings provisioned (Tooling or ToolingLight). These ought to come commonplace with each SageMaker challenge besides these provisioned with the SQL & Gen AI blueprints. If that you must arrange SageMaker Unified Studio, see Getting began with Amazon SageMaker Unified Studio.
- A House with Distant Entry enabled. Both a JupyterLab or Code Editor House works. The occasion will need to have at the very least 8 GiB of reminiscence (for instance,
ml.t3.massiveor bigger). The defaultml.t3.medium(4 GiB) can’t allow Distant Entry. You could improve the occasion sort first, then toggle Distant Entry to Enabled within the Configure House dialog. - A VS Code-compatible editor. Kiro, VS Code, Cursor, or one other VS Code-based IDE put in in your native machine. This walkthrough makes use of Kiro, however the Distant Entry connection has been examined with VS Code and Cursor as effectively.
- AWS Toolkit v4.1.0 or later. Kiro ships with the AWS Toolkit pre-installed. For VS Code and Cursor, set up the AWS Toolkit extension and confirm your model is 4.1.0 or later (
Cmd+Shift+Xand seek for “AWS Toolkit”). - AWS credentials. You have to be authenticated within the SageMaker Unified Studio panel of the AWS Toolkit with the identical identification (AWS IAM Identification Middle or AWS Identification and Entry Administration (IAM)) that you just use to entry SageMaker Unified Studio within the browser.
- Community connectivity. Your House will need to have web entry (PublicInternetOnly mode, or digital personal cloud (VPC) with a NAT gateway or HTTP proxy that enables VS Code and Open VSX endpoints).
The next screenshots present the SageMaker Unified Studio portal and the Configure House dialog. Navigate to your challenge, choose your House, and confirm the configuration. Distant Entry is disabled when the occasion has lower than 8 GiB of reminiscence. Choose an occasion with at the very least 8 GiB, equivalent to ml.t3.massive, then allow Distant Entry. It is a one-time configuration per House.

Determine 4 — SMUS challenge Areas overview within the portal

Determine 5 — Configure House dialog displaying occasion sort choice

Determine 6 — Enabling Distant Entry on a House with 8 GiB or extra
Connecting your editor to your SageMaker House
There are two methods to attach: immediately from the SageMaker Unified Studio portal, or out of your native IDE utilizing the AWS Toolkit.
Methodology 1: Join from the SageMaker Unified Studio portal
To launch your IDE immediately from the portal, navigate to your challenge’s Code Areas web page, discover your House, and select Open in to pick your editor (Kiro, VS Code, or Cursor):

Determine 7 — Open in Native IDE from the Code Areas listing
It’s also possible to launch from inside a House’s particulars web page:

Determine 8 — Open in Native IDE from the House particulars web page
Or from throughout the JupyterLab or Code Editor browser surroundings:

Determine 9 — Open in Native IDE from JupyterLab
Your browser will immediate you to permit opening the IDE. Affirm, and the editor launches with an SSH connection to your House already established through the AWS Toolkit. No extra configuration is often required.
Methodology 2: Join out of your IDE through the AWS Toolkit
- Open your editor in your native machine. Then, within the AWS Toolkit panel, select Sign up. Authenticate together with your IAM Identification Middle or IAM credentials, the identical identification you employ to entry SageMaker Unified Studio within the browser. The next screenshots present Kiro, however the steps are the identical in VS Code and Cursor.

Determine 10 — AWS Toolkit button in Kiro
Determine 11 — AWS Toolkit panel expanded

Determine 12 — AWS Toolkit Sign up dialog
- Select your AWS profile. You could have a profile configured within the AWS CLI with the proper account and AWS Area set.
- Within the Toolkit panel, browse your SageMaker Unified Studio domains and initiatives. Choose the challenge that you just wish to work in.

Determine 13 — Shopping SMUS domains and initiatives in Kiro
Vital: The credentials that you just use within the AWS Toolkit should match the identification that you just use within the SageMaker Unified Studio portal. The Toolkit validates that your identification has entry to the House.
AI steering: How SageMaker Unified Studio pre-seeds AI context
The true worth of the characteristic comes from what you don’t have to do. When related to Kiro SageMaker Unified Studio robotically generates steering recordsdata that information your AI assistant with challenge context, so you possibly can give attention to constructing analytics slightly than configuring connections. If you open a SageMaker Unified Studio challenge, SageMaker Unified Studio presents a immediate to create steering recordsdata: an AGENTS.md file that references a newly created smus-context.md. These recordsdata present context about your challenge surroundings, equivalent to challenge configuration, surroundings particulars, and utilities for locating your information catalog and challenge construction. Kiro detects and applies these recordsdata robotically; in different editors, you possibly can reference them as context in your AI options.

Determine 14 — SMUS popup providing to create steering recordsdata

Determine 15 — Generated AGENTS.md and smus-context.md steering recordsdata
With out these steering recordsdata, your AI assistant would wish a number of back-and-forth prompts to find what information you have got and the way to entry it. With them, the assistant understands your challenge from the primary immediate: the way to uncover your databases, how your surroundings is configured, and what instruments can be found. The steering recordsdata additionally assist correctly configure MCP servers, which you arrange within the subsequent part.
Exploring your challenge
After you’re related, the challenge construction expands into Information and Compute sections within the sidebar, as it will within the SageMaker Unified Studio portal.

Determine 16 — Venture Information and Compute sections within the Kiro sidebar
You’ll be able to discover your information catalog and S3 buckets immediately from the sidebar:

Determine 17 — Exploring the information catalog and S3 buckets from the sidebar
It’s also possible to distant right into a suitable House for direct growth. Hover over a House and choose the distant icon on the precise:

Determine 18 — Distant connection icon on a suitable House
After a second, the House opens in a brand new Kiro window:

Determine 19 — House opened in a brand new Kiro window
You could sign up once more, after which belief the authors of the recordsdata within the House:

Determine 20 — Belief authors dialog for the House recordsdata
You’re now related to your House. The Toolkit works on the House the best way it does regionally, besides the sources are scoped to the challenge’s permissions.

Determine 21 — Related to the SMUS House with the Toolkit energetic
Organising MCP servers
Earlier than you should utilize AI-assisted growth successfully, you need to give Kiro entry to your information providers by Mannequin Context Protocol (MCP) servers. MCP servers lengthen the Kiro agent with instruments: the flexibility to question catalogs, run SQL, handle credentials, and extra.
Out of the field, Kiro has no MCP servers configured:

Determine 22 — Kiro MCP servers panel with no servers configured
Immediate Kiro to search out and configure the MCP servers that ship pre-installed in your SageMaker House. Utilizing the steering file context, Kiro situated the servers and generated the configuration. If a server fails to attach, choose the failed entry and Kiro will counsel fixes. You may want extra prompts to get the smus_spark_upgrade server (a pre-installed MCP server for managing Spark session upgrades) working accurately.

Determine 23 — Kiro discovering and configuring SMUS MCP servers

Determine 24 — MCP servers after iterating on configuration fixes
For extra deterministic outcomes, you too can configure the MCP servers manually. Here’s a pattern configuration:
Be aware: Your MCP configuration may differ relying in your SageMaker Unified Studio surroundings. Use the previous configuration as a place to begin and let your editor modify if a server fails to attach.
Subsequent, add the AWS Information Processing MCP server to get catalog data and Athena question capabilities. This isn’t strictly required (Kiro can use Python or AWS CLI for a similar duties), however it offers the agent native instruments for catalog and question operations.

Determine 25 — AWS Information Processing MCP server instruments with Amazon EMR instruments disabled
You’ll be able to listing the instruments that every MCP server supplies. As a result of the AWS Information Processing MCP server consists of instruments for a lot of providers, we suggest disabling instruments that you just don’t want for a given challenge to save lots of mannequin context. For this walkthrough, disable the Amazon EMR instruments to give attention to AWS Glue and Amazon Athena.
Exploring information with notebooks
Kiro helps Jupyter notebooks in your SageMaker House with the identical language and connection selectors that you’d discover in SageMaker JupyterLab or Code Editor. Open the command palette (Cmd+Shift+P) and create a brand new Jupyter pocket book:

Determine 26 — Command palette to create a brand new Jupyter pocket book

Determine 27 — New Jupyter pocket book opened in Kiro with language and connection selectors in a pocket book cell
As in SageMaker JupyterLab, you get language and connection selectors within the backside proper of every cell. Select the connection selector to see your accessible connections:

Determine 28 — SageMaker connection selector
Choose PySpark to fill within the magic instructions in your cell. Write your code (on this case, enter spark and press Shift+Enter) to confirm the session begins:

Determine 29 — PySpark magic command and spark verification code

Determine 30 — Working the PySpark cell
If that is your first time utilizing Jupyter with Kiro, you’re prompted to put in the Jupyter extension. After it’s put in, choose the kernel from Python Environments → Base:

Determine 31 — Jupyter kernel choice immediate

Determine 32 — Deciding on the Python kernel from the Base surroundings
Re-run your cell. After just a few moments, AWS Glue provisions a PySpark session:

Determine 33 — AWS Glue provisioning a PySpark session in a Jupyter pocket book in Kiro
You see outcomes the best way you’ll in JupyterLab within the SageMaker Unified Studio portal:

Determine 34 — PySpark code working in a Jupyter pocket book in Kiro
The pocket book generate button
You’ll discover a Generate button beneath pocket book cells. Let’s take a look at it with a easy immediate:

Determine 35 — Utilizing the Generate button with a pure language immediate

Determine 36 — Generated PySpark code from the immediate
This immediate builder, like different pocket book era options, doesn’t have good context on the encircling cells. You have to be specific about what you need as a result of it received’t learn different code or cells as enter.
Whereas the Kiro pocket book generate button works for easy edits, for severe code era, we suggest that you just use Kiro agent mode. This mode has full challenge and SageMaker context, as demonstrated within the “See it in motion” walkthrough earlier on this put up.
What’s taking place below the hood
If you join your editor to a SageMaker Unified Studio House, the AWS Toolkit extension establishes a safe SSH tunnel between your native IDE and your cloud-based House.
Key particulars:
- SSH tunnel. The connection is managed totally by the AWS Toolkit (v4.1.0+) or VS Code’s built-in SSH extension. No separate Distant SSH extension is required; the aptitude is inbuilt.
- File system entry. Your editor sees the House’s persistent storage at
/house/sagemaker-user/, together with shared challenge recordsdata and notebooks or scripts you create. - SageMaker Unified Studio steering context. The mixing generates
AGENTS.mdandsmus-context.mdrecordsdata that present your AI assistant with context about your challenge surroundings and utilities for understanding your information. That is what makes the assistant efficient from the primary immediate. - MCP server integration. MCP servers like
smus_local(for challenge metadata and surroundings utilities) andaws-dataprocessing(for AWS Glue Information Catalog and Amazon Athena) lengthen your editor’s AI with direct entry to your information providers. Your personal MCP servers might be equally useful right here. - Credential stream. The Toolkit makes use of your present AWS identification (IAM Identification Middle or IAM) to authenticate to the House. No separate SSH keys to handle. The
aws_context_providerinstrument from thesmus_localMCP server handles credential discovery for agent operations.
Greatest practices
To work successfully together with your IDE and SageMaker Unified Studio:
- Discover your information earlier than constructing. Begin each session by asking your AI assistant to find your catalog, pattern your information, and perceive the schema. This single step helps cut back the commonest supply of errors in AI-assisted information work: the LLM making assumptions about information it has not seen. See the “See it in motion” walkthrough earlier on this put up for a concrete instance of the distinction this makes.
- Use the SageMaker Unified Studio steering recordsdata. When prompted to create
AGENTS.mdandsmus-context.md, settle for. These recordsdata are the inspiration that makes all the pieces else work: surroundings context, MCP server configuration, and challenge understanding. With out them, your AI assistant begins from zero on each immediate. Kiro detects these robotically; in different editors, add them as context. - Disable unused MCP instruments. The AWS Information Processing MCP server consists of instruments for AWS Glue, Amazon EMR, Amazon Athena, and extra. Disable the providers that you just’re not utilizing for a given challenge to save lots of mannequin context and cut back noise.
- Be particular in your prompts. The extra element you give your AI (column names, question patterns you like, output codecs), the nearer the primary go might be. “Run information high quality analysis utilizing Athena SQL” will get you higher code than “test my information.”
- All the time take a look at interactively first. Whether or not in notebooks or the terminal, validate code earlier than deploying it. AI brokers can iterate rapidly, however catching points in an interactive session is quicker than debugging a failed AWS Glue job. Athena PySpark and the SageMaker
sqlutilsandsparkutilspackages are nice for this. - Cease your House when idle. Your House runs on compute (the identical occasion varieties as Code Editor and JupyterLab). If idle, the House will terminate after 60 minutes and shut your distant connection. Shut the distant window and reconnect to proceed.
Issues to know
- Pocket book agent mode. For notebook-heavy analytics workflows the place you need agentic AI to generate and run cells immediately, SageMaker Notebooks with Information Agent in SageMaker Unified Studio is the really useful choice at present. Present pocket book help in native editors covers enhancing, working, and producing code in particular person cells.
- MCP setup takes iteration. Configuring MCP servers could require iteration, particularly for servers with advanced authentication. Many AI-enabled editors can self-correct when a server fails. For extra deterministic outcomes, use the previous MCP configuration JSON as a place to begin slightly than relying solely on auto-discovery.
- CLI choice. AI brokers usually choose the AWS CLI and bash even when MCP instruments can be found. This doesn’t have an effect on outcomes, however you possibly can steer your assistant towards MCP instruments utilizing a steering doc in the event you choose consistency.
Safety and governance boundaries
A core advantage of this integration is that your present safety and governance controls stay enforced. Your editor connects to your SageMaker House by a safe SSH tunnel managed by the AWS Toolkit. It doesn’t bypass your group’s entry controls. Information entry is ruled by the identical AWS Lake Formation permissions and IAM Identification Middle authentication that apply whenever you work within the SageMaker Unified Studio portal immediately. Your project-level permissions, database grants, and column-level safety insurance policies apply persistently whether or not a question originates from an AI agent, a pocket book cell, or the SageMaker console. Information entry is ruled by the boundaries you outline in your SageMaker Unified Studio area and challenge configuration.
Clear up
To keep away from ongoing fees from billable sources (SageMaker House compute fees per hour, AWS Glue classes cost per DPU-hour, Amazon Athena queries cost per TB scanned):
- Cease your House – Within the SageMaker Unified Studio portal, navigate to your challenge’s Areas and cease the House you used for this walkthrough.
- Disconnect: Shut the distant connection in your editor (File → Shut Distant Connection).
- Confirm AWS Glue classes are terminated – In case you ran PySpark queries throughout this walkthrough, confirm that the classes are stopped. Within the SageMaker Unified Studio portal, navigate to Information processing and make sure no energetic AWS Glue classes stay. Periods auto-terminate when the House stops, however confirm to keep away from sudden fees.
- Delete demo sources (non-compulsory) – File deletion is everlasting and can’t be undone. Again up any work that you just wish to retain earlier than continuing. In case you created scripts or recordsdata throughout this walkthrough that you just now not want, delete them from
/house/sagemaker-user/. For instance, delete any take a look at notebooks, Python scripts, or generated information recordsdata. The patternsagemaker_sample_db.churndataset is read-only and doesn’t want cleanup.
Conclusion
This put up confirmed what occurs when agentic AI meets ruled information, and walked by the way to set it up your self.
Three key insights emerged from this hands-on expertise:
- SageMaker Unified Studio steering recordsdata remodel the developer expertise. Your AI assistant is project-aware from the primary immediate, understanding your surroundings and accessible information with out handbook setup.
- MCP servers bridge “AI that writes code” with “AI that queries your information”. The
smus_localandaws-dataprocessingservers are important for efficient agentic information work. - The “discover first” sample pays instant dividends. When your AI assistant understands your information earlier than writing code, it makes smarter engine decisions and produces appropriate analytics on the primary go.
This integration brings collectively two capabilities which are stronger collectively: your IDE handles the AI-assisted coding and iteration, whereas SageMaker Unified Studio handles information governance, entry management, and compute administration. You get the productiveness of an agentic AI coding assistant with out compromising on the controls your group requires.
To get began, obtain Kiro, set up VS Code or Cursor, and add the AWS Toolkit for Visible Studio Code (v4.1.0 or later). Then go to the Amazon SageMaker Unified Studio documentation and the AWS Information Processing MCP Server to arrange your first House. For associated studying, see Pace up supply of ML workloads utilizing Code Editor in Amazon SageMaker Unified Studio.
In regards to the authors
