Geminiの検索機能実装、他

README.md

@@ -22,6 +22,12 @@
    <a href="https://github.com/sponsors/tegnike"><img alt="GitHub Sponsor" src="https://img.shields.io/badge/Sponsor-GitHub-ea4aaa?style=flat&logo=github"/></a>
 </p>
 
+<div align="center">
+   <h3>
+      🌟 <a href="https://aituberkit.com">デモサイトへ</a> 🌟
+   </h3>
+</div>
+
 <h3 align="center">
    <a href="./docs/README_en.md">English</a>｜
    <a href="./docs/README_zh.md">中文</a>｜

docs/README_en.md

@@ -23,6 +23,12 @@
    <a href="https://github.com/sponsors/tegnike"><img alt="GitHub Sponsor" src="https://img.shields.io/badge/Sponsor-GitHub-ea4aaa?style=flat&logo=github"/></a>
 </p>
 
+<div align="center">
+   <h3>
+      🌟 <a href="https://aituberkit.com">Demo Site</a> 🌟
+   </h3>
+</div>
+
 <h3 align="center">
    <a href="../README.md">【Japanese】</a>｜
    <a href="./README_zh.md">【Chinese】</a>｜

docs/README_ko.md

@@ -23,6 +23,12 @@
    <a href="https://github.com/sponsors/tegnike"><img alt="GitHub Sponsor" src="https://img.shields.io/badge/Sponsor-GitHub-ea4aaa?style=flat&logo=github"/></a>
 </p>
 
+<div align="center">
+   <h3>
+      🌟 <a href="https://aituberkit.com">Demo Site</a> 🌟
+   </h3>
+</div>
+
 <h3 align="center">
    <a href="../README.md">【일본어】</a>｜
    <a href="./README_en.md">【영어】</a>｜

docs/README_zh.md

@@ -23,6 +23,12 @@
    <a href="https://github.com/sponsors/tegnike"><img alt="GitHub Sponsor" src="https://img.shields.io/badge/Sponsor-GitHub-ea4aaa?style=flat&logo=github"/></a>
 </p>
 
+<div align="center">
+   <h3>
+      🌟 <a href="https://aituberkit.com">Demo Site</a> 🌟
+   </h3>
+</div>
+
 <h3 align="center">
    <a href="../README.md">【日本語】</a>｜
    <a href="./README_en.md">【英語】</a>｜

locales/en/translation.json

@@ -173,6 +173,7 @@
   "AudioMode": "Audio Mode (Beta)",
   "InputText": "Text",
   "InputAudio": "Audio",
+  "SearchGrounding": "Use Search Grounding",
   "UpdateRealtimeAPISettings": "Update Realtime API Settings",
   "UpdateRealtimeAPISettingsInfo": "When updating the API key, Azure Endpoint, voice type, model, or system prompt, please press the update button to start a new WebSocket session.",
   "AzureEndpoint": "Azure Endpoint",

locales/ja/translation.json

@@ -174,6 +174,7 @@
   "AudioMode": "オーディオモード（ベータ版）",
   "InputText": "テキスト",
   "InputAudio": "音声",
+  "SearchGrounding": "検索機能を利用する",
   "UpdateRealtimeAPISettings": "リアルタイムAPI設定を更新",
   "UpdateRealtimeAPISettingsInfo": "APIキー、Azure Endpoint、ボイスタイプ、モデル、システムプロンプトを更新した際は更新ボタンを押して、新しいWebSocketセッションを開始してください。",
   "AzureEndpoint": "Azure Endpoint",

locales/ko/translation.json

@@ -173,6 +173,7 @@
   "AudioMode": "오디오 모드 (베타 버전)",
   "InputText": "텍스트",
   "InputAudio": "음성",
+  "SearchGrounding": "검색 기능을 사용하기",
   "UpdateRealtimeAPISettings": "리얼타임 API 설정 업데이트",
   "UpdateRealtimeAPISettingsInfo": "API 키, Azure Endpoint, 보이스 타입, 모델, 시스템 프롬프트를 업데이트한 경우 업데이트 버튼을 눌러 새로운 WebSocket 세션을 시작하십시오.",
   "AzureEndpoint": "Azure Endpoint",

locales/zh/translation.json

@@ -173,6 +173,7 @@
   "AudioMode": "音訊模式（測試版）",
   "InputText": "文字",
   "InputAudio": "音訊",
+  "SearchGrounding": "使用搜索功能",
   "UpdateRealtimeAPISettings": "更新實時API設定",
   "UpdateRealtimeAPISettingsInfo": "當更新 API 金鑰、Azure Endpoint、語音型別、模型或系統提示時，請按更新按鈕以開始新的 WebSocket 會話。",
   "AzureEndpoint": "Azure Endpoint",

src/components/live2DViewer.tsx

@@ -65,7 +65,7 @@ export default function Live2DViewer() {
 
   console.log('Rendering Live2DViewer')
   return (
-    <div className="fixed bottom-0 right-0 w-screen h-screen">
+    <div className="fixed bottom-0 right-0 w-screen h-screen z-5">
       <Script
         key={`cubismcore-${scriptLoadRetries.cubismcore}`}
         src="/scripts/live2dcubismcore.min.js"

src/components/settings/modelProvider.tsx

@@ -38,6 +38,7 @@ const ModelProvider = () => {
   const perplexityKey = settingsStore((s) => s.perplexityKey)
   const fireworksKey = settingsStore((s) => s.fireworksKey)
   const difyKey = settingsStore((s) => s.difyKey)
+  const useSearchGrounding = settingsStore((s) => s.useSearchGrounding)
 
   const selectAIService = settingsStore((s) => s.selectAIService)
   const selectAIModel = settingsStore((s) => s.selectAIModel)
@@ -480,6 +481,22 @@ const ModelProvider = () => {
                   </option>
                 </select>
               </div>
+              <div className="my-24">
+                <div className="my-16 typography-20 font-bold">
+                  {t('SearchGrounding')}
+                </div>
+                <div className="my-8">
+                  <TextButton
+                    onClick={() => {
+                      settingsStore.setState({
+                        useSearchGrounding: !useSearchGrounding,
+                      })
+                    }}
+                  >
+                    {useSearchGrounding ? t('StatusOn') : t('StatusOff')}
+                  </TextButton>
+                </div>
+              </div>
             </>
           )
         } else if (selectAIService === 'azure') {

src/features/chat/vercelAIChat.ts

@@ -18,6 +18,7 @@ const getAIConfig = () => {
     selectAIService: aiService,
     selectAIModel: ss.selectAIModel,
     azureEndpoint: ss.azureEndpoint,
+    useSearchGrounding: ss.useSearchGrounding,
   }
 }
 
@@ -28,8 +29,13 @@ function handleApiError(errorCode: string): string {
 }
 
 export async function getVercelAIChatResponse(messages: Message[]) {
-  const { aiApiKey, selectAIService, selectAIModel, azureEndpoint } =
-    getAIConfig()
+  const {
+    aiApiKey,
+    selectAIService,
+    selectAIModel,
+    azureEndpoint,
+    useSearchGrounding,
+  } = getAIConfig()
 
   try {
     const response = await fetch('/api/aiChat', {
@@ -44,6 +50,7 @@ export async function getVercelAIChatResponse(messages: Message[]) {
         model: selectAIModel,
         azureEndpoint: azureEndpoint,
         stream: false,
+        useSearchGrounding: useSearchGrounding,
       }),
     })
 
@@ -67,8 +74,13 @@ export async function getVercelAIChatResponse(messages: Message[]) {
 export async function getVercelAIChatResponseStream(
   messages: Message[]
 ): Promise<ReadableStream<string>> {
-  const { aiApiKey, selectAIService, selectAIModel, azureEndpoint } =
-    getAIConfig()
+  const {
+    aiApiKey,
+    selectAIService,
+    selectAIModel,
+    azureEndpoint,
+    useSearchGrounding,
+  } = getAIConfig()
 
   const response = await fetch('/api/aiChat', {
     method: 'POST',
@@ -82,6 +94,7 @@ export async function getVercelAIChatResponseStream(
       model: selectAIModel,
       azureEndpoint: azureEndpoint,
       stream: true,
+      useSearchGrounding: useSearchGrounding,
     }),
   })
 

src/features/stores/settings.ts

@@ -136,6 +136,7 @@ interface General {
   slideMode: boolean
   messageReceiverEnabled: boolean
   clientId: string
+  useSearchGrounding: boolean
 }
 
 interface ModelType {
@@ -288,6 +289,7 @@ const settingsStore = create<SettingsState>()(
       slideMode: process.env.NEXT_PUBLIC_SLIDE_MODE === 'true',
       messageReceiverEnabled: false,
       clientId: '',
+      useSearchGrounding: false,
 
       // NijiVoice settings
       nijivoiceApiKey: '',

src/pages/api/aiChat.ts

@@ -41,8 +41,15 @@ export default async function handler(req: NextRequest) {
     )
   }
 
-  const { messages, apiKey, aiService, model, azureEndpoint, stream } =
-    await req.json()
+  const {
+    messages,
+    apiKey,
+    aiService,
+    model,
+    azureEndpoint,
+    stream,
+    useSearchGrounding,
+  } = await req.json()
 
   let aiApiKey = apiKey
   if (!aiApiKey) {
@@ -128,10 +135,14 @@ export default async function handler(req: NextRequest) {
   const instance = aiServiceInstance()
   const modifiedMessages: Message[] = modifyMessages(aiService, messages)
 
+  const isUseSearchGrounding = aiService === 'google' && useSearchGrounding
+  const options = isUseSearchGrounding ? { useSearchGrounding: true } : {}
+  console.log('options', options)
+
   try {
     if (stream) {
       const result = await streamText({
-        model: instance(modifiedModel),
+        model: instance(modifiedModel, options),
         messages: modifiedMessages as CoreMessage[],
       })