generated from amazon-archives/__template_Apache-2.0
-
Notifications
You must be signed in to change notification settings - Fork 616
Add doc for scratchpad tools #11167
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
natebower
merged 12 commits into
opensearch-project:main
from
mingshl:main-add-scratchpad
Oct 13, 2025
Merged
Add doc for scratchpad tools #11167
Changes from all commits
Commits
Show all changes
12 commits
Select commit
Hold shift + click to select a range
0dfe51d
draft for scratchpad tools
mingshl a224707
add details to scratch pads tools
mingshl 75e15d2
add tool execute example
mingshl 2d26b92
explain register paramters and execute parameters in two tables
mingshl ce212ee
executing parameters is required
mingshl 5740d61
Apply suggestions from code review
natebower f9dc998
Doc review
kolchfa-aws 8279885
Reorganize sections
kolchfa-aws a1ccc52
Small updates
kolchfa-aws 8e07294
Add missing tools to index page
kolchfa-aws d4a4a15
Apply suggestions from code review
natebower d22d513
Merge branch 'main-add-scratchpad' of github.com:mingshl/documentatio…
kolchfa-aws File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
338 changes: 338 additions & 0 deletions
338
_ml-commons-plugin/agents-tools/tools/scratchpad-tools.md
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,338 @@ | ||
--- | ||
layout: default | ||
title: Scratchpad tools | ||
has_children: false | ||
has_toc: false | ||
nav_order: 60 | ||
parent: Tools | ||
grand_parent: Agents and tools | ||
--- | ||
|
||
<!-- vale off --> | ||
# Scratchpad tools | ||
**Introduced 3.3** | ||
{: .label .label-purple } | ||
<!-- vale on --> | ||
|
||
The scratchpad tools consist of `WriteToScratchPadTool` and `ReadFromScratchPadTool`, which enable agents to store and retrieve intermediate thoughts and results during runtime. These tools serve as temporary memory for a single agent execution session, allowing agents to take notes and store important findings during tool executions. | ||
|
||
The scratchpad acts as runtime memory that persists only during a single agent execution. When you call the agent's `_execute` API, a new scratchpad is created for that session. All notes and data, except `persistent_notes`, are cleared when the execution completes, ensuring each execution starts with a fresh scratchpad. | ||
{: .important} | ||
|
||
## Use cases | ||
|
||
- **Task decomposition**: Store research plans, intermediate findings, and progress notes during multi-step operations within a single execution. | ||
- **Temporary state management**: Maintain context and accumulated knowledge during the current agent execution session. | ||
- **Multi-step workflows**: Save key findings after searches to build comprehensive responses in complex tasks. | ||
- **Execution planning**: Store and reference step-by-step plans during complex operations. | ||
|
||
## Scratchpad lifecycle | ||
|
||
The scratchpad follows a simple lifecycle: | ||
|
||
1. **Creation**: A new, empty scratchpad is created when an agent execution begins. | ||
2. **Usage**: During execution, the agent can read from and write to the scratchpad multiple times. | ||
3. **Cleanup**: The scratchpad is automatically cleared when execution completes. | ||
|
||
Each call to the agent's `_execute` API creates a fresh scratchpad, ensuring executions are isolated from each other. | ||
|
||
## Best practices | ||
|
||
- **Structured notes**: Encourage agents to maintain organized, structured notes in the scratchpad. | ||
- **Regular updates**: Have agents update the scratchpad after each significant step or finding. | ||
- **Session awareness**: Remember that scratchpad content is temporary and specific to the current execution. | ||
- **Efficient usage**: Use the scratchpad for intermediate results that need to be referenced multiple times during execution. | ||
|
||
## Example: Building a research agent with scratchpad tools | ||
|
||
Use the following steps to build a research agent with scratchpad tools. | ||
|
||
### Step 1: Register and deploy a model | ||
|
||
Register a conversational model that supports the agent framework. The following example uses Anthropic Claude: | ||
|
||
```json | ||
POST /_plugins/_ml/models/_register?deploy=true | ||
{ | ||
"name": "Claude Sonnet for Research Agent", | ||
"function_name": "remote", | ||
"description": "Claude model for research agent with scratchpad", | ||
"connector": { | ||
"name": "Bedrock Claude Sonnet Connector", | ||
"description": "Amazon Bedrock connector for Claude Sonnet", | ||
"version": 1, | ||
"protocol": "aws_sigv4", | ||
"parameters": { | ||
"region": "us-east-1", | ||
"service_name": "bedrock", | ||
"model": "anthropic.claude-3-5-sonnet-20241022-v2:0" | ||
}, | ||
"credential": { | ||
"access_key": "${AWS_ACCESS_KEY_ID}", | ||
"secret_key": "${AWS_SECRET_ACCESS_KEY}", | ||
"session_token": "${AWS_SESSION_TOKEN}" | ||
}, | ||
"actions": [ | ||
{ | ||
"action_type": "predict", | ||
"method": "POST", | ||
"url": "https://bedrock-runtime.${parameters.region}.amazonaws.com/model/${parameters.model}/converse", | ||
"headers": { | ||
"content-type": "application/json" | ||
}, | ||
"request_body": """{"system": [{"text": "${parameters.system_prompt}"}], "messages": ${parameters.messages}, "inferenceConfig": {"maxTokens": 8000, "temperature": 0}}""" | ||
} | ||
] | ||
} | ||
} | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
### Step 2: Register an agent with scratchpad tools | ||
|
||
Register a conversational agent that includes both scratchpad tools and other research tools: | ||
|
||
```json | ||
POST /_plugins/_ml/agents/_register | ||
{ | ||
"name": "Research Agent with Scratchpad", | ||
"type": "conversational", | ||
"description": "Research assistant with persistent scratchpad memory", | ||
"app_type": "rag", | ||
"llm": { | ||
"model_id": "your-model-id", | ||
"parameters": { | ||
"max_iteration": 50, | ||
"system_prompt": "You are a sophisticated research assistant with access to OpenSearch indices and a persistent scratchpad for note-taking.\n\nYour Research Workflow:\n1. Check Scratchpad: Before starting a new research task, check your scratchpad to see if you have any relevant information already saved\n2. Create Research Plan: Create a structured research plan\n3. Write to Scratchpad: Save the research plan and any important information to your scratchpad\n4. Use Search: Gather information using OpenSearch search queries\n5. Update Scratchpad: After each search, update your scratchpad with new findings\n6. Iterate: Repeat searching and updating until you have comprehensive information\n7. Complete Task: Provide a thorough response based on your accumulated research\n\nRemember: Your scratchpad is temporary memory for this execution session only. Use it to organize your thoughts and findings during this task.", | ||
"prompt": "${parameters.question}" | ||
} | ||
}, | ||
"memory": { | ||
"type": "conversation_index" | ||
}, | ||
"parameters": { | ||
"_llm_interface": "bedrock/converse/claude" | ||
}, | ||
"tools": [ | ||
{ | ||
"type": "SearchIndexTool" | ||
}, | ||
{ | ||
"type": "ListIndexTool" | ||
}, | ||
{ | ||
"type": "IndexMappingTool" | ||
}, | ||
{ | ||
"type": "ReadFromScratchPadTool", | ||
"name": "ReadFromScratchPadTool", | ||
"parameters": { | ||
"persistent_notes": "You are a helpful researcher. Before making searches, use the ListIndexTool to discover available indices. Write down important notes after using tools." | ||
} | ||
}, | ||
{ | ||
"type": "WriteToScratchPadTool", | ||
"name": "WriteToScratchPadTool" | ||
} | ||
] | ||
} | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
### Step 3: Execute the agent | ||
|
||
Execute the agent with a research question: | ||
|
||
```json | ||
POST /_plugins/_ml/agents/<your-agent-id>/_execute?async=true | ||
{ | ||
"parameters": { | ||
"question": "How many residents are in New York?" | ||
} | ||
} | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
|
||
The agent will: | ||
1. Read from its scratchpad to check for existing relevant information (starts empty for new executions). | ||
2. Create and save a research plan to the scratchpad. | ||
3. Execute searches and update the scratchpad with findings. | ||
4. Provide a comprehensive answer based on accumulated research. | ||
|
||
When using the `agents/<your-agent-id>/_execute` API, you will get a `parent_interaction_id` and `memory_id` in the response. Note the `parent_interaction_id` for later tracing steps. For more information, see [Viewing scratchpad activity](#viewing-scratchpad-activity). | ||
|
||
## Tool parameters | ||
|
||
The following are the parameters for the scratchpad tools. | ||
|
||
### ReadFromScratchPadTool | ||
|
||
The following are the **registration parameters** used when adding to an agent. | ||
|
||
Parameter | Type | Required/Optional | Description | ||
:--- | :--- | :--- | :--- | ||
`persistent_notes` | String | Optional | Initial notes or instructions to store in the scratchpad when first created. | ||
|
||
The following are the **execution parameters** used when calling the tool directly. | ||
|
||
Parameter | Type | Required/Optional | Description | ||
:--- | :--- |:------------------| :--- | ||
`persistent_notes` | String | Required | Initial notes or instructions to store in the scratchpad. | ||
|
||
### WriteToScratchPadTool | ||
|
||
The following **registration parameters** are used when adding to an agent. | ||
|
||
Parameter | Type | Required/Optional | Description | ||
:--- | :--- | :--- | :--- | ||
`return_history` | Boolean | Optional | When set to `true`, returns the full scratchpad content after writing. When `false` or omitted (default), returns the newly added note with confirmation. | ||
|
||
The following **execution parameters** are used when calling the tool directly. | ||
|
||
Parameter | Type | Required/Optional | Description | ||
:--- | :--- | :--- | :--- | ||
`notes` | String | Required | The content to write to the scratchpad. | ||
`return_history` | Boolean | Optional | When set to `true`, returns the full scratchpad content after writing. When `false` or omitted (default), returns the newly added note with confirmation. | ||
|
||
|
||
## Testing the tools | ||
|
||
You can use the Tools API directly to execute both scratchpad tools and test their responses before registering them with your agents. | ||
|
||
### Testing the ReadFromScratchPadTool | ||
|
||
```json | ||
POST /_plugins/_ml/tools/_execute/ReadFromScratchPadTool | ||
{ | ||
"parameters": { | ||
"persistent_notes": "You are a helpful researcher to conduct searches in OpenSearch cluster. Before making the search, please remember to use the listIndexTool to figure out what are the available indices first. When using listIndexTool, remember the index name has to be in an array format. Please write down important notes after tool used." | ||
} | ||
} | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
When provided `persistent_notes`, the tool attempts to show the persistent notes in the response: | ||
|
||
```json | ||
{ | ||
"inference_results": [ | ||
{ | ||
"output": [ | ||
{ | ||
"name": "response", | ||
"result": """Notes from scratchpad: | ||
- You are a helpful researcher to conduct searches in OpenSearch cluster. Before making the search, please remember to use the listIndexTool to figure out what are the available indices first. When using listIndexTool, remember the index name has to be in an array format. Please write down important notes after tool used.""" | ||
} | ||
] | ||
} | ||
] | ||
} | ||
``` | ||
|
||
You can also test with an empty `persistent_notes` field: | ||
|
||
```json | ||
POST /_plugins/_ml/tools/_execute/ReadFromScratchPadTool | ||
{ | ||
"parameters": { | ||
"persistent_notes": "" | ||
} | ||
} | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
The response indicates that the scratchpad is empty: | ||
|
||
```json | ||
{ | ||
"inference_results": [ | ||
{ | ||
"output": [ | ||
{ | ||
"name": "response", | ||
"result": "Scratchpad is empty." | ||
} | ||
] | ||
} | ||
] | ||
} | ||
``` | ||
|
||
### Testing the WriteToScratchPadTool | ||
|
||
You can use the Tools API directly to execute the `WriteToScratchPadTool` and test the tool response before registering it with your agents. | ||
|
||
```json | ||
POST /_plugins/_ml/tools/_execute/WriteToScratchPadTool | ||
{ | ||
"parameters": { | ||
"notes": "Research Plan for OpenSearch History and ML Evolution:\\n\\n1. OpenSearch version history, major releases after v2.0\\n2. For each major release:\\n a. Key architectural upgrades\\n b. New machine learning capabilities, especially ML Commons Agent framework \\n c. Descriptions of major Agent tools added\\n d. GitHub issue IDs tied to Agent framework features\\n3. Look for official OpenSearch documentation, release notes, blogs\\n4. Search code repositories for more technical details on ML changes\\n" | ||
} | ||
} | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
The following is the example response from the tool output: | ||
|
||
```json | ||
{ | ||
"inference_results": [ | ||
{ | ||
"output": [ | ||
{ | ||
"name": "response", | ||
"result": "Wrote to scratchpad: Research Plan for OpenSearch History and ML Evolution:\\n\\n1. OpenSearch version history, major releases after v2.0\\n2. For each major release:\\n a. Key architectural upgrades\\n b. New machine learning capabilities, especially ML Commons Agent framework \\n c. Descriptions of major Agent tools added\\n d. GitHub issue IDs tied to Agent framework features\\n3. Look for official OpenSearch documentation, release notes, blogs\\n4. Search code repositories for more technical details on ML changes\\n" | ||
} | ||
] | ||
} | ||
] | ||
} | ||
``` | ||
|
||
You can set the `return_history` parameter to `true` to get the full scratchpad content after writing: | ||
|
||
```json | ||
POST /_plugins/_ml/tools/_execute/WriteToScratchPadTool | ||
{ | ||
"parameters": { | ||
"notes": "Research Plan for OpenSearch History and ML Evolution:\\n\\n1. OpenSearch version history, major releases after v2.0\\n2. For each major release:\\n a. Key architectural upgrades\\n b. New machine learning capabilities, especially ML Commons Agent framework \\n c. Descriptions of major Agent tools added\\n d. GitHub issue IDs tied to Agent framework features\\n3. Look for official OpenSearch documentation, release notes, blogs\\n4. Search code repositories for more technical details on ML changes\\n", | ||
"return_history": true | ||
} | ||
} | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
The response contains the full scratchpad content: | ||
|
||
```json | ||
{ | ||
"inference_results": [ | ||
{ | ||
"output": [ | ||
{ | ||
"name": "response", | ||
"result": """Scratchpad updated. Full content: | ||
- Research Plan for OpenSearch History and ML Evolution:\n\n1. OpenSearch version history, major releases after v2.0\n2. For each major release:\n a. Key architectural upgrades\n b. New machine learning capabilities, especially ML Commons Agent framework \n c. Descriptions of major Agent tools added\n d. GitHub issue IDs tied to Agent framework features\n3. Look for official OpenSearch documentation, release notes, blogs\n4. Search code repositories for more technical details on ML changes\n""" | ||
} | ||
] | ||
} | ||
] | ||
} | ||
``` | ||
|
||
## Viewing scratchpad activity | ||
|
||
You can monitor how the agent uses the scratchpad by examining the execution traces: | ||
|
||
```json | ||
GET /_plugins/_ml/memory/message/<parent_interaction_id>/traces?next_token=0 | ||
``` | ||
{% include copy-curl.html %} | ||
|
||
The traces show the sequence of scratchpad reads and writes, demonstrating how the agent accumulates knowledge during the execution session. | ||
|
||
## Related documentation | ||
|
||
- [Agents and tools]({{site.url}}{{site.baseurl}}/ml-commons-plugin/agents-tools/) | ||
- [Conversational agents]({{site.url}}{{site.baseurl}}/ml-commons-plugin/agents-tools/agents/) |
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.