7|hddlZddlZddlZddlmZddlmZmZddlZddl m Z ddl m Z GddeZ Gdd e Zy) N)Enum)ListOptional)Document) BaseLoaderc\eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZy)BlockchainTypez(Enumerator of the supported blockchains.z eth-mainnetz eth-goerliz eth-sepoliaz eth-holeskyzpolygon-mainnetzpolygon-mumbaiz polygon-amoyz arb-mainnetz arb-sepoliaz opt-mainnetz opt-sepoliaz base-mainnetz base-sepoliaz blast-mainnetz blast-sepoliazzksync-mainnetzzksync-sepoliaz zora-mainnetz zora-sepoliaN)__name__ __module__ __qualname____doc__ ETH_MAINNET ETH_GOERLI ETH_SEPOLIA ETH_HOLESKYPOLYGON_MAINNETPOLYGON_MUMBAI POLYGON_AMOY ARB_MAINNET ARB_SEPOLIA OP_MAINNET OP_SEPOLIA BASE_MAINNET BASE_SEPOLIA BLAST_MAINNET BLAST_SEPOLIAZKSYNC_MAINNETZKSYNC_SEPOLIA ZORA_MAINNET ZORA_SEPOLIAn/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/document_loaders/blockchain.pyr r sd2KJKK'O%N!LKKJJ!L!L#M#M%N%N!L!Lr"r ceZdZdZej ddddfdededed ed ed ee f d Z d e e fdZ ded efdZeded efdZy)BlockchainDocumentLoadera9Load elements from a blockchain smart contract. See supported blockchains here: https://python.langchain.com/v0.2/api_reference/community/document_loaders/langchain_community.document_loaders.blockchain.BlockchainType.html If no BlockchainType is specified, the default is Ethereum mainnet. The Loader uses the Alchemy API to interact with the blockchain. ALCHEMY_API_KEY environment variable must be set to use this loader. The API returns 100 NFTs per request and can be paginated using the startToken parameter. If get_all_tokens is set to True, the loader will get all tokens on the contract. Note that for contracts with a large number of tokens, this may take a long time (e.g. 10k tokens is 100 requests). Default value is false for this reason. The max_execution_time (sec) can be set to limit the execution time of the loader. Future versions of this loader can: - Support additional Alchemy APIs (e.g. getTransactions, etc.) - Support additional blockchain APIs (e.g. Infura, Opensea, etc.) z docs-demoFNcontract_addressblockchainTypeapi_key startTokenget_all_tokensmax_execution_timecL||_|j|_tjj dxs||_||_||_||_ |j s tdtjd|jstd|jy)au Args: contract_address: The address of the smart contract. blockchainType: The blockchain type. api_key: The Alchemy API key. startToken: The start token for pagination. get_all_tokens: Whether to get all tokens on the contract. max_execution_time: The maximum execution time (sec). ALCHEMY_API_KEYzAlchemy API key not provided.z^0x[a-fA-F0-9]{40}$zInvalid contract address N) r'valuer(osenvirongetr)r*r+r, ValueErrorrematch)selfr'r(r)r*r+r,s r#__init__z!BlockchainDocumentLoader.__init__?s&!1,22zz~~&78CG $,"4||<= =xx.0E0EF89N9N8OPQ QGr"returncg}|j}tj} d|jd|jd|jd|}t j |}|jdk7rtd|j|jd}|sn|D]K}t|}|dd } |j|j| d } |jt|| M|jsn^|j|d jd }|j /tj|z |j kDr t#d 9|std|j|S)Nzhttps://z.g.alchemy.com/nft/v2/z8/getNFTsForCollection?withMetadata=True&contractAddress=z &startToken=z Request failed with status code nftsidtokenId)source blockchainr=) page_contentmetadataz/Execution time exceeded the allowed time limit.z#No NFTs found for contract address )r*timer(r)r'requestsr2 status_coder3jsonstrappendrr+_get_next_tokenIdrAr, RuntimeError) r6resultcurrent_start_token start_timeurlresponseitemsitemcontentr=rAs r#loadzBlockchainDocumentLoader.load_s"ooYY[ 4..//E<<.!((,(=(='>235  ||C(H##s* 6x7K7K6LMMMOF+E Qd)t*Y/"33"&"5"5&  hGhOP Q&&#'"8"89L9LY9W"X ''3YY[:-1H1HH"#TUUSV5d6K6K5LM  r"r=c d|j|}|dk(r t|d}n |dk(rt|ddd}n t|}|dz}|dk(r*dt|dtt |dz zdzzS|dk(r*d t|dtt |d z zdzzSt|S) Nhex_0xhex_0xbf0x0x0xbf)_detect_value_typeintformatrGlen)r6r= value_type value_intrKs r#rIz*BlockchainDocumentLoader._get_next_tokenIds,,W5  !GR(I : %GABK,IG IQ  !&s3wrns7 !-@"T"0QzQr"