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

refactor: Use user_tenants table to update

Browse source
This commit is contained in:
Igor Ilic 2025-11-05 16:14:37 +01:00
parent fa4c50f972
commit c4807a0c67
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
alembic/versions/c946955da633_multi_tenant_support.py
View file

@ -95,7 +95,9 @@ def upgrade() -> None:
)
else:
conn = op.get_bind()
conn.execute(dataset.update().values(tenant_id=tenant_id_from_user, user_id=user.c.id))
conn.execute(
user_tenants.update().values(tenant_id=tenant_id_from_user, user_id=user.c.id)
)
tenant_id_column = _get_column(insp, "datasets", "tenant_id")
if not tenant_id_column: