o Zh@sxddlZddlZddlZddlmZddlmZmZddlZddl m Z ddl m Z GdddeZ Gdd d e ZdS) N)Enum)ListOptional)Document) BaseLoaderc@s\eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdS)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_MAINNETZ ETH_GOERLIZ ETH_SEPOLIAZ ETH_HOLESKYZPOLYGON_MAINNETZPOLYGON_MUMBAIZ POLYGON_AMOYZ ARB_MAINNETZ ARB_SEPOLIAZ OP_MAINNETZ OP_SEPOLIAZ BASE_MAINNETZ BASE_SEPOLIAZ BLAST_MAINNETZ BLAST_SEPOLIAZZKSYNC_MAINNETZZKSYNC_SEPOLIAZ ZORA_MAINNETZ ZORA_SEPOLIAr r f/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/document_loaders/blockchain.pyr s*rc@s~eZdZdZejddddfdededed ed ed ee f d d Z de e fddZ dedefddZededefddZdS)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_timecCsb||_|j|_tjdp||_||_||_||_ |js t dt d|js/t d|jdS)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). ZALCHEMY_API_KEYzAlchemy API key not provided.z^0x[a-fA-F0-9]{40}$zInvalid contract address N) rvaluerosenvirongetrrrr ValueErrorrematch)selfrrrrrrr r r__init__?sz!BlockchainDocumentLoader.__init__returnc Csg}|j}t} d|jd|jd|jd|}t|}|jdkr-td|j| d}|s6nA|D]}t |}|d d } |j|j| d } | t || d q8|j sZn||d jd }|jdurvt||jkrvtdq |std|j|S)NTzhttps://z.g.alchemy.com/nft/v2/z8/getNFTsForCollection?withMetadata=True&contractAddress=z &startToken=z Request failed with status code ZnftsidtokenId)sourceZ blockchainr#)Z page_contentmetadataz/Execution time exceeded the allowed time limit.z#No NFTs found for contract address )rtimerrrrequestsr status_coderjsonstrappendrr_get_next_tokenIdr%r RuntimeError) rresultZcurrent_start_token start_timeurlresponseitemsitemcontentr#r%r r rload_sP       + zBlockchainDocumentLoader.loadr#cCs||}|dkrt|d}n|dkrt|ddd}nt|}|d}|dkr:dt|dtt|ddS|dkrOd t|dtt|d dSt|S) Nhex_0xhex_0xbf0x0x0xbf)_detect_value_typeintformatr+len)rr#Z value_typeZ value_intr/r r rr-s  ""z*BlockchainDocumentLoader._get_next_tokenIdcCs.t|trdS|drdS|drdSdS)NrBr<r7r?r9) isinstancerB startswith)r#r r rrAs   z+BlockchainDocumentLoader._detect_value_type)rr r r rr r+boolrrBrrrr6r- staticmethodrAr r r rr%s0  :r)rrr'enumrtypingrrr(Zlangchain_core.documentsrZ)langchain_community.document_loaders.baserrrr r r rs