feat: improve error handling in web search

components/DeepResearch.vue

@@ -8,12 +8,14 @@
   import type { TreeNode } from './Tree.vue'
   import { marked } from 'marked'
 
-  const { t, locale } = useI18n()
-  const { config } = storeToRefs(useConfigStore())
   const emit = defineEmits<{
     (e: 'complete', results: ResearchResult): void
   }>()
 
+  const toast = useToast()
+  const { t, locale } = useI18n()
+  const { config } = storeToRefs(useConfigStore())
+
   const tree = ref<TreeNode>({
     id: '0',
     label: t('webBrowsing.startNode.label'),
@@ -41,8 +43,8 @@
           // 创建新节点
           node = {
             id: nodeId,
-            label: t('webBrowsing.generating'),
-            researchGoal: t('webBrowsing.generating'),
+            label: '',
+            researchGoal: '',
             learnings: [],
             children: [],
           }
@@ -63,7 +65,7 @@
         }
         // 更新节点的查询内容
         if (step.result) {
-          node.label = step.result.query ?? t('webBrowsing.generating')
+          node.label = step.result.query ?? ''
           node.researchGoal = step.result.researchGoal
         }
         break
@@ -102,6 +104,15 @@
 
       case 'error':
         console.error(`Research error on node ${nodeId}:`, step.message)
+        node!.error = step.message
+        toast.add({
+          title: t('webBrowsing.nodeFailedToast', {
+            label: node!.label ?? nodeId,
+          }),
+          description: step.message,
+          color: 'error',
+          duration: 8000,
+        })
         break
 
       case 'complete':
@@ -186,8 +197,19 @@
         <Tree :node="tree" :selected-node="selectedNode" @select="selectNode" />
       </div>
       <div v-if="selectedNode" class="p-4">
-        <USeparator :label="t('webBrowsing.nodeDetails')" />
-        <h2 class="text-xl font-bold my-2">{{ selectedNode.label }}</h2>
+        <USeparator :label="$t('webBrowsing.nodeDetails')" />
+        <UAlert
+          v-if="selectedNode.error"
+          class="my-2"
+          :title="$t('webBrowsing.nodeFailed')"
+          :description="selectedNode.error"
+          color="error"
+          variant="soft"
+          :duration="8000"
+        />
+        <h2 class="text-xl font-bold my-2">
+          {{ selectedNode.label ?? $t('webBrowsing.generating') }}
+        </h2>
 
         <!-- Root node has no additional information -->
         <p v-if="selectedNode.id === '0'">

components/Tree.vue

@@ -14,6 +14,7 @@
     visitedUrls?: string[]
     status?: TreeNodeStatus
     children: TreeNode[]
+    error?: string
   }
 
   const props = defineProps<{

i18n/en.json

@@ -64,7 +64,9 @@
     "researchGoal": "Research Goal",
     "visitedUrls": "Visited URLs",
     "learnings": "Learnings",
-    "generating": "Generating..."
+    "generating": "Generating...",
+    "nodeFailed": "Search failed",
+    "nodeFailedToast": "Search node \"{label}\" failed"
   },
   "researchReport": {
     "title": "4. Research Report",

i18n/zh.json

@@ -29,7 +29,9 @@
       "providerHelp": "目前仅支持 Tavily，每个月可以免费搜索 1000 次。\n请在 {0} 生成一个 API 密钥。",
       "apiKey": "API 密钥",
       "queryLanguage": "使用语言",
-      "queryLanguageHelp": "修改搜索词的语言。如果你想获取不同的搜索结果（比如查询高质量的英文资料），可以在这里修改。\nAI 模型在总结的时候仍然会使用当前网页的语言。"
+      "queryLanguageHelp": "修改搜索词的语言。如果你想获取不同的搜索结果（比如查询高质量的英文资料），可以在这里修改。\nAI 模型在总结的时候仍然会使用当前网页的语言。",
+      "nodeFailed": "搜索失败",
+      "nodeFailedToast": "搜索步骤 “{label}” 失败"
     }
   },
   "researchTopic": {

lib/deep-research.ts

@@ -381,9 +381,20 @@ export async function deepResearch({
               }
             }
           } catch (e: any) {
-            throw new Error(
-              `Error searching for ${searchQuery.query}, depth ${currentDepth}\nMessage: ${e.message}`,
+            const id = childNodeId(nodeId, i)
+            console.error(
+              `Error in node ${id} for query ${searchQuery.query}`,
+              e,
             )
+            onProgress({
+              type: 'error',
+              message: e.message,
+              nodeId: id,
+            })
+            return {
+              learnings: [],
+              visitedUrls: [],
+            }
           }
         }),
       ),