o ȳgW @sRddlmZddlmZmZmZmZddlmZer ddl m Z GdddZ dS)) annotations) TYPE_CHECKINGIteratorListOptional) get_from_envODPSc@sDeZdZdZdddZeddddddZdddZdddZdS)MaxComputeAPIWrapperz7Interface for querying Alibaba Cloud MaxCompute tables.clientr cCs ||_dS)zvInitialize MaxCompute document loader. Args: client: odps.ODPS MaxCompute client object. N)r )selfr r e/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/max_compute.py__init__s zMaxComputeAPIWrapper.__init__N) access_idsecret_access_keyendpointstrprojectr Optional[str]rreturnc Cszddlm}Wnty}ztd|d}~ww|p tdd}|p'tdd}|||||d }||s=td |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/.NrMAX_COMPUTE_ACCESS_IDrMAX_COMPUTE_SECRET_ACCESS_KEY)rrrrz The project "z" does not exist.)odpsr ImportErrorr exist_project ValueError)clsrrrrr exr r r r from_paramss. z MaxComputeAPIWrapper.from_paramsqueryIterator[dict]ccsd|j|}|jdkrtd|D] }dd|DVqWddS1s+wYdS)NrzTable contains no data.cSsi|]\}}||qSr r ).0kvr r r Isz3MaxComputeAPIWrapper.lazy_query..)r execute_sql open_readercountr)r r readerrecordr r r lazy_queryCs "zMaxComputeAPIWrapper.lazy_query List[dict]cCst||S)N)listr+)r r r r rr KszMaxComputeAPIWrapper.query)r r ) rrrrrrrrrr )r rrr!)r rrr,) __name__ __module__ __qualname____doc__r classmethodrr+r r r r rr s  ,r N) __future__rtypingrrrrlangchain_core.utilsrrr r r r r rs