Skip to content

feat(bedrock_agent): add new Amazon Bedrock Agents Functions Resolver #6564

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 32 commits into from
Jun 3, 2025

Conversation

anafalcao
Copy link
Contributor

@anafalcao anafalcao commented Apr 25, 2025

Issue number: #6300

Summary

This PR introduces support for Bedrock Agents Functions, providing the user a similar experience from the current Bedrock Agents with OpenAPI Resolver.

Changes

  • Bedrock Agent Functions data class
  • BedrockFunctionsResponseBuilder
  • BedrockAgentFunctionResolver
  • Optional response fields - sessionAttributes, promptSessionAttributes, knowledgeBasesConfiguration
  • Parser
  • Additional validations
  • name on Tool decorator
  • Documentation with examples

User experience

from aws_lambda_powertools.event_handler import BedrockAgentFunctionResolver
from aws_lambda_powertools.utilities.typing import LambdaContext

app = BedrockAgentFunctionResolver()

@app.tool(name="currentTime", description="Gets the current UTC time")
def get_current_time():
    ... 
@app.tool(description="Greets the user using name")
def greet_user(name):
...
@app.tool(description="Add two numbers")
def simple_calculator(a, b, operation):
...
def lambda_handler(event: dict, context: LambdaContext) → dict:
    return app.resolve(event, context)

Checklist

If your change doesn't seem to apply, please leave them unchecked.

Is this a breaking change?

RFC issue number:

Checklist:

  • Migration process documented
  • Implement warnings (if it can live side by side)

Acknowledgment

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

Disclaimer: We value your time and bandwidth. As such, any pull requests created on non-triaged issues might not be successful.

@boring-cyborg boring-cyborg bot added the tests label Apr 25, 2025
@pull-request-size pull-request-size bot added the size/L Denotes a PR that changes 100-499 lines, ignoring generated files. label Apr 25, 2025
@github-actions github-actions bot added the feature New feature or functionality label Apr 25, 2025
@anafalcao anafalcao self-assigned this Apr 25, 2025
@anafalcao anafalcao linked an issue Apr 25, 2025 that may be closed by this pull request
2 tasks
Copy link

codecov bot commented Apr 25, 2025

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 96.15%. Comparing base (9c16b81) to head (13d5569).
Report is 1 commits behind head on develop.

Additional details and impacted files
@@             Coverage Diff             @@
##           develop    #6564      +/-   ##
===========================================
+ Coverage    96.09%   96.15%   +0.05%     
===========================================
  Files          254      256       +2     
  Lines        12179    12350     +171     
  Branches       905      916      +11     
===========================================
+ Hits         11704    11875     +171     
  Misses         373      373              
  Partials       102      102              

☔ 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.

@pull-request-size pull-request-size bot added size/XL Denotes a PR that changes 500-999 lines, ignoring generated files. and removed size/L Denotes a PR that changes 100-499 lines, ignoring generated files. labels Apr 25, 2025
@anafalcao anafalcao changed the title feat(bedrock_agent): create bedrock agents functions data class feat(bedrock_agent): Add new Amazon Bedrock Agents Functions Apr 25, 2025
@anafalcao anafalcao changed the title feat(bedrock_agent): Add new Amazon Bedrock Agents Functions feat(bedrock_agent): WIP - Add new Amazon Bedrock Agents Functions Apr 25, 2025
@pull-request-size pull-request-size bot added size/L Denotes a PR that changes 100-499 lines, ignoring generated files. and removed size/XL Denotes a PR that changes 500-999 lines, ignoring generated files. labels May 1, 2025
@pull-request-size pull-request-size bot added size/XL Denotes a PR that changes 500-999 lines, ignoring generated files. and removed size/L Denotes a PR that changes 100-499 lines, ignoring generated files. labels May 2, 2025
Copy link
Contributor

⚠️Large PR detected⚠️

Please consider breaking into smaller PRs to avoid significant review delays. Ignore if this PR has naturally grown to this size after reviews.

@github-actions github-actions bot removed the documentation Improvements or additions to documentation label May 28, 2025
@boring-cyborg boring-cyborg bot added the documentation Improvements or additions to documentation label May 28, 2025
Copy link
Contributor

⚠️Large PR detected⚠️

Please consider breaking into smaller PRs to avoid significant review delays. Ignore if this PR has naturally grown to this size after reviews.

@github-actions github-actions bot removed the documentation Improvements or additions to documentation label May 28, 2025
@boring-cyborg boring-cyborg bot added the documentation Improvements or additions to documentation label May 28, 2025
Copy link
Contributor

⚠️Large PR detected⚠️

Please consider breaking into smaller PRs to avoid significant review delays. Ignore if this PR has naturally grown to this size after reviews.

dreamorosi
dreamorosi previously approved these changes May 28, 2025
@boring-cyborg boring-cyborg bot added the documentation Improvements or additions to documentation label Jun 2, 2025
Copy link
Contributor

github-actions bot commented Jun 2, 2025

⚠️Large PR detected⚠️

Please consider breaking into smaller PRs to avoid significant review delays. Ignore if this PR has naturally grown to this size after reviews.

@github-actions github-actions bot removed the documentation Improvements or additions to documentation label Jun 2, 2025
@boring-cyborg boring-cyborg bot added the documentation Improvements or additions to documentation label Jun 2, 2025
@github-actions github-actions bot removed the documentation Improvements or additions to documentation label Jun 2, 2025
Copy link
Contributor

github-actions bot commented Jun 2, 2025

⚠️Large PR detected⚠️

Please consider breaking into smaller PRs to avoid significant review delays. Ignore if this PR has naturally grown to this size after reviews.

@boring-cyborg boring-cyborg bot added the documentation Improvements or additions to documentation label Jun 3, 2025
@github-actions github-actions bot removed the documentation Improvements or additions to documentation label Jun 3, 2025
Copy link
Contributor

github-actions bot commented Jun 3, 2025

⚠️Large PR detected⚠️

Please consider breaking into smaller PRs to avoid significant review delays. Ignore if this PR has naturally grown to this size after reviews.

Copy link

sonarqubecloud bot commented Jun 3, 2025

@dreamorosi dreamorosi self-requested a review June 3, 2025 09:22
@leandrodamascena leandrodamascena merged commit b66665c into develop Jun 3, 2025
14 checks passed
@leandrodamascena leandrodamascena deleted the feat/bedrock_functions branch June 3, 2025 09:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
event_handlers feature New feature or functionality size/XXL Denotes a PR that changes 1000+ lines, ignoring generated files. tests
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Feature request: Amazon Bedrock Agents Functions
3 participants