a bgO @sddlZddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZmZeeZGddde ZdS) N)Path)DictIteratorListUnion)BaseChatLoader) ChatSession) AIMessage HumanMessagec@sleZdZdZeeefdddZee e e dddZ ee jee edd d Zee d d d ZdS)SlackChatLoaderz0Load `Slack` conversations from a dump zip file.)pathcCs8t|tr|nt||_|js4td|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 rt/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/chat_loaders/slack.py__init__s  zSlackChatLoader.__init__)messagesreturnc Csg}d}|D]}t|tsq |dd}|dd}|dd}|sFq tjdtjd}||rbq ||kr|djd|7_|djd  d |in | t |||d |igd d |}q t |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_session)zip_file file_pathrcCsV||d}t|}Wdn1s,0Yt|tsRtdt||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_json)rccsftt|jd@}|D]&}|dr|||}||VqWdn1sX0YdS)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