o ZhO @sddlZddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZmZeeZGddde ZdS) N)Path)DictIteratorListUnion)BaseChatLoader) ChatSession) AIMessage HumanMessagec@sreZdZdZdeeeffddZede e de fddZ ed e jd ede efd d Zdee fd dZdS)SlackChatLoaderz0Load `Slack` conversations from a dump zip file.pathcCs8t|tr|nt||_|jstd|jddS)z Initialize the chat loader with the path to the exported Slack dump zip file. :param path: Path to the exported Slack dump zip file. zFile z not foundN) isinstancerzip_pathexistsFileNotFoundError)selfr r]/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/chat_loaders/slack.py__init__s zSlackChatLoader.__init__messagesreturnc Csg}d}|D]Y}t|tsq|dd}|dd}|dd}|s#qtjdtjd}||r1q||krM|djd|7_|djd  d |in| t |||d |igd d |}qt |d S)Ntexttsuserz<@U\d+> has joined the channel)flagsz eventsZ message_time)senderr)Zrolecontentadditional_kwargs)r) r dictgetrecompile IGNORECASEmatchrr appendr r)rresultsZprevious_sendermessager timestamprZ skip_patternrrr_load_single_chat_sessions>      z)SlackChatLoader._load_single_chat_sessionzip_file file_pathcCsV||d }t|}Wdn1swYt|ts)tdt||S)z"Read JSON data from a zip subfile.rNz#Expected list of dictionaries, got )openjsonloadr list ValueErrortype)r,r-fdatarrr _read_jsonCs   zSlackChatLoader._read_jsonccsjtt|jd!}|D]}|dr"|||}||VqWddS1s.wYdS)z Lazy load the chat sessions from the Slack dump file and yield them in the required format. :return: Iterator of chat sessions containing messages. r.z.jsonN)zipfileZipFilestrrnamelistendswithr7r+)rr,r-rrrr lazy_loadLs    "zSlackChatLoader.lazy_loadN)__name__ __module__ __qualname____doc__rr:rr staticmethodrrrr+r8r9r!r7rr=rrrrr s   #r )r0loggingr#r8pathlibrtypingrrrrZlangchain_core.chat_loadersrZlangchain_core.chat_sessionsrZlangchain_core.messagesr r getLoggerr>loggerr rrrrs