a bg @sdZddlmZddlmZmZmZddlmZddl m Z m Z dee eddd Z eeeed d d Zee ed ddZeee eed ddZdS)zUtilities for chat loaders.)deepcopy)IterableIteratorList) ChatSession) AIMessage BaseMessage ) chat_session delimiterreturncCsg}|dD]}t|jts,td|j|s@|t|q t|t|dr|djddur|djd|jdkrt|djtstd|dj|dj||j |d_|djdg |jdpgq |t|q t |dS)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 r messagert/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/chat_loaders/utils.pymerge_chat_runs_in_session s>      r) chat_sessionsr ccs|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 rrrmerge_chat_runs6s r!)r rr cCs`g}d}|dD]D}|jd|krJt|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_convertedrrrrmap_ai_messages_in_sessionEs   r'ccs|D]}t||VqdS)r"N)r')r rr rrrmap_ai_messagesXsr(N)r )__doc__r%rtypingrrrZlangchain_core.chat_sessionsrZlangchain_core.messagesrrrrr!r'r(rrrrs   ,