add better print options

This commit is contained in:
2026-02-06 10:56:09 +08:00
parent aa4c37909a
commit cd867761fe

View File

@@ -3,6 +3,7 @@ import psycopg
from typing import List, Dict, Union
from enum import Enum
import os
from loguru import logger
from langchain_core.messages import HumanMessage, AIMessage, ToolMessage, SystemMessage, BaseMessage
@@ -96,6 +97,7 @@ class ConversationPrinter:
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:]
print("\n")
for msg in to_print_msg:
msg.pretty_print()
@@ -110,3 +112,8 @@ CONV_STORE = ConversationStore()
def use_printer():
global CONV_STORE
CONV_STORE = ConversationPrinter()
def print_store_type():
global CONV_STORE
logger.info(CONV_STORE)