Skip to content

Commit

Permalink
community: add Kendra DocumentRelevanceOverrideConfigurations request…
Browse files Browse the repository at this point in the history
… parameter (langchain-ai#20695)

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

Co-authored-by: Simone Caserini <[email protected]>
Co-authored-by: Bagatur <[email protected]>
  • Loading branch information
3 people authored and dglogo committed May 8, 2024
1 parent 36d91ab commit 33ebad9
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions libs/community/langchain_community/retrievers/kendra.py
Original file line number Diff line number Diff line change
Expand Up @@ -344,6 +344,10 @@ class AmazonKendraRetriever(BaseRetriever):
attribute_filter: Additional filtering of results based on metadata
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
allowing access to all result item attributes. By default, it uses
the item's title and excerpt.
Expand All @@ -367,6 +371,7 @@ class AmazonKendraRetriever(BaseRetriever):
credentials_profile_name: Optional[str] = None
top_k: int = 3
attribute_filter: Optional[Dict] = None
document_relevance_override_configurations: Optional[List[Dict]] = None
page_content_formatter: Callable[[ResultItem], str] = combined_text
client: Any
user_context: Optional[Dict] = None
Expand Down Expand Up @@ -421,6 +426,10 @@ def _kendra_query(self, query: str) -> Sequence[ResultItem]:
}
if self.attribute_filter is not None:
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:
kendra_kwargs["UserContext"] = self.user_context

Expand Down

0 comments on commit 33ebad9

Please sign in to comment.