Merge branch 'nsc/__experimental_streamSteps'

This commit is contained in:
Nicolas 2025-01-14 02:00:58 -03:00
commit 23d3257a57
2 changed files with 15 additions and 2 deletions

View File

@ -54,6 +54,19 @@ export async function updateExtract(
extract.steps = [...current.steps, ...extract.steps];
}
// Limit links in steps to 500
if (extract.steps) {
extract.steps = extract.steps.map(step => {
if (step.discoveredLinks && step.discoveredLinks.length > 500) {
return {
...step,
discoveredLinks: step.discoveredLinks.slice(0, 500)
};
}
return step;
});
}
await redisConnection.set(
"extract:" + id,
JSON.stringify({ ...current, ...extract }),

View File

@ -256,7 +256,7 @@ export async function performExtraction(
step: ExtractStep.MULTI_ENTITY_SCRAPE,
startedAt: Date.now(),
finishedAt: Date.now(),
discoveredLinks: [],
discoveredLinks: links,
},
],
});
@ -510,7 +510,7 @@ export async function performExtraction(
step: ExtractStep.EXTRACT,
startedAt: Date.now(),
finishedAt: Date.now(),
discoveredLinks: [],
discoveredLinks: links,
},
],
});