o Zh@sPddlZddlmZddlmZmZddlmZddlm Z Gddde Z dS)N)Path)IteratorUnion)Document) BaseLoaderc@s8eZdZdZdeeeffddZdee fddZ 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 `/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/document_loaders/toml.py__init__szTomlLoader.__init__returnc csddl}|jr|jjdkr|jg}n|jr#t|jd}ntd|D]R}|jdddA}| }z| |}t t |d t|id }|VWn|jyk}ztd |d |WYd}~nd}~wwWdn1svwYq)dS) 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 ValueErroropenreadloadsrjsondumpsstrTOMLDecodeErrorprint) r rfiles file_pathfilecontentdatadocer r r lazy_loads2      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