U h@sPddlZddlmZddlmZmZddlmZddlm Z Gddde Z dS)N)Path)IteratorUnion)Document) BaseLoaderc@s8eZdZdZeeefdddZee dddZ dS) TomlLoaderziLoad `TOML` files. It can load a single source file or several files in a single directory. )sourcecCst||_dS)z:Initialize the TomlLoader with a source file or directory.N)rr)selfrr M/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/document_loaders/toml.py__init__szTomlLoader.__init__)returnc csddl}|jr(|jjdkr(|jg}n$|jrDt|jd}ntd|D]}|jdddv}| }z.| |}t t |d t|id }|VWn8|jk r}ztd |d |W5d}~XYnXW5QRXqPdS) zALazily load the TOML documents from the source file or directory.rNz.tomlz **/*.tomlz Invalid source path or file typerzutf-8)encodingr)Z page_contentmetadatazError parsing TOML file z: )tomliris_filesuffixis_dirlistglob ValueErroropenreadloadsrjsondumpsstrZTOMLDecodeErrorprint) r rfiles file_pathfilecontentdatadocer r r lazy_loads$     zTomlLoader.lazy_loadN) __name__ __module__ __qualname____doc__rrrr rrr&r r r r r sr) rpathlibrtypingrrZlangchain_core.documentsrZ)langchain_community.document_loaders.baserrr r r r s