@quatrain/ai-gemini 1.1.1

package/dist/GeminiAdapter.d.ts

@@ -0,0 +1,11 @@
+import { AbstractAiAdapter } from '@quatrain/ai';
+import { GoogleGenAI } from '@google/genai';
+export declare class GeminiAdapter extends AbstractAiAdapter {
+    protected _ai: GoogleGenAI | null;
+    protected _apiKey: string;
+    constructor(apiKey: string);
+    init(): void;
+    generateText(prompt: string, options?: any): Promise<string>;
+    generateStructured(prompt: string, schema: any, options?: any): Promise<any>;
+}
+//# sourceMappingURL=GeminiAdapter.d.ts.map

package/dist/GeminiAdapter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"GeminiAdapter.d.ts","sourceRoot":"","sources":["../src/GeminiAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAA;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAA;AAE3C,qBAAa,aAAc,SAAQ,iBAAiB;IACjD,SAAS,CAAC,GAAG,EAAE,WAAW,GAAG,IAAI,CAAO;IACxC,SAAS,CAAC,OAAO,EAAE,MAAM,CAAA;gBAEb,MAAM,EAAE,MAAM;IAK1B,IAAI,IAAI,IAAI;IAIN,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC;IAY5D,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;CAmBpF"}

package/dist/GeminiAdapter.js

@@ -0,0 +1,45 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GeminiAdapter = void 0;
+const ai_1 = require("@quatrain/ai");
+const genai_1 = require("@google/genai");
+class GeminiAdapter extends ai_1.AbstractAiAdapter {
+    _ai = null;
+    _apiKey;
+    constructor(apiKey) {
+        super();
+        this._apiKey = apiKey;
+    }
+    init() {
+        this._ai = new genai_1.GoogleGenAI({ apiKey: this._apiKey });
+    }
+    async generateText(prompt, options) {
+        if (!this._ai)
+            this.init();
+        const model = options?.model || 'gemini-2.5-flash';
+        const response = await this._ai.models.generateContent({
+            model,
+            contents: prompt,
+        });
+        return response.text || '';
+    }
+    async generateStructured(prompt, schema, options) {
+        if (!this._ai)
+            this.init();
+        const model = options?.model || 'gemini-2.5-flash';
+        const response = await this._ai.models.generateContent({
+            model,
+            contents: prompt,
+            config: {
+                responseMimeType: 'application/json',
+                responseSchema: schema,
+            }
+        });
+        if (!response.text) {
+            throw new Error('No text returned from Gemini API');
+        }
+        return JSON.parse(response.text);
+    }
+}
+exports.GeminiAdapter = GeminiAdapter;
+//# sourceMappingURL=GeminiAdapter.js.map

package/dist/GeminiAdapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GeminiAdapter.js","sourceRoot":"","sources":["../src/GeminiAdapter.ts"],"names":[],"mappings":";;;AAAA,qCAAgD;AAChD,yCAA2C;AAE3C,MAAa,aAAc,SAAQ,sBAAiB;IACvC,GAAG,GAAuB,IAAI,CAAA;IAC9B,OAAO,CAAQ;IAEzB,YAAY,MAAc;QACvB,KAAK,EAAE,CAAA;QACP,IAAI,CAAC,OAAO,GAAG,MAAM,CAAA;IACxB,CAAC;IAED,IAAI;QACD,IAAI,CAAC,GAAG,GAAG,IAAI,mBAAW,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAA;IACvD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,OAAa;QAC7C,IAAI,CAAC,IAAI,CAAC,GAAG;YAAE,IAAI,CAAC,IAAI,EAAE,CAAA;QAE1B,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,kBAAkB,CAAA;QAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAI,CAAC,MAAM,CAAC,eAAe,CAAC;YACrD,KAAK;YACL,QAAQ,EAAE,MAAM;SAClB,CAAC,CAAA;QAEF,OAAO,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAA;IAC7B,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,MAAc,EAAE,MAAW,EAAE,OAAa;QAChE,IAAI,CAAC,IAAI,CAAC,GAAG;YAAE,IAAI,CAAC,IAAI,EAAE,CAAA;QAE1B,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,IAAI,kBAAkB,CAAA;QAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAI,CAAC,MAAM,CAAC,eAAe,CAAC;YACrD,KAAK;YACL,QAAQ,EAAE,MAAM;YAChB,MAAM,EAAE;gBACL,gBAAgB,EAAE,kBAAkB;gBACpC,cAAc,EAAE,MAAM;aACxB;SACH,CAAC,CAAA;QAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAA;QACtD,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;IACnC,CAAC;CACH;AA5CD,sCA4CC"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+export { GeminiAdapter } from './GeminiAdapter';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAA"}

