dhZSSKJr SSKJrJr SSKJr SSKJr SSK J r "SS\ 5r g) ) annotations)AnyList)CallbackManagerForRetrieverRun)Document) BaseRetrievercl\rSrSr%SrS\S'S\S'\S S Sjj5rS SjrS r g )KayAiRetriever z Retriever for Kay.ai datasets. To work properly, expects you to have KAY_API_KEY env variable set. You can get one for free at https://kay.ai/. rclientint num_contextsc^SSKJn U"X5nU"XSS9$![a [S5ef=f)a Create a KayRetriever given a Kay dataset id and a list of datasources. Args: dataset_id: A dataset id category in Kay, like "company" data_types: A list of datasources present within a dataset. For "company" the corresponding datasources could be ["10-K", "10-Q", "8-K", "PressRelease"]. num_contexts: The number of documents to retrieve on each query. Defaults to 6. r) KayRetrieverzNCould not import kay python package. Please install it with `pip install kay`.)r r)kay.rag.retrieversr ImportError)cls dataset_id data_typesrrr s Z/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/retrievers/kay.pycreateKayAiRetriever.createsD$  7j5&<< %  s,cURRXRS9n/nUH5nURSS5nUcMUR [ U0UES95 M7 U$)N)query num_contextchunk_embed_text) page_contentmetadata)r rrpopappendr)selfr run_managerctxsdocsctxrs r_get_relevant_documents&KayAiRetriever._get_relevant_documents2sh{{  u:K:K LC77#5tr:s "C-32]2r(