@langchain/google 0.1.11 → 0.2.0

package/dist/utils/stream_events.js

@@ -0,0 +1,173 @@
+import { finalizeContentBlock } from "@langchain/core/language_models/compat";
+//#region src/utils/stream_events.ts
+/**
+* Converts Gemini stream responses into LangChain ChatModelStreamEvents.
+*
+* @module
+*/
+async function* convertGoogleGeminiStream(source, options = {}) {
+	const shouldStreamUsage = options.streamUsage ?? true;
+	const blockAccumulators = /* @__PURE__ */ new Map();
+	const blockKeyToIndex = /* @__PURE__ */ new Map();
+	let nextBlockIndex = 0;
+	let messageStarted = false;
+	let usageSnapshot;
+	let finishReason = "stop";
+	const getOrCreateBlockIndex = (key, initial) => {
+		const existing = blockKeyToIndex.get(key);
+		if (existing !== void 0) return {
+			index: existing,
+			isNew: false
+		};
+		const index = nextBlockIndex++;
+		blockKeyToIndex.set(key, index);
+		blockAccumulators.set(index, { ...initial });
+		return {
+			index,
+			isNew: true
+		};
+	};
+	for await (const response of source) {
+		if (!messageStarted) {
+			messageStarted = true;
+			yield { event: "message-start" };
+		}
+		const usageMetadata = response.usageMetadata ?? response.usage_metadata;
+		if (shouldStreamUsage && usageMetadata) {
+			const input = usageMetadata.promptTokenCount ?? 0;
+			const output = usageMetadata.candidatesTokenCount ?? 0;
+			usageSnapshot = {
+				input_tokens: input,
+				output_tokens: output,
+				total_tokens: usageMetadata.totalTokenCount ?? input + output
+			};
+			yield {
+				event: "usage",
+				usage: usageSnapshot
+			};
+		}
+		const candidate = response.candidates?.[0];
+		if (candidate?.finishReason) finishReason = mapGeminiFinishReason(candidate.finishReason);
+		const parts = candidate?.content?.parts;
+		if (!parts) continue;
+		let toolIdx = 0;
+		for (const part of parts) if (part.text) if (part.thought) {
+			const { index, isNew } = getOrCreateBlockIndex("reasoning", {
+				type: "reasoning",
+				reasoning: ""
+			});
+			if (isNew) yield {
+				event: "content-block-start",
+				index,
+				content: {
+					type: "reasoning",
+					reasoning: ""
+				}
+			};
+			const acc = blockAccumulators.get(index);
+			acc.reasoning = (acc.reasoning ?? "") + part.text;
+			yield {
+				event: "content-block-delta",
+				index,
+				delta: {
+					type: "reasoning-delta",
+					reasoning: part.text
+				}
+			};
+		} else {
+			const { index, isNew } = getOrCreateBlockIndex("text", {
+				type: "text",
+				text: ""
+			});
+			if (isNew) yield {
+				event: "content-block-start",
+				index,
+				content: {
+					type: "text",
+					text: ""
+				}
+			};
+			const acc = blockAccumulators.get(index);
+			acc.text = (acc.text ?? "") + part.text;
+			yield {
+				event: "content-block-delta",
+				index,
+				delta: {
+					type: "text-delta",
+					text: part.text
+				}
+			};
+		}
+		else if (part.functionCall) {
+			const key = `tool:${toolIdx}`;
+			const args = JSON.stringify(part.functionCall.args ?? {});
+			const { index, isNew } = getOrCreateBlockIndex(key, {
+				type: "tool_call_chunk",
+				name: part.functionCall.name,
+				args: "",
+				index: toolIdx
+			});
+			if (isNew) yield {
+				event: "content-block-start",
+				index,
+				content: {
+					type: "tool_call_chunk",
+					name: part.functionCall.name,
+					args: "",
+					index: toolIdx
+				}
+			};
+			const acc = blockAccumulators.get(index);
+			acc.args = args;
+			yield {
+				event: "content-block-delta",
+				index,
+				delta: {
+					type: "block-delta",
+					fields: {
+						type: "tool_call_chunk",
+						name: acc.name,
+						args: acc.args
+					}
+				}
+			};
+			toolIdx += 1;
+		}
+	}
+	for (const [index, acc] of blockAccumulators) yield {
+		event: "content-block-finish",
+		index,
+		content: finalizeContentBlock(acc)
+	};
+	yield {
+		event: "message-finish",
+		reason: finishReason,
+		...usageSnapshot ? { usage: usageSnapshot } : {},
+		responseMetadata: { model_provider: "google" }
+	};
+}
+function mapGeminiFinishReason(reason) {
+	switch (reason.toLowerCase()) {
+		case "max_tokens":
+		case "max-token":
+		case "max_token": return "length";
+		case "safety":
+		case "recitation":
+		case "language":
+		case "blocklist":
+		case "prohibited_content":
+		case "prohibited-content":
+		case "spii":
+		case "image_safety":
+		case "image-safety":
+		case "image_prohibited_content":
+		case "image-prohibited-content":
+		case "image_recitation":
+		case "image-recitation": return "content_filter";
+		default: return "stop";
+	}
+}
+//#endregion
+export { convertGoogleGeminiStream };
+
+//# sourceMappingURL=stream_events.js.map

