shudao-main/shudao-chat-py/test_db_check.py

import sys
import os

# Append the directory to sys.path so we can import modules
sys.path.append("/Users/fanhong/UGIT/shudao-main/shudao-chat-py")

from database import SessionLocal
from models.chat import AIConversation, AIMessage

def check_db():
    db = SessionLocal()
    try:
        conv = db.query(AIConversation).filter(AIConversation.id == 11453).first()
        if conv:
            print("Conversation:", conv.content)
        
        messages = db.query(AIMessage).filter(AIMessage.ai_conversation_id == 11453).order_by(AIMessage.id).all()
        for msg in messages:
            print(f"\n--- {msg.type} ---")
            print(msg.content[:500])
            print("...")
    finally:
        db.close()

if __name__ == "__main__":
    check_db()