Create NodeJS package

scripts/main.js

@@ -0,0 +1,17 @@
+import Tests from './prompt.js';
+
+class main {
+    constructor() {
+        this.platform = new Tests();
+        this.first();
+    }
+
+    first () {
+        let result = this.platform.generate("gemini-1.5-flash", "What is a TPU?");
+        result.then((log) => {
+            console.log(log.text)
+        })
+    }
+}
+
+let runtime = new main();

scripts/prompt.js

@@ -0,0 +1,22 @@
+import { GoogleGenAI } from "@google/genai";
+
+class Tests {
+    #key;
+    #toolkit;
+
+    constructor () {
+        this.#key = process.env.GoogleGenAIKey
+        this.#toolkit = new GoogleGenAI({ apiKey: this.#key });
+    }
+
+    async generate (model, contents) {
+        const response = await this.#toolkit.models.generateContent({
+          "model": model,
+          "contents": contents
+        });
+
+        return response;
+    }
+}
+
+export default Tests;