github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 17:08:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: add Kendra DocumentRelevanceOverrideConfigurations request parameter (#20695)

Browse Source 
- **Description:** add **DocumentRelevanceOverrideConfigurations**
request parameter to Kendra retriever

Co-authored-by: Simone Caserini <simone.caserini@klarna.com>
Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
This commit is contained in:
scaserini 2024-05-06 23:26:36 +02:00 committed by GitHub
parent 0345bcf4ef
commit a6cdf6572f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
libs/community/langchain_community/retrievers/kendra.py
View File

@ -344,6 +344,10 @@ class AmazonKendraRetriever(BaseRetriever):
attribute_filter: Additional filtering of results based on metadata attribute_filter: Additional filtering of results based on metadata
See: https://docs.aws.amazon.com/kendra/latest/APIReference See: https://docs.aws.amazon.com/kendra/latest/APIReference
document_relevance_override_configurations: Overrides relevance tuning
configurations of fields/attributes set at the index level
See: https://docs.aws.amazon.com/kendra/latest/APIReference
page_content_formatter: generates the Document page_content page_content_formatter: generates the Document page_content
allowing access to all result item attributes. By default, it uses allowing access to all result item attributes. By default, it uses
the item's title and excerpt. the item's title and excerpt.
@ -367,6 +371,7 @@ class AmazonKendraRetriever(BaseRetriever):
credentials_profile_name: Optional[str] = None credentials_profile_name: Optional[str] = None
top_k: int = 3 top_k: int = 3
attribute_filter: Optional[Dict] = None attribute_filter: Optional[Dict] = None
document_relevance_override_configurations: Optional[List[Dict]] = None
page_content_formatter: Callable[[ResultItem], str] = combined_text page_content_formatter: Callable[[ResultItem], str] = combined_text
client: Any client: Any
user_context: Optional[Dict] = None user_context: Optional[Dict] = None
@ -421,6 +426,10 @@ class AmazonKendraRetriever(BaseRetriever):
} }
if self.attribute_filter is not None: if self.attribute_filter is not None:
kendra_kwargs["AttributeFilter"] = self.attribute_filter kendra_kwargs["AttributeFilter"] = self.attribute_filter
if self.document_relevance_override_configurations is not None:
kendra_kwargs[
"DocumentRelevanceOverrideConfigurations"
] = self.document_relevance_override_configurations
if self.user_context is not None: if self.user_context is not None:
kendra_kwargs["UserContext"] = self.user_context kendra_kwargs["UserContext"] = self.user_context