package/dist/utils/stream_events.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream_events.js","names":[],"sources":["../../src/utils/stream_events.ts"],"sourcesContent":["/**\n * Converts Gemini stream responses into LangChain ChatModelStreamEvents.\n *\n * @module\n */\n\nimport { finalizeContentBlock } from \"@langchain/core/language_models/compat\";\nimport type {\n  ChatModelStreamEvent,\n  FinishReason,\n} from \"@langchain/core/language_models/event\";\nimport type { ContentBlock, UsageMetadata } from \"@langchain/core/messages\";\nimport type { Gemini } from \"../chat_models/api-types.js\";\n\nexport type GeminiStreamResponse = Gemini.GenerateContentResponse;\n\nexport interface ConvertGoogleGeminiStreamOptions {\n  streamUsage?: boolean;\n}\n\ntype BlockKey = \"text\" | \"reasoning\" | `tool:${number}`;\n\nexport async function* convertGoogleGeminiStream(\n  source: AsyncIterable<GeminiStreamResponse>,\n  options: ConvertGoogleGeminiStreamOptions = {}\n): AsyncGenerator<ChatModelStreamEvent> {\n  const shouldStreamUsage = options.streamUsage ?? true;\n  const blockAccumulators = new Map<\n    number,\n    // oxlint-disable-next-line @typescript-eslint/no-explicit-any\n    Record<string, any>\n  >();\n  const blockKeyToIndex = new Map<BlockKey, number>();\n  let nextBlockIndex = 0;\n  let messageStarted = false;\n  let usageSnapshot: UsageMetadata | undefined;\n  let finishReason: FinishReason = \"stop\";\n\n  const getOrCreateBlockIndex = (\n    key: BlockKey,\n    initial: Record<string, unknown>\n  ): { index: number; isNew: boolean } => {\n    const existing = blockKeyToIndex.get(key);\n    if (existing !== undefined) {\n      return { index: existing, isNew: false };\n    }\n    const index = nextBlockIndex++;\n    blockKeyToIndex.set(key, index);\n    blockAccumulators.set(index, { ...initial });\n    return { index, isNew: true };\n  };\n\n  for await (const response of source) {\n    if (!messageStarted) {\n      messageStarted = true;\n      yield { event: \"message-start\" as const };\n    }\n\n    const usageMetadata = response.usageMetadata ?? response.usage_metadata;\n    if (shouldStreamUsage && usageMetadata) {\n      const input = usageMetadata.promptTokenCount ?? 0;\n      const output = usageMetadata.candidatesTokenCount ?? 0;\n      usageSnapshot = {\n        input_tokens: input,\n        output_tokens: output,\n        total_tokens: usageMetadata.totalTokenCount ?? input + output,\n      };\n      yield { event: \"usage\" as const, usage: usageSnapshot };\n    }\n\n    const candidate = response.candidates?.[0];\n    if (candidate?.finishReason) {\n      finishReason = mapGeminiFinishReason(candidate.finishReason);\n    }\n\n    const parts = candidate?.content?.parts;\n    if (!parts) continue;\n\n    let toolIdx = 0;\n    for (const part of parts) {\n      if (part.text) {\n        if (part.thought) {\n          const key: BlockKey = \"reasoning\";\n          const { index, isNew } = getOrCreateBlockIndex(key, {\n            type: \"reasoning\",\n            reasoning: \"\",\n          });\n          if (isNew) {\n            yield {\n              event: \"content-block-start\" as const,\n              index,\n              content: { type: \"reasoning\", reasoning: \"\" } as ContentBlock,\n            };\n          }\n          const acc = blockAccumulators.get(index)!;\n          acc.reasoning = (acc.reasoning ?? \"\") + part.text;\n          yield {\n            event: \"content-block-delta\" as const,\n            index,\n            delta: { type: \"reasoning-delta\" as const, reasoning: part.text },\n          };\n        } else {\n          const key: BlockKey = \"text\";\n          const { index, isNew } = getOrCreateBlockIndex(key, {\n            type: \"text\",\n            text: \"\",\n          });\n          if (isNew) {\n            yield {\n              event: \"content-block-start\" as const,\n              index,\n              content: { type: \"text\", text: \"\" } as ContentBlock,\n            };\n          }\n          const acc = blockAccumulators.get(index)!;\n          acc.text = (acc.text ?? \"\") + part.text;\n          yield {\n            event: \"content-block-delta\" as const,\n            index,\n            delta: { type: \"text-delta\" as const, text: part.text },\n          };\n        }\n      } else if (part.functionCall) {\n        const key: BlockKey = `tool:${toolIdx}`;\n        const args = JSON.stringify(part.functionCall.args ?? {});\n        const { index, isNew } = getOrCreateBlockIndex(key, {\n          type: \"tool_call_chunk\",\n          name: part.functionCall.name,\n          args: \"\",\n          index: toolIdx,\n        });\n        if (isNew) {\n          yield {\n            event: \"content-block-start\" as const,\n            index,\n            content: {\n              type: \"tool_call_chunk\",\n              name: part.functionCall.name,\n              args: \"\",\n              index: toolIdx,\n            } as ContentBlock,\n          };\n        }\n        const acc = blockAccumulators.get(index)!;\n        acc.args = args;\n        yield {\n          event: \"content-block-delta\" as const,\n          index,\n          delta: {\n            type: \"block-delta\" as const,\n            fields: {\n              type: \"tool_call_chunk\",\n              name: acc.name,\n              args: acc.args,\n            },\n          },\n        };\n        toolIdx += 1;\n      }\n    }\n  }\n\n  for (const [index, acc] of blockAccumulators) {\n    yield {\n      event: \"content-block-finish\" as const,\n      index,\n      content: finalizeContentBlock(acc as ContentBlock),\n    };\n  }\n\n  yield {\n    event: \"message-finish\" as const,\n    reason: finishReason,\n    ...(usageSnapshot ? { usage: usageSnapshot } : {}),\n    responseMetadata: { model_provider: \"google\" },\n  };\n}\n\nfunction mapGeminiFinishReason(reason: string): FinishReason {\n  switch (reason.toLowerCase()) {\n    case \"max_tokens\":\n    case \"max-token\":\n    case \"max_token\":\n      return \"length\";\n    case \"safety\":\n    case \"recitation\":\n    case \"language\":\n    case \"blocklist\":\n    case \"prohibited_content\":\n    case \"prohibited-content\":\n    case \"spii\":\n    case \"image_safety\":\n    case \"image-safety\":\n    case \"image_prohibited_content\":\n    case \"image-prohibited-content\":\n    case \"image_recitation\":\n    case \"image-recitation\":\n      return \"content_filter\";\n    default:\n      return \"stop\";\n  }\n}\n"],"mappings":";;;;;;;AAsBA,gBAAuB,0BACrB,QACA,UAA4C,EAAE,EACR;CACtC,MAAM,oBAAoB,QAAQ,eAAe;CACjD,MAAM,oCAAoB,IAAI,KAI3B;CACH,MAAM,kCAAkB,IAAI,KAAuB;CACnD,IAAI,iBAAiB;CACrB,IAAI,iBAAiB;CACrB,IAAI;CACJ,IAAI,eAA6B;CAEjC,MAAM,yBACJ,KACA,YACsC;EACtC,MAAM,WAAW,gBAAgB,IAAI,IAAI;AACzC,MAAI,aAAa,KAAA,EACf,QAAO;GAAE,OAAO;GAAU,OAAO;GAAO;EAE1C,MAAM,QAAQ;AACd,kBAAgB,IAAI,KAAK,MAAM;AAC/B,oBAAkB,IAAI,OAAO,EAAE,GAAG,SAAS,CAAC;AAC5C,SAAO;GAAE;GAAO,OAAO;GAAM;;AAG/B,YAAW,MAAM,YAAY,QAAQ;AACnC,MAAI,CAAC,gBAAgB;AACnB,oBAAiB;AACjB,SAAM,EAAE,OAAO,iBAA0B;;EAG3C,MAAM,gBAAgB,SAAS,iBAAiB,SAAS;AACzD,MAAI,qBAAqB,eAAe;GACtC,MAAM,QAAQ,cAAc,oBAAoB;GAChD,MAAM,SAAS,cAAc,wBAAwB;AACrD,mBAAgB;IACd,cAAc;IACd,eAAe;IACf,cAAc,cAAc,mBAAmB,QAAQ;IACxD;AACD,SAAM;IAAE,OAAO;IAAkB,OAAO;IAAe;;EAGzD,MAAM,YAAY,SAAS,aAAa;AACxC,MAAI,WAAW,aACb,gBAAe,sBAAsB,UAAU,aAAa;EAG9D,MAAM,QAAQ,WAAW,SAAS;AAClC,MAAI,CAAC,MAAO;EAEZ,IAAI,UAAU;AACd,OAAK,MAAM,QAAQ,MACjB,KAAI,KAAK,KACP,KAAI,KAAK,SAAS;GAEhB,MAAM,EAAE,OAAO,UAAU,sBAAsB,aAAK;IAClD,MAAM;IACN,WAAW;IACZ,CAAC;AACF,OAAI,MACF,OAAM;IACJ,OAAO;IACP;IACA,SAAS;KAAE,MAAM;KAAa,WAAW;KAAI;IAC9C;GAEH,MAAM,MAAM,kBAAkB,IAAI,MAAM;AACxC,OAAI,aAAa,IAAI,aAAa,MAAM,KAAK;AAC7C,SAAM;IACJ,OAAO;IACP;IACA,OAAO;KAAE,MAAM;KAA4B,WAAW,KAAK;KAAM;IAClE;SACI;GAEL,MAAM,EAAE,OAAO,UAAU,sBAAsB,QAAK;IAClD,MAAM;IACN,MAAM;IACP,CAAC;AACF,OAAI,MACF,OAAM;IACJ,OAAO;IACP;IACA,SAAS;KAAE,MAAM;KAAQ,MAAM;KAAI;IACpC;GAEH,MAAM,MAAM,kBAAkB,IAAI,MAAM;AACxC,OAAI,QAAQ,IAAI,QAAQ,MAAM,KAAK;AACnC,SAAM;IACJ,OAAO;IACP;IACA,OAAO;KAAE,MAAM;KAAuB,MAAM,KAAK;KAAM;IACxD;;WAEM,KAAK,cAAc;GAC5B,MAAM,MAAgB,QAAQ;GAC9B,MAAM,OAAO,KAAK,UAAU,KAAK,aAAa,QAAQ,EAAE,CAAC;GACzD,MAAM,EAAE,OAAO,UAAU,sBAAsB,KAAK;IAClD,MAAM;IACN,MAAM,KAAK,aAAa;IACxB,MAAM;IACN,OAAO;IACR,CAAC;AACF,OAAI,MACF,OAAM;IACJ,OAAO;IACP;IACA,SAAS;KACP,MAAM;KACN,MAAM,KAAK,aAAa;KACxB,MAAM;KACN,OAAO;KACR;IACF;GAEH,MAAM,MAAM,kBAAkB,IAAI,MAAM;AACxC,OAAI,OAAO;AACX,SAAM;IACJ,OAAO;IACP;IACA,OAAO;KACL,MAAM;KACN,QAAQ;MACN,MAAM;MACN,MAAM,IAAI;MACV,MAAM,IAAI;MACX;KACF;IACF;AACD,cAAW;;;AAKjB,MAAK,MAAM,CAAC,OAAO,QAAQ,kBACzB,OAAM;EACJ,OAAO;EACP;EACA,SAAS,qBAAqB,IAAoB;EACnD;AAGH,OAAM;EACJ,OAAO;EACP,QAAQ;EACR,GAAI,gBAAgB,EAAE,OAAO,eAAe,GAAG,EAAE;EACjD,kBAAkB,EAAE,gBAAgB,UAAU;EAC/C;;AAGH,SAAS,sBAAsB,QAA8B;AAC3D,SAAQ,OAAO,aAAa,EAA5B;EACE,KAAK;EACL,KAAK;EACL,KAAK,YACH,QAAO;EACT,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK;EACL,KAAK,mBACH,QAAO;EACT,QACE,QAAO"}

