iu9~SrSSKrSSKrSSKJr SSKJrJrJrJr SSK r Sr "SS5r g!\ a Sr Sr Nf=f) z>JSON-based memory store for user feedback and session history.N)datetime)DictAnyListOptionalTFc H\rSrSrSrS%S\S\S\4SjjrS\\\44Sjr S \S\4S jr S \\\44S jr S\\\44S jr S \\\44Sjr S\S\\\4S\\\\4S\\\44SjrS\\\\44SjrS\S\R$4SjrS\S\R$4SjrS\S\\\\44SjrS&S\S\\\S\\\4SjjrS\S\\\\\44SjrS\S\R$S \\\44S!jrS\S\\\\44S"jrS\S\\\\44S#jrS$rg)' MemoryStorezFJSON-based memory store for storing user feedback and session history.pathuser_memory_pathintent_memory_pathcXlX lX0l[R"[RR U5SS9 [R"[RR U5SS9 [R"[RR U5SS9 [RR U5(d-[USSS9n[R"S/0USSS 9 S S S 5 [RR U5(d+[USSS9n[R"0USSS 9 S S S 5 [RR U5(d,[USSS9n[R"0USSS 9 S S S 5 g g !,(df  N=f!,(df  Np=f!,(df  g =f) zInitialize 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 ?/home/ubuntu/codebase/yexijia/保研/iCoLoc/src/memory/store.py__init__MemoryStore.__init__sD 0"4 BGGOOD)D9 BGGOO$45E BGGOO$67$Gww~~d##dC'2a :r*AaeL3ww~~.//&g>! "a>?ww~~011(#@A "a>A@232?>A@s$F)F+8F< F(+ F9< G returnc[URSSS9n[R"U5sSSS5 $!,(df  g=f![RGapnSSKnUR [5nURSURS35 [URSSS9nUR5nSSS5 O!,(df  O=fSSK nURSS W5nURS S U5n[R"U5nS U;aOURS /5H9nS U;dM US H%n [U [ 5(dMS U ;dM"U S M' M; UR#U5 UR%S5 UsSnA$![Ra UR'S5 S /0ssSnA$f=fSnAff=f)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) r r!er-loggercontentr3datasessionpatterns r"r+MemoryStore.load1sz % (diiw71yy|877##" ( &&x0F NN4QUUG;QR Sdiiw71&&(877 ff>GLGff3RAG (zz'*%#'88J#;+w6+23C+D#-gt#<#<PWAW(/ (<,E$< $ 78 '' ( NO"B'' (?" (sA7 A AAAG A G*C; G C 1G5F9FF .FG )G=G>G GGG objc [(an[U[R5(aUR 5$[U[R [R 45(aUR5$[U[5(a4UR5VVs0sHup#X RU5_M snn$[U[5(a!UVs/sHo@RU5PM sn$U$s snnfs snf)zRecursively clean object to make it JSON serializable. Args: obj: Object to clean Returns: JSON-serializable object ) NUMPY_AVAILABLEr7npndarraytolistintegerfloatingitemr8items_clean_for_jsonlist)r rCkvrKs r"rMMemoryStore._clean_for_json^s ?#rzz**zz|#C"**bkk!:;;xxz! c4 ;>99;G;41A++A..;G G T " ";>?34((.3? ?J H@s "C;Dr?cURU5n[URSSS9n[R"X#SSS9 SSS5 g!,(df  g=f)zRSave history to JSON file. Args: data: Dictionary containing session history rrrrFrN)rMrr rr)r r? cleaned_datar!s r"r9MemoryStore.savexsB++D1 $))S7 3q IIlae D4 3 3s A Ac[URSSS9n[R"U5sSSS5 $!,(df  g=f)z\Load user memory (vectors) from JSON file. Returns: Dictionary containing user vectors r'rrN)rr rr+)r r!s r"load_user_memoryMemoryStore.load_user_memorys0 $''w ?199Q<@ ? ?s 6 Ac[URSSS9n[R"XSSS9 SSS5 g!,(df  g=f)z]Save user memory (vectors) to JSON file. Args: data: Dictionary containing user vectors rrrrFrN)rr rr)r r?r!s r"save_user_memoryMemoryStore.save_user_memorys5 $''w ?1 IIdae <@ ? ?s 6 AqueryparamspatternsfeedbackcUR5n/nUHSnUR5nSU;aUS SU;a SU;aUS(aUSSSO/US'US URU5 MU 0n UR5Hup[ U [ 5(a/X'U Hn [ U [ 5(a7U R5n SU ;aU S SU ;aU S XRU 5 MO[ U [ 5(aXRU 5 MyXRU 5 M MXU 'M [R"5RS5UUUU S.nUSRU5 URU5 g) 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)timer[r\r)r^r) r+copyappendrLr7rNr8rnowstrftimer9)r r[r\r]r^r?cleaned_patternsrA pattern_copycleaned_feedbackkeyvaluerK item_copyr@s r" add_sessionMemoryStore.add_sessionsyy{G"<<>Ll* - L0+<?dprCeD\J[=\]_^_=`JLL!9: !23  # #L 1 "..*JC%&&(* %!D!$--$(IIK &)3 )+ 6, 9 )*; <(-44Y?#D$//(-44T:(-44T:").%%+*LLN++,?@.(   Z( $cFUR5nURS/5$)z@Get all historical sessions. Returns: List of all sessions r)r+r6)r r?s r"get_all_sessionsMemoryStore.get_all_sessionss yy{xx B''rquser_idvectorcUR5nX;a///S.X1'[U[R5(aUR 5OUnX1SR U5 UR U5 g)zAdd a positive (liked) pattern vector to user memory. Args: user_id: User identifier vector: Pattern embedding vector positivenegativehistoryryNrVr7rFrGrHrfrYr rurvmemory vector_lists r" add_positiveMemoryStore.add_positivel&&(  FO*4FBJJ)G)GfmmoV  #**;7 f%rqcUR5nX;a///S.X1'[U[R5(aUR 5OUnX1SR U5 UR U5 g)zAdd a negative (disliked) pattern vector to user memory. Args: user_id: User identifier vector: Pattern embedding vector rxrzNr|r}s r" add_negativeMemoryStore.add_negativerrqcDUR5nURU5$)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 rVr6)r rur~s r"get_user_profileMemoryStore.get_user_profiles!&&(zz'""rqNlikeddislikedcUR5nX;a//S.XA'U(a [U5O/U(a [U5O/S.XAS'URU5 g)zSave 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 )ryrz)likedislikepreference_featuresN)rVrNrY)r rurrr~s r"save_preference_features$MemoryStore.save_preference_featuressY&&(  +-2>FO#(DKb)1tH~r2 -. f%rqcvUR5nURU5nU(dgURS5$)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)r rur~profiles r"get_preference_features#MemoryStore.get_preference_features%s6&&(**W%{{011rqu_llm intent_jsonc[RRUR5(a5[ URSSS9n[ R "U5nSSS5 O0n[U[R5(aUR5OUnUU[R"5RS5S.WU'[ URSSS9n[ R"XTSS S 9 SSS5 S SKnUR!["5nUR%S U35 g!,(df  N=f!,(df  NM=f![&a6n S SKnUR!["5nUR)S U 3SS9 Sn A gSn A ff=f)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'rrNrc)rintent created_atrrFrrzSaved intent for user zError saving intent: Texc_info)rr rr rrr+r7rFrGrHrrgrhrr-r.r/r: Exceptionr;) r rurrr! intent_memory u_llm_listr-r=r<s r" save_intentMemoryStore.save_intent4sA Eww~~d5566$113IQ$(IIaLMJI!# ,6eRZZ+H+HeJ$%&lln556IJ&M' "d--sWE -15IF &&x0F KK0 : ;+JI FE E &&x0F LL04tL D EsIAED%A>$"9"9::d--sWE $ ! F!$$W- -FE   &&x0F LL1!5L E  s4.BBA6B6 BB C,CCcrUR5nURU5nUc/$URS/5$)zGet iteration round history for a user (for Stage4). Args: user_id: User identifier Returns: List of round history dictionaries roundsr)r rur~ user_profiles r"get_round_historyMemoryStore.get_round_historyss@&&(zz'*  I"--rq)r r r )zmemory/history.jsonzmemory/user_memory.jsonzmemory/intent_memory.json)NN)r/ __module__ __qualname____firstlineno____doc__strr#rrr+rMr9rVrYrrorsrFrGrrrrrrrrr__static_attributes__rqr"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 / <<