Fix Bedrock Knowledge Base retrieval by adding overrideSearchType parameter #163
+4
−1
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
…ameter This commit fixes an issue with the Strands Agent's retrieve tool when accessing Amazon Bedrock Knowledge Bases. The primary changes include: 1. Updated the retrieve.py implementation to include the required "overrideSearchType" parameter in the vectorSearchConfiguration object when calling the Bedrock Agent Runtime API 2. Added explicit AWS region configuration to ensure the tool connects to the correct region where the Knowledge Base is deployed. os.environ["AWS_REGION"] = "us-east-1" The issue was causing "ResourceNotFoundException" errors even when the Knowledge Base existed because the API call was missing a required parameter in the retrievalConfiguration. This fix ensures compatibility with the current Bedrock Agent Runtime API specification. Testing: - Verified Knowledge Base existence using AWS CLI - Successfully retrieved content from the Knowledge Base using the updated tool - Confirmed proper scoring and filtering of results This change maintains backward compatibility while ensuring the tool works correctly with the current API version.
|
Thank you for your contribution! We truly appreciate community members like you taking the time to help improve the project. Based on bedrock docs, overrideSearchType is not a required parameter. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR fixes an issue with the Strands Agent's retrieve tool when accessing Amazon Bedrock Knowledge Bases. The problem was causing "ResourceNotFoundException" errors even when the Knowledge Base existed because the API call was missing a required parameter in the retrievalConfiguration.
The fix ensures compatibility with the current Bedrock Agent Runtime API specification by adding the required "overrideSearchType" parameter to the vectorSearchConfiguration object.
Changes Made
Testing Performed
-Root Cause Analysis
The issue was caused by changes in the Bedrock Agent Runtime API that now requires the "overrideSearchType" parameter in the vectorSearchConfiguration object. Without this parameter, the API returns a ResourceNotFoundException error even when the Knowledge Base exists.
Related Issues
"ResourceNotFoundException when using retrieve tool with Bedrock Knowledge Base"
Type of Change
[x] Bug fix
[ ] New Tool
[ ] Breaking change
[ ] Other (please describe):
Screenshots/Logs
Before fix:
Error during retrieval: An error occurred (ResourceNotFoundException) when calling the Retrieve operation: Knowledge Base with id XXXXXXXXX does not existAfter fix: