From 37f09d3cb870155afc23b730e08880c4e6ee4ad6 Mon Sep 17 00:00:00 2001
From: MyPrototypeWhat <43230886+MyPrototypeWhat@users.noreply.github.com>
Date: Wed, 30 Apr 2025 00:16:58 +0800
Subject: [PATCH] fix: update webSearch type and results structure in upgrade
 logic (#5512)

---
 src/renderer/src/databases/upgrades.ts | 2 +-
 src/renderer/src/types/index.ts        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/databases/upgrades.ts b/src/renderer/src/databases/upgrades.ts
index 3303ae467a..b9449b1bfd 100644
--- a/src/renderer/src/databases/upgrades.ts
+++ b/src/renderer/src/databases/upgrades.ts
@@ -222,7 +222,7 @@ export async function upgradeToV7(tx: Transaction): Promise<void> {
       if (oldMessage.metadata?.webSearch) {
         hasCitationData = true
         citationDataToCreate.response = {
-          results: oldMessage.metadata.webSearch?.results,
+          results: oldMessage.metadata.webSearch,
           source: WebSearchSource.WEBSEARCH
         }
       }
diff --git a/src/renderer/src/types/index.ts b/src/renderer/src/types/index.ts
index 6180ada35a..889333f49a 100644
--- a/src/renderer/src/types/index.ts
+++ b/src/renderer/src/types/index.ts
@@ -87,7 +87,7 @@ export type LegacyMessage = {
     // Zhipu or Hunyuan
     webSearchInfo?: any[]
     // Web search
-    webSearch?: WebSearchResponse
+    webSearch?: WebSearchProviderResponse
     // MCP Tools
     mcpTools?: MCPToolResponse[]
     // Generate Image