@ragpipe/plugin-gemini 0.0.1

package/dist/index.cjs

@@ -0,0 +1,162 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  geminiEmbedding: () => geminiEmbedding,
+  geminiGeneration: () => geminiGeneration
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/embedding.ts
+function geminiEmbedding(options) {
+  const model = options.model ?? "gemini-embedding-001";
+  return {
+    name: "gemini",
+    dimensions: 3072,
+    rateLimit: { delayMs: 800 },
+    async embed(text) {
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:embedContent?key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ content: { parts: [{ text }] } })
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini embedding error: ${res.status} ${await res.text()}`
+        );
+      }
+      const data = await res.json();
+      return data.embedding.values;
+    },
+    async embedMany(texts) {
+      const requests = texts.map((text) => ({
+        model: `models/${model}`,
+        content: { parts: [{ text }] }
+      }));
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:batchEmbedContents?key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ requests })
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini batch embedding error: ${res.status} ${await res.text()}`
+        );
+      }
+      const data = await res.json();
+      return data.embeddings.map((e) => e.values);
+    }
+  };
+}
+
+// src/generation.ts
+function geminiGeneration(options) {
+  const model = options.model ?? "gemini-2.5-flash";
+  function buildBody(question, context, opts) {
+    const systemPrompt = opts?.systemPrompt ?? options.systemPrompt ?? "Answer based on the provided context.";
+    let userPrompt = `Context:
+${context}
+
+Question: ${question}`;
+    if (opts?.history) {
+      userPrompt = `Conversation history:
+${opts.history}
+
+${userPrompt}`;
+    }
+    return {
+      contents: [{ parts: [{ text: userPrompt }] }],
+      systemInstruction: { parts: [{ text: systemPrompt }] }
+    };
+  }
+  return {
+    name: "gemini",
+    async generate(question, context, opts) {
+      const body = buildBody(question, context, opts);
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:generateContent?key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify(body)
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini generation error: ${res.status} ${await res.text()}`
+        );
+      }
+      const data = await res.json();
+      return data.candidates[0].content.parts[0].text;
+    },
+    async *generateStream(question, context, opts) {
+      const body = buildBody(question, context, opts);
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:streamGenerateContent?alt=sse&key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify(body)
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini generation stream error: ${res.status} ${await res.text()}`
+        );
+      }
+      const reader = res.body?.getReader();
+      if (!reader) throw new Error("No response body");
+      const decoder = new TextDecoder();
+      let buffer = "";
+      try {
+        for (; ; ) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          buffer += decoder.decode(value, { stream: true });
+          const lines = buffer.split("\n");
+          buffer = lines.pop() ?? "";
+          for (const line of lines) {
+            if (!line.startsWith("data: ")) continue;
+            try {
+              const data = JSON.parse(line.slice(6));
+              const text = data.candidates?.[0]?.content?.parts?.[0]?.text;
+              if (text) yield text;
+            } catch {
+            }
+          }
+        }
+      } finally {
+        reader.releaseLock();
+      }
+    }
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  geminiEmbedding,
+  geminiGeneration
+});

package/dist/index.d.cts

@@ -0,0 +1,16 @@
+import { EmbeddingPlugin, GenerationPlugin } from 'ragpipe';
+
+interface GeminiEmbeddingOptions {
+    apiKey: string;
+    model?: string;
+}
+declare function geminiEmbedding(options: GeminiEmbeddingOptions): EmbeddingPlugin;
+
+interface GeminiGenerationOptions {
+    apiKey: string;
+    model?: string;
+    systemPrompt?: string;
+}
+declare function geminiGeneration(options: GeminiGenerationOptions): GenerationPlugin;
+
+export { type GeminiEmbeddingOptions, type GeminiGenerationOptions, geminiEmbedding, geminiGeneration };

package/dist/index.d.ts

@@ -0,0 +1,16 @@
+import { EmbeddingPlugin, GenerationPlugin } from 'ragpipe';
+
+interface GeminiEmbeddingOptions {
+    apiKey: string;
+    model?: string;
+}
+declare function geminiEmbedding(options: GeminiEmbeddingOptions): EmbeddingPlugin;
+
+interface GeminiGenerationOptions {
+    apiKey: string;
+    model?: string;
+    systemPrompt?: string;
+}
+declare function geminiGeneration(options: GeminiGenerationOptions): GenerationPlugin;
+
+export { type GeminiEmbeddingOptions, type GeminiGenerationOptions, geminiEmbedding, geminiGeneration };

package/dist/index.js

