完善模型测试与参考文件历史能力

backend/database.py

@@ -1,4 +1,5 @@
 from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker, AsyncSession
+from sqlalchemy import inspect, text
 from sqlalchemy.orm import DeclarativeBase
 from config import settings
 
@@ -24,5 +25,18 @@ async def init_db():
     from models.ai_model import AiModel
     from models.document import Document
     from models.generation_log import GenerationLog
+    from models.reference_file import ReferenceFile
     async with engine.begin() as conn:
         await conn.run_sync(Base.metadata.create_all)
+        dialect_name = conn.dialect.name
+        columns = await conn.run_sync(lambda sync_conn: [column["name"] for column in inspect(sync_conn).get_columns("ai_models")])
+        if "supports_streaming" not in columns:
+            if dialect_name == "sqlite":
+                await conn.execute(text("ALTER TABLE ai_models ADD COLUMN supports_streaming BOOLEAN DEFAULT 0"))
+            else:
+                await conn.execute(text("ALTER TABLE ai_models ADD COLUMN supports_streaming TINYINT(1) DEFAULT 0"))
+        if "enable_reasoning" not in columns:
+            if dialect_name == "sqlite":
+                await conn.execute(text("ALTER TABLE ai_models ADD COLUMN enable_reasoning BOOLEAN DEFAULT 0"))
+            else:
+                await conn.execute(text("ALTER TABLE ai_models ADD COLUMN enable_reasoning TINYINT(1) DEFAULT 0"))