File tree Expand file tree Collapse file tree 1 file changed +10
-3
lines changed
Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Original file line number Diff line number Diff line change 22import difflib
33import json
44import re
5+ import uuid
56
67from datetime import datetime
78from typing import TYPE_CHECKING , Any , Literal
@@ -236,6 +237,7 @@ def _single_add_operation(
236237 else :
237238 to_add_memory = new_memory_item .model_copy (deep = True )
238239
240+ to_add_memory .id = str (uuid .uuid4 ())
239241 if to_add_memory .metadata .memory_type == "PreferenceMemory" :
240242 to_add_memory .metadata .preference = new_memory_item .memory
241243
@@ -359,9 +361,14 @@ def semantics_feedback(
359361 lang = detect_lang ("" .join (memory_item .memory ))
360362 template = FEEDBACK_PROMPT_DICT ["compare" ][lang ]
361363 if current_memories == []:
362- # retrieve
363- last_user_index = max (i for i , d in enumerate (chat_history_list ) if d ["role" ] == "user" )
364- last_qa = " " .join ([item ["content" ] for item in chat_history_list [last_user_index :]])
364+ user_indices = [i for i , d in enumerate (chat_history_list ) if d ["role" ] == "user" ]
365+ if user_indices :
366+ last_user_index = max (user_indices )
367+ last_qa = " " .join (
368+ [item ["content" ] for item in chat_history_list [last_user_index :]]
369+ )
370+ else :
371+ last_qa = " " .join ([item ["content" ] for item in chat_history_list ])
365372 supplementary_retrieved = self ._retrieve (last_qa , info = info , user_name = user_name )
366373 feedback_retrieved = self ._retrieve (memory_item .memory , info = info , user_name = user_name )
367374
You can’t perform that action at this time.
0 commit comments