Skip to content

Conversation

@ylwu-amzn
Copy link
Collaborator

@ylwu-amzn ylwu-amzn commented Nov 18, 2025

Description

Restore context after running connector executor.

Related Issues

Resolves #[Issue number to be closed when this PR is merged]

Check List

  • New functionality includes testing.
  • New functionality has been documented.
  • API changes companion pull request created.
  • Commits are signed per the DCO using --signoff.
  • Public documentation issue/PR created.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.
For more information on following Developer Certificate of Origin and signing off your commits, please check here.

@ylwu-amzn ylwu-amzn added the v3.4.0 Issues targeting release v3.4.0 label Nov 18, 2025
@ylwu-amzn ylwu-amzn force-pushed the agent_mem_integ_pr branch 2 times, most recently from a710f11 to cbd6e56 Compare November 18, 2025 06:32
dhrubo-os
dhrubo-os previously approved these changes Nov 18, 2025
pyek-bot
pyek-bot previously approved these changes Nov 18, 2025
ylwu-amzn and others added 2 commits November 18, 2025 11:48
…opensearch-project#4430)

* Add support for custom named connector actions; add PUT/DELETE action

Signed-off-by: Yaliang Wu <ylwu@amazon.com>

* add more comment

Signed-off-by: Yaliang Wu <ylwu@amazon.com>

---------

Signed-off-by: Yaliang Wu <ylwu@amazon.com>
…rch Agents (opensearch-project#4432)

* add hooks in ml-commons (opensearch-project#4326)

Signed-off-by: Xun Zhang <xunzh@amazon.com>

* initiate context management api with hook implementation (opensearch-project#4345)

* initiate context management api with hook implementation

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* apply spotless

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

---------

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* Add Context Manager to PER (opensearch-project#4379)

* add pre_llm hook to per agent

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

change context management passing from query parameters to payload

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

pass hook registery into PER

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

apply spotless

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

initiate context management api with hook implementation

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* add comment

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* format

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* add validation

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

---------

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* add inner create context management to agent register api

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* add code coverage

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* allow context management hook register in during agent execute

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* add code coverage

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* add more code coverage

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* add validation check

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* adapt to inplace update for context

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* Allow context management inline create in register agent without storing in index (opensearch-project#4403)

* allow inline create context management without storing in agent register

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* make ML_COMMONS_MULTI_TENANCY_ENABLED default is false

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

---------

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* Update the POST_TOOL hook emit saving to agentic memory (opensearch-project#4408)

* Fix POST_TOOL hook interaction updates and add tenant ID support
Signed-off-by: Mingshi Liu <mingshl@amazon.com>

- Fix POST_TOOL hook to return full ContextManagerContext like PRE_LLM hook
- Update MLChatAgentRunner to properly handle interaction updates from POST_TOOL hook
- Ensure interactions list and tmpParameters.INTERACTIONS stay synchronized
- Add tenant ID support to MLPredictionTaskRequest in ModelGuardrail and SummarizationManager

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* fix error message escaping

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* consolicate post_hook logic

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

---------

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

* bump supported version to 3.4

Signed-off-by: Mingshi Liu <mingshl@amazon.com>

---------

Signed-off-by: Xun Zhang <xunzh@amazon.com>
Signed-off-by: Mingshi Liu <mingshl@amazon.com>
Co-authored-by: Xun Zhang <xunzh@amazon.com>
@ylwu-amzn ylwu-amzn dismissed stale reviews from pyek-bot and dhrubo-os via fbe3ca4 November 18, 2025 21:45
Signed-off-by: Yaliang Wu <ylwu@amazon.com>
Signed-off-by: Yaliang Wu <ylwu@amazon.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

backport 3.3 v3.4.0 Issues targeting release v3.4.0

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants