Create a KayRetriever given a Kay dataset id and a list of datasources.
create(
cls,
dataset_id: str,
data_types: List[str],
num_contexts: int = 6
) -> KayAiRetriever| Name | Type | Description |
|---|---|---|
dataset_id* | str | A dataset id category in Kay, like "company" |
data_types* | List[str] | A list of datasources present within a dataset. For "company" the corresponding datasources could be ["10-K", "10-Q", "8-K", "PressRelease"]. |
num_contexts | int | Default: 6The number of documents to retrieve on each query. Defaults to 6. |