~ i 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 delimiterreturnc`g}|dD]}t|jtrId}|jD]8}t|tr||ddpdz }3||z }9||_t|jt st d|j|s#|t|t|t|dr|dj d|dj d|j dkrt|djt st d|dj|dj|z|jz |d_|dj dg |j dpg|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. messagestextzAChat Loaders only support messages with content type string, got senderNeventsr) isinstancecontentlistdictgetstr ValueErrorappendrtypeadditional_kwargsstripextendr)r r rmessagerrs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langchain_community/chat_loaders/utils.pymerge_chat_runs_in_sessionr$ sC#%H +"/"/ got , , #D"? $ $gt,,$GKK339r9DDGODD"GO'/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 r#merge_chat_runsr(>s8&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_convertedr"s r#map_ai_messages_in_sessionr0Ms HM ,!!  $ ( ( 2 2f < <")";"@"@"B"BD99G Q M     ) ) ))r%c#8K|D]}t||VdS)r*N)r0)r&rr s r#map_ai_messagesr2`s:&?? (v>>>>>>??r%N)r )__doc__r-rtypingrrrlangchain_core.chat_sessionsrlangchain_core.messagesrr rr$r(r0r2r%r#r8s1!!++++++++++444444::::::::171*1*1**-1*1*1*1*1*h 78K#8 7Xk=R 7 7 7 7*k*3*;****&?K(?25? k??????r%