From c13b0d66ab834acdc174ffa3ba334c0256b68ba6 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Sun, 7 Sep 2025 13:46:04 +0800
Subject: [PATCH] 'commit'

---
 .../VikingDBMemoryService.cpython-310.pyc     | Bin 14017 -> 16332 bytes
 dsLightRag/Volcengine/T3_ChatWithMemory.py    |  42 +++++-----
 .../Volcengine/T4_OnlyChatWithMemory.py       |  77 ++++++++++++++++++
 3 files changed, 98 insertions(+), 21 deletions(-)
 create mode 100644 dsLightRag/Volcengine/T4_OnlyChatWithMemory.py

diff --git a/dsLightRag/Volcengine/Kit/__pycache__/VikingDBMemoryService.cpython-310.pyc b/dsLightRag/Volcengine/Kit/__pycache__/VikingDBMemoryService.cpython-310.pyc
index 9f9f8e5343c071a58c02a37c5b7010fa971301fd..2f61a84a0684f87ea6348e5e0e25be4926fb6b8b 100644
GIT binary patch
delta 5220
zcmaJ^dvp}#6`z@%oqgy1AS8he7{UUEmw-}4IVeX2@d4ChTDoW58N!Cm?(*#f3C^<A
zfPw-F(^{3#CWuJ1@^C3~6j6`;qgo%e538fK4_N}Iw$<Z_Rn(r|-#6rWXp^(Qz2EEJ
z``!Ed?)U90t#4!tl7WC%pg-s67uTE!?=Gm3O2pk$KC3TLOScG3^=g?~PTv#M3bm5H
z8`M#16@5=srR73sv^h?EUu?>KBQ6q43P=en1QZkW2&~jx;W%1}Mo45O7$_ZpGP7E~
zN~|zf%9G}lqqlNkXvie=1qjB`C#?{;1+VCL^1kTq;<DW9@@Y}5Hn%JFVvTu3sVW+=
zIGB4+c@#64d!blvo^UTNfj(A5U^&t{3wc?IS?r1U_`))bm75vQ6W3Re7?N2vdKxiX
zS))Z%jj^d1aN7I17Nh8}9BZ|9OOEMU>#$8Wd-vQlfySF0kr<;W%q2M_F>Hl7AuDV)
z<T(HgvhM)q0WJrO*w2*QBHu!(E`%Wm0jY%fk<%J;PX%U*$|4NiV18V%poY)7g4#`m
zDW;_&Obat@U7JQBxW&A-uw1;{{BGfd?11$qUs{joZRv0{Nt>PlMKcL3S=Tn`mZBxl
zA9C|BSdK_4W+|jJ(wb%~F_*_gWR|0i#Vwg>>C{LUr_;1JebQqHyrNsA#EQAE3j?BW
z1vE?oOg2l4>xzaGi%l@ED_)w#dWH>g=PYMD8Bx;=M6ncA<6aMBn$5zzCIX}z@+!a&
z0C3lGkcaGQNN$qCE$L)}jj%dA5mv6S<U3;aPFfQ3aLt3huqgDx6Q%-&yFt@Oc%Wet
zjI-5%|8F<Ytm$EGLsU!Yv1IPG(x6y%BdzLj4CaX2hZ$2U-%`#u$xRH|fw{N5b6O16
zmH-maAlrT=(Q)ZT%@ubxtf%3SV;SR%m-a7acR?Cz;M0mKrJXuA)|<DEuQ2N?BeuO3
zOk4{<@>wh3CIEc6$S1~qSZ5xqEd3wue^lAC5|b<^QJA$;_G7IvqUGvXT&E+tY(SxF
z$fh#r%>p(7F1^POM;#H9=8meyYi=G;Vzb=#>K^oV0JZ?O0xqd3_f?fHihX9HdamQb
z)n4;h^=$LyYNephO-p)zbm;81=jI0c%)aVp#m~({H50`i^PQUUb9$t~MD|FQ)Huy@
zv<@#SwBGX0i^tM>G})S(|2JxXZceVfk&Lacy_)cN?KQLeP<9tAKKsn$`OZUUH}5;Q
zxudbM(NfawX{u2@Qj^S*am{i?lL=i*=$71;*4PCzG^TbE4LV!tbjR&|ilm5ZTe|e7
zb4NSK_56Wj=Z-%C?^9!z+Z#G_>d`Z&j?WER<~w6<b6Ik9O+>fMNn^*%gdeOEfWu0a
zV*}Y!Lw=qZ%NvikhNJNqTyVjKa7?w#yT^`iv5J`%*Vac8dYJZ~WHBvm1#IubDMlOD
ztpYyKP9=)=utDeUT3!StuF>g8ScQ>vn!*GfRP*BP_9+`eQ+PDW%mhc}g29_bN<pgb
zg!n_VVO(O(0l54GCOBdVojs4{y?`eH*h@cKkC1Hn2TjFPb_`u!J(-F{!-+_%#yD*_
zWB5TI;)40@xLWb^T=n=RlK7z6T>sb&+|&kG;id*086*ZTIF>UV*R+%!RwuDO$#|U$
zUF7_2=HdyH%U^+>kv4PNC$xy(D=~Bx!3Fc%h7fJOVdA`Me&2bhlWrznrv_$O=BA0Y
zMTfCyi2J#xC(f6O_<>DfC=JlfuO_iO`4$Gf=e3#_;x?!sdZ>o1#B#W61^A9A_vv(`
zS!;4fnHJHtur|#2o0AtxMEK&yrCCnVi*$dev-kM^GbeWi8~VyfS3|IO@5%huT<>FB
z&K*9Qf9YO22@JGe!F~gK?-0<r<9YO5<doi?(x@czjg2TAx6*XW8BMkkXMJ>Cw8iys
zyoPwjcJYyPG|4p5hZg&YpwIlWu_n;xBJu3@)A?;j>EzT+9lh#;6b$wiP;X;!qU~Y+
z`Ga$V7aWZ<S6^_1<^~s?muQq!Jw#uY5=*FbPWnbcPv=v%Dtn#1t&ca~W4WeJqrVw5
z`^-nC-a?@{KXr1pC%^{Nu$fld{81vQ6dgs&5!VuyM@zI3PbtJhXf~KK&dFcW(on&W
zii(Ia9vxIvE5J0JwTB}*RTw&+0lON9Z?GQ*{Ax`l6KTzI51eSlj-Msc`@xR9P1fH*
z^AMV$c`*0Ev{yytw~${o56oCJ>-P{x9Fkw54~~Z<+tZQcH^<IQXD`F7dccI?(17N|
zexT=-0hvDrCH698bkGgBY#9yy0dg%vMpzYKG=b&O*AQ9NSfbgImnNfY*%%B30p);&
z0^LHE9}lcpv{lpBBvm#A1C4;`fM&oN01gY|M`<IZy8)X4ycj(P=`?`v`)x=(DhDBb
zLtq8MRA8w(MYVW59JYgMKa=NUP8p!u?7Cuk7H1!Cmw`VgNeX=y(kH!$Ko>HCE@nhs
z%1FAy5Dd|fnjIZtx>%Q0u|vE^)Sc*6CDozIJLQZ+S5&7Vsme|%BO4AwUX7m(t~Nv@
zzinUs$p`W~9zNT-Gk<?~e)^srbj9Qk@29J=_vs^TIA{5nww~{Pfv&LLT}Sf!JMs@c
za{j&t&zyYb+~L!`FCI2eHPmMp+_ogV;FiVVCD-3U2Y?M^mLBg{k_KMkhCH%gL*MHJ
zR409qy#PPp*MK(wgH_NuM9G^NdJAwEVAGR7D0mPN{DV|h(m@hrNfu?XKyr(cB#HLF
z@{OcGQvTzX8D>Xig8n}|6Y=4}ewg&=LYJtDhGVawO8Xq$q9~}2jO-CMZl<7%s#8^(
zCDoPflrj|a7Kw7<Fr2De^-$cEW+CG;lrD$rjS1Znefv~LH)Un5%oZDp>a7w~f0b~Z
zu&N^CRs)86J^fHHNUFlOt4p?J!UCFE*eSvMx*3LxZ8c=BWd(Pys5?soTr8&4yE;Vj
zw#4>UR}7b0+98pV+V4vlPsYpVl^GtIQ*L;<>WaOxTB*BQ+|(xH_;sm{-bs7jr*M<1
z;ksQ|Zd<5EkKL99-P8V>C}ezDQP?<L_Zq$qG2>Tj3_tDU2>Dn#N65(BV67pO!7&$|
z@^$&uu~3vK;-f*sN5=U6nuXhm?so`kUB<V`x13u2a>8l%-6h^7tn-J)v3Ds^?2mx#
zd%zL5340%%s1ED{0I1A91pEoG3-A%(&w#%G{tCcbj(rR`1IPo;0(uD+v2zg51O5*9
z1d#m)@SlKx0X`*I#y%qq1#Ge=(R~EKOLZTlF981rd<pmpZ~^c&fS1fmmS7q^1&TM3
zP(l=!M295G=BLwcEoCS<QIw1Rzks=NbX}H$Z--Zxs5D3@0v$w9;xZAGO;TkdCvka;
zU?|rK9b&hbaT9M`hMSV+-shk+x_F{INZQ7r>ZMU%C!GMpSxF?ORP$D)kntE|i`?RD
zQHanUwSY>F*YIXrTxwyJ(Bf|KWPG;^i5jAgw|y;9$4}~P4ip*wekavp%%Z%PWc-`_
zlz0A(ymKc?2iamcyL|oT)iT4!-QYYjh&!Qjb|1xoZR3xq?GWz=JV0RXs;QdEcA&Wv
zup6)kFyhAhG4*{x;bsq$uyG0=s%E6NEX#g^$&UaY1w01W3)lzvDd1-S(2{ilx(SxC
z{SY4~p!fFy$avuL#IZw=dH_!Wo(B9JPz>k<Tmg7S5Sj<5#p~c8zm8(UF~GBc9N;*h
z1MmyL2>|EUh`l~X1D}9e^GNVYMN~V@slnM1hJj(U@?Zjec%eK+f+43pJUc`lDw1@=
z@D^1=qL6H#doDY5y0y9--9$6Ww<Du~j^<~<(puNXDypF})lkKdiD2%X^2M^|F%+||
zt~%>%Uqzg86K4v!c0UMe6PxoTka2Hv!?wyLFovBZ=ja_h80i<NW5`bl1eAWLzlGkx
zC}K-?6kmj7|8%$!;-%ttng)vKnaEzk475Ta%M}i*$td0SGpKF3;>qS_dX2DeG39x)
za@J|F$$WQKNIbA-_V<=NiC&&tTtD{R_ckOPXdZt1g)x{gE|t-!<(@~s6x-t3d@wD2
YrMYMJ<po7vdf#N-hP%Q&&41Z{0ZrTFc>n+a

delta 3170
zcmaJ@eQXrR75D7jt<Pu9UvU1wcjk<1?8R_4IDy81DWQP?hNPjXs!f{B=5DYT_HK{a
zU23}@M*|5Vl_FqhrGROa5283~6DNln)TXpi@<-G}iYir+I)7CyHl$Vmpnp|rRrS4D
zu+J{7rTh8Ky!YmP%zJP4)6xg|Iwu~FRl(omiC>MK-g~w#sr<d_?7CYK<zSW8#P276
zAflX9c9Euf5Mm7oO#stXYz6m1GcCB#2wRSf74wkbTf+B|7Vd}FZf?fi)_GQi5bo6h
ztcIVf0z|7~WLZS+WzROQDE}pVjgTatR=Y`xf2OuIEK%+(|3iHdCC){6l4kySbWanW
z#~J}d$n}^OW6Sx58nad^CXm<6FV$Rrpan=R%#yg%kJ9QGHFK1)0c1picy=HcH$><N
zoe*KmqxnTao{F8@-vfL-WLgZeumzPBhM}+s(kk1G^p_Ek$+jYFL&zX3iD!NJQ0-2o
zb1kw$0B|IlUn<sL{%w4asJoH1hllESB&F27u-5B~j22Bs4Mvap6jJat{$zcEWcacA
zHKlp$dadi3p6?o22ZA2NgYE$kVULb`LZt=V*P=2FB4idVp~9hN-euoFS(y{UM962h
z2s7#ym%6wS#BBJv9RM*BC2+B%+};o;wU6QneF*FLU}I;);=*D*Jl8l<LVbn-N#{hw
zcFdg1x?n@7IV!!LziGA!<sJjT>qhzs1Ooxx6(R7D4I>o=$~f#g1-8WM;)@XF{-!OY
zlm!*7MxMOj3(MiT=n10;i`@Y6B_13?AvDfL5&pm3_$c)ZI-aFP&vMFNEKifReV{cR
zTDT)=AI<R9iIWK#q%=_x0sm!UYTz(h8$odJgmB<j5jXCxbgKEOo^JzN3msrmdH)?{
zHjY%KfrgegbrLowyZO<sX8u5n`Aw;E1O@TlSstN)Z~y^)xLr?7`oM74Meq=OguDCo
z@0RIDQAI?cpxFe}f|a+RNDo=I2MzlWYRcoQD?<D}?jA=dA>94sf3?0%iu|W-nWl$P
zXpxWnR@?HG&;C!{a@$Ie5TC!$J`|eV7UREb-^BmeuGaZcP`Ppbjk)LE8%p~;k$ji9
z{NrRV(fF-o*Jc<F^UbCyLdgt>P=4`H0eunMYFn<Cb@IjGKf&7N`%-(s*pE}&0Dqs_
zzRAZ4w`0%T&t9FKdSmYRrJKi3WHK3{x)UzUC{5AB%(1Dc&N>B;7CaI5UCJi;o{rR7
zU`F!L-8MT3XD|XFPa7H8e{<#pxIX*ZJ2$UfM(@vb91H@w{{D;C-@h`H*7>cDN2>L3
zcFgp2zIRo}AdY`_1_7HDCdW9~%W2ExRfw2v=Z&mwp$paMf??%!esNXTVZD)2n;tU@
zo&oVY%%ZLy54<;u3<CG`I?0%T5d|@fd(vG!hDot0bVot2H(eLfglE9avnPTst0+_S
zsFhzzx0b(>eu%&ooay{Id7AHDT^KuuE^`zJSp|>1gv(P1KSID*?o{hVp!H=7rmP&B
z!mXI+6s@dLF!Pk1K&598WTRKOz*Ai*;+D5`?Ni9}{8aaa2c@YAl$ECD9a$(0IdDY8
zwJ9wIX^lXgJ50_w?RJ$P<kp(L#H)9+S)N{Vn8fZw))2rXPxWdL{O;bZ?ee{=flYXs
z<eZw9()mw%Qw<kTslxs8C%wZ;gKXHN$O?LuxAn27WC#o1W6FX-d<>5do{pf^D>P9^
z4&u0s2=3l>%~7gHvy7S^HRvMaf9u<+K*4XwjFe=F-Y)kuQ#Y=>a{cNH=^h_l?MdIb
zcy;!f@{J28Z@x7%`|D?+6J%L`oxP3rX3!sn$$PB5T~ppfk;3yY16wjG9szL?$vQsN
zY%M$&zU>)y66%bdmK-<hFbe14ioFNm^Va@k+^+_5?)0_U=VqXDcJ#0O)})e7`}jI!
z(!GI)v%h$KC_Nd<3?7;cX+!B<UnszZqkahgglZLX&?i2ADa}p285|Yp1!4X9_|tmz
zh5`87IOy}M{SQNGY6kjBTAVFh!w*0^s2|lV7NHeI$fgBRLkm9CDWtff%oeUp*5vEx
zOkKqc#f&kTof1%boKcTW7^Vjk2HG<o%wOZd;KM*ZtwpEc!e!p+XrTuA6U5ThAoD?k
zyo)DSlo>A1m;X8NAyMB!`ZE7$<F0$lNG@rTH*pUghqTE8{;!7HKIoRZQEClB&tmAs
zrFIA4^Xhv@%a1{Yy@E7uzzcZb05Gp0onmdczY?JxK-73+P^>wtFe<_$PWA{}g{)45
zW`wU+LAt~;{J^rZdFqWhIkp}d{RkTo#t<w7Y!)V4={rb$7eU^k97dOL`7(lx`z@qo
zR?Z{!A%KV*FkoRiWvOi&Mv&CtGkH78BoLDPH}@R;6865<EvgM8=VU>25Y?+~XLJ-o
zVIQI3OI#VgMg;$0SR?27Uxy#tEYpwq4!(2b{8+d+Nudc5-3o7tZ_{A{D|a0~yM4W~
GBK99w<M`hI

diff --git a/dsLightRag/Volcengine/T3_ChatWithMemory.py b/dsLightRag/Volcengine/T3_ChatWithMemory.py
index 6df47877..7538b99e 100644
--- a/dsLightRag/Volcengine/T3_ChatWithMemory.py
+++ b/dsLightRag/Volcengine/T3_ChatWithMemory.py
@@ -16,28 +16,27 @@ if not logger.handlers:
     logger.addHandler(handler)
 
 
-
 def main():
     logger.info("开始测试大模型记忆功能...")
-    
+
     try:
         # 使用initialize_services函数初始化服务和LLM客户端
         memory_service, llm_client = initialize_services()
-        
+
         collection_name = MEMORY_COLLECTION_NAME
-        user_id = "liming"
-        assistant_id = "assistant"
-        
+        user_id = "liming"  # 用户李明
+        assistant_id = "assistant"  # 助手ID:助手
+
         # 告知大模型用户信息
         logger.info("告知大模型用户信息...")
         user_info = "李明，男生，15岁，家住长春"
-        
+
         # 记录信息到记忆体
         logger.info("记录用户信息到记忆体...")
-        
+
         # 准备对话历史
         conversation_history = []
-        
+
         # 使用正确的handle_conversation_turn方法参数
         response = handle_conversation_turn(
             memory_service=memory_service,
@@ -47,9 +46,9 @@ def main():
             user_message=f"请记住以下用户信息：{user_info}",
             conversation_history=conversation_history
         )
-        
+
         logger.info(f"模型回复: {response}")
-        
+
         # 归档对话
         archive_conversation(
             memory_service=memory_service,
@@ -59,15 +58,15 @@ def main():
             conversation_history=conversation_history,
             topic_name="user_info"
         )
-        
+
         # 等待一段时间确保索引更新
         logger.info("等待索引更新...")
         time.sleep(5)
-        
+
         # 验证大模型是否记住个人信息
         logger.info("验证大模型是否记住个人信息...")
         test_conversation_history = []
-        
+
         test_response = handle_conversation_turn(
             memory_service=memory_service,
             llm_client=llm_client,
@@ -76,22 +75,23 @@ def main():
             user_message="请告诉我李明的个人信息",
             conversation_history=test_conversation_history
         )
-        
+
         logger.info(f"测试回复: {test_response}")
-        
+
         # 检查回复中是否包含关键信息
         keywords = ["李明", "男", "15", "长春"]
         found_keywords = [kw for kw in keywords if kw in test_response]
-        
+
         if len(found_keywords) == len(keywords):
             logger.info("✅ 大模型成功记住了用户信息！")
         else:
             logger.info(f"❌ 大模型可能没有完全记住用户信息。找到的关键词: {found_keywords}")
-            
+
             # 尝试直接搜索记忆
             logger.info("尝试直接搜索记忆...")
             filter_params = {
                 "user_id": [user_id],
+                "assistant_id": assistant_id,  # 添加这一行
                 "memory_type": ["sys_event_v1", "sys_profile_v1"]
             }
             search_result = memory_service.search_memory(
@@ -100,13 +100,13 @@ def main():
                 filter=filter_params,
                 limit=5
             )
-            
+
             logger.info(f"搜索结果: {search_result}")
-            
+
     except Exception as e:
         logger.error(f"操作失败: {e}")
         sys.exit(1)
 
 
 if __name__ == "__main__":
-    main()
\ No newline at end of file
+    main()
diff --git a/dsLightRag/Volcengine/T4_OnlyChatWithMemory.py b/dsLightRag/Volcengine/T4_OnlyChatWithMemory.py
new file mode 100644
index 00000000..e5325f3c
--- /dev/null
+++ b/dsLightRag/Volcengine/T4_OnlyChatWithMemory.py
@@ -0,0 +1,77 @@
+import logging
+import sys
+import time
+
+from Volcengine.Kit.VikingDBMemoryService import MEMORY_COLLECTION_NAME, initialize_services, \
+    handle_conversation_turn, archive_conversation
+
+# 控制日志输出
+logger = logging.getLogger('ChatWithMemory')
+logger.setLevel(logging.INFO)
+
+# 只添加一次处理器，避免重复日志
+if not logger.handlers:
+    handler = logging.StreamHandler()
+    handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
+    logger.addHandler(handler)
+
+
+
+def main():
+    logger.info("开始测试大模型记忆功能...")
+    
+    try:
+        # 使用initialize_services函数初始化服务和LLM客户端
+        memory_service, llm_client = initialize_services()
+        
+        collection_name = MEMORY_COLLECTION_NAME
+        user_id = "liming" # 用户李明
+        assistant_id = "assistant" # 助手ID:助手
+        
+        # 验证大模型是否记住个人信息
+        logger.info("验证大模型是否记住个人信息...")
+        test_conversation_history = []
+        
+        test_response = handle_conversation_turn(
+            memory_service=memory_service,
+            llm_client=llm_client,
+            collection_name=collection_name,
+            user_id=user_id,
+            user_message="请告诉我李明的个人信息",
+            conversation_history=test_conversation_history
+        )
+        
+        logger.info(f"测试回复: {test_response}")
+        
+        # 检查回复中是否包含关键信息
+        keywords = ["李明", "男", "15", "长春"]
+        found_keywords = [kw for kw in keywords if kw in test_response]
+        
+        if len(found_keywords) == len(keywords):
+            logger.info("✅ 大模型成功记住了用户信息！")
+        else:
+            logger.info(f"❌ 大模型可能没有完全记住用户信息。找到的关键词: {found_keywords}")
+            
+            # 尝试直接搜索记忆
+            logger.info("尝试直接搜索记忆...")
+            filter_params = {
+                "user_id": [user_id],
+                "assistant_id": assistant_id,  # 添加这一行
+                "memory_type": ["sys_event_v1", "sys_profile_v1"]
+            }
+            search_result = memory_service.search_memory(
+                collection_name=collection_name,
+                query="李明 15岁 长春 男生",
+                filter=filter_params,
+                limit=5
+            )
+            
+            logger.info(f"搜索结果: {search_result}")
+            
+    except Exception as e:
+        logger.error(f"操作失败: {e}")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file