feat(memory): add episodic memory strategy support #208

aneeshsabarad · 2026-01-08T20:37:25Z

Add EPISODIC to StrategyType, MemoryStrategyTypeEnum, OverrideType enums
Add EPISODIC to DEFAULT_NAMESPACES
Add episodic wrapper keys for extraction, consolidation, reflection
Add add_episodic_strategy() and add_episodic_strategy_and_wait()
Add add_custom_episodic_strategy() and add_custom_episodic_strategy_and_wait()
Update _wrap_configuration() to handle EPISODIC_OVERRIDE
Add unit tests for all new episodic functionality

Issue #, if available:

Description of changes:

Enable support for using Episodic Strategies with AgentCore Memory

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

abhivr-amazon · 2026-01-09T17:08:08Z

src/bedrock_agentcore/memory/client.py

+        name: str,
+        description: Optional[str] = None,
+        namespaces: Optional[List[str]] = None,
+        reflection_namespaces: Optional[List[str]] = None,


should this be optional if we require a reflection namespace?

We should leave it to match the same behavior as what is defined in the cli and model, as validation of this will come from the service side, not the client side

Discussed offline and it is changed now

abhivr-amazon · 2026-01-09T17:10:44Z

src/bedrock_agentcore/memory/client.py

+        name: str,
+        description: Optional[str] = None,
+        namespaces: Optional[List[str]] = None,
+        reflection_namespaces: Optional[List[str]] = None,


same comment as above. This should be required.

abhivr-amazon · 2026-01-09T18:55:42Z

tests/bedrock_agentcore/memory/test_client.py

            except Exception as e:
                assert "Unexpected error" in str(e)
+
+


nit: We should also add negative test cases to check for errors being thrown

We would like negative tests too

aidandaly24 · 2026-01-12T15:19:08Z

src/bedrock_agentcore/memory/client.py

        self.add_user_preference_strategy(memory_id, name, description, namespaces)
        return self._wait_for_memory_active(memory_id, max_wait, poll_interval)

