From 9598bd5905da60df74f49deb3db5d27f49c0a7d6 Mon Sep 17 00:00:00 2001
From: Mark <348207283@qq.com>
Date: Tue, 28 Apr 2026 13:44:05 +0800
Subject: [PATCH] [modify] migration script

---
 api/migrations/versions/1f85dce125e5_202604271530.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/api/migrations/versions/1f85dce125e5_202604271530.py b/api/migrations/versions/1f85dce125e5_202604271530.py
index cc0caa3a..c35cf6e5 100644
--- a/api/migrations/versions/1f85dce125e5_202604271530.py
+++ b/api/migrations/versions/1f85dce125e5_202604271530.py
@@ -28,6 +28,11 @@ def upgrade() -> None:
                existing_comment="模型能力列表（如['vision', 'audio', 'video']）",
                existing_nullable=False)
     # ### end Alembic commands ###
+    op.execute("""
+        UPDATE files
+        SET file_key = 'kb/' || kb_id::text || '/' || parent_id::text || '/' || id::text || file_ext
+        WHERE file_ext != 'folder' AND file_key IS NULL
+    """)
 
 
 def downgrade() -> None: