[AI] AG-UI compliant Langgraph ReAct agent implementation by goyamegh · Pull Request #10612 · opensearch-project/OpenSearch-Dashboards

goyamegh · 2025-09-30T21:25:02Z

Description

This PR introduces the osd-agents package - an experimental reference implementation of an AG-UI compliant ReAct (Reasoning and Acting) agent built with LangGraph and AWS Bedrock.

What does this agent do?

The agent implements a reasoning loop that:

Reasons through problems step-by-step
Acts by dynamically discovering and calling tools via Model Context Protocol (MCP)
Observes tool results to inform next actions
Iterates until the task is complete

Key Features

✅ AG-UI Protocol Compliance: Implements AG-UI standard for agent interactions
✅ ReAct Pattern: LangGraph-based reasoning and action loop
✅ Dynamic Tool Use: Runtime tool discovery via MCP servers (local stdio & remote HTTP)
✅ Dual Operation Modes:
- Interactive CLI for development/testing
- HTTP server with streaming SSE for integration
✅ AWS Bedrock Integration: Uses Claude models for reasoning
✅ Comprehensive Logging: Audit logging, metrics, and LLM request tracking

Architecture Highlights

Agent Factory Pattern: Extensible architecture supporting multiple agent types (currently: ReAct)
MCP Integration Layer: Unified interface for local and remote MCP servers
AG-UI Adapters: Protocol translation between AG-UI messages and agent internals
Message Lifecycle Management: Proper handling of tool calls, interruptions, and streaming

What's Included

Complete ReAct agent implementation (src/agents/langgraph/)
MCP client infrastructure (stdio & HTTP)
AG-UI server with SSE streaming
Configuration management for MCP servers
Comprehensive logging and metrics
CLI and HTTP server entry points

Experimental Status

⚠️ This is an experimental reference implementation:

Not production-ready
Intended for learning, testing, and development
API stability not guaranteed
Limited testing coverage

Issues Resolved

RFC #10585
Can be used in conjunction with chat interface and client tools of context provider in #10600

Screenshot

TBD

Testing the changes

Tested with PR #10600

The agent can be tested via:

# Interactive CLI mode
npm start

# AG-UI HTTP server mode
npm run start:ag-ui

Requires:
- AWS Bedrock access with Claude model permissions
- MCP server configuration in configuration/mcp_config.json

Changelog

feat: [AI] AG-UI compliant Langgraph ReAct agent implementation

Check List

All tests pass
- yarn test:jest
- yarn test:jest_integration
New functionality includes testing.
New functionality has been documented.
Update CHANGELOG.md
Commits are signed per the DCO using --signoff

github-actions · 2025-09-30T21:25:35Z

❌ Empty Changelog Section

The Changelog section in your PR description is empty. Please add a valid changelog entry or entries. If you did add a changelog entry, check to make sure that it was not accidentally included inside the comment block in the Changelog section.

codecov · 2025-09-30T21:44:11Z

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 60.14%. Comparing base (3ac73b1) to head (4a99470).
⚠️ Report is 3 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main   #10612   +/-   ##
=======================================
  Coverage   60.14%   60.14%           
=======================================
  Files        4425     4425           
  Lines      118370   118370           
  Branches    19443    19443           
=======================================
  Hits        71188    71188           
  Misses      42267    42267           
  Partials     4915     4915

Flag	Coverage Δ
Linux_1	`26.65% <ø> (ø)`
Linux_2	`38.83% <ø> (ø)`
Linux_3	`38.62% <ø> (+<0.01%)`	⬆️
Linux_4	`32.59% <ø> (ø)`
Windows_1	`26.67% <ø> (ø)`
Windows_2	`38.80% <ø> (ø)`
Windows_3	`38.63% <ø> (+<0.01%)`	⬆️
Windows_4	`32.59% <ø> (ø)`

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

joshuali925 · 2025-09-30T21:28:32Z

packages/osd-agents/configuration/default_model.json

