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