23 lines
792 B
Python
23 lines
792 B
Python
import asyncio
|
|
from sqlalchemy.future import select
|
|
from sqlalchemy import delete
|
|
from app.db.session import async_session_maker
|
|
from app.models.tool import Tool
|
|
|
|
async def run():
|
|
async with async_session_maker() as session:
|
|
result = await session.execute(select(Tool).where(Tool.name == "generate_image"))
|
|
tool = result.scalar_one_or_none()
|
|
if tool:
|
|
await session.execute(
|
|
f"DELETE FROM agent_tools WHERE tool_id = '{tool.id}'"
|
|
)
|
|
await session.delete(tool)
|
|
await session.commit()
|
|
print("Successfully deleted old 'generate_image' tool and its references.")
|
|
else:
|
|
print("Tool 'generate_image' not found in database.")
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(run())
|