7|h dZddlmZddlmZmZmZddlmZddl m Z m Z  ddede defd Z d eedeefd Zd ed e defd Zd eed e deefdZy)zUtilities for chat loaders.)deepcopy)IterableIteratorList) ChatSession) AIMessage BaseMessage chat_session delimiterreturncFg}|dD]}t|jtstd|j|s|j t |St|t |dr|djjd|djd|jjdk(rt|djtstd|dj|dj|z|jzj|d_|djjdgj|jjdxsgr|j t |t|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. messageszAChat Loaders only support messages with content type string, got sendereventsr) isinstancecontentstr ValueErrorappendrtypeadditional_kwargsgetstripextendr)r r rmessages e/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/chat_loaders/utils.pymerge_chat_runs_in_sessionr s#%H +/'//3/')  OOHW- . wXb\ 2 3 ..228<H ..x8((,,X67hrl22C8 #B<//02  $$y07??Beg RL  RL * * . .x < C C))--h7=2  OOHW- .5/6  )) chat_sessionsc#4K|D]}t|yw)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 s rmerge_chat_runsr#6s"&7 (667src g}d}|dD]q}|jjd|k(r@t|j|jj t |dd}|dz }|j |st|S) Convert messages from the specified 'sender' to AI messages. This is useful for fine-tuning the AI to adapt to your voice. rrrexampleN)rrr&r)rrrrcopygetattrrr)r!rr num_convertedrs rmap_ai_messages_in_sessionr+Es HM ,!  $ $ ( ( 2f <")";";"@"@"BD9G Q M !  ))r c#6K|D]}t||yw)r%N)r+)r!rr s rmap_ai_messagesr-Xs$&? (v>>?sN)z )__doc__r(rtypingrrrlangchain_core.chat_sessionsrlangchain_core.messagesrr rrr#r+r-r rr3s!++4:17)*)**-)*)*X 78K#8 7Xk=R 7*k*3*;*&?K(?25? k?r