@@ -0,0 +1,3 @@
+{
+  "modelId": "us.anthropic.claude-sonnet-4-20250514-v1:0"


does it perform better using claude sonnet 4.5?

will try it and share feedback

joshuali925 · 2025-09-30T21:40:58Z

packages/osd-agents/src/ag_ui/base_ag_ui_adapter.ts

+
+    // Check for custom system prompt file path from environment variable
+    let customSystemPrompt: string | undefined;
+    const systemPromptPath = process.env.SYSTEM_PROMPT;


why not configuration?

can take up as fat-follow

joshuali925 · 2025-09-30T22:00:16Z

packages/osd-agents/src/agents/langgraph/react_graph_builder.ts

+ */
+
+import { StateGraph, START, END, Annotation } from '@langchain/langgraph';
+import { SqliteSaver } from '@langchain/langgraph-checkpoint-sqlite';


can this use https://langchain-ai.github.io/langgraphjs/agents/agents/?h=memory+saver#5-add-memory

import { StateGraph, START, END, Annotation, MemorySaver } from '@langchain/langgraph';

and not introduce sqlite lib dependency?

joshuali925 · 2025-09-30T22:28:05Z

packages/osd-agents/src/agents/langgraph/bedrock_client.ts

+      hasAwsAccessKey: !!process.env.AWS_ACCESS_KEY_ID,
+      hasAwsSecretKey: !!process.env.AWS_SECRET_ACCESS_KEY,
+      hasAwsProfile: !!process.env.AWS_PROFILE,
+      hasAwsSessionToken: !!process.env.AWS_SESSION_TOKEN,


these are unnecessary logs? sdk should figure out where credentials come from

joshuali925 · 2025-09-30T22:31:02Z

packages/osd-agents/src/agents/langgraph/prompt_manager.ts

+`;
+      prompt = prompt.replace('{{CLIENT_STATE}}', stateContent);
+    } else {
+      prompt = prompt.replace('{{CLIENT_STATE}}', '// No client state provided');


what is {{CLIENT_STATE}}? can this placeholder come from some constant?

would clientState ever be a complex/circular object that can't be stringified?

joshuali925 · 2025-09-30T22:32:39Z

packages/osd-agents/src/agents/langgraph/prompt_manager.ts

+          }`;
+
+          if (tool.inputSchema.required && tool.inputSchema.required.length > 0) {
+            toolDescription += `\n  - Required parameters: ${tool.inputSchema.required.join(', ')}`;


the langgraph framework doesn't auto do this and requires us to manually build tool prompt?

can check this and get back on it

Lint fixes and changing file to snake case Fixing imports for snake case feat: Implement TextMessageManager for improved message lifecycle management and tool call interruption in AG UI feat: Refactor tool result handling to improve message lifecycle management in BaseAGUIAdapter feat: Improve response handling by logging warnings for empty content blocks and filtering out empty text blocks to prevent ValidationException Renaming text_message_manager and imports Signed-off-by: Megha Goyal <goyamegh@amazon.com> Adding default observability prompt Signed-off-by: Megha Goyal <goyamegh@amazon.com> Removing jarvis and coact agents Signed-off-by: Megha Goyal <goyamegh@amazon.com>

Signed-off-by: Megha Goyal <goyamegh@amazon.com>

joshuali925

didn't get to go through everything, but approving since it's a separate experimental package

joshuali925 · 2025-10-01T16:39:28Z

packages/osd-agents/package.json

+    "build": "tsc",
+    "clean": "find . -name '*.js' -not -path './node_modules/*' -delete",
+    "opensearch:ingest": "ts-node src/opensearch-ingestion.ts",
+    "opensearch:watch": "ts-node src/opensearch-ingestion.ts --watch"


file is opensearch_ingestion with underscore

ananzh · 2025-10-01T16:59:34Z

Failed windows tests are known CI/CD issue that we observe in my PRs lately. There is a tracking fix #10613. Should not block PR due to these CI/CD issues.

Wait for the last cypress to re-run then we could merge the PR in.

…-project#10612) * AG-UI compliant Langgraph ReAct agent implementation Lint fixes and changing file to snake case Fixing imports for snake case feat: Implement TextMessageManager for improved message lifecycle management and tool call interruption in AG UI feat: Refactor tool result handling to improve message lifecycle management in BaseAGUIAdapter feat: Improve response handling by logging warnings for empty content blocks and filtering out empty text blocks to prevent ValidationException Renaming text_message_manager and imports Signed-off-by: Megha Goyal <goyamegh@amazon.com> Adding default observability prompt Signed-off-by: Megha Goyal <goyamegh@amazon.com> Removing jarvis and coact agents Signed-off-by: Megha Goyal <goyamegh@amazon.com> * Changeset file for PR opensearch-project#10612 created/updated * Fixing build failures due to langgraph-checkpoint-sqlite Signed-off-by: Megha Goyal <goyamegh@amazon.com> * Fixing yarn.lock using yarn v1.22.10 Signed-off-by: Megha Goyal <goyamegh@amazon.com> * Fix yarn.lock aain for 1.22.10 after rebase Signed-off-by: Megha Goyal <goyamegh@amazon.com> --------- Signed-off-by: Megha Goyal <goyamegh@amazon.com> Co-authored-by: opensearch-changeset-bot[bot] <154024398+opensearch-changeset-bot[bot]@users.noreply.github.com> Signed-off-by: Nathan Yang <yanatha@amazon.com>

github-actions bot added the repeat-contributor label Sep 30, 2025

github-actions bot added the failed changeset label Sep 30, 2025

opensearch-changeset-bot bot added a commit to ananzh/OpenSearch-Dashboards that referenced this pull request Sep 30, 2025

Changeset file for PR opensearch-project#10612 created/updated

eeb67da

github-actions bot removed the failed changeset label Sep 30, 2025

joshuali925 reviewed Sep 30, 2025

View reviewed changes

goyamegh marked this pull request as ready for review September 30, 2025 22:57

goyamegh requested review from Maosaic, angle943, d-buckner, ps48, raintygao and wanglam as code owners September 30, 2025 22:57

peterzhuamazon mentioned this pull request Oct 1, 2025

[RELEASE] Release version 3.3.0 opensearch-project/opensearch-build#5693

Closed

66 tasks

ps48 previously approved these changes Oct 1, 2025

View reviewed changes

goyamegh dismissed ps48’s stale review via d929b43 October 1, 2025 03:14

ananzh previously approved these changes Oct 1, 2025

View reviewed changes

goyamegh and others added 4 commits September 30, 2025 23:57

Changeset file for PR opensearch-project#10612 created/updated

26a8d04

Fixing build failures due to langgraph-checkpoint-sqlite

c04ecae

Signed-off-by: Megha Goyal <goyamegh@amazon.com>

Fixing yarn.lock using yarn v1.22.10

82483d3

Signed-off-by: Megha Goyal <goyamegh@amazon.com>

goyamegh dismissed ananzh’s stale review via 82483d3 October 1, 2025 07:06

goyamegh force-pushed the osd-agents branch from 3363b90 to 82483d3 Compare October 1, 2025 07:06

ananzh approved these changes Oct 1, 2025

View reviewed changes

ananzh added ai Changes related to AI chat, context provider and artifact v3.3.0 Issues targeting release v3.3.0 OSD Changes being merged by the OSD team labels Oct 1, 2025

Fix yarn.lock aain for 1.22.10 after rebase

4a99470

Signed-off-by: Megha Goyal <goyamegh@amazon.com>

goyamegh force-pushed the osd-agents branch from 91a882f to 4a99470 Compare October 1, 2025 07:33

joshuali925 approved these changes Oct 1, 2025

View reviewed changes

ananzh merged commit 681c9e6 into opensearch-project:main Oct 1, 2025
212 of 242 checks passed

goyamegh mentioned this pull request Oct 10, 2025

[PROPOSAL]OpenSearch Agent service for specialized opensearch agents opensearch-project/.github#398

Open

Desvelao mentioned this pull request Oct 17, 2025

Compatibility with OpenSearch dashboard 3.3.0 wazuh/wazuh-dashboard#966

Closed

4 tasks

This was referenced Jan 10, 2026

docs: add OpenSearch Dashboards AI Chat report for v3.3.0 tkykenmt/opensearch-feature-explorer#1458

Merged

[feature] OpenSearch Dashboards AI Chat tkykenmt/opensearch-feature-explorer#1451

Closed

		@@ -0,0 +1,3 @@
		{
		"modelId": "us.anthropic.claude-sonnet-4-20250514-v1:0"

Conversation

goyamegh commented Sep 30, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Description

What does this agent do?

Key Features

Architecture Highlights

What's Included

Experimental Status

Issues Resolved

Screenshot

Testing the changes

Changelog

Check List

Uh oh!

github-actions bot commented Sep 30, 2025

❌ Empty Changelog Section

Uh oh!

codecov bot commented Sep 30, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Codecov Report

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!

joshuali925 left a comment

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

ananzh commented Oct 1, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants

goyamegh commented Sep 30, 2025 •

edited

Loading

codecov bot commented Sep 30, 2025 •

edited

Loading

ananzh commented Oct 1, 2025 •

edited

Loading