Skip to content
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

Add min_score_confidence support for the Bedrock KB retriver #49

Merged

Conversation

ihmaws
Copy link
Contributor

@ihmaws ihmaws commented May 17, 2024

Add min_score_confidence support for the Bedrock KB retriver

  • add min_score_confidence field to AmazonKnowledgeBasesRetriever class
  • add unit tests and integration tests to test score
  • update kendra retriver _filter_by_score_confidence doc string

Closes: #48

  - add min_score_confidence field to AmazonKnowledgeBasesRetriever class
  - add unit tests and integration tests to test score
  - update kendra retriver _filter_by_score_confidence doc string
@3coins
Copy link
Collaborator

3coins commented May 17, 2024

cc @rozerarenu to review this PR.

@ihmaws
Copy link
Contributor Author

ihmaws commented May 21, 2024

Thanks for having a look and approving @rozerarenu and @3coins!

Any thoughts on when this would be merged and put into a release? I'm currently extending the AmazonKnowledgeBasesRetriever and using the code in this PR in an application today, but will switch to directly using it from langchain once released.

Let me know if there is anything else I can help with to get this into a release.

Copy link
Collaborator

@3coins 3coins left a comment

Choose a reason for hiding this comment

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

@ihmaws
Looks good! Thanks for making these updates.

@3coins 3coins merged commit 3bc0c39 into langchain-ai:main May 21, 2024
12 checks passed
@ihmaws ihmaws deleted the dev/ihm/add-score-filtering-for-bedrock-kb branch May 21, 2024 17:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add functionality to filter Bedrock Knowledge base results by a minimum confidence score
3 participants