iu9|dZddlZddlZddlmZddlmZmZmZmZ ddl Z dZ GddZ y#e $rdZ dZ YwxYw) z>JSON-based memory store for user feedback and session history.N)datetime)DictAnyListOptionalTFc eZdZdZ d#dededefdZdeeeffdZdedefd Z d eeeffd Z deeeffd Z d eeeffd Z dedeeefde eeefdeeeffdZde eeeffdZdedej"fdZdedej"fdZdedeeeeffdZ d$dedee edee efdZdedeeee effdZdedej"deeeffd Zdedeeeeffd!Zdede eeeffd"Zy)% MemoryStorezFJSON-based memory store for storing user feedback and session history.pathuser_memory_pathintent_memory_pathc||_||_||_tjtjj |dtjtjj |dtjtjj |dtjj |s2t|dd5}tjdgi|dd d d d tjj |s0t|dd5}tji|dd d d d tjj |s1t|dd5}tji|dd d d d y y #1swYxYw#1swYfxYw#1swYy xYw) aInitialize memory store. Args: path: Path to the JSON file storing history user_memory_path: Path to the JSON file storing user vectors intent_memory_path: Path to the JSON file storing intent vectors T)exist_okwutf-8encodingsessionsFindent ensure_asciiN) r r r osmakedirsdirnameexistsopenjsondump)selfr r r fs C/home/ubuntu/codebase/yexijia/保研/colocation_mvp/memory/store.py__init__zMemoryStore.__init__sA 0"4 BGGOOD)D9 BGGOO$45E BGGOO$67$Gww~~d#dC'2a :r*AaeL3ww~~./&g>! "a>?ww~~01(#@A "a>A@232?>A@s$F#0F/?F;#F,/F8;Greturncl t|jdd5}tj|cdddS#1swYyxYw#tj$rZ}ddl}|j t}|jd|jdt|jdd5}|j}dddn #1swYnxYwddl }|jdd }|jd d |} tj|}d |vr?|jd gD]*}d |vs|d D]} t| t sd | vs| d =,|j#||j%d|cYd}~S#tj$r|j'dd gicYcYd}~SwxYwd}~wwxYw)zqLoad history from JSON file. Returns: Dictionary containing session history rrrNrzJSON file corrupted at z, attempting to fix...z\"embedding\":\s*[^,\n}]*,?\n?z\"embedding\":\s*\nrshown_patterns embeddingzJSON file fixed and savedz/Cannot fix JSON file, returning empty structure)rr rloadJSONDecodeErrorlogging getLogger__name__warningposreadresubloadsget isinstancedictsaveinfoerror) rr er+loggercontentr1datasessionpatterns r!r)zMemoryStore.load1sw % (diiw71yy|877##" ( &&x0F NN4QUUG;QR Sdiiw71&&(877 ff>GLGff3RAG (zz'*%#'88J#;+w6+23C+D#-gt#<PWAW(/ (<,E$< $ 78 '' ( NO"B'' (?" (sA9 AAAAF3AF.)C: F.C 1F.:2E;-E;E; *E;5F3;(F+#F.$F3*F++F..F3objctrdt|tjr|j St|tj tj fr|jSt|tr3|jDcic]\}}||j|c}}St|tr|Dcgc]}|j|c}S|Scc}}wcc}w)zRecursively clean object to make it JSON serializable. Args: obj: Object to clean Returns: JSON-serializable object ) NUMPY_AVAILABLEr5npndarraytolistintegerfloatingitemr6items_clean_for_jsonlist)rr@kvrHs r!rJzMemoryStore._clean_for_json^s #rzz*zz|#C"**bkk!:;xxz! c4 ;>99;G;41aAt++A..;G G T ";>?34D((.3? ?J H@s CC%r=c|j|}t|jdd5}tj||dddddy#1swYyxYw)zrSave history to JSON file. Args: data: Dictionary containing session history rrrrFrN)rJrr rr)rr= cleaned_datar s r!r7zMemoryStore.savexsD++D1 $))S7 3q IIlAae D4 3 3s A  Act|jdd5}tj|cdddS#1swYyxYw)z|Load user memory (vectors) from JSON file. Returns: Dictionary containing user vectors r%rrN)rr rr))rr s r!load_user_memoryzMemoryStore.load_user_memorys0 $''w ?199Q<@ ? ?s 8Act|jdd5}tj||dddddy#1swYyxYw)z}Save user memory (vectors) to JSON file. Args: data: Dictionary containing user vectors rrrrFrN)rr rr)rr=r s r!save_user_memoryzMemoryStore.save_user_memorys7 $''w ?1 IIdAae <@ ? ?s <Aqueryparamspatternsfeedbackc|j}g}|D]G}|j}d|vr|d=d|vrd|vr|dr|dddng|d<|d=|j|Ii} |jD]\} } t | t rg| | <| D]~} t | t r3| j} d| vr| d=d| vr| d=| | j| Ft | t r| | j| k| | j| | | | <tjjd|||| d}|dj||j|y) zAdd a new session to history. Args: query: User's query params: Extracted mining parameters patterns: Patterns shown to user feedback: User feedback (like/dislike) r(table_instancestable_instances_sampleN%Y-%m-%d %H:%M:%S)timerTrUr'rWr) r)copyappendrIr5rKr6rnowstrftimer7)rrTrUrVrWr=cleaned_patternsr? pattern_copycleaned_feedbackkeyvaluerH item_copyr>s r! add_sessionzMemoryStore.add_sessionsyy{G"<<>Ll* - L0+<?dprCeD\J[=\]_^_=`JLL!9: !23  # #L 1 "..*JC%&(* %!D!$-$(IIK &)3 )+ 6, 9 )*; <(-44Y?#D$/(-44T:(-44T:"). %%+*LLN++,?@.(   Z( $cF|j}|jdgS)z`Get all historical sessions. Returns: List of all sessions r)r)r4)rr=s r!get_all_sessionszMemoryStore.get_all_sessionss yy{xx B''riuser_idvectorc|j}||vr gggd||<t|tjr|j n|}||dj ||j |y)zAdd a positive (liked) pattern vector to user memory. Args: user_id: User identifier vector: Pattern embedding vector positivenegativehistoryrpNrQr5rCrDrEr_rSrrlrmmemory vector_lists r! add_positivezMemoryStore.add_positiveo&&( & F7O*4FBJJ)GfmmoV w #**;7 f%ric|j}||vr gggd||<t|tjr|j n|}||dj ||j |y)zAdd a negative (disliked) pattern vector to user memory. Args: user_id: User identifier vector: Pattern embedding vector rorqNrsrts r! add_negativezMemoryStore.add_negativerxricD|j}|j|S)zGet user profile (positive and negative vectors). Args: user_id: User identifier Returns: Dictionary with 'positive' and 'negative' lists of vectors, or None if user not found rQr4)rrlrus r!get_user_profilezMemoryStore.get_user_profiles!&&(zz'""riNlikeddislikedc|j}||vrggd||<|r t|ng|r t|ngd||d<|j|y)a$Save user preference feature names (for preference-weighted scoring, e.g. 2025 adaptive fusion). Args: user_id: User identifier liked: List of feature names user likes (e.g. POI types) disliked: List of feature names user dislikes )rprq)likedislikepreference_featuresN)rQrKrS)rrlr~rrus r!save_preference_featuresz$MemoryStore.save_preference_featuress_&&( & +-2>F7O#(DKb)1tH~r2 w-. f%ricl|j}|j|}|sy|jdS)zGet user preference feature names (for preference-weighted scoring). Args: user_id: User identifier Returns: Dict with 'like' and 'dislike' lists of feature names, or None if not set Nrr|)rrlruprofiles r!get_preference_featuresz#MemoryStore.get_preference_features%s6&&(**W%{{011riu_llm intent_jsonc tjj|jr7t |jdd5}t j |}dddni}t|tjr|jn|}||tjjdd|<t |jdd5}t j||dd dddd dl}|j!t"}|j%d |y#1swYxYw#1swYCxYw#t&$r9} d dl}|j!t"}|j)d | dYd} ~ yd} ~ wwxYw)zSave intent vector and intent JSON to intent memory. Args: user_id: User identifier u_llm: LLM-generated initial user preference vector intent_json: Intent JSON from IntentEncoder r%rrNr\)rintent created_atrrFrrzSaved intent for user zError saving intent: Texc_info)rr rr rrr)r5rCrDrErr`rarr+r,r-r8 Exceptionr9) rrlrrr intent_memory u_llm_listr+r;r:s r! save_intentzMemoryStore.save_intent4s@ Eww~~d556$113IQ$(IIaLMJI!# ,6eRZZ+HeJ$%&lln556IJ&M' "d--sWE -15IF &&x0F KK0 : ;+JI FE E &&x0F LL04tL D D EsIAD:D"A9D:D.,5D:"D+'D:.D73D:: E</E77E<c tjj|jsyt |jdd5}t j |}dddj|S#1swYxYw#t$r9}ddl }|jt}|jd|dYd}~yd}~wwxYw) zLoad intent vector and intent JSON for a user. Args: user_id: User identifier Returns: Dictionary with 'u_llm' (vector as list) and 'intent' (JSON), or None if not found Nr%rrrzError loading intent: Tr) rr rr rrr)r4rr+r,r-r9)rrlr rr:r+r;s r! load_intentzMemoryStore.load_intent[s 77>>$"9"9:d--sWE $ ! F!$$W- -FE   &&x0F LL1!5L E  s4)A?A?A3A?3A<8A?? C/B<<Ccp|j}|j|}|gS|jdgS)zGet iteration round history for a user (for Stage4). Args: user_id: User identifier Returns: List of round history dictionaries roundsr|)rrlru user_profiles r!get_round_historyzMemoryStore.get_round_historyss@&&(zz'*  I"--ri)zmemory/history.jsonzmemory/user_memory.jsonzmemory/intent_memory.json)NN)r- __module__ __qualname____doc__strr"rrr)rJr7rQrSrrhrkrCrDrwrzrr}rrrrrrir!r r sPRk+F?S?C?%(?<+(d38n+(Z334 Ec3h E $sCx. =T#s(^=::d38n:!$sCx.1:=A#s(^:x($tCH~"6(&C&&,&C&&, # #c3h0H #SWBF&&HT#YrsKD ,,O r.r. O Bs / ;;