Skip to content

Commit 5c37ae2

Browse files
authored
Merge pull request #45449 from tabito-hara/f-aws_bedrockagent_agent-add_max_recent_sessions
[Enhancement] aws_bedrockagent_agent: Add `session_summary_configuration.max_recent_sessions` argument
2 parents 7316bc0 + 9e30718 commit 5c37ae2

File tree

4 files changed

+25
-2
lines changed

4 files changed

+25
-2
lines changed

.changelog/45449.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
```release-note:enhancement
2+
resource/aws_bedrockagent_agent: Add `session_summary_configuration.max_recent_sessions` argument
3+
```

internal/service/bedrockagent/agent.go

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -732,8 +732,13 @@ type guardrailConfigurationModel struct {
732732
}
733733

734734
type memoryConfigurationModel struct {
735-
EnabledMemoryTypes fwtypes.ListValueOf[fwtypes.StringEnum[awstypes.MemoryType]] `tfsdk:"enabled_memory_types"`
736-
StorageDays types.Int32 `tfsdk:"storage_days"`
735+
EnabledMemoryTypes fwtypes.ListValueOf[fwtypes.StringEnum[awstypes.MemoryType]] `tfsdk:"enabled_memory_types"`
736+
SessionSummaryConfiguration fwtypes.ListNestedObjectValueOf[sessionSummaryConfigurationModel] `tfsdk:"session_summary_configuration"`
737+
StorageDays types.Int32 `tfsdk:"storage_days"`
738+
}
739+
740+
type sessionSummaryConfigurationModel struct {
741+
MaxRecentSessions types.Int64 `tfsdk:"max_recent_sessions"`
737742
}
738743

739744
type promptOverrideConfigurationModel struct {

internal/service/bedrockagent/agent_test.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -497,6 +497,11 @@ func TestAccBedrockAgentAgent_memoryConfiguration(t *testing.T) {
497497
resource.TestCheckResourceAttr(resourceName, "agent_name", rName),
498498
resource.TestCheckResourceAttr(resourceName, names.AttrDescription, "basic claude"),
499499
resource.TestCheckResourceAttr(resourceName, "memory_configuration.#", "1"),
500+
resource.TestCheckResourceAttr(resourceName, "memory_configuration.0.enabled_memory_types.#", "1"),
501+
resource.TestCheckResourceAttr(resourceName, "memory_configuration.0.enabled_memory_types.0", "SESSION_SUMMARY"),
502+
resource.TestCheckResourceAttr(resourceName, "memory_configuration.0.storage_days", "15"),
503+
resource.TestCheckResourceAttr(resourceName, "memory_configuration.0.session_summary_configuration.#", "1"),
504+
resource.TestCheckResourceAttr(resourceName, "memory_configuration.0.session_summary_configuration.0.max_recent_sessions", "5"),
500505
resource.TestCheckResourceAttr(resourceName, "skip_resource_in_use_check", acctest.CtTrue),
501506
),
502507
},
@@ -940,6 +945,9 @@ resource "aws_bedrockagent_agent" "test" {
940945
memory_configuration {
941946
enabled_memory_types = ["SESSION_SUMMARY"]
942947
storage_days = 15
948+
session_summary_configuration {
949+
max_recent_sessions = 5
950+
}
943951
}
944952
}
945953
`, rName, model, description))

website/docs/r/bedrockagent_agent.html.markdown

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,8 +103,15 @@ The `guardrail_configuration` configuration block supports the following argumen
103103
The `memory_configuration` configuration block supports the following arguments:
104104

105105
* `enabled_memory_types` - (Required) The type of memory being stored by the agent. See [AWS API documentation](https://docs.aws.amazon.com/bedrock/latest/APIReference/API_agent_MemoryConfiguration.html) for possible values.
106+
* `session_summary_configuration` - (Optional) Configuration block for `SESSION_SUMMARY` memory type enabled for the agent. See [`session_summary_configuration` Block](#session_summary_configuration-block) for details.
106107
* `storage_days` - (Optional) The number of days the agent is configured to retain the conversational context. Minimum value of 0, maximum value of 30.
107108

109+
### `session_summary_configuration` Block
110+
111+
The `session_summary_configuration` configuration block supports the following arguments:
112+
113+
* `max_recent_sessions` - (Optional) Maximum number of recent session summaries to include in the agent's prompt context.
114+
108115
### `prompt_override_configuration` Block
109116

110117
The `prompt_override_configuration` configuration block supports the following arguments:

0 commit comments

Comments
 (0)