o Zh @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 delimiterreturncCsg}|dD]u}t|jtstd|j|s |t|qt|t|drt|djddurt|djd|jdkrtt|djtsRtd|dj|dj||j |d_|djdg |jdpqgq|t|qt |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 messager]/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/chat_loaders/utils.pymerge_chat_runs_in_session s<     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 rrrmerge_chat_runs6s 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_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,   ,