25 lines
850 B
Python
25 lines
850 B
Python
import asyncio
|
|
import sys
|
|
import os
|
|
|
|
# Add the parent directory to sys.path to allow importing from app
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from sqlalchemy import text
|
|
from app.database import engine
|
|
|
|
async def add_column():
|
|
async with engine.begin() as conn:
|
|
try:
|
|
print("Attempting to add 'stock_code' column to 'llm_usage_logs'...")
|
|
# Using raw SQL to add the column safely
|
|
await conn.execute(text("ALTER TABLE llm_usage_logs ADD COLUMN IF NOT EXISTS stock_code VARCHAR(50);"))
|
|
print("Success: Column 'stock_code' verified/added in 'llm_usage_logs'.")
|
|
except Exception as e:
|
|
print(f"Error executing migration: {e}")
|
|
finally:
|
|
await engine.dispose()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(add_column())
|