47 lines
No EOL
1.4 KiB
JavaScript
47 lines
No EOL
1.4 KiB
JavaScript
import { GoogleGenAI } from "@google/genai"
|
|
import fs from "fs"
|
|
|
|
const ai = new GoogleGenAI({ apiKey: import.meta.env.GEMINI_API_KEY })
|
|
|
|
async function uploadLocalPDFs() {
|
|
var pdfList = fs.readdirSync("public/pdfs")
|
|
|
|
// Upload each file in /public
|
|
pdfList.forEach(async (path) => {
|
|
console.log("file names: " + path)
|
|
console.log("file names: " + path.slice(0, path.length - 4))
|
|
|
|
console.log("UPLOADING")
|
|
const file = await ai.files.upload({
|
|
file: "public/pdfs/" + path,
|
|
config: {
|
|
displayName: path.slice(0, path.length - 4)
|
|
}
|
|
})
|
|
|
|
console.log("FETCHING: public/pdfs/" + path)
|
|
|
|
// Wait for the file to be processed
|
|
let getFile = await ai.files.get({
|
|
name: file.name
|
|
})
|
|
|
|
while (getFile.state === "PROCESSING") {
|
|
let getFile = await ai.files.get({
|
|
name: file.name
|
|
})
|
|
console.log(`Current file status: ${getFile.state}`)
|
|
console.log("File is currently processing, retrying in 5 seconds")
|
|
|
|
await new Promise((resolve) => {
|
|
setTimeout(resolve, 5000) // Checks every 5 seconds
|
|
})
|
|
|
|
// Error handling
|
|
if (getFile.state === "FAILED") {
|
|
throw new Error("File has failed to process!")
|
|
}
|
|
return file
|
|
}
|
|
})
|
|
} |