Python: feat: add Amazon Bedrock Knowledge Base tool and context provider

[PVidyadhar]

feat: add Amazon Bedrock Knowledge Base tool and context provider

Motivation & Context

Enables Agent Framework agents to retrieve context from Amazon Bedrock Knowledge Bases. This adds RAG capabilities using AWS's managed infrastructure without requiring users to manage their own vector stores or embedding pipelines.

Description & Review Guide

• What are the major changes?

  • BedrockKnowledgeBaseTool — standalone tool with async run() + get_tool_definition() for agents to call on-demand
  • BedrockKnowledgeBaseProvider — ContextProvider subclass with before_run() for automatic context injection on every agent invocation
  • Both support managed search (managedSearchConfiguration) and agentic retrieval with fallback

• What is the impact of these changes?

  • Additive only — new files in python/packages/tools/, no existing code modified
  • Users get two integration patterns: explicit tool calling or automatic context injection

• What do you want reviewers to focus on?

  • Whether the tool/provider pattern matches the framework's conventions
  • The ContextProvider integration with before_run()

Related Issue

N/A — new feature adding Amazon Bedrock Knowledge Base integration.

Contribution Checklist

• The code builds clean without any errors or warnings
• All unit tests pass, and I have added new tests where possible
• The PR follows the Contribution Guidelines
• This PR is linked to an issue and there is no other open PR for this issue (see Related Issue above).
• This is not a breaking change.

Testing

• Tool.run() returned 5 results from live managed KB in us-west-2
• Provider._retrieve() returned 5 passages
• _format_context() produced 7399 chars of formatted context
• SDK requirement: boto3 >= 1.43
• Microsoft CLA: pre-approved (cla.opensource.microsoft.com)

[Copilot]

Pull request overview

Adds Amazon Bedrock Knowledge Base retrieval capabilities to the Python tools package, enabling both explicit tool-based retrieval and automatic context injection via a ContextProvider.

Changes:

• Introduces BedrockKnowledgeBaseTool with async run() + get_tool_definition() for on-demand KB retrieval.
• Introduces BedrockKnowledgeBaseProvider (ContextProvider.before_run) to inject KB context automatically per agent invocation.
• Adds a design/usage doc plus unit tests for the new tool/provider.

Reviewed changes

Copilot reviewed 4 out of 4 changed files in this pull request and generated 17 comments.

File	Description
python/packages/tools/agent_framework_tools/bedrock_knowledge_base.py	Adds the Bedrock Knowledge Base tool implementation and tool definition metadata.
python/packages/tools/agent_framework_tools/bedrock_knowledge_base_provider.py	Adds a context provider that retrieves and injects KB passages before runs.
python/packages/tools/tests/test_bedrock_knowledge_base.py	Adds unit tests covering tool retrieval, default config behavior, and provider retrieval.
python/packages/tools/agent_framework_tools/BEDROCK_MANAGED_KB.md	Adds documentation describing usage, configuration, and permissions.

[PVidyadhar]

@microsoft-github-policy-service agree company="Amazon"