NgbddlmZddlmZmZddlmZddlmZddl m Z Gdde Z dS) ) annotations)AnyList)CallbackManagerForRetrieverRun)Document) BaseRetrievercLeZdZUdZded<ded<e ddd ZddZdS)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_contexts dataset_idstr data_types List[str]returnc~ ddlm}n#t$rtdwxYw|||}|||S)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)clsrrr rr s ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/retrievers/kay.pycreatezKayAiRetriever.createsv$  7 7 7 7 7 7 7   %   j*55s&|<<<r7s""""""CCCCCC------33333322222]22222r*