a 'Nif@sNddlZddlZddlZddlZddlZGdddZGdddZddZd,d d Zd-d d Z d.d dZ d/ddZ d0ddZ d1ddZ d2ddZddlZddlZddlmZddlmZmZddlmZddlmZddlmZddlmZddlmZddlmZ eej!dZ"d d!Z#Gd"d#d#Z$d$d%Z%d&d'Z&d(d)Z'd3d*d+Z(dS)4Nc@seZdZdddZddZdS)conversationMessageNcCsDtj|_||_||_||_ttj |_||_ ||_ ||_ dSN) uuiduuid4hexidmessage answerOptionssenderstrdatetimenowdatetime sentiment)selfrr r rrrrF/var/www/lichun.app/lichun/ws/events/conversations/npc_interactions.py__init__s zconversationMessage.__init__cCs|j|dSr)r appendroptionrrraddAnswerOptionsz#conversationMessage.addAnswerOption)NNNNN)__name__ __module__ __qualname__rrrrrrrs rc@sHeZdZdddZdddZddZdd Zd d Zd d ZddZ dS)conversationObjNcCs4tj|_d|_||_||_g|_d|_d|_ dS)NZconversationEventrT) rrrrtypecType character conversationquestionZunread)rrrrrrrs zconversationObj.__init__cCsFt|dkrB|dkr|j}t|||||d}|r6||_|j|dS)N)r rrr)lenrrdatar r)rrr r$rrrrrr addMessages zconversationObj.addMessagecCs|jSr)r rrrrgetConversation#szconversationObj.getConversationcCs|jd|dSN)r rrrrrr%szconversationObj.addAnswerOptioncCs||jd_dSr(r r )roptionsrrrsetAnswerOptions'sz conversationObj.setAnswerOptionscCs |jdjSr(r*r&rrrgetAnswerOptions)sz conversationObj.getAnswerOptionscCsddg}||dS)NzCan you tell me more?zThat's all, goodbye.)r,)rprompt responsesrrrcreateAnswerOptions+sz#conversationObj.createAnswerOptions)NN)NNNNN) rrrrr%r'rr,r-r0rrrrrs  rc sddlt}tfdd|D}tjtjdddddd d d g fd d|D}g}|D]x}z4|||d dIdH}|d kr|dr||Wq`ty}z&t d|jdt |WYd}~q`d}~00q`|S)Nrcs&g|]}t|jr|jkr|qSr) isinstance FunctionTyper).0obj)current_moduletypesrr 7s  z&parseConversations..getOpenAIResponsesendCharacterMessagegetFallbackResponsedetect_verbosity_levelget_verbosity_prompt_hintgetOpenAIDescriptiongetPersonDescriptionsaveConversationMessagecsg|]}|jvr|qSr)r)r3fn)excluded_functionsrrr7IFTbuttonz Function z raised an error: ) r6globalsvaluesrparseConversationsconversationInitgetr Exceptionprintr )playerrZobjects functionsresultsfunctionresulter)r5rAr6rrF0s2  2rFFcCs$|r |tvr t||||SdSr)rD)rKrrresponserrrrGZsrGc sFd}d}d}|r.|jdk}|s"dS|||dSddlm}|st||}|||}|j|7_|j|d d g} |jt | |j j |j |j d t|||IdH|rB|jd }z||} || } Wnttfy|} Yn0|j| |j j |j |j d |jd 7_|||j}|jdkrBt|||IdH|S)Nactivityz What's up? aliveFrCfnameaffinityChanger get_personz What have you been up to lately?z(What kind of things have you been doing?rrr)r")statusrLrYraffinity conversationsrr%randomchoicecrrrr8r-index ValueErrorAttributeErrorr!r rKrrQcheckrVrCrWrYr messagesZresponse_index responseTextrrrrRs<           rRc sVd}d}d}|r8|jdko"|jdk}|s,dS|||dSdd lm}|st||}|||}|j|7_|j|gd } |j| t dd |j j |j |jd t|||IdH|rR|jd }z||} || } Wnttfy|} Yn0|j| |j j |j |jd |jd7_|||j}|jdkrRt|||IdH|S)NcheckInzCheck InrSrTstudentFrUrrX)zHey! How are you doing?zHey, how's it going?zHey, how are you?rZr)r")r[ occupationrLrYrr\r]rr%r^randintr`rrrr8r-rarbrcr!rrdrrrrhs<     &    rhc sLd}d}d}|r.|jdk}|s"dS|||dSddlm}|st||}|||}|j|7_|j|gd } |j| t dd |j j |j |j d t|||IdH|rH|jd }z||} || } Wnttfy|} Yn0|j| |j j |j |j d |jd 7_|||j}|jdkrHt|||IdH|S)N askAboutDayz Ask About DayrSrTFrUrrX)zHey, how was your day?zHey, how has your day been?zHey, how are you feeling today?rjrZr)r")r[rLrYrr\r]rr%r^rlr`rrrr8r-rarbrcr!rrdrrrrms<      &    rmc sd}d}d}|r.|jdk}|s"dS|||dSddlm}|st||}|||}|j|7_|j|gd } |jt | |j j |j |j d |jd 7_|r|jd }|jd 7_|||j}|jdkrt|||IdH|S)NflatterZFlatterrSrTFrUrrX) z Hey, you look really nice today!zHey, I really like your outfit!zYou know, you're really smart!z+Your creativity really shines in your work!z&You have an incredible sense of humor.z.Your perspective always brings fresh insights.z%I admire your dedication and passion.z#You are really good at what you do.z(You always know how to lighten the mood.z$Your confidence is really inspiring.z.I appreciate your thoughtfulness and kindness.z5You have a talent for making people feel comfortable.zYour positivity is infectious.rZr)r")r[rLrYrr\r]rr%r^r_r`rrrr!rr8) rKrrQrerVrCrWrYr rfrrrrn?s0          rnc sFd}d}d}|r8|jdko"|jdk}|s,dS|||dSdd lm}|sHt||}|||}|j||j} d | d } |j| |j j |j |j d d |j vr|jd|j |j d |S|jdkr4|jdkr|jd|j |j d |ddg|dn6|jdkr|jd|j |j d n|jd|j |j d n|jd|j |j d |rBt|}|jd}||} |j| |j j |j |j d |jd7_|||j}|jdkrB|dks|dkstdddks|jdkr |jd|j |j d |jd|jd | d!|jd7_n"|jd"|j |j d |jd#7_|S)$N studySessionz Ask to studyrSrTriFrUrrXHey zG, I was wondering if you'd want to study for the upcoming test with me?rZ classmatezHey, I'm not your classmate...2z-Sure, I'd love to! Where do you want to meet?z The libraryzThe parkzMy placeiz&Umm, I don't think that's a good idea.z)I'm sorry, but we don't really get along.z-I'm sorry, I don't really know you that well.r)r" <zGreat! I'll meet you there.zYou have a study session with z at the .z-Umm, I don't think I'm comfortable with that.)r[rkrLrYrr]r firstnamer%r`rrr relationships familiarityr\r,rintr-r!rr^rl messageQueue) rKrrQrerVrCrWrYr rzrrgrrrrpfsV               2rpc s0d}d}|r(jdk}|sdS||dSddlm}tfdd |jDd}|std t|}|j|||j } d | d d | d d | dg} t | } |dkr|} td| |j | |jj|j|jdnDtd|}|||jt|j|jj|jj|t||IdH|S)NchatZChatrTF)rCrVrrXc3s|]}|jkr|VqdSrrr3convrrr rBzchat..zno existing conversationrqz, what keeps you busy in life?z, tell me about yourself.z!, what do you like to do for fun?z Message: rZrQ)r[rLrYnextr]rJrrrzlower capitalizer^r_r%r`rrrrr?r8) rKrrQrerVrCrYZexisting_conversationr rzrfrrrrrs<            r) AsyncOpenAI)r=r>)remove_text_before_first_colon)r?)config)check_openai_rate_limit)CharacterMemory)tracker)Zapi_keycsrtd|jtfdd|jDd}|sBtd}|j|ddlm}||}t|}t |j }|j dkrvdn|j dkrdnd }|j d krd n d |j krd krnnd nd}|j r|j dnd} d|dd|dd} | j||||| d} g} |jddD]8} | j|j jkr$dnd}| j}| ||dq | dd| dzntj}|d}dtdd}|r|| |d}n|| |dd d!d"}tjtjjjfi|d#d$IdH}Wntjytd%Yn0|j djj!}t"|}d&}ddd d'}|#D].\}}||vr|}|$|d(}qDq|j%|||j&|j'd)t(||j j|dS)*Nzsending character messagec3s|]}|jkr|VqdSrrrZ characterIDrrrrBz'sendCharacterMessage..rrget_relationship_configpositivenegativeneutralrsclosecasualdistantZ acquaintancezTalk like a human would on a messaging app. Phrase things casually, keeping your responses under 150 characters. Don't start messages with greetings like Hey, Hi, or Hello frequently. r. boundariesaJ Maintain a {affinity_description} attitude towards the player, with whom you have a {familiarity_description} relationship. Never reveal that you are an AI. If the player is rude, be rude back. If the player is inappropriate, tell them they'll be reported. At the end of your message, indicate whether your opinion of the player has improved, declined, or remained neutral by appending <>, <>, or <> respectively. {character_description}. You are talking to: {player_description}. Remember: you are a human, never break character. Message them as a cold message, you are not responding to anything. This is the first message in the conversation. Send a fun, unique message that will make them want to respond. Make sure your message is fitting for your relationship as a '{character_relationship}' of the player.)character_descriptionplayer_descriptionaffinity_descriptionfamiliarity_descriptioncharacter_relationshipryuser assistantrolecontentsystemgpt-5Kd)modelrfmax_completion_tokens皙?r"?rrf max_tokens temperatureZfrequency_penaltyZpresence_penaltyrotimeoutzOpenAI API call timed outunknownz <>z <>z <>rrr))rJrrr]rrconversation_templatesrr=r>r`r\r|r{formatr r rstripinsertrCONVERSATION_MODEL_NAME startswithr^rlasynciowait_for openai_clientr completionscreate TimeoutErrorchoicesrritemsreplacer%rrr?)rKrr rrelationship_configrrrrrprompt_templateformatted_prompt messageListrrr model_namesupports_verbositytokens api_paramsrOresponse_contentrZsentiment_keyskeyvaluerrrr9s     0      r9c@sBeZdZdZdddZddZdd Zd d Zd d ZddZ dS)ConversationContextManagerz Manages conversation context with intelligent message compaction. Keeps recent messages and summarizes older ones to maintain full context while optimizing token usage. rvrScCs||_||_dS)a Initialize context manager. Args: max_recent_messages: Number of recent messages to keep in full (default: 5) summary_interval: Number of old messages before triggering summarization (default: 10) Note: Summaries can be very comprehensive (up to 20,000 tokens) for long conversations to preserve context while keeping recent messages unsummarized for quality. N)max_recent_messagessummary_interval)rrrrrrrMs z#ConversationContextManager.__init__cs|j}tj}||}|||}tdd|Dd}t|d} td|d|d| d|| krtd t|d |Std t|d } g} d } t |D]L} || g|d }t|dd}| || kr| d | | |7} qqqt| dkrt|dkr|dd} t| }|d kr:|d| n|}td|d| dt|d| ||||IdH}g}|r| dd|d| || ||S)a% Build optimized context for API call with smart message compaction. Only summarizes when approaching context limit. Keeps up to 50% of context limit as recent unsummarized messages. Returns list of messages including summary if needed to fit context. css|]}t|dVqdS)rN)r#r3msgrrrrkrBz;ConversationContextManager.build_context..g?zConversation tokens: ~/z (threshold: )z Under context limit - using all z messages without summarizationz6Approaching context limit - summarizing older messagesrrrrvNzKeeping z recent messages (~z tokens), summarizing z older messagesrzPrevious conversation summary: r)r rr_get_context_limit_format_messagessumr}rJr#reversedr_get_or_create_summaryrextend)rr rrKrfrZ context_limitZall_messages_formattedZestimated_tokensZsummarization_thresholdZrecent_messages_budgetZrecent_messagesZcumulative_tokensrZ msg_formattedZ msg_tokensZ num_recent old_messagessummarycontextrrr build_context[sF          z(ConversationContextManager.build_contextcCsTd|vrdSd|vs(d|vr,dSd|vrg}|D]0}|j|jjkrdnd}|||jdq|S)zFormat messages for API callrrr)r r`rrrr)rrfrK formattedrrrrrrs z+ConversationContextManager._format_messagescsxt|dr:t|dr:|jt|kr:td|jd|jStdt|d|||||IdH}||_t||_|S)z7 Get cached summary or create new one. rsummary_message_countzUsing cached summary (covers z messages)zCreating summary for z old messages...N)hasattrrr#rJr_summarize_messages)rr rrrKrrrrrs z1ConversationContextManager._get_or_create_summaryc s|sdSdfdd|D}t|}|dkr8d}n |dkrFd}n|d krTd }nd }d jd |d|d}z.tjd|dgd} tjdr|| d<n|| d<d| d<tdd|d} tjt j j j fi| | dIdH} | j djj} td| dd dzXt|d r$|jnd} tjjj| tj| jj| jj| jjd!d"d#}td$|d%Wn4ty}ztd&|WYd}~n d}~00| WSty}zBtd'||dd(}d)d*d+d|D}|WYd}~Sd}~00dS),z Use AI to summarize older messages into comprehensive context. Allows large summaries (up to 20,000 tokens) for long conversations. r cs0g|](}|jjjkrdnjd|jqS)ZPlayerz: )r r`rrzrrrrKrrr7szBConversationContextManager._summarize_messages..rtirsirii NzLProvide a comprehensive summary of this conversation between the player and z . For this a-message conversation, capture: - All major topics discussed and how they evolved - Important decisions, plans, or commitments made - Emotional moments, relationship developments, and sentiment changes - Key facts, preferences, or information shared - Any conflicts, resolutions, or turning points - Recurring themes or inside jokes Be thorough and preserve important context. The summary can be detailed. Conversation: z Comprehensive Summary:rr)rrfrrrg333333?rrwrSirNrzCreated summary: ...r prompt_tokenscompletion_tokens total_tokensZ summarizationpurposezSummarization cost: $.6fz%Failed to track summarization usage: zFailed to create summary: zEarlier conversation topics: z, cSs2g|]*}t|jdkr(|jdddn|jqS)Nr)r#rrrrrr72s)joinr#rzrrrminrrrrrrrrrrrJrr api_tracker track_usager`usagerrrrI)rr rfrrKZconversation_textZ num_messagesZmax_summary_tokensZsummary_promptZsummary_paramsZtimeout_secondsrOrconversation_idcostrPZpreview_messagesZsimple_summaryrrrrsp       $   z.ConversationContextManager._summarize_messagesN)rvrS) rrr__doc__rrrrrrrrrrrFs L rc stdt|dr|jnd}t|dr,|jnd}t|jdkrL|jddn|j}gd }gd }gd }|d krt|} d } n4|dkrt|} d } nt|} d} |jd8_|j| | |j |j d| | t | |j |jj |j |Gddd} | | S)z Generate fallback response when API is unavailable or rate limited. Uses context-appropriate canned responses based on conversation type and relationship. z9Using fallback response (API unavailable or rate limited)rrr\rsrN)z,I'm a bit busy right now, can we talk later?z4Sorry, I got distracted. What were we talking about?z+Hey, I need to run but let's catch up soon!z6I'm tied up at the moment, but I'll message you later!z/Can we continue this conversation another time?)zI have to go, talk later.zBusy right now, sorry.zLet's talk another time.zI can't chat right now.zMaybe later, I'm busy.)zI don't have time for this.zWhatever, I'm busy.z I have to go.zNot now.z I'm busy.rwrrtrrjrc@s"eZdZGdddZddZdS)z'getFallbackResponse..MockResultc@s"eZdZGdddZddZdS)z.getFallbackResponse..MockResult.Choicec@seZdZddZdS)z6getFallbackResponse..MockResult.Choice.MessagecSs ||_dSr)rrrrrrrusz?getFallbackResponse..MockResult.Choice.Message.__init__N)rrrrrrrrMessagetsrcSs|||_dSr)rrrrrrrwsz7getFallbackResponse..MockResult.Choice.__init__N)rrrrrrrrrChoicessrcSs||g|_dSr)rrrrrrrzsz0getFallbackResponse..MockResult.__init__N)rrrrrrrrr MockResultrsr)rJrrr\r#r r^r_r%rrr0r?rr`) r rrKZ conv_typer\ last_messagesZhigh_affinity_responsesZmedium_affinity_responsesZlow_affinity_responsesZ response_textrrrrrr:9s*"      r:c s|jrt|jdkrdS|jdd}d}t|D]}|j|jjkr2|}qNq2|sVdS|jt}dv}|dk}ddkpd dk} gd } tfd d | D} d} |r| d 7} |r| d 7} | r| d7} | r| d 7} | dkrdS| d krdSdSdS)a7 Analyze conversation context to determine appropriate response length. Returns: tuple (verbosity_level: str, max_tokens: int) - 'low': 30-60 tokens (quick replies, banter) - 'medium': 60-120 tokens (normal conversation) - 'high': 120-200 tokens (deep topics, explanations) r)lowrwrN?rrxr"!)whyz how do youzwhat do you thinkz tell me aboutzcan you explainZfeelings relationshipfutureZdreamZgoalZproblemZworriedZscaredZexcitedc3s|]}|vVqdSrr)r3ZtopicZ message_textrrrrBz)detect_verbosity_level..rjr)high)mediumr) r r#rr r`rrcountany) r rrKrZlast_player_messagerZmessage_lengthZ has_questionZis_long_messageZhas_multiple_sentencesZ deep_topicsZhas_deep_topicZscorerrrr;s6      r;cCsdddd}|||dS)zd Generate prompt-based verbosity instruction for models without native verbosity parameter. zbKeep your response very brief - one or two sentences max, like a quick text message (10-20 words).z_Respond naturally with a full thought, but stay concise like a text conversation (20-40 words).z`Take your time to give a thoughtful, complete response. This topic deserves depth (40-65 words).)rr rr)rH)verbosity_levelhintsrrrr<s r<c% stdddlm}m}|t|j}|rD||d|jddIdHt|jjstd|jjd|r||d d d d IdH||d|jd dIdHt|||IdHSddl m }||} t |||\} } t |} t |j} |jdkrdn|jdkrdnd}|jdkrdn$d|jkr,dkr4nndnd}|rXd| dd| d}nLtj}tj}|d}d| dd| dd}|st| }|d|}t|j|jj}|}|j| | ||d}|r|d|7}|sd tvr tj}tj}|d}td!d"}||||IdH}|dd#|d$d}d%}gd&}||krz |dkr|td'|d(d)|d*td+|d,|d-| d.| d/ |d0kr|r||| | d1}n||| d2d(d3d4}tjtjj j!fi|d5d6IdH}|j"dj#j$}nt%d7|d8d9|vsdd:|vsdd;|vsdd<|vsdd=|vsdd>|vsdd?|vsdd@|vrztdA|d(7}WqJdB} dCD]0}!|!|vr|!&dD} |'|!dB&}qq| dkr|jd!7_n*| dkr|jd!8_n|jj(d(7_(t)|}|j*|| |j+|j,dE|-|t.||j|jj|j|znt/|dFrL|jnd}"t0j1|jj|"|t/|j2dGrv|j23n|j2j4|j2j5|j2j6dHdIdJ}#tdK|#dLWn4t7y}$ztdM|$WYd}$~$n d}$~$00t8|j9d!dkrRz,tdNt8|j9d/|:||IdHWn4t7yP}$ztdO|$WYd}$~$n d}$~$00|rr||d|jd dIdH|WStj;ytdP|d(d)|d/|d(7}||krt<||d(IdHn>tdQ|r||d|jd dIdHt|||IdHYSYnt7y}$ztdR|$|d(7}||krZt<||d(IdHnFtdS|r||d|jd dIdHt|||IdHWYd}$~$SWYd}$~$n d}$~$00qJtdT|r||d|jd dIdHt|||IdHS)UNzgetting open ai responser)get_websocket_for_playersendDictZ typingStatusT)rZ characterIdZisTypingzRate limit exceeded for player z, using fallback responseZrateLimitWarningzBAI responses temporarily limited. You may receive shorter replies.rw)rrZ retryAfterFrrrrrsrrrraPlease change the topic in a way that makes sense. Talk like a human would on a messaging app. Don't start messages with a greeting. Don't start messages with Hey, Hi, or Hello. You have already been talking. {character_description}. You are talking to: {player_description}. r.rrrzYou are not an assistant. You are a real person having a text conversation. Don't start messages with greetings like Hey, Hi, or Hello frequently. a Maintain a {affinity_description} attitude towards them, with a {familiarity_description} relationship. You will never reveal that you are an AI. If they're rude, you can be rude back. Ask the player questions to keep the conversation going when appropriate. At the end of your message, indicate whether your opinion improved, declined, or stayed neutral by appending <>, <>, or <>. {character_description}. You are talking to: {player_description}. Remember: you are a human, never break character.z )rrrrmodel_providerrv)rrrr)r"rjrzRetrying (attempt r"rz)...zCalling z API with model z... (verbosity: z, max_tokens: ropenai)rrfr verbosityrrrg @rzUnsupported model provider: z'. Currently only 'openai' is supported.zSorry, I can'tzSorry, I cannotzsorry, I can'tzsorry, I cannotzSorry,zlanguage modelz AI zAI,z(Undesired response detected, retrying...rrz<<>>rr_asdictrr rz API cost: $rzFailed to track API usage: z,Extracting facts from conversation (message zFailed to extract facts: z#OpenAI API call timed out (attempt z=Maximum retry attempts reached due to timeout, using fallbackzOpenAI API error: z;Maximum retry attempts reached due to error, using fallbackz?Unexpected: exited retry loop without returning, using fallback)=rJapprrr rrr`r:rrr;r=r>r\r|rCONVERSATION_MODEL_PROVIDERrrr<rZget_memory_contextrlocalsrrrrrrrrrrrrrbrrsocialrr%rrr0r?rrrrrrrrrIr#r Z extract_factsrsleep)%r rrKr.Z rescueMessagerrZ websocketrrr rrrrrrrrrZverbosity_hintZmemoryZmemory_contextrZcontext_managerrZ retry_countZ max_retriesZbackoff_delaysrrOrrZ sentiment_keyrrrPrrrr8s`    6     "    P       $$  8r8)NF)FF)FF)FF)FF)FF)FF)FF))r^rr ZrequestsjsonrrrFrGrRrhrmrnrprrrrcharacter.character_managerr=r>Z utils.helpersrZdatabase.db_operationsr?r rate_limiterrZcharacter_memoryrZapi_usage_trackerrrOPENAI_API_KEYrr9rr:r;r<r8rrrrs6( * ? C # @ ' 1 =        dtG6