From 552380172c092aaa4ad6b394c000bfcfa73fa9a2 Mon Sep 17 00:00:00 2001 From: goulustis Date: Fri, 30 Jan 2026 11:43:40 +0800 Subject: [PATCH] implemented dummy --- lang_agent/components/conv_store.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/lang_agent/components/conv_store.py b/lang_agent/components/conv_store.py index 1645761..5b12dee 100644 --- a/lang_agent/components/conv_store.py +++ b/lang_agent/components/conv_store.py @@ -87,6 +87,17 @@ class ConversationPrinter: def __init__(self): self.id_dic = {} - - -CONV_STORE = ConversationStore() \ No newline at end of file + def record_message_list(self, conv_id:str, inp:List[BaseMessage]): + inp = [e for e in inp if not isinstance(e, SystemMessage)] + curr_len = self.id_dic.get(conv_id, 0) + to_print_msg = inp[curr_len:] + for msg in to_print_msg: + msg.pretty_print() + + if curr_len == 0: + self.id_dic[conv_id] = len(inp) + else: + self.id_dic[conv_id] += len(to_print_msg) + +# CONV_STORE = ConversationStore() +CONV_STORE = ConversationPrinter() \ No newline at end of file