28 lines
936 B
Python
28 lines
936 B
Python
|
import asyncio
|
|||
|
|
from sqlalchemy import create_engine, text
|
|||
|
|
from chainlit.data.sql_alchemy import SQLAlchemyDataLayer
|
|||
|
|
|
|||
|
|
DATABASE_URL = "postgresql+asyncpg://ai_user:secure_password_here@postgres:5432/ai_station"
|
|||
|
|
|
|||
|
|
async def init_database():
|
|||
|
|
"""Inizializza le tabelle per Chainlit"""
|
|||
|
|
print("🔧 Inizializzazione database...")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# Crea data layer
|
|||
|
|
data_layer = SQLAlchemyDataLayer(conninfo=DATABASE_URL)
|
|||
|
|
|
|||
|
|
# Forza creazione tabelle
|
|||
|
|
if hasattr(data_layer, '_create_database'):
|
|||
|
|
await data_layer._create_database()
|
|||
|
|
print("✅ Database inizializzato con successo")
|
|||
|
|
else:
|
|||
|
|
print("⚠️ Metodo _create_database non disponibile")
|
|||
|
|
print("ℹ️ Le tabelle verranno create automaticamente al primo utilizzo")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ Errore: {e}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
asyncio.run(init_database())
|