@lobehub/chat 1.62.10 → 1.63.0

package/src/features/ChatInput/ActionBar/Search/index.tsx

@@ -0,0 +1,76 @@
+import { ActionIcon } from '@lobehub/ui';
+import { Popover } from 'antd';
+import { useTheme } from 'antd-style';
+import { Globe } from 'lucide-react';
+import { memo } from 'react';
+import { useTranslation } from 'react-i18next';
+import { Flexbox } from 'react-layout-kit';
+
+import { useIsMobile } from '@/hooks/useIsMobile';
+import { useAgentStore } from '@/store/agent';
+import { agentSelectors } from '@/store/agent/selectors';
+import { aiModelSelectors, aiProviderSelectors, useAiInfraStore } from '@/store/aiInfra';
+
+import AINetworkSettings from './SwitchPanel';
+
+const Search = memo(() => {
+  const { t } = useTranslation('chat');
+  const [isLoading, isAgentEnableSearch] = useAgentStore((s) => [
+    agentSelectors.isAgentConfigLoading(s),
+    agentSelectors.isAgentEnableSearch(s),
+  ]);
+  const [model, provider] = useAgentStore((s) => [
+    agentSelectors.currentAgentModel(s),
+    agentSelectors.currentAgentModelProvider(s),
+  ]);
+
+  const [isModelHasBuiltinSearch] = useAiInfraStore((s) => [
+    aiModelSelectors.isModelHasBuiltinSearchConfig(model, provider)(s),
+    aiProviderSelectors.isProviderHasBuiltinSearchConfig(provider)(s),
+  ]);
+
+  const isMobile = useIsMobile();
+
+  const theme = useTheme();
+  if (isLoading)
+    return (
+      <ActionIcon
+        icon={Globe}
+        placement={'bottom'}
+        style={{
+          cursor: 'not-allowed',
+        }}
+      />
+    );
+
+  return (
+    isModelHasBuiltinSearch && (
+      <Flexbox>
+        <Popover
+          arrow={false}
+          content={<AINetworkSettings />}
+          styles={{
+            body: {
+              minWidth: isMobile ? undefined : 200,
+              padding: 4,
+              width: isMobile ? '100vw' : undefined,
+            },
+          }}
+        >
+          <ActionIcon
+            color={
+              isAgentEnableSearch ? (theme.isDarkMode ? theme.cyan7 : theme.cyan10) : undefined
+            }
+            icon={Globe}
+            placement={'bottom'}
+            title={t('search.title')}
+          />
+        </Popover>
+      </Flexbox>
+    )
+  );
+});
+
+Search.displayName = 'Search';
+
+export default Search;

package/src/features/ChatInput/ActionBar/config.ts

@@ -2,8 +2,9 @@ import STT from '../STT';
 import Clear from './Clear';
 import History from './History';
 import Knowledge from './Knowledge';
-import ModelSwitch from './ModelSwitch';
+import Model from './Model';
 import Params from './Params';
+import Search from './Search';
 import { MainToken, PortalToken } from './Token';
 import Tools from './Tools';
 import Upload from './Upload';
