gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Resolve SQLite migration issue

Browse source
This commit is contained in:
Igor Ilic 2025-11-06 14:51:25 +01:00
parent ac6dd08855
commit ac751bacf0
1 changed files with 17 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
alembic/versions/c946955da633_multi_tenant_support.py
View file

@ -97,13 +97,23 @@ def upgrade() -> None:
# Insert into user_tenants table # Insert into user_tenants table
if user_data: if user_data:
op.bulk_insert( if op.get_context().dialect.name == "sqlite":
user_tenants, insert_stmt = user_tenants.insert().values(
[ [
{"user_id": user_id, "tenant_id": tenant_id, "created_at": _now()} {"user_id": user_id, "tenant_id": tenant_id, "created_at": _now()}
for user_id, tenant_id in user_data for user_id, tenant_id in user_data
], ]
) )
conn.execute(insert_stmt)
conn.commit()
else:
op.bulk_insert(
user_tenants,
[
{"user_id": user_id, "tenant_id": tenant_id, "created_at": _now()}
for user_id, tenant_id in user_data
],
)
tenant_id_column = _get_column(insp, "datasets", "tenant_id") tenant_id_column = _get_column(insp, "datasets", "tenant_id")
if not tenant_id_column: if not tenant_id_column: