U hW @sRddlmZddlmZmZmZmZddlmZer@ddl m Z GdddZ dS)) annotations) TYPE_CHECKINGIteratorListOptional) get_from_envODPSc@s`eZdZdZddddZedddddd d dd d d Zdd dddZdddddZdS)MaxComputeAPIWrapperz7Interface for querying Alibaba Cloud MaxCompute tables.r clientcCs ||_dS)zvInitialize MaxCompute document loader. Args: client: odps.ODPS MaxCompute client object. Nr )selfr rM/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/max_compute.py__init__szMaxComputeAPIWrapper.__init__N) access_idsecret_access_keystrz Optional[str])endpointprojectrrreturnc Cszddlm}Wn,tk r<}ztd|W5d}~XYnX|pJtdd}|pXtdd}|||||d }||std |d ||S) azConvenience constructor that builds the odsp.ODPS MaxCompute client from given parameters. Args: endpoint: MaxCompute endpoint. project: A project is a basic organizational unit of MaxCompute, which is similar to a database. access_id: MaxCompute access ID. Should be passed in directly or set as the environment variable `MAX_COMPUTE_ACCESS_ID`. secret_access_key: MaxCompute secret access key. Should be passed in directly or set as the environment variable `MAX_COMPUTE_SECRET_ACCESS_KEY`. rrzCould not import pyodps python package. Please install it with `pip install pyodps` or refer to https://pyodps.readthedocs.io/.NrZMAX_COMPUTE_ACCESS_IDrZMAX_COMPUTE_SECRET_ACCESS_KEY)rrrrz The project "z" does not exist.)odpsr ImportErrorrZ exist_project ValueError)clsrrrrr exr rrr from_paramss, z MaxComputeAPIWrapper.from_paramszIterator[dict])queryrc csL|j|2}|jdkr$td|D]}dd|DVq(W5QRXdS)NrzTable contains no data.cSsi|]\}}||qSrr).0kvrrr Isz3MaxComputeAPIWrapper.lazy_query..)r Z execute_sqlZ open_readercountr)r rreaderrecordrrr lazy_queryCs  zMaxComputeAPIWrapper.lazy_queryz List[dict]cCst||S)N)listr%)r rrrrrKszMaxComputeAPIWrapper.query) __name__ __module__ __qualname____doc__r classmethodrr%rrrrrr s,r N) __future__rtypingrrrrZlangchain_core.utilsrrr r rrrrs