package/node.cjs

@@ -0,0 +1 @@
+module.exports = require("./dist/node.cjs");

package/node.d.cts

@@ -0,0 +1 @@
+export * from "./dist/node.js";

package/node.d.ts

@@ -0,0 +1 @@
+export * from "./dist/node.js";

package/node.js

@@ -0,0 +1 @@
+export * from "./dist/node.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/google",
-  "version": "0.1.11",
+  "version": "0.2.0",
   "description": "Google integrations for LangChain.js",
   "author": "LangChain",
   "license": "MIT",
@@ -14,9 +14,9 @@
     "directory": "libs/providers/langchain-google"
   },
   "dependencies": {
-    "eventsource-parser": "^3.0.6",
+    "eventsource-parser": "^3.1.0",
     "google-auth-library": "^10.6.2",
-    "jose": "^6.1.3"
+    "jose": "^6.2.3"
   },
   "peerDependencies": {
     "@langchain/core": "^1.0.0"
@@ -25,11 +25,11 @@
     "@tsconfig/recommended": "^1.0.10",
     "dpdm": "^3.14.0",
     "sparktype": "^0.1.3",
-    "typescript": "~5.8.3",
-    "vitest": "^4.1.2",
+    "typescript": "~6.0.3",
+    "vitest": "^4.1.8",
     "zod": "^3.25.76 || ^4",
     "zod-to-json-schema": "^3.25.2",
-    "@langchain/core": "^1.1.42",
+    "@langchain/core": "^1.2.0",
     "@langchain/standard-tests": "0.0.23"
   },
   "publishConfig": {
@@ -64,7 +64,11 @@
     "dist/",
     "CHANGELOG.md",
     "README.md",
-    "LICENSE"
+    "LICENSE",
+    "node.d.cts",
+    "node.cjs",
+    "node.d.ts",
+    "node.js"
   ],
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -82,6 +86,7 @@
     "typegen": "pnpm run typegen:sparktype && pnpm run typegen:profiles",
     "typegen:sparktype": "sparktype generate",
     "typegen:profiles": "pnpm --filter @langchain/model-profiles make --config profiles.toml",
-    "typegen:check": "sparktype check"
+    "typegen:check": "sparktype check",
+    "typegen:fetch": "curl -o src/openapi/gemini-openapi.json 'https://generativelanguage.googleapis.com/$discovery/OPENAPI3_0?version=v1beta'"
   }
 }