UniqAI/hexabot
1
0
Fork 0
mirror of https://github.com/hexastack/hexabot synced 2025-06-26 18:27:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: simplify prepareBlocksOutOfCategoryUpdateScope logic and add error handling

Browse Source
This commit is contained in:
medchedli 2025-06-11 11:58:09 +01:00
parent 79f866e406
commit 6a128541df
1 changed files with 39 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
api/src/chat/repositories/block.repository.ts
View File

@ -178,18 +178,25 @@ export class BlockRepository extends BaseRepository<
});
for (const { id, nextBlocks, attachedBlock } of blocks) {
const updatedNextBlocks = nextBlocks.filter((nextBlock) =>
ids.includes(nextBlock),
);
try {
const updatedNextBlocks = nextBlocks.filter((nextBlock) =>
ids.includes(nextBlock),
);
const updatedAttachedBlock = ids.includes(attachedBlock || '')
? attachedBlock
: null;
const updatedAttachedBlock = ids.includes(attachedBlock || '')
? attachedBlock
: null;
await this.updateOne(id, {
nextBlocks: updatedNextBlocks,
attachedBlock: updatedAttachedBlock,
});
await this.updateOne(id, {
nextBlocks: updatedNextBlocks,
attachedBlock: updatedAttachedBlock,
});
} catch (error) {
this.logger?.error(
`Failed to update block ${id} during in-category scope update.`,
error,
);
}
}
}
@ -206,21 +213,30 @@ export class BlockRepository extends BaseRepository<
ids: string[],
): Promise<void> {
for (const block of otherBlocks) {
// Handle attached block references
if (block.attachedBlock && ids.includes(block.attachedBlock)) {
await this.updateOne(block.id, { attachedBlock: null });
}
try {
// If block has an attachedBlock, we only need to check that
if (block.attachedBlock) {
if (ids.includes(block.attachedBlock)) {
await this.updateOne(block.id, { attachedBlock: null });
}
continue; // Skip nextBlocks check since it can't have both
}
// Handle nextBlocks references
const filteredNextBlocks = block.nextBlocks?.filter(
(nextBlock) => !ids.includes(nextBlock),
);
// Only check nextBlocks if there is no attachedBlock
const filteredNextBlocks = block.nextBlocks?.filter(
(nextBlock) => !ids.includes(nextBlock),
);
// If the filtered nextBlocks length is different from the original, update the block
if (filteredNextBlocks?.length !== block.nextBlocks?.length) {
await this.updateOne(block.id, {
nextBlocks: filteredNextBlocks || [],
});
if (filteredNextBlocks?.length !== block.nextBlocks?.length) {
await this.updateOne(block.id, {
nextBlocks: filteredNextBlocks || [],
});
}
} catch (error) {
this.logger?.error(
`Failed to update block ${block.id} during out-of-category scope update.`,
error,
);
}
}
}