7|hZddlmZddlmZmZddlmZddlmZddl m Z Gdde Z y) ) annotations)AnyList)CallbackManagerForRetrieverRun)Document) BaseRetrieverc^eZdZUdZded<ded<e d d dZ d dZy) KayAiRetrieverz 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_contextscf ddlm}|||}|||S#t$r tdwxYw)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_typesr rr s a/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/retrievers/kay.pycreatezKayAiRetriever.createsH$  7j*5&|<< %  s0c|jj||j}g}|D]5}|jdd}||j t |i|7|S)N)query num_contextchunk_embed_text) page_contentmetadata)r rr popappendr)selfr run_managerctxsdocsctxrs r_get_relevant_documentsz&KayAiRetriever._get_relevant_documents2sq{{  u$:K:K L OC77#5tr5s "C-32]2r%