@lobehub/chat 1.97.6 → 1.97.7

package/CHANGELOG.md

@@ -2,6 +2,31 @@
 
 # Changelog
 
+### [Version 1.97.7](https://github.com/lobehub/lobe-chat/compare/v1.97.6...v1.97.7)
+
+<sup>Released on **2025-07-10**</sup>
+
+#### 💄 Styles
+
+- **misc**: Add google search grounding for Vertex AI.
+
+<br/>
+
+<details>
+<summary><kbd>Improvements and Fixes</kbd></summary>
+
+#### Styles
+
+- **misc**: Add google search grounding for Vertex AI, closes [#8313](https://github.com/lobehub/lobe-chat/issues/8313) ([afd5900](https://github.com/lobehub/lobe-chat/commit/afd5900))
+
+</details>
+
+<div align="right">
+
+[![](https://img.shields.io/badge/-BACK_TO_TOP-151515?style=flat-square)](#readme-top)
+
+</div>
+
 ### [Version 1.97.6](https://github.com/lobehub/lobe-chat/compare/v1.97.5...v1.97.6)
 
 <sup>Released on **2025-07-10**</sup>

package/changelog/v1.json

@@ -1,4 +1,13 @@
 [
+  {
+    "children": {
+      "improvements": [
+        "Add google search grounding for Vertex AI."
+      ]
+    },
+    "date": "2025-07-10",
+    "version": "1.97.7"
+  },
   {
     "children": {
       "improvements": [

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lobehub/chat",
-  "version": "1.97.6",
+  "version": "1.97.7",
   "description": "Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.",
   "keywords": [
     "framework",

package/src/config/aiModels/vertexai.ts

@@ -6,6 +6,7 @@ const vertexaiChatModels: AIChatModelCard[] = [
     abilities: {
       functionCall: true,
       reasoning: true,
+      search: true,
       vision: true,
     },
     contextWindowTokens: 1_048_576 + 65_536,
@@ -20,6 +21,10 @@ const vertexaiChatModels: AIChatModelCard[] = [
       output: 10, // prompts <= 200k tokens
     },
     releasedAt: '2025-06-17',
+    settings: {
+      searchImpl: 'params',
+      searchProvider: 'google',
+    },
     type: 'chat',
   },
   {
@@ -65,6 +70,7 @@ const vertexaiChatModels: AIChatModelCard[] = [
     abilities: {
       functionCall: true,
       reasoning: true,
+      search: true,
       vision: true,
     },
     contextWindowTokens: 1_048_576 + 65_536,
@@ -78,6 +84,10 @@ const vertexaiChatModels: AIChatModelCard[] = [
       output: 2.5,
     },
     releasedAt: '2025-06-17',
+    settings: {
+      searchImpl: 'params',
+      searchProvider: 'google',
+    },
     type: 'chat',
   },
   {
@@ -117,6 +127,10 @@ const vertexaiChatModels: AIChatModelCard[] = [
       output: 0.4,
     },
     releasedAt: '2025-06-17',
+    settings: {
+      searchImpl: 'params',
+      searchProvider: 'google',
+    },
     type: 'chat',
   },
   {

package/src/libs/model-runtime/utils/streams/google-ai.test.ts

@@ -752,10 +752,6 @@ describe('GoogleGenerativeAIStream', () => {
         'event: text',
         'data: "123"\n',
 
-        'id: chat_1',
-        'event: grounding',
-        'data: {}\n',
-
         'id: chat_1',
         'event: text',
         'data: "45678"\n',

package/src/libs/model-runtime/utils/streams/google-ai.ts

@@ -86,9 +86,8 @@ const transformGoogleGenerativeAIStream = (
     }
 
     // return the grounding
-    if (candidate.groundingMetadata) {
-      const { webSearchQueries, groundingChunks } = candidate.groundingMetadata;
-
+    const { groundingChunks, webSearchQueries } = candidate.groundingMetadata ?? {};
+    if (groundingChunks) {
       return [
         { data: text, id: context.id, type: 'text' },
         {

package/src/libs/model-runtime/utils/streams/vertex-ai.ts

@@ -1,6 +1,7 @@
 import { GenerateContentResponse } from '@google/genai';
 
 import { ModelTokensUsage } from '@/types/message';
+import { GroundingSearch } from '@/types/search';
 import { nanoid } from '@/utils/uuid';
 
 import { type GoogleAIStreamOptions } from './google-ai';
@@ -95,6 +96,29 @@ const transformVertexAIStream = (
       ];
     }
 
+    // return the grounding
+    const { groundingChunks, webSearchQueries } = candidate.groundingMetadata ?? {};
+    if (groundingChunks) {
+      return [
+        !!part?.text ? { data: part.text, id: context?.id, type: 'text' } : undefined,
+        {
+          data: {
+            citations: groundingChunks?.map((chunk) => ({
+              // google 返回的 uri 是经过 google 自己处理过的 url，因此无法展现真实的 favicon
+              // 需要使用 title 作为替换
+              favicon: chunk.web?.title,
+              title: chunk.web?.title,
+              url: chunk.web?.uri,
+            })),
+            searchQueries: webSearchQueries,
+          } as GroundingSearch,
+          id: context.id,
+          type: 'grounding',
+        },
+        ...usageChunks,
+      ].filter(Boolean) as StreamProtocolChunk[];
+    }
+
     if (candidate.finishReason) {
       if (chunk.usageMetadata) {
         return [