github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2026-03-22 15:39:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove save image function

Browse Source
This commit is contained in:
Sha Zhou
2025-06-10 17:40:35 +08:00
parent 16d858f15e
commit 609687aaeb
3 changed files with 43 additions and 375 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/utils/image-tools.ts
View File

@@ -4,8 +4,6 @@ import { TokenTracker } from './token-tracker';
import { ImageObject } from '../types';
import { cosineSimilarity } from '../tools/cosine';
export type { Canvas, Image } from '@napi-rs/canvas';
import { Storage } from '@google-cloud/storage';
import { randomUUID } from 'crypto';
export const downloadFile = async (uri: string) => {
const resp = await fetch(uri);
@@ -52,13 +50,13 @@ const _loadImage = async (input: string | Buffer) => {
throw new Error('Invalid input');
}
const img = await canvas.loadImage(buff);
const img = await canvas.loadImage(buff).catch((err) => {
throw new Error('Loading image failed: ' + err.message);
});
Reflect.set(img, 'contentType', contentType);
return {
img,
buff,
contentType,
};
}
@@ -107,7 +105,7 @@ export const canvasToBuffer = (canvas: canvas.Canvas, mimeType?: 'image/png' | '
export const processImage = async (url: string, tracker: TokenTracker): Promise<ImageObject | undefined> => {
try {
const { img, buff, contentType } = await loadImage(url);
const { img } = await loadImage(url);
if (!img) {
return;
}
@@ -117,7 +115,6 @@ export const processImage = async (url: string, tracker: TokenTracker): Promise<
return;
}
const newUrl = await saveImageToFirebase(buff, contentType);
const canvas = fitImageToSquareBox(img, 256);
const base64Data = (await canvasToDataUrl(canvas)).split(',')[1];
@@ -127,7 +124,7 @@ export const processImage = async (url: string, tracker: TokenTracker): Promise<
});
return {
url: newUrl ?? url,
url,
embedding: embeddings,
};
@@ -194,40 +191,4 @@ export const dedupImagesWithEmbeddings = (
// Return all new images if there is an error
return newImages;
}
}
export const saveImageToFirebase = async (
buffer: Buffer,
mimeType?: string | null,
): Promise<string | undefined> => {
if (!process.env.GCLOUD_PROJECT) {
console.error('GCLOUD_PROJECT environment variable is not set');
return;
}
const firebaseDefaultBucket = new Storage().bucket(`${process.env.GCLOUD_PROJECT}.appspot.com`);
try {
let extension = 'png';
const finalMimeType = mimeType || 'image/png';
if (!finalMimeType.startsWith('image/')) {
return;
} else {
extension = finalMimeType?.split('/')[1] || 'png';
}
const fileName = `readImages/${randomUUID()}.${extension}`;
const file = firebaseDefaultBucket.file(fileName);
await file.save(buffer, {
contentType: finalMimeType,
public: true,
});
return file.publicUrl();
} catch (error) {
console.error('Error saving image to Firebase Storage:', error);
return;
}
};
}