Skip to content

feat: add Data Engineering Agent A2A client example#14283

Open
xtan999 wants to merge 1 commit into
GoogleCloudPlatform:mainfrom
xtan999:data-engineering-agent-a2a-client
Open

feat: add Data Engineering Agent A2A client example#14283
xtan999 wants to merge 1 commit into
GoogleCloudPlatform:mainfrom
xtan999:data-engineering-agent-a2a-client

Conversation

@xtan999
Copy link
Copy Markdown

@xtan999 xtan999 commented Jun 4, 2026

Description

Adds a sample Python client implementation using the Agent-to-Agent (A2A) protocol to interact with the Google Cloud Data Engineering Agent.

Key features:

  • A2A SDK integration for discovery and client creation.
  • ADC authentication support.
  • Multi-turn conversation state persistence via local tokens.
  • Handling of DEADLINE_EXCEEDED with automated resumption.
  • Support for custom agent instructions.

Fixes: #14252

Checklist

Testing

  • [ x] I have tested this change on a live environment and verified it works as intended.

Compliance & Style


Post-Approval Actions

  • [ x] Please merge this PR for me once it is approved

@xtan999 xtan999 requested review from a team as code owners June 4, 2026 21:58
@snippet-bot
Copy link
Copy Markdown

snippet-bot Bot commented Jun 4, 2026

Here is the summary of changes.

You are about to add 1 region tag.

This comment is generated by snippet-bot.
If you find problems with this result, please file an issue at:
https://github.com/googleapis/repo-automation-bots/issues.
To update this comment, add snippet-bot:force-run label or use the checkbox below:

  • Refresh this comment

@product-auto-label product-auto-label Bot added the samples Issues that are directly related to samples. label Jun 4, 2026
Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a Python-based Agent-to-Agent (A2A) client example for the Google Cloud Data Engineering Agent (DEA), complete with documentation, unit tests, and dependency specifications. While the implementation is solid, several key issues should be addressed: a trailing internal path in the test file causes a syntax error, invalid instruction paths are silently ignored rather than raising an error, a timeout mismatch between the HTTP client and the call context could lead to premature failures, and a potential bug in token extraction could inadvertently break session persistence by overwriting the conversation token with null.

Comment thread geminidataanalytics/dataengineeringagent/dea_a2a_client_test.py Outdated
Comment thread geminidataanalytics/dataengineeringagent/dea_a2a_client.py
Comment thread geminidataanalytics/dataengineeringagent/dea_a2a_client.py
Comment thread geminidataanalytics/dataengineeringagent/dea_a2a_client.py
@xtan999 xtan999 force-pushed the data-engineering-agent-a2a-client branch from c5fab08 to a96dd03 Compare June 4, 2026 22:13
Copy link
Copy Markdown
Author

@xtan999 xtan999 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Updated based on review comments

@xtan999 xtan999 force-pushed the data-engineering-agent-a2a-client branch from a96dd03 to d4f8f3c Compare June 4, 2026 22:29
Adds a sample Python client implementation using the Agent-to-Agent (A2A)
protocol to interact with the Google Cloud Data Engineering Agent.

Key features:
- A2A SDK integration for discovery and client creation.
- ADC authentication support.
- Multi-turn conversation state persistence via local tokens.
- Handling of DEADLINE_EXCEEDED with automated resumption.
- Support for custom agent instructions.

Fixes: GoogleCloudPlatform#14252
@xtan999 xtan999 force-pushed the data-engineering-agent-a2a-client branch from d4f8f3c to 27fea27 Compare June 5, 2026 00:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

samples Issues that are directly related to samples.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

feat(geminidataanalytics): Add Python A2A client sample for Data Engineering Agent

1 participant