j6dZddlmZddlmZmZddlmZdddddd ZeGd d Z ddZ ddZ ddZ d dZ dS)!aConvert OpenAI-style message lists into a single prompt for the muse web chat. muse.ai's web UI is a single text box, so multi-turn context and system prompts are serialised into one message. Image parts are extracted separately so that drivers can attach them as files. ) annotations) dataclassfield)AnySystemUser AssistantTool)system developeruser assistanttoolc<eZdZUded<eeZded<dS) FlatPromptstrtext)default_factory list[str]imagesN)__name__ __module__ __qualname____annotations__rlistr0C:\Projects\Muse2API\src\muse2api\core\prompt.pyrrs9 IIId333F3333WWrrcontentrrrreturnrc.|dSt|tr|Sg}|D]}t|ts#|t|:|d}|dvr*||dd}|dvrV|d}t|tr|d}|r||dd |DS) Ntype)r input_text output_textr) image_url input_imager&url c3K|]}||V dS)Nr).0ps r z#_content_to_text..0s'++1+Q++++++r) isinstancerdictappendgetjoin)rrpartspartptyperefs r_content_to_textr7sr'3E # #$%%  LLT # # #    9 9 9 LL&"-- . . . . 2 2 2((;''C#t$$ %ggenn # c""" 99+++++ + ++rmessageslist[dict[str, Any]]list[tuple[str, str]]cg}|D]d}t|dg}|r+||dd|fe|S)zHRole/text pairs, using the same text extraction as ``flatten_messages``.rroler )r7r1stripr0)r8turnsmsgrs r message_turnsr@3sp#%E:: 2 2B77==??  : LL#''&&1148 9 9 9 Lrpreviousincoming str | Nonec|r t|t|krdS|dt|t|krdSt|t|d}|r,|dddvr|dd}|r|dddvg}|D]#\}}|dkrdS||$|sdSd|S)zText to send when ``incoming`` continues a conversation already on the page. Returns only the new user text. ``None`` means the histories do not line up, so the caller must open a fresh thread and send the full prompt instead. Nr)rrr  )lenrr0r2)rArBextra user_textr<rs r followup_textrJ=s  s8}}H 55t#h-- DNN22t #h--//* + +E E!HQK#888abb  E!HQK#888I d 6>>44 t ;;y ! !!rcg}g}|D]f}|dd}t|d|}|r|||fg|st d|St |dkr/|dddkrt |dd|Sd|D}|d t d ||S) Nr<r rr")rrrErczg|]8\}}dt||d|9S)[z] ) ROLE_LABELSr1title)r+r<rs r z$flatten_messages..dsC Y Y YZT4@+//$ 55@@$@@ Y Y Yrzy[Instruction] Continue the conversation above and reply as the Assistant to the last User message. Output only the reply.rF)r1r7r=r0rrGr2)r8rr>r?r<rblockss rflatten_messagesrRTsF#%E''wwvv&& 2 2F;;AACC  ' LL$ & & & 2r&1111 5zzQ58A;&00uQx{6:::: Y YSX Y Y YF MM / 6;;v..v > > >>rN)rrrrr r)r8r9r r:)rAr:rBr:r rC)r8r9r r)__doc__ __future__r dataclassesrrtypingrrNrr7r@rJrRrrrrWs#"""""((((((((     XXXXXXX X ,,,,,"""".??????r