from src import db, create_app
from src.models import Agente, User, Chat

def check_db():
    app = create_app()
    with app.app_context():
        try:
            print("\n=== Tabelas no banco de dados ===")
            for table in db.metadata.tables:
                print(f"- {table}")
            
            print("\n=== Colunas da tabela agente ===")
            agente_columns = Agente.__table__.columns
            for column in agente_columns:
                print(f"- {column.name}: {column.type}")
            
            print("\n=== Agentes existentes ===")
            agentes = Agente.query.all()
            for agente in agentes:
                print(f"\nAgente: {agente.nome}")
                print(f"ID: {agente.id}")
                print(f"Descrição: {agente.descricao}")
                print(f"System Prompt: {agente.system_prompt}")
                print(f"Example Prompt: {agente.example_prompt}")
                print(f"Temperature: {agente.temperature}")
                print(f"Max Tokens: {agente.max_tokens}")
            
        except Exception as e:
            print(f"Erro ao verificar banco de dados: {str(e)}")

if __name__ == '__main__':
    check_db()