a bg"@slddlZddlZddlZddlZddlmZmZmZmZddl Z ddl m Z ddl m Z Gddde ZdS)N)IteratorListLiteralOptional)Document) BaseLoaderc @speZdZdZdddddddeedeeeeeeeee ddd Z ee d d d Z e e d d dZdS)MintbaseDocumentLoaderaLoad elements from a blockchain smart contract. The supported blockchains are: Near mainnet, Near testnet. If no BlockchainType is specified, the default is Near mainnet. The Loader uses the Mintbase API to interact with the blockchain. MB_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 Mintbase APIs (e.g. getTokens, etc.) Example: .. code-block:: python contractAddress = "nft.yearofchef.near" # Year of chef contract address blockchainLoader = MintbaseDocumentLoader( contract_address=contractAddress, blockchain_type="mainnet",api_key="omni-site" ) NF)api_keytableselectfieldsget_all_tokensmax_execution_time)ZmainnetZtestnet)contract_addressblockchain_typer r r r rrc Cs|||_||_tjdp||_dp$||_dp.||_|ptd|j|S)Nx query MyQuery { table(select) { fields } } r rr r  application/jsonz mb-api-keyz Content-TypeMyQueryquery variablesZ operationNamehttps://graph.mintbase.xyz/headersdata Request failed with status code r0rrsourceZ blockchaintokenIdZ page_contentmetadata/Execution time exceeded the allowed time limit.z#No NFTs found for contract address )timereplacer rr joinr r rrequestspostjsondumps status_coderstrappendrrr RuntimeError) r result start_timeoperations_docr/r0urlresponseitemsitemcontentrr7r!r!r"loadsV       zMintbaseDocumentLoader.loadc cs"t}d}|d|j}|d|j}|d|j}|dd|j}|jdd}|id d }d |j}t j ||t |d }|j d krtd|j | dd}|sq|D]2}t|} |d} |j|j| d} t| | dVq|jsq|jdurt||jkrtdqdS)Nr%r rr r r&r'r(r)r*r-r.r1r2r0rrr3r6r8)r9r:r rr r;r r rr<r=r>r?r@rrArrrrC) r rErFr/r0rGrHrIrJrKr5r7r!r!r" lazy_loadsJ     z MintbaseDocumentLoader.lazy_load)__name__ __module__ __qualname____doc__rArrrboolintr#rrLrrMr!r!r!r"r s&%  SIr)r>rrr9typingrrrrr<Zlangchain_core.documentsrZ)langchain_community.document_loaders.baserrr!r!r!r"s