@@ -14,9 +15,10 @@ export const actionMap = {
   history: History,
   knowledgeBase: Knowledge,
   mainToken: MainToken,
-  model: ModelSwitch,
+  model: Model,
   params: Params,
   portalToken: PortalToken,
+  search: Search,
   stt: STT,
   temperature: Params,
   tools: Tools,

package/src/features/Conversation/Messages/Assistant/SearchGrounding.tsx

@@ -0,0 +1,153 @@
+import { Icon, SearchResultCards, Tag } from '@lobehub/ui';
+import { createStyles } from 'antd-style';
+import { AnimatePresence, motion } from 'framer-motion';
+import { ChevronDown, ChevronRight, Globe } from 'lucide-react';
+import Image from 'next/image';
+import { rgba } from 'polished';
+import { memo, useState } from 'react';
+import { useTranslation } from 'react-i18next';
+import { Flexbox } from 'react-layout-kit';
+
+import { GroundingSearch } from '@/types/search';
+
+const useStyles = createStyles(({ css, token, isDarkMode }) => ({
+  container: css`
+    width: fit-content;
+    padding-block: 4px;
+    padding-inline: 8px;
+    border-radius: 6px;
+
+    color: ${token.colorTextTertiary};
+
+    &:hover {
+      background: ${isDarkMode ? token.colorFillQuaternary : token.colorFillTertiary};
+    }
+  `,
+  expand: css`
+    background: ${isDarkMode ? token.colorFillQuaternary : token.colorFillTertiary} !important;
+  `,
+  shinyText: css`
+    color: ${rgba(token.colorText, 0.45)};
+
+    background: linear-gradient(
+      120deg,
+      ${rgba(token.colorTextBase, 0)} 40%,
+      ${token.colorTextSecondary} 50%,
+      ${rgba(token.colorTextBase, 0)} 60%
+    );
+    background-clip: text;
+    background-size: 200% 100%;
+
+    animation: shine 1.5s linear infinite;
+
+    @keyframes shine {
+      0% {
+        background-position: 100%;
+      }
+
+      100% {
+        background-position: -100%;
+      }
+    }
+  `,
+  title: css`
+    overflow: hidden;
+    display: -webkit-box;
+    -webkit-box-orient: vertical;
+    -webkit-line-clamp: 1;
+
+    font-size: 12px;
+    text-overflow: ellipsis;
+  `,
+}));
+
+const SearchGrounding = memo<GroundingSearch>(({ searchQueries, citations }) => {
+  const { t } = useTranslation('chat');
+  const { styles, cx, theme } = useStyles();
+
+  const [showDetail, setShowDetail] = useState(false);
+
+  return (
+    <Flexbox
+      className={cx(styles.container, showDetail && styles.expand)}
+      gap={16}
+      style={{ width: showDetail ? '100%' : undefined }}
+    >
+      <Flexbox
+        distribution={'space-between'}
+        flex={1}
+        gap={8}
+        horizontal
+        onClick={() => {
+          setShowDetail(!showDetail);
+        }}
+        style={{ cursor: 'pointer' }}
+      >
+        <Flexbox align={'center'} gap={8} horizontal>
+          <Icon icon={Globe} />
+          <Flexbox horizontal>{t('search.grounding.title', { count: citations?.length })}</Flexbox>
+          {!showDetail && (
+            <Flexbox horizontal>
+              {citations?.slice(0, 8).map((item, index) => (
+                <Image
+                  alt={item.title || item.url}
+                  height={16}
+                  key={`${item.url}-${index}`}
+                  src={`https://icons.duckduckgo.com/ip3/${new URL(item.url).host}.ico`}
+                  style={{
+                    background: theme.colorBgContainer,
+                    borderRadius: 8,
+                    marginInline: -2,
+                    padding: 2,
+                    zIndex: 100 - index,
+                  }}
+                  unoptimized
+                  width={16}
+                />
+              ))}
+            </Flexbox>
+          )}
+        </Flexbox>
+
+        <Flexbox gap={4} horizontal>
+          <Icon icon={showDetail ? ChevronDown : ChevronRight} />
+        </Flexbox>
+      </Flexbox>
+
+      <AnimatePresence initial={false}>
+        {showDetail && (
+          <motion.div
+            animate="open"
+            exit="collapsed"
+            initial="collapsed"
+            style={{ overflow: 'hidden', width: '100%' }}
+            transition={{
+              duration: 0.2,
+              ease: [0.4, 0, 0.2, 1], // 使用 ease-out 缓动函数
+            }}
+            variants={{
+              collapsed: { height: 0, opacity: 0, width: 'auto' },
+              open: { height: 'auto', opacity: 1, width: 'auto' },
+            }}
+          >
+            <Flexbox gap={12}>
+              {searchQueries && (
+                <Flexbox gap={4} horizontal>
+                  {t('search.grounding.searchQueries')}
+                  <Flexbox gap={8} horizontal>
+                    {searchQueries.map((query, index) => (
+                      <Tag key={index}>{query}</Tag>
+                    ))}
+                  </Flexbox>
+                </Flexbox>
+              )}
+              {citations && <SearchResultCards dataSource={citations} />}
+            </Flexbox>
+          </motion.div>
+        )}
+      </AnimatePresence>
+    </Flexbox>
+  );
+});
+
+export default SearchGrounding;

package/src/features/Conversation/Messages/Assistant/index.tsx

@@ -9,13 +9,14 @@ import { ChatMessage } from '@/types/message';
 import { DefaultMessage } from '../Default';
 import FileChunks from './FileChunks';
 import Reasoning from './Reasoning';
+import SearchGrounding from './SearchGrounding';
 import Tool from './Tool';
 
 export const AssistantMessage = memo<
   ChatMessage & {
     editableContent: ReactNode;
   }
->(({ id, tools, content, chunksList, ...props }) => {
+>(({ id, tools, content, chunksList, search, ...props }) => {
   const editing = useChatStore(chatSelectors.isMessageEditing(id));
   const generating = useChatStore(chatSelectors.isMessageGenerating(id));
 
@@ -23,6 +24,8 @@ export const AssistantMessage = memo<
 
   const isReasoning = useChatStore(aiChatSelectors.isMessageInReasoning(id));
 
+  const showSearch = !!search && !!search.citations?.length;
+
   // remove \n to avoid empty content
   // refs: https://github.com/lobehub/lobe-chat/pull/6153
   const showReasoning =
@@ -38,6 +41,9 @@ export const AssistantMessage = memo<
     />
   ) : (
     <Flexbox gap={8} id={id}>
+      {showSearch && (
+        <SearchGrounding citations={search?.citations} searchQueries={search?.searchQueries} />
+      )}
       {!!chunksList && chunksList.length > 0 && <FileChunks data={chunksList} />}
       {showReasoning && <Reasoning {...props.reasoning} id={id} />}
       {content && (

package/src/features/ModelSelect/index.tsx

@@ -4,7 +4,7 @@ import { memo, useMemo } from 'react';
 
 import { ModelItemRender, ProviderItemRender } from '@/components/ModelSelect';
 import { useEnabledChatModels } from '@/hooks/useEnabledChatModels';
-import { EnabledProviderWithModels } from '@/types/aiModel';
+import { EnabledProviderWithModels } from '@/types/aiProvider';
 
 const useStyles = createStyles(({ css, prefixCls }) => ({
   select: css`

package/src/features/ModelSwitchPanel/index.tsx

@@ -14,7 +14,7 @@ import { useEnabledChatModels } from '@/hooks/useEnabledChatModels';
 import { useIsMobile } from '@/hooks/useIsMobile';
 import { useAgentStore } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/slices/chat';
-import { EnabledProviderWithModels } from '@/types/aiModel';
+import { EnabledProviderWithModels } from '@/types/aiProvider';
 
 const useStyles = createStyles(({ css, prefixCls }) => ({
   menu: css`
@@ -68,7 +68,7 @@ const ModelSwitchPanel = memo<PropsWithChildren>(({ children }) => {
       if (items.length === 0)
         return [
           {
-            key: 'empty',
+            key: `${provider.id}-empty`,
             label: (
               <Flexbox gap={8} horizontal style={{ color: theme.colorTextTertiary }}>
                 {t('ModelSwitchPanel.emptyModel')}
@@ -114,7 +114,6 @@ const ModelSwitchPanel = memo<PropsWithChildren>(({ children }) => {
         },
       }}
       placement={isMobile ? 'top' : 'topLeft'}
-      trigger={['click']}
     >
       <div className={styles.tag}>{children}</div>
     </Dropdown>

package/src/hooks/useEnabledChatModels.ts

@@ -4,7 +4,7 @@ import { isDeprecatedEdition } from '@/const/version';
 import { useAiInfraStore } from '@/store/aiInfra';
 import { useUserStore } from '@/store/user';
 import { modelProviderSelectors } from '@/store/user/selectors';
-import { EnabledProviderWithModels } from '@/types/aiModel';
+import { EnabledProviderWithModels } from '@/types/aiProvider';
 
 export const useEnabledChatModels = (): EnabledProviderWithModels[] => {
   const enabledList = useUserStore(modelProviderSelectors.modelProviderListForModelSelect, isEqual);

package/src/libs/agent-runtime/azureai/index.ts

@@ -29,10 +29,29 @@ export class LobeAzureAI implements LobeRuntimeAI {
     const { messages, model, ...params } = payload;
     // o1 series models on Azure OpenAI does not support streaming currently
     const enableStreaming = model.includes('o1') ? false : (params.stream ?? true);
+
+    // Convert 'system' role to 'user' or 'developer' based on the model
+    const systemToUserModels = new Set([
+      'o1-preview',
+      'o1-preview-2024-09-12',
+      'o1-mini',
+      'o1-mini-2024-09-12',
+    ]);
+
+    const updatedMessages = messages.map((message) => ({
+      ...message,
+      role:
+        (model.includes('o1') || model.includes('o3')) && message.role === 'system'
+          ? [...systemToUserModels].some((sub) => model.includes(sub))
+            ? 'user'
+            : 'developer'
+          : message.role,
+    }));
+    
     try {
       const response = this.client.path('/chat/completions').post({
         body: {
-          messages: messages as OpenAI.ChatCompletionMessageParam[],
+          messages: updatedMessages as OpenAI.ChatCompletionMessageParam[],
           model,
           ...params,
           stream: enableStreaming,
@@ -98,7 +117,7 @@ export class LobeAzureAI implements LobeRuntimeAI {
 
   private maskSensitiveUrl = (url: string) => {
     // 使用正则表达式匹配 'https://' 后面和 '.azure.com/' 前面的内容
-    const regex = /^(https:\/\/)([^.]+)(\.azure\.com\/.*)$/;
+    const regex = /^(https:\/\/)([^.]+)(\.cognitiveservices\.azure\.com\/.*)$/;
 
     // 使用替换函数
     return url.replace(regex, (match, protocol, subdomain, rest) => {

package/src/libs/agent-runtime/google/index.test.ts

@@ -74,6 +74,145 @@ describe('LobeGoogleAI', () => {
       // 额外的断言可以加入，比如验证返回的流内容等
     });
 
+    it('should withGrounding', () => {
+      const data = [
+        {
+          candidates: [{ content: { parts: [{ text: 'As' }], role: 'model' } }],
+          usageMetadata: { promptTokenCount: 8, totalTokenCount: 8 },
+          modelVersion: 'gemini-2.0-flash',
+        },
+        {
+          candidates: [
+            {
+              content: { parts: [{ text: ' of February 22, 2025, "Ne Zha ' }], role: 'model' },
+              safetyRatings: [
+                { category: 'HARM_CATEGORY_HATE_SPEECH', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_DANGEROUS_CONTENT', probability: 'MEDIUM' },
+                { category: 'HARM_CATEGORY_HARASSMENT', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', probability: 'NEGLIGIBLE' },
+              ],
+            },
+          ],
+          usageMetadata: { promptTokenCount: 8, totalTokenCount: 8 },
+          modelVersion: 'gemini-2.0-flash',
+        },
+        {
+          candidates: [
+            {
+              content: {
+                parts: [{ text: '2" has grossed the following:\n\n*   **Worldwide:** $1' }],
+                role: 'model',
+              },
+              safetyRatings: [
+                { category: 'HARM_CATEGORY_HATE_SPEECH', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_DANGEROUS_CONTENT', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_HARASSMENT', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', probability: 'NEGLIGIBLE' },
+              ],
+            },
+          ],
+          usageMetadata: { promptTokenCount: 8, totalTokenCount: 8 },
+          modelVersion: 'gemini-2.0-flash',
+        },
+        {
+          candidates: [
+            {
+              content: {
+                parts: [
+                  {
+                    text: '.66 billion\n*   **China:** $1.82 billion (CN¥12.35 billion)\n*   **US &',
+                  },
+                ],
+                role: 'model',
+              },
+              safetyRatings: [
+                { category: 'HARM_CATEGORY_HATE_SPEECH', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_DANGEROUS_CONTENT', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_HARASSMENT', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', probability: 'NEGLIGIBLE' },
+              ],
+            },
+          ],
+          usageMetadata: { promptTokenCount: 8, totalTokenCount: 8 },
+          modelVersion: 'gemini-2.0-flash',
+        },
+        {
+          candidates: [
+            {
+              content: { parts: [{ text: ' Canada:** $24,744,753\n' }], role: 'model' },
+              finishReason: 'STOP',
+              safetyRatings: [
+                { category: 'HARM_CATEGORY_HATE_SPEECH', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_DANGEROUS_CONTENT', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_HARASSMENT', probability: 'NEGLIGIBLE' },
+                { category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', probability: 'NEGLIGIBLE' },
+              ],
+              groundingMetadata: {
+                searchEntryPoint: {
+                  renderedContent:
+                    '<style>\n.container {\n  align-items: center;\n  border-radius: 8px;\n  display: flex;\n  font-family: Google Sans, Roboto, sans-serif;\n  font-size: 14px;\n  line-height: 20px;\n  padding: 8px 12px;\n}\n.chip {\n  display: inline-block;\n  border: solid 1px;\n  border-radius: 16px;\n  min-width: 14px;\n  padding: 5px 16px;\n  text-align: center;\n  user-select: none;\n  margin: 0 8px;\n  -webkit-tap-highlight-color: transparent;\n}\n.carousel {\n  overflow: auto;\n  scrollbar-width: none;\n  white-space: nowrap;\n  margin-right: -12px;\n}\n.headline {\n  display: flex;\n  margin-right: 4px;\n}\n.gradient-container {\n  position: relative;\n}\n.gradient {\n  position: absolute;\n  transform: translate(3px, -9px);\n  height: 36px;\n  width: 9px;\n}\n@media (prefers-color-scheme: light) {\n  .container {\n    background-color: #fafafa;\n    box-shadow: 0 0 0 1px #0000000f;\n  }\n  .headline-label {\n    color: #1f1f1f;\n  }\n  .chip {\n    background-color: #ffffff;\n    border-color: #d2d2d2;\n    color: #5e5e5e;\n    text-decoration: none;\n  }\n  .chip:hover {\n    background-color: #f2f2f2;\n  }\n  .chip:focus {\n    background-color: #f2f2f2;\n  }\n  .chip:active {\n    background-color: #d8d8d8;\n    border-color: #b6b6b6;\n  }\n  .logo-dark {\n    display: none;\n  }\n  .gradient {\n    background: linear-gradient(90deg, #fafafa 15%, #fafafa00 100%);\n  }\n}\n@media (prefers-color-scheme: dark) {\n  .container {\n    background-color: #1f1f1f;\n    box-shadow: 0 0 0 1px #ffffff26;\n  }\n  .headline-label {\n    color: #fff;\n  }\n  .chip {\n    background-color: #2c2c2c;\n    border-color: #3c4043;\n    color: #fff;\n    text-decoration: none;\n  }\n  .chip:hover {\n    background-color: #353536;\n  }\n  .chip:focus {\n    background-color: #353536;\n  }\n  .chip:active {\n    background-color: #464849;\n    border-color: #53575b;\n  }\n  .logo-light {\n    display: none;\n  }\n  .gradient {\n    background: linear-gradient(90deg, #1f1f1f 15%, #1f1f1f00 100%);\n  }\n}\n</style>\n<div class="container">\n  <div class="headline">\n    <svg class="logo-light" width="18" height="18" viewBox="9 9 35 35" fill="none" xmlns="http://www.w3.org/2000/svg">\n      <path fill-rule="evenodd" clip-rule="evenodd" d="M42.8622 27.0064C42.8622 25.7839 42.7525 24.6084 42.5487 23.4799H26.3109V30.1568H35.5897C35.1821 32.3041 33.9596 34.1222 32.1258 35.3448V39.6864H37.7213C40.9814 36.677 42.8622 32.2571 42.8622 27.0064V27.0064Z" fill="#4285F4"/>\n      <path fill-rule="evenodd" clip-rule="evenodd" d="M26.3109 43.8555C30.9659 43.8555 34.8687 42.3195 37.7213 39.6863L32.1258 35.3447C30.5898 36.3792 28.6306 37.0061 26.3109 37.0061C21.8282 37.0061 18.0195 33.9811 16.6559 29.906H10.9194V34.3573C13.7563 39.9841 19.5712 43.8555 26.3109 43.8555V43.8555Z" fill="#34A853"/>\n      <path fill-rule="evenodd" clip-rule="evenodd" d="M16.6559 29.8904C16.3111 28.8559 16.1074 27.7588 16.1074 26.6146C16.1074 25.4704 16.3111 24.3733 16.6559 23.3388V18.8875H10.9194C9.74388 21.2072 9.06992 23.8247 9.06992 26.6146C9.06992 29.4045 9.74388 32.022 10.9194 34.3417L15.3864 30.8621L16.6559 29.8904V29.8904Z" fill="#FBBC05"/>\n      <path fill-rule="evenodd" clip-rule="evenodd" d="M26.3109 16.2386C28.85 16.2386 31.107 17.1164 32.9095 18.8091L37.8466 13.8719C34.853 11.082 30.9659 9.3736 26.3109 9.3736C19.5712 9.3736 13.7563 13.245 10.9194 18.8875L16.6559 23.3388C18.0195 19.2636 21.8282 16.2386 26.3109 16.2386V16.2386Z" fill="#EA4335"/>\n    </svg>\n    <svg class="logo-dark" width="18" height="18" viewBox="0 0 48 48" xmlns="http://www.w3.org/2000/svg">\n      <circle cx="24" cy="23" fill="#FFF" r="22"/>\n      <path d="M33.76 34.26c2.75-2.56 4.49-6.37 4.49-11.26 0-.89-.08-1.84-.29-3H24.01v5.99h8.03c-.4 2.02-1.5 3.56-3.07 4.56v.75l3.91 2.97h.88z" fill="#4285F4"/>\n      <path d="M15.58 25.77A8.845 8.845 0 0 0 24 31.86c1.92 0 3.62-.46 4.97-1.31l4.79 3.71C31.14 36.7 27.65 38 24 38c-5.93 0-11.01-3.4-13.45-8.36l.17-1.01 4.06-2.85h.8z" fill="#34A853"/>\n      <path d="M15.59 20.21a8.864 8.864 0 0 0 0 5.58l-5.03 3.86c-.98-2-1.53-4.25-1.53-6.64 0-2.39.55-4.64 1.53-6.64l1-.22 3.81 2.98.22 1.08z" fill="#FBBC05"/>\n      <path d="M24 14.14c2.11 0 4.02.75 5.52 1.98l4.36-4.36C31.22 9.43 27.81 8 24 8c-5.93 0-11.01 3.4-13.45 8.36l5.03 3.85A8.86 8.86 0 0 1 24 14.14z" fill="#EA4335"/>\n    </svg>\n    <div class="gradient-container"><div class="gradient"></div></div>\n  </div>\n  <div class="carousel">\n    <a class="chip" href="https://vertexaisearch.cloud.google.com/grounding-api-redirect/AQXblrycKK-4Q61T9-BeH_jYKcMfCwyI0-TGMMzPcvZuXVtBjnsxXJkWcxxay0giciDNQ5g4dfD8SdUuBIlBLFQE7Fuc8e50WZuKO9u3HVjQXMznQxtzcQ4fHUn1lDlsvKiurKnD-G-Sl6s7_8h3JNMJSsObKg79sP0vQ_f9N7ib5s3tuF35FglH1NLaiTvdpM1DVhaHZc2In94_hV3W-_k=">Nezha Reborn 2 box office</a>\n  </div>\n</div>\n',
+                },
+                groundingChunks: [
+                  {
+                    web: {
+                      uri: 'https://vertexaisearch.cloud.google.com/grounding-api-redirect/AQXblrz3Up-UZrEsLlT8zPkpwbakcjDZbojH5RuXL0HAa_0rHfG1WE5h6jADFSzcMxKNZcit_n7OaxnTvZNjp9WFL4NNJmjkqQRJoK_XdeVsnbshWJpm9TJL7KNNwzAl254th8cHxTsQIOPoNxsnrXeebIlMDVb8OuFWfCWUToiRxhv1_Vo=',
+                      title: 'screenrant.com',
+                    },
+                  },
+                  {
+                    web: {
+                      uri: 'https://vertexaisearch.cloud.google.com/grounding-api-redirect/AQXblry4I3hWcwVL-mI75BJYSy72Lb97KF50N2p5PWvH8vuLQQgekFmlw9PDiJ3KouByidcMsja_7IJ3F1S0PguLC0r_uxbcAGfFvJzbiMNdWOhQ7xDSJqObd_mCUa-VFpYzm6cd',
+                      title: 'imdb.com',
+                    },
+                  },
+                ],
+                groundingSupports: [
+                  {
+                    segment: {
+                      startIndex: 64,
+                      endIndex: 96,
+                      text: '*   **Worldwide:** $1.66 billion',
+                    },
+                    groundingChunkIndices: [0],
+                    confidenceScores: [0.95218265],
+                  },
+                  {
+                    segment: {
+                      startIndex: 146,
+                      endIndex: 178,
+                      text: '*   **US & Canada:** $24,744,753',
+                    },
+                    groundingChunkIndices: [1],
+                    confidenceScores: [0.7182074],
+                  },
+                ],
+                retrievalMetadata: {},
+                webSearchQueries: ['Nezha Reborn 2 box office'],
+              },
+            },
+          ],
+          usageMetadata: {
+            promptTokenCount: 7,
+            candidatesTokenCount: 79,
+            totalTokenCount: 86,
+            promptTokensDetails: [{ modality: 'TEXT', tokenCount: 7 }],
+            candidatesTokensDetails: [{ modality: 'TEXT', tokenCount: 79 }],
+          },
+          modelVersion: 'gemini-2.0-flash',
+        },
+      ];
+      const mockStream = new ReadableStream({
+        start(controller) {
+          controller.enqueue('Hello, world!');
+          controller.close();
+        },
+      });
+
+      vi.spyOn(instance['client'], 'getGenerativeModel').mockReturnValue({
+        generateContentStream: vi.fn().mockResolvedValueOnce(mockStream),
+      } as any);
+    });
+
     it('should call debugStream in DEBUG mode', async () => {
       // 设置环境变量以启用DEBUG模式
       process.env.DEBUG_GOOGLE_CHAT_COMPLETION = '1';
@@ -368,51 +507,19 @@ describe('LobeGoogleAI', () => {
       it('get default result with gemini-pro', async () => {
         const messages: OpenAIChatMessage[] = [{ content: 'Hello', role: 'user' }];
 
-        const contents = await instance['buildGoogleMessages'](messages, 'gemini-pro');
+        const contents = await instance['buildGoogleMessages'](messages);
 
         expect(contents).toHaveLength(1);
         expect(contents).toEqual([{ parts: [{ text: 'Hello' }], role: 'user' }]);
       });
 
-      it('messages should end with user if using gemini-pro', async () => {
-        const messages: OpenAIChatMessage[] = [
-          { content: 'Hello', role: 'user' },
-          { content: 'Hi', role: 'assistant' },
-        ];
-
-        const contents = await instance['buildGoogleMessages'](messages, 'gemini-1.0');
-
-        expect(contents).toHaveLength(3);
-        expect(contents).toEqual([
-          { parts: [{ text: 'Hello' }], role: 'user' },
-          { parts: [{ text: 'Hi' }], role: 'model' },
-          { parts: [{ text: '' }], role: 'user' },
-        ]);
-      });
-
-      it('should include system role if there is a system role prompt', async () => {
-        const messages: OpenAIChatMessage[] = [
-          { content: 'you are ChatGPT', role: 'system' },
-          { content: 'Who are you', role: 'user' },
-        ];
-
-        const contents = await instance['buildGoogleMessages'](messages, 'gemini-1.0');
-
-        expect(contents).toHaveLength(3);
-        expect(contents).toEqual([
-          { parts: [{ text: 'you are ChatGPT' }], role: 'user' },
-          { parts: [{ text: '' }], role: 'model' },
-          { parts: [{ text: 'Who are you' }], role: 'user' },
-        ]);
-      });
-
       it('should not modify the length if model is gemini-1.5-pro', async () => {
         const messages: OpenAIChatMessage[] = [
           { content: 'Hello', role: 'user' },
           { content: 'Hi', role: 'assistant' },
         ];
 
-        const contents = await instance['buildGoogleMessages'](messages, 'gemini-1.5-pro-latest');
+        const contents = await instance['buildGoogleMessages'](messages);
 
         expect(contents).toHaveLength(2);
         expect(contents).toEqual([
@@ -431,10 +538,9 @@ describe('LobeGoogleAI', () => {
             role: 'user',
           },
         ];
-        const model = 'gemini-1.5-flash-latest';
 
         // 调用 buildGoogleMessages 方法
-        const contents = await instance['buildGoogleMessages'](messages, model);
+        const contents = await instance['buildGoogleMessages'](messages);
 
         expect(contents).toHaveLength(1);
         expect(contents).toEqual([