Ng dZddlmZddlmZmZmZddlmZddl m Z m Z  ddede d efd Z d eed eefd Zd ed e d efdZd eed e d eefdZdS)zUtilities for chat loaders.)deepcopy)IterableIteratorList) ChatSession) AIMessage BaseMessage chat_session delimiterreturncg}|dD]}t|jtstd|j|s#|t |Yt|t |dr|djd|djd|jdkrt|djtstd|dj|dj|z|jz |d_|djdg |jdpg|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 senderNeventsr) isinstancecontentstr ValueErrorappendrtypeadditional_kwargsgetstripextendr)r r rmessages b/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_loaders/utils.pymerge_chat_runs_in_sessionr s#%H +//'/3// )))  / OOHW-- . . . . wXb\ 2 2 3 3 / .228<<H .x8(,,X6677hrl2C88  2#B</22  $y07?Begg RL  RL * . .x < < C C)--h77=2     OOHW-- . . . .  ) ) )) chat_sessionsc#6K|D]}t|VdS)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$6s8&77 (66666677r!rc *g}d}|dD]w}|jd|krBt|j|jt |dd}|dz }||xt|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 ,!!  $ ( ( 2 2f < <")";"@"@"B"BD99G Q M     ) ) ))r!c#8K|D]}t||VdS)r&N)r,)r"rr s rmap_ai_messagesr.Xs:&?? (v>>>>>>??r!N)r )__doc__r)rtypingrrrlangchain_core.chat_sessionsrlangchain_core.messagesrr rr r$r,r.r!rr4s1!!++++++++++444444::::::::17)*)*)**-)*)*)*)*)*X 78K#8 7Xk=R 7 7 7 7*k*3*;****&?K(?25? k??????r!