o |q i @sdZddlmZddlmZmZmZddlmZddl m Z m Z  ddede d efd d Z d eed eefd dZd ede d efddZd eede d eefddZdS)zUtilities for chat loaders.)deepcopy)IterableIteratorList) ChatSession) AIMessage BaseMessage chat_session delimiterreturncCsLg}|dD]}t|jtr-d}|jD]}t|tr%||ddp"d7}q||7}q||_t|jts;td|j|sE|t|qt|t |dr|dj ddur|dj d|j dkrt|djtswtd|dj|dj||j |d_|dj dg |j dpgq|t|qt |d S) zMerge chat runs together in a chat session. A chat run is a sequence of messages from the same sender. Args: chat_session: A chat session. Returns: A chat session with merged chat runs. messagestextzAChat Loaders only support messages with content type string, got senderNeventsr ) isinstancecontentlistdictgetstr ValueErrorappendrtypeadditional_kwargsstripextendr)r r r messagerrr!m/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/langchain_community/chat_loaders/utils.pymerge_chat_runs_in_session sJ         r# chat_sessionsccs|D]}t|VqdS)zMerge chat runs together. A chat run is a sequence of messages from the same sender. Args: chat_sessions: A list of chat sessions. Returns: A list of chat sessions with merged chat runs. N)r#)r$r r!r!r"merge_chat_runs>s r%rcCs`g}d}|dD]"}|jd|kr%t|j|jt|ddd}|d7}||qt|dS) Convert messages from the specified 'sender' to AI messages. This is useful for fine-tuning the AI to adapt to your voice. rr rexampleN)rrr'r)rrrrcopygetattrrr)r$rr Z num_convertedr r!r!r"map_ai_messages_in_sessionMs    r+ccs|D]}t||VqdS)r&N)r+)r$rr r!r!r"map_ai_messages`sr,N)r )__doc__r)rtypingrrrZlangchain_core.chat_sessionsrZlangchain_core.messagesrrrr#r%r+r,r!r!r!r"s,   4