Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-20 02:10:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

feat: enhance agent permissions migration with DocumentDB compatibility and add dry-run script

Browse source
This commit is contained in:
Atef Bellaaj 2025-06-18 14:37:51 +02:00 • committed by Danny Avila
parent c62a23fafc
commit f9994d1547
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
2 changed files with 23 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

34
config/migrate-agent-permissions.js
View file

@ -32,32 +32,36 @@ async function migrateAgentPermissionsEnhanced({ dryRun = true, batchSize = 100
logger.info(`Found ${globalAgentIds.size} agents in global project`);
// Find agents without ACL entries
// Find agents without ACL entries using DocumentDB-compatible approach
const agentsToMigrate = await Agent.aggregate([
{
$lookup: {
from: 'aclentries',
let: { agentId: '$_id' },
pipeline: [
{
$match: {
$expr: {
$and: [
{ $eq: ['$resourceType', 'agent'] },
{ $eq: ['$resourceId', '$$agentId'] },
{ $eq: ['$principalType', 'user'] },
],
},
localField: '_id',
foreignField: 'resourceId',
as: 'aclEntries',
},
},
{
$addFields: {
userAclEntries: {
$filter: {
input: '$aclEntries',
as: 'aclEntry',
cond: {
$and: [
{ $eq: ['$$aclEntry.resourceType', 'agent'] },
{ $eq: ['$$aclEntry.principalType', 'user'] },
],
},
},
],
as: 'aclEntries',
},
},
},
{
$match: {
author: { $exists: true, $ne: null },
$expr: { $eq: [{ $size: '$aclEntries' }, 0] },
userAclEntries: { $size: 0 },
},
},
{