a 'NiO)@s~dZddlZddlmZddlmZddlmZeejdZzddl m Z Wne ybdZ Yn0Gdd d Z d d ZdS) z Character Memory System for BaoLife Provides long-term memory for NPCs by extracting and storing important facts from conversations. Enables characters to remember player preferences, past events, and relationship history across sessions. N) AsyncOpenAI)get_database_connection)config)api_key)trackerc@sTeZdZdZddZddZddZdd d Zdd d ZddZ ddZ ddZ dS)CharacterMemoryzr Persistent memory system for NPC characters. Extracts and stores important facts from conversations. cCs||_||_g|_d|_dS)z Initialize character memory. Args: character_id: ID of the character player_id: ID of the player FN) character_id player_idfactsloaded)selfrr r 1/var/www/lichun.app/lichun/ws/character_memory.py__init__szCharacterMemory.__init__c st|jdkrdS|jdd}|||}d|jd|d}z8tjtjjj dd|d gd d d d dIdH}|j dj j }trzRt|dr|jnd}tj|j|d|jj|jj|jjddd} td| dWn4ty} ztd| WYd} ~ n d} ~ 00|dkrx|rxdd|dD} | rx|| ||j| tdt| d|j| WSWn4ty} ztd| WYd} ~ n d} ~ 00gS)a Extract important facts from recent conversation messages. Called periodically (e.g., every 5 messages) to update memory. Args: conversation: conversationObj instance character: personClass instance Nz.Extract key facts from this conversation that z2 should remember about the player. Conversation: a Extract facts in this format (one per line): - [Fact about player's preferences, plans, or important information] Only extract genuinely important facts (max 3). If nothing important, respond with "None". Facts should be specific and actionable for future conversations. Facts:z gpt-4o-miniuser)rolecontentg?)modelmessages max_tokens temperatureg@)timeoutrid) prompt_tokenscompletion_tokens total_tokensZfact_extraction)purposezFact extraction cost: $z.6fz'Failed to track fact extraction usage: nonecSs.g|]&}|dr|ddqS)-N)strip startswith).0fr r r psz1CharacterMemory.extract_facts.. z Extracted z new facts for zFailed to extract facts: )len conversation_format_messages_for_extraction firstnameasynciowait_for openai_clientchat completionscreatechoicesmessagerr# api_trackerhasattrr track_usager usagerrrprint Exceptionlowersplit _save_factsr extend) r r* characterrecent_messagesconversation_textZextraction_promptresult facts_textconversation_idcoster r r r extract_facts,s\       $   $zCharacterMemory.extract_factscCsDg}|D]0}|j|jkrdn|j}||d|jqd|S)z#Format messages for fact extractionPlayerz: r()senderrr,appendr4join)r rr? formattedmsgZspeakerr r rr+s z/CharacterMemory._format_messages_for_extraction c Cs |jr|jr|jSt}zz|jddr}|d|j|j|f|}dd|D|_d|_tdt |jd|j|jWdWW| S1s0YWnBt y}z(td |gWYd}~W| Sd}~00W| n | 0dS) z Load character's memory facts from database. Args: limit: Maximum number of facts to load (most recent) Returns: List of fact strings T) dictionarya SELECT fact, importance, learned_date FROM character_memory WHERE character_id = %s AND player_id = %s ORDER BY importance DESC, learned_date DESC LIMIT %s cSsg|] }|dqS)factr )r%rowr r rr'z.CharacterMemory.load_facts..zLoaded z facts for character Nz!Failed to load character memory: ) r r rcursorexecuterr fetchallr9r)closer:)r limitmydbrSresultsrFr r r load_factss*  zCharacterMemory.load_factsrcCs|js||jd|S)z Get relevant facts for context injection. Args: max_facts: Maximum number of facts to return Returns: List of fact strings N)r rZr )r Z max_factsr r rget_relevant_factss z"CharacterMemory.get_relevant_factscCs$|}|sdSd|}d|S)z Get formatted memory context for prompt injection. Returns: String containing memory context, or empty string if no memories z; z,You remember these things about the player: )r[rK)r r rCr r rget_memory_contexts  z"CharacterMemory.get_memory_contextc Cst}zz|}g}|D]D}tdtdt|d}||j|j||t|drX|j ndfq| d|| t dt|dWdn1s0YWn@t y}z(t d|t d |WYd}~n d}~00W|n |0dS) z Save extracted facts to database. Args: facts: List of fact strings conversation: conversationObj instance for metadata rNrNz INSERT INTO character_memory (character_id, player_id, fact, importance, conversation_id, learned_date) VALUES (%s, %s, %s, %s, %s, NOW()) zSaved z facts to databasez"Failed to save facts to database: z Facts were: )rrSminmaxr)rJrr r6r executemanycommitr9r:rV) r r r*rXrSvaluesrPZ importancerFr r rr=s,  6&zCharacterMemory._save_factsc Cst}zzd|H}|d|j|jf|g|_d|_td|jWdn1s`0YWn2t y}ztd|WYd}~n d}~00W| n | 0dS)z6Clear all facts for this character-player relationshipz DELETE FROM character_memory WHERE character_id = %s AND player_id = %s Fz Cleared all facts for character NzFailed to clear facts: ) rrSrTrr rbr r r9r:rV)r rXrSrFr r r clear_factss  2&zCharacterMemory.clear_factsN)rN)r) __name__ __module__ __qualname____doc__rrGr+rZr[r]r=rdr r r rrs V ' +rc Cst}zzF|*}|d|tdWdn1sB0YWn2ty}ztd|WYd}~n d}~00W|n |0dS)zn Create character_memory table if it doesn't exist. Call this during server startup or migration. a~ CREATE TABLE IF NOT EXISTS character_memory ( id INT AUTO_INCREMENT PRIMARY KEY, character_id VARCHAR(36) NOT NULL, player_id VARCHAR(36) NOT NULL, fact TEXT NOT NULL, learned_date DATETIME NOT NULL, importance TINYINT DEFAULT 5, conversation_id VARCHAR(36), INDEX idx_char_player (character_id, player_id), INDEX idx_importance (importance DESC), INDEX idx_learned_date (learned_date DESC) ) z'Character memory table created/verifiedNz)Failed to create character_memory table: )rrSrTrbr9r:rV)rXrSrFr r rcreate_character_memory_tables  *&ri)rhr-openairdatabase.db_operationsrrOPENAI_API_KEYr/api_usage_trackerrr5 ImportErrorrrir r r rs      y