@@ -0,0 +1,134 @@
+// src/embedding.ts
+function geminiEmbedding(options) {
+  const model = options.model ?? "gemini-embedding-001";
+  return {
+    name: "gemini",
+    dimensions: 3072,
+    rateLimit: { delayMs: 800 },
+    async embed(text) {
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:embedContent?key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ content: { parts: [{ text }] } })
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini embedding error: ${res.status} ${await res.text()}`
+        );
+      }
+      const data = await res.json();
+      return data.embedding.values;
+    },
+    async embedMany(texts) {
+      const requests = texts.map((text) => ({
+        model: `models/${model}`,
+        content: { parts: [{ text }] }
+      }));
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:batchEmbedContents?key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ requests })
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini batch embedding error: ${res.status} ${await res.text()}`
+        );
+      }
+      const data = await res.json();
+      return data.embeddings.map((e) => e.values);
+    }
+  };
+}
+
+// src/generation.ts
+function geminiGeneration(options) {
+  const model = options.model ?? "gemini-2.5-flash";
+  function buildBody(question, context, opts) {
+    const systemPrompt = opts?.systemPrompt ?? options.systemPrompt ?? "Answer based on the provided context.";
+    let userPrompt = `Context:
+${context}
+
+Question: ${question}`;
+    if (opts?.history) {
+      userPrompt = `Conversation history:
+${opts.history}
+
+${userPrompt}`;
+    }
+    return {
+      contents: [{ parts: [{ text: userPrompt }] }],
+      systemInstruction: { parts: [{ text: systemPrompt }] }
+    };
+  }
+  return {
+    name: "gemini",
+    async generate(question, context, opts) {
+      const body = buildBody(question, context, opts);
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:generateContent?key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify(body)
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini generation error: ${res.status} ${await res.text()}`
+        );
+      }
+      const data = await res.json();
+      return data.candidates[0].content.parts[0].text;
+    },
+    async *generateStream(question, context, opts) {
+      const body = buildBody(question, context, opts);
+      const res = await fetch(
+        `https://generativelanguage.googleapis.com/v1beta/models/${model}:streamGenerateContent?alt=sse&key=${options.apiKey}`,
+        {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify(body)
+        }
+      );
+      if (!res.ok) {
+        throw new Error(
+          `Gemini generation stream error: ${res.status} ${await res.text()}`
+        );
+      }
+      const reader = res.body?.getReader();
+      if (!reader) throw new Error("No response body");
+      const decoder = new TextDecoder();
+      let buffer = "";
+      try {
+        for (; ; ) {
+          const { done, value } = await reader.read();
+          if (done) break;
+          buffer += decoder.decode(value, { stream: true });
+          const lines = buffer.split("\n");
+          buffer = lines.pop() ?? "";
+          for (const line of lines) {
+            if (!line.startsWith("data: ")) continue;
+            try {
+              const data = JSON.parse(line.slice(6));
+              const text = data.candidates?.[0]?.content?.parts?.[0]?.text;
+              if (text) yield text;
+            } catch {
+            }
+          }
+        }
+      } finally {
+        reader.releaseLock();
+      }
+    }
+  };
+}
+export {
+  geminiEmbedding,
+  geminiGeneration
+};

package/package.json

@@ -0,0 +1,60 @@
+{
+	"name": "@ragpipe/plugin-gemini",
+	"version": "0.0.1",
+	"description": "Gemini embedding and generation plugin for ragpipe",
+	"type": "module",
+	"license": "MIT",
+	"author": {
+		"name": "yungblud",
+		"url": "https://github.com/yungblud"
+	},
+	"repository": {
+		"type": "git",
+		"url": "https://github.com/yungblud/ragpipe",
+		"directory": "packages/plugin-gemini"
+	},
+	"bugs": {
+		"url": "https://github.com/yungblud/ragpipe/issues"
+	},
+	"homepage": "https://github.com/yungblud/ragpipe#readme",
+	"publishConfig": {
+		"access": "public"
+	},
+	"exports": {
+		".": {
+			"types": "./dist/index.d.ts",
+			"import": "./dist/index.js",
+			"require": "./dist/index.cjs"
+		}
+	},
+	"main": "./dist/index.cjs",
+	"module": "./dist/index.js",
+	"types": "./dist/index.d.ts",
+	"files": ["dist"],
+	"scripts": {
+		"build": "tsup",
+		"dev": "tsup --watch",
+		"typecheck": "tsc --noEmit",
+		"test": "vitest run",
+		"test:watch": "vitest",
+		"test:coverage": "vitest run --coverage"
+	},
+	"keywords": [
+		"ragpipe",
+		"rag",
+		"gemini",
+		"google-ai",
+		"embedding",
+		"generation",
+		"plugin"
+	],
+	"peerDependencies": {
+		"ragpipe": ">=0.0.1"
+	},
+	"devDependencies": {
+		"ragpipe": "workspace:*",
+		"tsup": "^8.4.0",
+		"typescript": "^5.8.3",
+		"vitest": "^3.1.1"
+	}
+}