AI/reader
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/jina-ai/reader synced 2025-08-17 03:25:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Yanlong Wang 2024-04-13 08:07:55 +08:00
parent 5199b00eeb
commit 950338261a
No known key found for this signature in database
GPG Key ID: C0A623C0BADF9F37
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/functions/src/cloud-functions/crawler.ts
View File

@ -90,6 +90,10 @@ ${this.content}
try {
for await (const scrapped of this.puppeteerControl.scrap(urlToCrawl.toString(), noCache)) {
if (!scrapped) {
continue;
}
const formatted = this.formatSnapshot(scrapped);
if (scrapped.screenshot && screenshotEnabled) {

8
backend/functions/src/services/puppeteer.ts
View File

@ -15,7 +15,7 @@ export interface PageSnapshot {
href: string;
html: string;
text: string;
parsed: {
parsed?: {
title: string;
content: string;
textContent: string;
@ -78,7 +78,7 @@ export class PuppeteerControl extends AsyncService {
timeout: 10_000
}).catch((err) => {
this.logger.error(`Unknown firebase issue, just die fast, quitting process.`, { err });
process.nextTick(()=> {
process.nextTick(() => {
process.exit(1);
});
return Promise.reject(err);
@ -153,7 +153,7 @@ function giveSnapshot() {
return page;
}
async *scrap(url: string, noCache: string | boolean = false): AsyncGenerator<PageSnapshot> {
async *scrap(url: string, noCache: string | boolean = false): AsyncGenerator<PageSnapshot | undefined> {
const parsedUrl = new URL(url);
// parsedUrl.search = '';
parsedUrl.hash = '';
@ -236,7 +236,7 @@ function giveSnapshot() {
while (true) {
await Promise.race([nextSnapshotDeferred.promise, gotoPromise]);
if (finalized) {
yield { ...snapshot, screenshot };
yield { ...snapshot, screenshot } as PageSnapshot;
break;
}
yield snapshot;