package/dist/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GeminiAdapter = void 0;
+var GeminiAdapter_1 = require("./GeminiAdapter");
+Object.defineProperty(exports, "GeminiAdapter", { enumerable: true, get: function () { return GeminiAdapter_1.GeminiAdapter; } });
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,iDAA+C;AAAtC,8GAAA,aAAa,OAAA"}

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "@quatrain/ai-gemini",
+  "version": "1.1.1",
+  "license": "AGPL-3.0-only",
+  "description": "AI adapter for Google Gemini",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "bun": "src/index.ts",
+  "files": [
+    "LICENSE.md",
+    "src/",
+    "dist/",
+    "README.md",
+    "NOTICE.md"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Quatrain/Core.git",
+    "directory": "packages/ai-gemini"
+  },
+  "author": "Quatrain Développement SAS <developers@quatrain.com>",
+  "dependencies": {
+    "@google/genai": "latest",
+    "@quatrain/ai": "^1.1.1",
+    "@quatrain/core": "^1.1.52"
+  },
+  "devDependencies": {
+    "@tsconfig/recommended": "^1.0.1",
+    "@types/jest": "^29.5.12",
+    "@types/node": "^22.10.1",
+    "jest": "^29.7.0",
+    "jest-node-exports-resolver": "^1.1.6",
+    "jest-serial-runner": "^1.2.1",
+    "trace-unhandled": "^2.0.1",
+    "ts-jest": "^29.4.6",
+    "ts-node": "^10.9.1",
+    "typescript": "^5.1.5"
+  },
+  "scripts": {
+    "test-ci": "jest --runInBand",
+    "build": "tsc",
+    "wbuild": "tsc --watch",
+    "bump-to": "yarn version"
+  },
+  "typedoc": {
+    "entryPoint": "src/index.ts"
+  }
+}

package/src/GeminiAdapter.ts

@@ -0,0 +1,48 @@
+import { AbstractAiAdapter } from '@quatrain/ai'
+import { GoogleGenAI } from '@google/genai'
+
+export class GeminiAdapter extends AbstractAiAdapter {
+   protected _ai: GoogleGenAI | null = null
+   protected _apiKey: string
+
+   constructor(apiKey: string) {
+      super()
+      this._apiKey = apiKey
+   }
+
+   init(): void {
+      this._ai = new GoogleGenAI({ apiKey: this._apiKey })
+   }
+
+   async generateText(prompt: string, options?: any): Promise<string> {
+      if (!this._ai) this.init()
+      
+      const model = options?.model || 'gemini-2.5-flash'
+      const response = await this._ai!.models.generateContent({
+         model,
+         contents: prompt,
+      })
+
+      return response.text || ''
+   }
+
+   async generateStructured(prompt: string, schema: any, options?: any): Promise<any> {
+      if (!this._ai) this.init()
+
+      const model = options?.model || 'gemini-2.5-flash'
+      const response = await this._ai!.models.generateContent({
+         model,
+         contents: prompt,
+         config: {
+            responseMimeType: 'application/json',
+            responseSchema: schema,
+         }
+      })
+
+      if (!response.text) {
+         throw new Error('No text returned from Gemini API')
+      }
+
+      return JSON.parse(response.text)
+   }
+}

package/src/__test__/GeminiAdapter.test.ts

@@ -0,0 +1,22 @@
+import { GeminiAdapter } from '../GeminiAdapter'
+
+jest.mock('@google/genai', () => ({
+   GoogleGenAI: jest.fn().mockImplementation(() => ({
+      models: {
+         generateContent: jest.fn().mockResolvedValue({ text: 'mock AI response' })
+      }
+   }))
+}))
+
+describe('GeminiAdapter', () => {
+   let adapter: GeminiAdapter
+
+   beforeEach(() => {
+      adapter = new GeminiAdapter('fake-api-key')
+   })
+
+   it('should instantiate and generate text', async () => {
+      const response = await adapter.generateText('Hello')
+      expect(response).toBe('mock AI response')
+   })
+})

package/src/index.ts

@@ -0,0 +1 @@
+export { GeminiAdapter } from './GeminiAdapter'