j ddlmZmZmZddlmZddlmZdeejde deeejedffd Z deejdejfd Z d ejde deejfd Z gd Zy))Any GeneratorList)types)build_conversation_turnsturns window_sizereturnNc#rKtt|D]}|td||z dz|dzyw)a Generates windows of conversation turns of a fixed size from a list of turns. This function creates a sliding window over the list of conversation turns. Each window includes the current turn and up to `window_size - 1` previous conversation turns. If there are fewer turns available than the `window_size`, the window will consist of all available turns up to the current turn. Args: turns: List of conversation turn objects representing the interactions in a conversation. window_size: Integer specifying the maximum number of turns to include in each window. Yields: A generator that produces lists of conversation turns, where each list represents a sliding window of turns. rrN)rangelenmax)r r is /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/evaluation/metrics/conversation/helpers.pyget_turns_in_sliding_windowrs<&3u: C1{?Q./!a%88s57cVg}|D]!}|j|j#|S)a Merges a list of conversation turns into a single conversation. This function takes a list of conversation turns and combines them into a single conversation by extending the output list with the data from each turn. Args: turns: A list of conversation turn objects to be combined. Returns: types.Conversation: A combined conversation object containing all the turns from the input list. )extendas_list)r outputturns r merge_turnsrs*"$F dlln% M conversationct|dk(r tdt|}t|dk(r tdt||Dcgc] }t |}}|Scc}w)ae Extracts a list of conversation windows based on turns using a sliding window approach. This function divides a conversation into consecutive overlapping windows, where each window contains a specified number of turns. Args: conversation: The input conversation from which turns will be processed. window_size: The number of turns to include in each sliding window. Returns: A list of conversations, each representing a window of turns specified by the given window size. Raises: ValueError: If the conversation is empty or if it has no turns. rzConversation is empty)rzConversation has no turns)r ValueErrorrrr)rr r turns_window turns_windowss r'extract_turns_windows_from_conversationr1sy" <A011 $, ?E 5zQ4558{K/KL L!K/  /s A#)rrr)typingrrrrconversation_turns_factoryrConversationTurnintr Conversationrr__all__rrr(s''@9 && '9699tE**+S$679.tE2238J8J&$$36 %  > r