+    def add_episodic_strategy(


nit: The parameter order differs from other strategy methods like add_user_preference_strategy(memory_id, name, description, namespaces). Having
reflection_namespaces (required) between name and description breaks the pattern. The only impact is that the inconsistent API may confuse users familiar with other strategy methods.

Yeah, decided to move reflection_namespaces before description since the required parameters should all come before the optional ones ideally

Sounds good to me, I also thought the same but then noticed that odd difference.

aidandaly24 · 2026-01-12T15:23:59Z

src/bedrock_agentcore/memory/client.py

+            "appendToPrompt": reflection_config["prompt"],
+            "modelId": reflection_config["modelId"],


suggestion: Should we consider adding input validation on reflection_config["prompt"] and reflection_config["modelId"]? I think this may raise a KeyError with an unclear error message if they are missing. We could instead raise a ValueError like:

if key not in reflection_config: raise ValueError(f"reflection_config missing required key: {key}")

Yeah this should be there. Prompt and modelId are required for all extraction, consolidation, and reflection config also so we can add a validation to check all of them like:

for config, config_name in [ (extraction_config, "extraction_config"), (consolidation_config, "consolidation_config"), (reflection_config, "reflection_config"), ]: for key in ("prompt", "modelId"): if key not in config: raise ValueError(f"{config_name} missing required key: {key}")

Also will move reflection to be in the strategy block to read cleaner like:

strategy = { StrategyType.CUSTOM.value: { "name": name, "configuration": { "episodicOverride": { "extraction": { "appendToPrompt": extraction_config["prompt"], "modelId": extraction_config["modelId"], }, "consolidation": { "appendToPrompt": consolidation_config["prompt"], "modelId": consolidation_config["modelId"], }, "reflection": { "appendToPrompt": reflection_config["prompt"], "modelId": reflection_config["modelId"], **( {"namespaces": reflection_config["namespaces"]} if "namespaces" in reflection_config else {} ), }, } }, } }

jariy17 · 2026-01-12T19:10:07Z

tests/bedrock_agentcore/memory/test_client.py

            except Exception as e:
                assert "Unexpected error" in str(e)
+
+


We would like negative tests too

jariy17 · 2026-01-12T19:48:06Z

src/bedrock_agentcore/memory/client.py

                        wrapped_config["extraction"] = {
                            "customExtractionConfiguration": {CUSTOM_EXTRACTION_WRAPPER_KEYS[override_enum]: extraction}
                        }
+            elif strategy_type == "CUSTOM" and override_type and any(key in extraction for key in custom_override_keys):


It's a little messy here to understand. What does this part of code achieve?

None of the other memory strategies have negative tests, so I just followed the same structure for the strategy related tests in this one. Should we still add negative tests just for episodic?

Refactoring this to be simpler, the wrapping was off before itself but it should be fine now. Tested it to match the behavior of the bedrock-agentcore-control cli.

- Add EPISODIC to StrategyType, MemoryStrategyTypeEnum, OverrideType enums - Add EPISODIC to DEFAULT_NAMESPACES - Add episodic wrapper keys for extraction, consolidation, reflection - Add add_episodic_strategy() and add_episodic_strategy_and_wait() - Add add_custom_episodic_strategy() and add_custom_episodic_strategy_and_wait() - Update _wrap_configuration() to handle EPISODIC_OVERRIDE - Add unit tests for all new episodic functionality

jariy17

Just need to address non-nit comments.

jariy17 · 2026-01-12T20:59:25Z

src/bedrock_agentcore/memory/constants.py

@@ -86,12 +90,18 @@ class MessageRole(Enum):
 CUSTOM_EXTRACTION_WRAPPER_KEYS: Dict[OverrideType, str] = {


nit: OVERRIDE_EXTRACTION_WRAPPERS_KEYS

jariy17 · 2026-01-12T21:07:26Z

src/bedrock_agentcore/memory/client.py

+                    wrapped_config["extraction"] = extraction
+            elif any(key in extraction for key in builtin_config_keys):
+                strategy_type_enum = MemoryStrategyTypeEnum(strategy_type)
+                if strategy_type in ("SEMANTIC", "USER_PREFERENCE"):


Does episodic strategy has a special wrapped_config dictionary like SEMANTIC and USER_PREFERENCES

Ran local integration tests to verify the behavior on update memory operations match what is expected from the service, and the wrapped values are accepted

aneeshsabarad temporarily deployed to auto-approve January 8, 2026 20:37 — with GitHub Actions Inactive

abhivr-amazon reviewed Jan 9, 2026

View reviewed changes

aneeshsabarad force-pushed the episodic_memory_strategy branch from 344b39d to 5528322 Compare January 9, 2026 17:23

aneeshsabarad temporarily deployed to auto-approve January 9, 2026 17:23 — with GitHub Actions Inactive

abhivr-amazon reviewed Jan 9, 2026

View reviewed changes

aidandaly24 mentioned this pull request Jan 9, 2026

Support for Episodic memory strategy #203

Closed

aidandaly24 reviewed Jan 12, 2026

View reviewed changes

aneeshsabarad force-pushed the episodic_memory_strategy branch from 5528322 to 4ab866c Compare January 12, 2026 15:53

aneeshsabarad temporarily deployed to auto-approve January 12, 2026 15:53 — with GitHub Actions Inactive

jariy17 reviewed Jan 12, 2026

View reviewed changes

aneeshsabarad force-pushed the episodic_memory_strategy branch from 4ab866c to 8bc4b00 Compare January 12, 2026 20:16

aneeshsabarad temporarily deployed to auto-approve January 12, 2026 20:16 — with GitHub Actions Inactive

jariy17 reviewed Jan 12, 2026

View reviewed changes

jariy17 approved these changes Jan 12, 2026

View reviewed changes

jariy17 merged commit 0df9757 into main Jan 12, 2026
18 of 19 checks passed

jariy17 deleted the episodic_memory_strategy branch January 12, 2026 21:38

		"appendToPrompt": reflection_config["prompt"],
		"modelId": reflection_config["modelId"],

		@@ -86,12 +90,18 @@ class MessageRole(Enum):
		CUSTOM_EXTRACTION_WRAPPER_KEYS: Dict[OverrideType, str] = {

feat(memory): add episodic memory strategy support #208

feat(memory): add episodic memory strategy support #208

Conversation

aneeshsabarad commented Jan 8, 2026

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

jariy17 left a comment

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants