koishi-plugin-chatluna-google-gemini-adapter 1.3.0-alpha.5 → 1.3.0-alpha.6

package/lib/index.cjs

@@ -55,7 +55,7 @@ var import_types = require("koishi-plugin-chatluna/llm-core/platform/types");
 var import_error2 = require("koishi-plugin-chatluna/utils/error");
 
 // src/requester.ts
-var import_messages2 = require("@langchain/core/messages");
+var import_messages = require("@langchain/core/messages");
 var import_outputs = require("@langchain/core/outputs");
 var import_api = require("koishi-plugin-chatluna/llm-core/platform/api");
 var import_error = require("koishi-plugin-chatluna/utils/error");
@@ -63,7 +63,6 @@ var import_sse = require("koishi-plugin-chatluna/utils/sse");
 var import_stream = require("koishi-plugin-chatluna/utils/stream");
 
 // src/utils.ts
-var import_messages = require("@langchain/core/messages");
 var import_zod_to_json_schema = require("zod-to-json-schema");
 var import_v1_shared_adapter = require("@chatluna/v1-shared-adapter");
 var import_string = require("koishi-plugin-chatluna/utils/string");
@@ -130,7 +129,8 @@ function parseJsonArgs(args) {
 }
 __name(parseJsonArgs, "parseJsonArgs");
 function processFunctionMessage(message) {
-  if (message instanceof import_messages.AIMessageChunk || message instanceof import_messages.AIMessage) {
+  if (message["tool_calls"]) {
+    message = message;
     const toolCalls = message.tool_calls;
     return {
       role: "model",
@@ -145,13 +145,14 @@ function processFunctionMessage(message) {
       })
     };
   }
+  const finalMessage = message;
   return {
     role: "user",
     parts: [
       {
         functionResponse: {
           name: message.name,
-          id: message.tool_call_id,
+          id: finalMessage.tool_call_id,
           response: parseJsonArgs(message.content)
         }
       }
@@ -808,7 +809,7 @@ var GeminiRequester = class extends import_api.ModelRequester {
     if (groundingContent.length > 0) {
       logger.debug(`grounding content: ${groundingContent}`);
       if (this._pluginConfig.groundingContentDisplay) {
-        const groundingMessage = new import_messages2.AIMessageChunk(
+        const groundingMessage = new import_messages.AIMessageChunk(
           `
 ${groundingContent}`
         );
@@ -821,7 +822,7 @@ ${groundingContent}`
     }
   }
   _createMessageChunk(content, functionCall, imagePart) {
-    const messageChunk = new import_messages2.AIMessageChunk({
+    const messageChunk = new import_messages.AIMessageChunk({
       content: content ?? "",
       tool_call_chunks: [functionCall].filter(Boolean)
     });

package/lib/index.mjs

@@ -40,7 +40,7 @@ import {
 
 // src/requester.ts
 import {
-  AIMessageChunk as AIMessageChunk2
+  AIMessageChunk
 } from "@langchain/core/messages";
 import { ChatGenerationChunk } from "@langchain/core/outputs";
 import {
@@ -54,10 +54,6 @@ import { checkResponse, sseIterable } from "koishi-plugin-chatluna/utils/sse";
 import { readableStreamToAsyncIterable } from "koishi-plugin-chatluna/utils/stream";
 
 // src/utils.ts
-import {
-  AIMessage,
-  AIMessageChunk
-} from "@langchain/core/messages";
 import { zodToJsonSchema } from "zod-to-json-schema";
 import {
   fetchImageUrl,
@@ -130,7 +126,8 @@ function parseJsonArgs(args) {
 }
 __name(parseJsonArgs, "parseJsonArgs");
 function processFunctionMessage(message) {
-  if (message instanceof AIMessageChunk || message instanceof AIMessage) {
+  if (message["tool_calls"]) {
+    message = message;
     const toolCalls = message.tool_calls;
     return {
       role: "model",
@@ -145,13 +142,14 @@ function processFunctionMessage(message) {
       })
     };
   }
+  const finalMessage = message;
   return {
     role: "user",
     parts: [
       {
         functionResponse: {
           name: message.name,
-          id: message.tool_call_id,
+          id: finalMessage.tool_call_id,
           response: parseJsonArgs(message.content)
         }
       }
@@ -808,7 +806,7 @@ var GeminiRequester = class extends ModelRequester {
     if (groundingContent.length > 0) {
       logger.debug(`grounding content: ${groundingContent}`);
       if (this._pluginConfig.groundingContentDisplay) {
-        const groundingMessage = new AIMessageChunk2(
+        const groundingMessage = new AIMessageChunk(
           `
 ${groundingContent}`
         );
@@ -821,7 +819,7 @@ ${groundingContent}`
     }
   }
   _createMessageChunk(content, functionCall, imagePart) {
-    const messageChunk = new AIMessageChunk2({
+    const messageChunk = new AIMessageChunk({
       content: content ?? "",
       tool_call_chunks: [functionCall].filter(Boolean)
     });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chatluna-google-gemini-adapter",
   "description": "google-gemini adapter for chatluna",
-  "version": "1.3.0-alpha.5",
+  "version": "1.3.0-alpha.6",
   "main": "lib/index.cjs",
   "module": "lib/index.mjs",
   "typings": "lib/index.d.ts",