koishi-plugin-chatluna-google-gemini-adapter 1.0.0-rc.3 → 1.0.0

package/lib/index.cjs

@@ -333,6 +333,10 @@ var GeminiRequester = class extends import_api.ModelRequester {
             {
               category: "HARM_CATEGORY_DANGEROUS_CONTENT",
               threshold: "BLOCK_NONE"
+            },
+            {
+              category: "HARM_CATEGORY_CIVIC_INTEGRITY",
+              threshold: "BLOCK_NONE"
             }
           ],
           generationConfig: {
@@ -340,6 +344,7 @@ var GeminiRequester = class extends import_api.ModelRequester {
             temperature: params.temperature,
             maxOutputTokens: params.model.includes("vision") ? void 0 : params.maxTokens,
             topP: params.topP
+            // thinkingConfig: { includeThoughts: true }
           },
           tools: params.tools != null || this._pluginConfig.googleSearch ? formatToolsToGeminiAITools(
             params.tools ?? [],
@@ -363,8 +368,11 @@ var GeminiRequester = class extends import_api.ModelRequester {
       };
       jsonParser.onValue = async ({ value }) => {
         const transformValue = value;
-        if (transformValue.candidates && transformValue.candidates[0]) {
-          const parts = transformValue.candidates[0]?.content?.parts;
+        if (!transformValue.candidates) {
+          return;
+        }
+        for (const candidate of transformValue.candidates) {
+          const parts = candidate.content?.parts;
           if (parts == null || parts.length < 1) {
             throw new Error(JSON.stringify(value));
           }
@@ -382,6 +390,7 @@ var GeminiRequester = class extends import_api.ModelRequester {
         0
       );
       let content = "";
+      let reasoningContent = "";
       let isOldVisionModel = params.model.includes("vision");
       const functionCall = {
         name: "",
@@ -392,6 +401,10 @@ var GeminiRequester = class extends import_api.ModelRequester {
         const messagePart = partAsType(chunk);
         const chatFunctionCallingPart = partAsType(chunk);
         if (messagePart.text) {
+          if (messagePart.thought) {
+            reasoningContent += messagePart.text;
+            continue;
+          }
           if (params.tools != null) {
             content = messagePart.text;
           } else {
@@ -450,6 +463,9 @@ var GeminiRequester = class extends import_api.ModelRequester {
           }
         }
       }
+      if (reasoningContent.length > 0) {
+        logger.debug(`reasoning content: ${reasoningContent}`);
+      }
     } catch (e) {
       if (e instanceof import_error.ChatLunaError) {
         throw e;

package/lib/index.d.ts

@@ -27,6 +27,7 @@ export interface ChatCompletionResponseMessage {
 export type ChatPart = ChatMessagePart | ChatUploadDataPart | ChatFunctionCallingPart | ChatFunctionResponsePart;
 export type ChatMessagePart = {
     text: string;
+    thought?: boolean;
 };
 export type ChatUploadDataPart = {
     inline_data: {

package/lib/index.mjs

@@ -327,6 +327,10 @@ var GeminiRequester = class extends ModelRequester {
             {
               category: "HARM_CATEGORY_DANGEROUS_CONTENT",
               threshold: "BLOCK_NONE"
+            },
+            {
+              category: "HARM_CATEGORY_CIVIC_INTEGRITY",
+              threshold: "BLOCK_NONE"
             }
           ],
           generationConfig: {
@@ -334,6 +338,7 @@ var GeminiRequester = class extends ModelRequester {
             temperature: params.temperature,
             maxOutputTokens: params.model.includes("vision") ? void 0 : params.maxTokens,
             topP: params.topP
+            // thinkingConfig: { includeThoughts: true }
           },
           tools: params.tools != null || this._pluginConfig.googleSearch ? formatToolsToGeminiAITools(
             params.tools ?? [],
@@ -357,8 +362,11 @@ var GeminiRequester = class extends ModelRequester {
       };
       jsonParser.onValue = async ({ value }) => {
         const transformValue = value;
-        if (transformValue.candidates && transformValue.candidates[0]) {
-          const parts = transformValue.candidates[0]?.content?.parts;
+        if (!transformValue.candidates) {
+          return;
+        }
+        for (const candidate of transformValue.candidates) {
+          const parts = candidate.content?.parts;
           if (parts == null || parts.length < 1) {
             throw new Error(JSON.stringify(value));
           }
@@ -376,6 +384,7 @@ var GeminiRequester = class extends ModelRequester {
         0
       );
       let content = "";
+      let reasoningContent = "";
       let isOldVisionModel = params.model.includes("vision");
       const functionCall = {
         name: "",
@@ -386,6 +395,10 @@ var GeminiRequester = class extends ModelRequester {
         const messagePart = partAsType(chunk);
         const chatFunctionCallingPart = partAsType(chunk);
         if (messagePart.text) {
+          if (messagePart.thought) {
+            reasoningContent += messagePart.text;
+            continue;
+          }
           if (params.tools != null) {
             content = messagePart.text;
           } else {
@@ -444,6 +457,9 @@ var GeminiRequester = class extends ModelRequester {
           }
         }
       }
+      if (reasoningContent.length > 0) {
+        logger.debug(`reasoning content: ${reasoningContent}`);
+      }
     } catch (e) {
       if (e instanceof ChatLunaError) {
         throw e;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-chatluna-google-gemini-adapter",
   "description": "google-gemini adapter for chatluna",
-  "version": "1.0.0-rc.3",
+  "version": "1.0.0",
   "main": "lib/index.cjs",
   "module": "lib/index.mjs",
   "typings": "lib/index.d.ts",
@@ -73,7 +73,7 @@
   },
   "peerDependencies": {
     "koishi": "^4.18.4",
-    "koishi-plugin-chatluna": "^1.0.0-rc.3"
+    "koishi-plugin-chatluna": "^1.0.0"
   },
   "koishi": {
     "description": {