Create NodeJS package
This commit is contained in:
parent
483c5bf4fd
commit
b6f1af0bb4
3 changed files with 65 additions and 0 deletions
26
package.json
Normal file
26
package.json
Normal file
|
@ -0,0 +1,26 @@
|
||||||
|
{
|
||||||
|
"name": "googlegeminiapitest",
|
||||||
|
"version": "0.0.1",
|
||||||
|
"description": "Some testing with the Google Gemini API using NodeJS",
|
||||||
|
"main": "scripts/main.js",
|
||||||
|
"type": "module",
|
||||||
|
"scripts": {
|
||||||
|
"test": "node --watch scripts/main.js",
|
||||||
|
"run": "node scripts/main.js"
|
||||||
|
},
|
||||||
|
"repository": {
|
||||||
|
"type": "git",
|
||||||
|
"url": "https://gitdab.com/buzzcode2007/GoogleGeminiAPITests_NodeJS.git"
|
||||||
|
},
|
||||||
|
"keywords": [
|
||||||
|
"AI",
|
||||||
|
"Google",
|
||||||
|
"Gemini",
|
||||||
|
"test"
|
||||||
|
],
|
||||||
|
"dependencies": {
|
||||||
|
"@google/genai": "^0.13.0",
|
||||||
|
"dotenv": "^8.2.0"
|
||||||
|
},
|
||||||
|
"author": "buzzcode2007"
|
||||||
|
}
|
17
scripts/main.js
Normal file
17
scripts/main.js
Normal file
|
@ -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();
|
22
scripts/prompt.js
Normal file
22
scripts/prompt.js
Normal file
|
@ -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;
|
Loading…
Add table
Add a link
Reference in a new issue