feat: refactor a2a_stream with common code by sriaradhyula · Pull Request #391 · cnoe-io/ai-platform-engineering

sriaradhyula · 2025-10-22T13:20:36Z

Summary

This PR refactors the a2a_stream functionality to use common code patterns across all agents, improving maintainability and consistency.

Changes Made

🏗️ Common Infrastructure

Moved common a2a functionality from common/ to utils/a2a_common/ directory
Created BaseAgent class with streaming support and common functionality:
- LLM initialization and tracing setup
- MCP client configuration
- Streaming response handling
- Agent execution patterns
Created BaseAgentExecutor for A2A protocol handling with streaming support
Added common streaming utilities and helper functions

🔄 Agent Refactoring

Refactored multiple agents to use the new common base classes (Komodor, Weather, Webex, etc.)
Updated imports across the codebase to use the new utils structure
Maintained backward compatibility while improving code organization

🐳 Docker & CI Fixes

Fixed Docker build context issues in GitHub Actions workflows
Updated all agent Dockerfiles to use correct relative paths for the new structure
Resolved CI build failures related to agent refactoring

🧹 Testing & Cleanup

Added comprehensive streaming test suite
Resolved test suite issues and enabled linting
Cleaned up unused files and old backup code
Applied automatic linting fixes across the codebase

Benefits

✅ Reduced code duplication across agents
✅ Consistent streaming patterns for all a2a agents
✅ Easier maintenance with centralized common functionality
✅ Better testability with shared base classes
✅ Improved developer experience with clear abstractions

Docker Image Prebuilding

This PR uses the prebuild/ branch prefix to trigger Docker image prebuilding, which will:

Build and cache Docker images for all affected agents
Speed up future builds and deployments
Validate Docker build fixes

Breaking Changes

None - this maintains backward compatibility while providing the new common infrastructure.

Note: This is a continuation of the closed PR #354 with additional fixes and the prebuild branch prefix for Docker image caching.

github-actions · 2025-10-22T13:22:44Z

🐳 Prebuild Docker Image Published

Agent: splunk
Repository: ghcr.io/cnoe-io/prebuild/mcp-splunk
Tag: a2a_stream_common_code-20

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-splunk:a2a_stream_common_code-20

github-actions · 2025-10-22T13:22:54Z

🐳 Prebuild Docker Image Published

Agent: argocd
Repository: ghcr.io/cnoe-io/prebuild/mcp-argocd
Tag: a2a_stream_common_code-20

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-argocd:a2a_stream_common_code-20

github-actions · 2025-10-22T13:24:54Z

🐳 Prebuild Docker Image Published

Agent: template
Repository: ghcr.io/cnoe-io/prebuild/agent-template
Tag: a2a_stream_common_code-20

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-template:a2a_stream_common_code-20

github-actions · 2025-10-22T13:26:06Z

🐳 Prebuild Docker Image Published

Repository: ghcr.io/cnoe-io/prebuild/ai-platform-engineering
Tag: a2a_stream_common_code-20

Usage

docker pull ghcr.io/cnoe-io/prebuild/ai-platform-engineering:a2a_stream_common_code-20

github-actions · 2025-10-22T13:31:26Z

🐳 Prebuild Docker Image Published

Component: webui
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-webui
Tag: a2a_stream_common_code-20

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-webui:a2a_stream_common_code-20

github-actions · 2025-10-22T13:34:34Z

🐳 Prebuild Docker Image Published

Component: agent-ontology
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-agent-ontology
Tag: a2a_stream_common_code-20

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-agent-ontology:a2a_stream_common_code-20

github-actions · 2025-10-22T13:37:59Z

🐳 Prebuild Docker Image Published

Component: server
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-server
Tag: a2a_stream_common_code-20

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-server:a2a_stream_common_code-20

github-actions · 2025-10-22T13:48:18Z

🐳 Prebuild Docker Image Published

Agent: argocd
Repository: ghcr.io/cnoe-io/prebuild/mcp-argocd
Tag: a2a_stream_common_code-21

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-argocd:a2a_stream_common_code-21

