o 5ti @szddlmZddlmZddlmZd#d$d d Zd%ddZd&ddZeGdddZ d'ddZ d(ddZ d)d*d!d"Z dS)+) annotations) dataclass)Any prefix str | Nonesuffix delimiterstrreturncCsD|s|pdS|s |p dS|ds|ds|||S||S)zXJoin prefix and suffix, adding delimiter only if neither has whitespace at the boundary.risspace)rrr rE/home/ubuntu/.local/lib/python3.10/site-packages/lm_eval/api/utils.py maybe_delimits   rboolcCs&|dkrdS|dp|d S)zBReturn True if neither string has whitespace at the join boundary.r Fr rr)rrrrrrequires_delimitersrscCst|o |dS)z/Return True if the string ends with whitespace.r )rr)rrrrends_with_whitespacerc@sBeZdZUdZded<ded<dZded<dd d Zdd d Zd S)MessageadA single message in a prompt, supporting both chat and plain-text formats. Used by `build_qa_turn()` and `fewshot_context()` to construct prompts that can be rendered either as plain text or as chat-formatted messages. Attributes: role (str): "system", "user", or "assistant". content (str): The message text. _delimiter (str): Suffix appended when rendering as plain text (via `to_text()`). Prefixed with `_` so it's excluded from the ` to_dict () ` output. Typically used for target delimiters (e.g., " ") or fewshot delimiters (e.g., "\n\n"). r rolecontentr _delimiterr dict[str, str]cCsdd|jDS)zIConvert to chat format dict, excluding internal fields like `_delimiter`.cSs i|] \}}|ds||qS)_) startswith).0kvrrr 5s z#Message.to_dict..)__dict__itemsselfrrrto_dict3rzMessage.to_dictcCs |j|jS)z-Render as plain text with delimiter appended.)rrr%rrrto_text7s zMessage.to_textN)r r)r r )__name__ __module__ __qualname____doc____annotations__rr'r(rrrrr s    rmessages list[Message]cCsddd|DS)zLConcatenate all messages into plain text, using each message's `_delimiter`.r cs|]}|VqdSNr(rmrrr >z#messages_to_text..)join)r.rrrmessages_to_text<rr8list[dict[str, Any]]cCs|djdkr|d|ddfnd|f\}}|djdkrB|dd}d|d_tddd d |Ddg|dg}nd|d_tddd d |Ddg}|ra|g|S|S) zYCollapse multi-turn messages into a single user message (plus optional system/assistant).rsystemNr assistantr usercsr0r1r2r3rrrr5Mr6z*multiturn_to_singleturn..csr0r1r2r3rrrr5Qr6)rrrr7r')r.r: _user_messageresrrrmultiturn_to_singleturnAs    "r@NrrtypercCs|s||dS|||dS)zACreate a chat message dict with role, content, and optional type.)rr)rArrr)rrrArrr format_turnVs   rB)r)rrrrr r r r )rr rr r r)rr r r)r.r/r r )r.r/r r9r1)rr rr rArr r) __future__r dataclassesrtypingrrrrrr8r@rBrrrrs