@lobehub/chat 1.62.11 → 1.63.0

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/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([

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

@@ -5,12 +5,13 @@ import {
   FunctionDeclaration,
   Tool as GoogleFunctionCallTool,
   GoogleGenerativeAI,
+  GoogleSearchRetrievalTool,
   Part,
   SchemaType,
 } from '@google/generative-ai';
 
-import type { ChatModelCard } from '@/types/llm';
 import { VertexAIStream } from '@/libs/agent-runtime/utils/streams/vertex-ai';
+import type { ChatModelCard } from '@/types/llm';
 import { imageUrlToBase64 } from '@/utils/imageToBase64';
 import { safeParseJSON } from '@/utils/safeParseJSON';
 
@@ -86,7 +87,7 @@ export class LobeGoogleAI implements LobeRuntimeAI {
       const payload = this.buildPayload(rawPayload);
       const model = payload.model;
 
-      const contents = await this.buildGoogleMessages(payload.messages, model);
+      const contents = await this.buildGoogleMessages(payload.messages);
 
       const geminiStreamResult = await this.client
         .getGenerativeModel(
@@ -123,7 +124,7 @@ export class LobeGoogleAI implements LobeRuntimeAI {
         .generateContentStream({
           contents,
           systemInstruction: payload.system as string,
-          tools: this.buildGoogleTools(payload.tools),
+          tools: this.buildGoogleTools(payload.tools, payload),
         });
 
       const googleStream = convertIterableToStream(geminiStreamResult.stream);
@@ -168,26 +169,30 @@ export class LobeGoogleAI implements LobeRuntimeAI {
       .map((model) => {
         const modelName = model.name.replace(/^models\//, '');
 
-        const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => modelName.toLowerCase() === m.id.toLowerCase());
+        const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
+          (m) => modelName.toLowerCase() === m.id.toLowerCase(),
+        );
 
         return {
           contextWindowTokens: model.inputTokenLimit + model.outputTokenLimit,
           displayName: model.displayName,
           enabled: knownModel?.enabled || false,
           functionCall:
-            modelName.toLowerCase().includes('gemini') && !modelName.toLowerCase().includes('thinking')
-            || knownModel?.abilities?.functionCall
-            || false,
+            (modelName.toLowerCase().includes('gemini') &&
+              !modelName.toLowerCase().includes('thinking')) ||
+            knownModel?.abilities?.functionCall ||
+            false,
           id: modelName,
           reasoning:
-            modelName.toLowerCase().includes('thinking')
-            || knownModel?.abilities?.reasoning
-            || false,
+            modelName.toLowerCase().includes('thinking') ||
+            knownModel?.abilities?.reasoning ||
+            false,
           vision:
-            modelName.toLowerCase().includes('vision')
-            || (modelName.toLowerCase().includes('gemini') && !modelName.toLowerCase().includes('gemini-1.0'))
-            || knownModel?.abilities?.vision
-            || false,
+            modelName.toLowerCase().includes('vision') ||
+            (modelName.toLowerCase().includes('gemini') &&
+              !modelName.toLowerCase().includes('gemini-1.0')) ||
+            knownModel?.abilities?.vision ||
+            false,
         };
       })
       .filter(Boolean) as ChatModelCard[];
@@ -266,43 +271,7 @@ export class LobeGoogleAI implements LobeRuntimeAI {
   };
 
   // convert messages from the OpenAI format to Google GenAI SDK
-  private buildGoogleMessages = async (
-    messages: OpenAIChatMessage[],
-    model: string,
-  ): Promise<Content[]> => {
-    // if the model is gemini-1.0 we need to pair messages
-    if (model.startsWith('gemini-1.0')) {
-      const contents: Content[] = [];
-      let lastRole = 'model';
-
-      for (const message of messages) {
-        // current to filter function message
-        if (message.role === 'function') {
-          continue;
-        }
-        const googleMessage = await this.convertOAIMessagesToGoogleMessage(message);
-
-        // if the last message is a model message and the current message is a model message
-        // then we need to add a user message to separate them
-        if (lastRole === googleMessage.role) {
-          contents.push({ parts: [{ text: '' }], role: lastRole === 'user' ? 'model' : 'user' });
-        }
-
-        // add the current message to the contents
-        contents.push(googleMessage);
-
-        // update the last role
-        lastRole = googleMessage.role;
-      }
-
-      // if the last message is a user message, then we need to add a model message to separate them
-      if (lastRole === 'model') {
-        contents.push({ parts: [{ text: '' }], role: 'user' });
-      }
-
-      return contents;
-    }
-
+  private buildGoogleMessages = async (messages: OpenAIChatMessage[]): Promise<Content[]> => {
     const pools = messages
       .filter((message) => message.role !== 'function')
       .map(async (msg) => await this.convertOAIMessagesToGoogleMessage(msg));
@@ -353,7 +322,13 @@ export class LobeGoogleAI implements LobeRuntimeAI {
 
   private buildGoogleTools(
     tools: ChatCompletionTool[] | undefined,
+    payload?: ChatStreamPayload,
   ): GoogleFunctionCallTool[] | undefined {
+    // 目前 Tools (例如 googleSearch) 无法与其他 FunctionCall 同时使用
+    if (payload?.enabledSearch) {
+      return [{ googleSearch: {} } as GoogleSearchRetrievalTool];
+    }
+
     if (!tools || tools.length === 0) return;
 
     return [

package/src/libs/agent-runtime/novita/__snapshots__/index.test.ts.snap

@@ -26,7 +26,7 @@ exports[`NovitaAI > models > should get models 1`] = `
     "contextWindowTokens": 8192,
     "description": "Meta's latest class of models, Llama 3.1, launched with a variety of sizes and configurations. The 8B instruct-tuned version is particularly fast and efficient. It has demonstrated strong performance in human evaluations, outperforming several leading closed-source models.",
     "displayName": "meta-llama/llama-3.1-8b-instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-8b-instruct",
     "reasoning": false,
@@ -36,7 +36,7 @@ exports[`NovitaAI > models > should get models 1`] = `
     "contextWindowTokens": 8192,
     "description": "Meta's latest class of models, Llama 3.1, has launched with a variety of sizes and configurations. The 70B instruct-tuned version is optimized for high-quality dialogue use cases. It has demonstrated strong performance in human evaluations compared to leading closed-source models.",
     "displayName": "meta-llama/llama-3.1-70b-instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-70b-instruct",
     "reasoning": false,
@@ -46,7 +46,7 @@ exports[`NovitaAI > models > should get models 1`] = `
     "contextWindowTokens": 32768,
     "description": "Meta's latest class of models, Llama 3.1, launched with a variety of sizes and configurations. This 405B instruct-tuned version is optimized for high-quality dialogue use cases. It has demonstrated strong performance compared to leading closed-source models, including GPT-4o and Claude 3.5 Sonnet, in evaluations.",
     "displayName": "meta-llama/llama-3.1-405b-instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-405b-instruct",
     "reasoning": false,

package/src/libs/agent-runtime/openrouter/__snapshots__/index.test.ts.snap

@@ -407,7 +407,7 @@ It has demonstrated strong performance compared to leading closed-source models
 
 To read more about the model release, [click here](https://ai.meta.com/blog/meta-llama-3/). Usage of this model is subject to [Meta's Acceptable Use Policy](https://llama.meta.com/llama3/use-policy/).",
     "displayName": "Meta: Llama 3.1 70B Instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-70b-instruct",
     "maxTokens": undefined,
@@ -439,7 +439,7 @@ It has demonstrated strong performance compared to leading closed-source models
 
 To read more about the model release, [click here](https://ai.meta.com/blog/meta-llama-3/). Usage of this model is subject to [Meta's Acceptable Use Policy](https://llama.meta.com/llama3/use-policy/).",
     "displayName": "Meta: Llama 3.1 8B Instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-8b-instruct",
     "maxTokens": undefined,
@@ -456,7 +456,7 @@ It has demonstrated strong performance compared to leading closed-source models
 
 To read more about the model release, [click here](https://ai.meta.com/blog/meta-llama-3/). Usage of this model is subject to [Meta's Acceptable Use Policy](https://llama.meta.com/llama3/use-policy/).",
     "displayName": "Meta: Llama 3.1 405B Instruct",
-    "enabled": true,
+    "enabled": false,
     "functionCall": false,
     "id": "meta-llama/llama-3.1-405b-instruct",
     "maxTokens": undefined,