github-actions · 2025-10-22T13:48:55Z

🐳 Prebuild Docker Image Published

Agent: splunk
Repository: ghcr.io/cnoe-io/prebuild/mcp-splunk
Tag: a2a_stream_common_code-21

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-splunk:a2a_stream_common_code-21

github-actions · 2025-10-22T13:50:52Z

🐳 Prebuild Docker Image Published

Agent: template
Repository: ghcr.io/cnoe-io/prebuild/agent-template
Tag: a2a_stream_common_code-21

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-template:a2a_stream_common_code-21

github-actions · 2025-10-22T13:52:52Z

🐳 Prebuild Docker Image Published

Repository: ghcr.io/cnoe-io/prebuild/ai-platform-engineering
Tag: a2a_stream_common_code-21

Usage

docker pull ghcr.io/cnoe-io/prebuild/ai-platform-engineering:a2a_stream_common_code-21

github-actions · 2025-10-22T13:56:32Z

🐳 Prebuild Docker Image Published

Component: agent-rag
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag
Tag: a2a_stream_common_code-21

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag:a2a_stream_common_code-21

sriaradhyula · 2025-10-22T15:16:26Z

+        # Check for documentation/knowledge base queries (direct to RAG)
+        # Only match explicit documentation requests, not operational queries
+        documentation_keywords = [
+            'documentation', 'docs',   # Documentation queries


make it a specific docs:, mention

github-actions · 2025-10-23T00:54:16Z

🐳 Prebuild Docker Image Published

Agent: argocd
Repository: ghcr.io/cnoe-io/prebuild/mcp-argocd
Tag: a2a_stream_common_code-22

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-argocd:a2a_stream_common_code-22

github-actions · 2025-10-23T00:54:41Z

🐳 Prebuild Docker Image Published

Agent: splunk
Repository: ghcr.io/cnoe-io/prebuild/mcp-splunk
Tag: a2a_stream_common_code-22

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-splunk:a2a_stream_common_code-22

github-actions · 2025-10-23T00:54:53Z

🐳 Prebuild Docker Image Published

Repository: ghcr.io/cnoe-io/prebuild/ai-platform-engineering
Tag: a2a_stream_common_code-22

Usage

docker pull ghcr.io/cnoe-io/prebuild/ai-platform-engineering:a2a_stream_common_code-22

github-actions · 2025-10-23T00:56:14Z

🐳 Prebuild Docker Image Published

Agent: template
Repository: ghcr.io/cnoe-io/prebuild/agent-template
Tag: a2a_stream_common_code-22

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-template:a2a_stream_common_code-22

github-actions · 2025-10-23T01:04:47Z

🐳 Prebuild Docker Image Published

Component: agent-rag
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag
Tag: a2a_stream_common_code-22

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag:a2a_stream_common_code-22

github-actions · 2025-10-23T14:10:23Z

🐳 Prebuild Docker Image Published

Agent: argocd
Repository: ghcr.io/cnoe-io/prebuild/mcp-argocd
Tag: a2a_stream_common_code-24

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-argocd:a2a_stream_common_code-24

github-actions · 2025-10-23T14:10:33Z

🐳 Prebuild Docker Image Published

Agent: splunk
Repository: ghcr.io/cnoe-io/prebuild/mcp-splunk
Tag: a2a_stream_common_code-24

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-splunk:a2a_stream_common_code-24

github-actions · 2025-10-23T14:12:38Z

🐳 Prebuild Docker Image Published

Agent: template
Repository: ghcr.io/cnoe-io/prebuild/agent-template
Tag: a2a_stream_common_code-24

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-template:a2a_stream_common_code-24

github-actions · 2025-10-23T14:15:44Z

🐳 Prebuild Docker Image Published

Repository: ghcr.io/cnoe-io/prebuild/ai-platform-engineering
Tag: a2a_stream_common_code-24

Usage

docker pull ghcr.io/cnoe-io/prebuild/ai-platform-engineering:a2a_stream_common_code-24

github-actions · 2025-10-23T14:18:42Z

🐳 Prebuild Docker Image Published

Component: agent-rag
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag
Tag: a2a_stream_common_code-24

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag:a2a_stream_common_code-24

github-actions · 2025-10-24T08:14:05Z

🐳 Prebuild Docker Image Published

Agent: splunk
Repository: ghcr.io/cnoe-io/prebuild/mcp-splunk
Tag: a2a_stream_common_code-25

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-splunk:a2a_stream_common_code-25

github-actions · 2025-10-24T08:14:40Z

🐳 Prebuild Docker Image Published

Agent: argocd
Repository: ghcr.io/cnoe-io/prebuild/mcp-argocd
Tag: a2a_stream_common_code-25

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-argocd:a2a_stream_common_code-25

github-actions · 2025-10-24T08:16:31Z

🐳 Prebuild Docker Image Published

Agent: template
Repository: ghcr.io/cnoe-io/prebuild/agent-template
Tag: a2a_stream_common_code-25

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-template:a2a_stream_common_code-25

Signed-off-by: Sri Aradhyula <sraradhy@cisco.com>

github-actions · 2025-10-31T15:26:56Z

🐳 Prebuild Docker Image Published

Agent: weather
Repository: ghcr.io/cnoe-io/prebuild/agent-weather
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-weather:a2a_stream_common_code-56

github-actions · 2025-10-31T15:26:57Z

🐳 Prebuild Docker Image Published

Agent: github
Repository: ghcr.io/cnoe-io/prebuild/agent-github
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-github:a2a_stream_common_code-56

github-actions · 2025-10-31T15:26:59Z

🐳 Prebuild Docker Image Published

Agent: template
Repository: ghcr.io/cnoe-io/prebuild/agent-template
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-template:a2a_stream_common_code-56

github-actions · 2025-10-31T15:27:08Z

🐳 Prebuild Docker Image Published

Agent: jira
Repository: ghcr.io/cnoe-io/prebuild/agent-jira
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-jira:a2a_stream_common_code-56

github-actions · 2025-10-31T15:27:27Z

🐳 Prebuild Docker Image Published

Agent: confluence
Repository: ghcr.io/cnoe-io/prebuild/agent-confluence
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-confluence:a2a_stream_common_code-56

github-actions · 2025-10-31T15:27:28Z

🐳 Prebuild Docker Image Published

Repository: ghcr.io/cnoe-io/prebuild/ai-platform-engineering
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/ai-platform-engineering:a2a_stream_common_code-56

github-actions · 2025-10-31T15:27:29Z

🐳 Prebuild Docker Image Published

Agent: webex
Repository: ghcr.io/cnoe-io/prebuild/agent-webex
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-webex:a2a_stream_common_code-56

github-actions · 2025-10-31T15:27:34Z

🐳 Prebuild Docker Image Published

Agent: pagerduty
Repository: ghcr.io/cnoe-io/prebuild/agent-pagerduty
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-pagerduty:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:11Z

🐳 Prebuild Docker Image Published

Agent: splunk
Repository: ghcr.io/cnoe-io/prebuild/agent-splunk
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-splunk:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:14Z

🐳 Prebuild Docker Image Published

Agent: backstage
Repository: ghcr.io/cnoe-io/prebuild/agent-backstage
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-backstage:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:27Z

🐳 Prebuild Docker Image Published

Agent: pagerduty
Repository: ghcr.io/cnoe-io/prebuild/mcp-pagerduty
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-pagerduty:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:30Z

🐳 Prebuild Docker Image Published

Agent: argocd
Repository: ghcr.io/cnoe-io/prebuild/agent-argocd
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-argocd:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:35Z

🐳 Prebuild Docker Image Published

Agent: backstage
Repository: ghcr.io/cnoe-io/prebuild/mcp-backstage
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-backstage:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:37Z

🐳 Prebuild Docker Image Published

Agent: splunk
Repository: ghcr.io/cnoe-io/prebuild/mcp-splunk
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-splunk:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:39Z

🐳 Prebuild Docker Image Published

Agent: slack
Repository: ghcr.io/cnoe-io/prebuild/agent-slack
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-slack:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:42Z

🐳 Prebuild Docker Image Published

Agent: jira
Repository: ghcr.io/cnoe-io/prebuild/mcp-jira
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-jira:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:46Z

🐳 Prebuild Docker Image Published

Agent: slack
Repository: ghcr.io/cnoe-io/prebuild/mcp-slack
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-slack:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:49Z

🐳 Prebuild Docker Image Published

Agent: komodor
Repository: ghcr.io/cnoe-io/prebuild/agent-komodor
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-komodor:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:54Z

🐳 Prebuild Docker Image Published

Agent: argocd
Repository: ghcr.io/cnoe-io/prebuild/mcp-argocd
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-argocd:a2a_stream_common_code-56

github-actions · 2025-10-31T15:28:56Z

🐳 Prebuild Docker Image Published

Agent: komodor
Repository: ghcr.io/cnoe-io/prebuild/mcp-komodor
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-komodor:a2a_stream_common_code-56

github-actions · 2025-10-31T15:29:17Z

🐳 Prebuild Docker Image Published

Agent: confluence
Repository: ghcr.io/cnoe-io/prebuild/mcp-confluence
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/mcp-confluence:a2a_stream_common_code-56

github-actions · 2025-10-31T15:34:58Z

🐳 Prebuild Docker Image Published

Component: webui
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-webui
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-webui:a2a_stream_common_code-56

github-actions · 2025-10-31T15:35:16Z

🐳 Prebuild Docker Image Published

Component: agent-ontology
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-agent-ontology
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-agent-ontology:a2a_stream_common_code-56

github-actions · 2025-10-31T15:36:54Z

🐳 Prebuild Docker Image Published

Agent: aws
Repository: ghcr.io/cnoe-io/prebuild/agent-aws
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/agent-aws:a2a_stream_common_code-56

github-actions · 2025-10-31T15:37:30Z

🐳 Prebuild Docker Image Published

Component: agent-rag
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-agent-rag:a2a_stream_common_code-56

github-actions · 2025-10-31T15:39:46Z

🐳 Prebuild Docker Image Published

Component: server
Repository: ghcr.io/cnoe-io/prebuild/caipe-rag-server
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/caipe-rag-server:a2a_stream_common_code-56

github-actions · 2025-10-31T15:40:19Z

🐳 Prebuild Docker Image Published

Component: Agent Forge Plugin
Repository: ghcr.io/cnoe-io/prebuild/backstage-plugin-agent-forge
Tag: a2a_stream_common_code-56

Usage

docker pull ghcr.io/cnoe-io/prebuild/backstage-plugin-agent-forge:a2a_stream_common_code-56

github-project-automation Bot added this to CAIPE (AI Platform Engineering) Project Backlog Oct 22, 2025

sriaradhyula requested a review from suwhang-cisco October 22, 2025 13:20

sriaradhyula commented Oct 22, 2025

View reviewed changes

suwhang-cisco approved these changes Oct 22, 2025

View reviewed changes

subbaksh approved these changes Oct 22, 2025

View reviewed changes

fix: lint

be08078

Signed-off-by: Sri Aradhyula <sraradhy@cisco.com>

sriaradhyula merged commit e22c633 into main Oct 31, 2025
50 of 54 checks passed

github-project-automation Bot moved this to Done in CAIPE (AI Platform Engineering) Project Backlog Oct 31, 2025

Uh oh!

Conversation

sriaradhyula commented Oct 22, 2025

Summary

Changes Made

🏗️ Common Infrastructure

🔄 Agent Refactoring

🐳 Docker & CI Fixes

🧹 Testing & Cleanup

Benefits

Docker Image Prebuilding

Breaking Changes

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 22, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

sriaradhyula Oct 22, 2025

Choose a reason for hiding this comment

Uh oh!

github-actions Bot commented Oct 23, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 23, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 23, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!

github-actions Bot commented Oct 23, 2025

🐳 Prebuild Docker Image Published

Usage

Uh oh!