Skip to content

Commit

Permalink
feat: add Kendra DocumentRelevanceOverrideConfigurations request para…
Browse files Browse the repository at this point in the history
…meter
  • Loading branch information
scaseriniklarna committed Apr 20, 2024
1 parent c909ae0 commit 8327d22
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 8327d22

Please sign in to comment.