P1i` dZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZmZmZmZdd lmZej4eZ d%d ed eeeefdedeedef dZ d&dedeededeedef dZ d'dededdfdZ! d'ded eeeefdedeedef dZ" d(dededeedeedeedeedefdZ# d%deded eedefd!Z$ d%d"ed eedefd#Z% d%d eedefd$Z&y))z` Utility functions for handling AI conversations about divination records (BaZi, LiuYao, etc.). N)DictAnyOptionalList)timezone)settings)LLMServiceFactory) get_ai_config)prepare_bazi_prompt)prepare_liuyao_prompt) ConversationMessageConversationSubjectConversationConfig)Persondivination_promptconversation_historycurrent_messagecontext_summaryreturnct|jd}|rt|jdnd}td|D}t|jd}t|dz}|rdnd} ||z|z|z|z| z} | S)ap Calculate the total context size in characters. Args: divination_prompt: Full divination prompt (BaZi, LiuYao, etc.) conversation_history: List of previous messages current_message: Current user message context_summary: Optional summary of older messages Returns: Total size in characters (bytes) utf-8rc3pK|].}t|jddjd0yw)contentrN)lengetencode.0msgs 7/home/cursorai/projects/iching/ai/utils/conversation.py z)calculate_context_size..+s3'C CGGIr " ) )' 23's462d)rrsum) rrrr prompt_size summary_size history_size message_sizeformatting_overheadsummary_overhead total_sizes r"calculate_context_sizer.s&'..w78KX` 1,2F2F2S2S1TUV%2%$"xx61V{  |2ckk]$??% 4 !LMaLbc5D N*"MMaLbc5D  N !<=Q.sD|3s{{+|s!r )rrrrCONVERSATION_CONTEXT_THRESHOLD順)r0r1r2r3)rcontext_summary_updated_atlast_summarized_message_id update_fieldsz7Generated incremental context summary for conversation z (summarized z new messages)z,Failed to generate summary for conversation r?) listmessagesorder_byrget_max_messagesrrLrIrJrKrHrMr r rNrr&r.intrgetattrrrgappendr\rnowrfsaveinfo Exceptionstr)r0r2 all_messages max_messagesrSr last_messagesr!recent_historyavg_message_size context_size thresholdr1start_id new_summaryes r"manage_conversation_contextrs --66|DEL&668L <L(    }\__$5_EF113K  }\__$55QRS((F2/ hO    * *h 61+Q 1,2F2F2S2S1TUV!,0M! Cckk2  ]isD|DDs} | jdd} | jdd} | dk(r | d| dz } 6| d | dz } @||d!| d|d"} | S)#a Prepare a prompt for conversation with divination context and managed history. Supports BaZi, LiuYao, and other divination types via ConversationSubject. Args: conversation: Conversation object (for accessing summary and subject) conversation_history: List of recent messages (last N, configurable) user_message: The current user message language: Language for the response ('zh-hans' or 'en') Returns: Formatted prompt string r5r6r7r8nameUnknownr9znYou are an expert in BaZi (Four Pillars of Destiny) analysis. You are having a conversation with a user about z-'s BaZi chart. Full BaZi Chart Information: z Please provide helpful, accurate analysis and answer questions about this BaZi chart. Be conversational and engaging while maintaining accuracy. Reference specific pillars and elements when relevant.uQ你是一位八字（四柱命理）分析专家。你正在与用户讨论关于u*的八字命盘。 完整八字信息： u 请提供有帮助、准确的分析，并回答关于这个八字命盘的问题。保持对话性和吸引力，同时保持准确性。在相关时引用具体的柱和五行。r:questionzYou are an expert in LiuYao (Six Lines Divination) analysis. You are having a conversation with a user about their LiuYao divination. Question: z& Full LiuYao Divination Information: z Please provide helpful, accurate analysis and answer questions about this LiuYao divination. Be conversational and engaging while maintaining accuracy. Reference specific hexagrams, lines, and elements when relevant.ub你是一位六爻分析专家。你正在与用户讨论关于他们的六爻占卜。 问题：u 完整六爻信息： u 请提供有帮助、准确的分析，并回答关于这个六爻占卜的问题。保持对话性和吸引力，同时保持准确性。在相关时引用具体的卦象、爻位和五行。r;rz! Previous Conversation Summary:  u 之前的对话摘要： rNr<rzUser: r@z Assistant: z Recent Conversation: z Assistant:) rH ValueErrorrKrLrMr hasattrrr rrr)r0rrr2rSr subject_name system_promptcontext_sectionrUr!rNr full_prompts r"prepare_conversation_promptr6sQ(113K =(99UVWW((F2/ hO+2;+G{''Y t !OP\O]]HKM!rr~q@uxM    * *h 61+Q/6{J/O{++U^ t ! .Y \M! A DM5l6J6J6W6W5XYZZO## t  D\EaEaDbbdeO >|?[?[>\\^_O'C7766*Dggi,Gv~!vgYd%;;!!{7)4%@@! (#OO#44NO`NaaghtguvDEK r/rBrCexisting_user_messagec d} t|||r;|}|j|k7r||_d|_d|_|j gdn#t j j|d|d}|r|s6|jjd}t|}|xs|d }|xs|d }tj} |jj|j j!d } t#| | d} | D cgc]} | j$| jd } } t'|| ||}tEjF|}|r|jI|t*jKd|j4|jM|dd}|s t3dt j j|d|jO||d|i }d!|_|j d"gtQjR|_*|j d#gt*jKd$|j4|Scc} w#t($r&}t*j-dt/||j1}|st3d|j4d|jjdk(rt7||}nI|jjdk(rt9||}n"t3d|jjt;t<dd}g}t?| D]/} | g|z}tA|||}||kr|jCd| /nt'||||}Yd}~:d}~wwxYw#t($r{}|rRt/|}d%|_||_|j d"d&gt*jWd'|j4d(|t*jWd)t/|d}~wwxYw)*a Send a user message and get AI response in a conversation. Includes context management. Args: conversation: Conversation object user_message: User message content provider: Optional AI provider override model: Optional AI model override language: Language for the conversation ('zh-hans' or 'en') existing_user_message: Optional existing Message object to reuse (for retries) Returns: Assistant Message object Nr8pending)rstatus error_messagerhr<)r0rNrrrArBrC)pkr^r_)r0rrr2zHFailed to prepare prompt with summary, falling back to limited history: r5r6r7r:r;rcrd)rrrrz.Sending conversation message for conversation gffffff?irDzEmpty response from LLM service assistantr2)r0rNrrBrCmetasentr updated_atz8Successfully created assistant message for conversation failedrzMarked user message z as failed: z$Error sending conversation message: ),rrrrrrrobjectscreaterLrMr rrmrkexcluderrlrjrNrrtrIwarningrurHrrKr r rorreversedr.insertr rOrPrsrQrRrrqrrJ)r0rrBrCr2ruser_msgrXrYrwrvrxr!rrErrSrr|limited_history test_history test_sizerZ ai_response assistant_msg error_msgs r"send_conversation_messagers.HD#L8D !,H</#/ 'HO%)H " MM(NM O--)$ .Hu*6*>*>*K*K)LM'Z $"#;: % $XX#++ 6$  * 0)%9)! FX(33H=   $ $U +  D\__DUVW!001 >? ? ..%%%'h' / ! XJ /#+,,. 7 N|N_`am #  NNefijkflemn o&99;K =0AA]!^__##00F:$7 h$W!%%22h>$9+PX$Y! #=l>R>R>_>_=`!abb*JFSI!O 45 #u6 2&7)5$0  )#**1c261)%4)! F=# T   AI&HO%.H " MM?(CM D LL/ }L T U ;CF8DE  sWDM; I'M;*I9D M;M; M8DM3-M;3M88M;; O?A6O::O?rM object_idtitlec vtj||}|j}|s9|dk(r|rd|j}n"|dk(r|rd|jdd}nd|d|}t j j|||}tjd|jd |jd |d||S) aQ Create a new conversation thread. Args: user: User object content_type: Type of divination ('bazi' or 'liuyao') object_id: ID of the divination record (Person ID for BaZi, LiuYao ID for LiuYao) title: Optional title for the conversation Returns: Conversation object r7zConversation about r:Nr$z #)r<rLrzCreated conversation z for user z about ) rget_or_create_subject get_objectrrr rrrIrsrK)r<rMrrrLrSr0s r"create_conversationr0s&"77 iPG$$&K  6 !k)+*:*:);*>s*C)DEE),r)EE''.. /L  KK' '8 477)7S_R``bclbmno r/personc2t|d|j|S)a Create a new BaZi conversation thread (convenience function). Args: user: User object person: Person (BaZi chart) object title: Optional title for the conversation Returns: Conversation object r7rrK)r<rrs r"create_bazi_conversationrYs tVVYY >>r/c2t|d|j|S)z Create a new LiuYao conversation thread (convenience function). Args: user: User object liuyao: LiuYao object title: Optional title for the conversation Returns: Conversation object r:r)r<r:rs r"create_liuyao_conversationrls tXvyy% @@r/ra)zh-hansN)r)NNrN)'__doc__loggingtypingrrrr django.utilsr django.confrai.services.factoryr ai.utils.configr ai.utils.bazir ai.utils.liuyao_analysisr ai.modelsr rrr bazi.modelsr getLogger__name__rIrurnr.r\rrrrrrr/r"rsL,,! 1)-:TT   8 $&* **tCH~.**c] *  *`&* LDLD=LDLDsm LD  LDbdididi diX( XXtCH~.XXsm X  X|#'/3 \\\sm\ C= \ sm \ $G, \ \F &&& C= &  &X ? ? C=? ?, A C=A Ar/