refactor(chat): Improve error handling and configuration retrieval for AI models across chat components

Author: mayneyao

components/ai-chat/ai-chat-new.tsx

@@ -104,7 +104,15 @@ export default function Chat() {
     }
   }, [reloadModel, aiModel])
 
-  const { getConfigByModel, hasAvailableModels } = useAiConfig()
+  const { getConfigByModel } = useAiConfig()
+  const config = useMemo(() => {
+    try {
+      return getConfigByModel(aiModel)
+    } catch (error) {
+      return {}
+    }
+  }, [aiModel, getConfigByModel])
+  
   const { messages, setMessages, reload, append, isLoading, stop } = useChat({
     onToolCall: async ({ toolCall }) => {
       const res = await handleToolsCall(toolCall.toolName, toolCall.args)
@@ -123,7 +131,7 @@ export default function Chat() {
       })
     },
     body: {
-      ...getConfigByModel(aiModel),
+      ...config,
       systemPrompt,
       model: aiModel,
       useTools: enableTools, // 使用 enableTools 状态控制

components/doc/plugins/AIToolsPlugin/ai-tools.tsx

@@ -132,6 +132,13 @@ export function AITools({
   }
 
   const { addScript } = useScript()
+  const config = useMemo(() => {
+    try {
+      return getConfigByModel(currentModel)
+    } catch (error) {
+      return {}
+    }
+  }, [currentModel, getConfigByModel])
   const { messages, setMessages, reload, isLoading, stop } = useChat({
     onError(error) {
       console.log("error:", error)
@@ -161,7 +168,7 @@ export function AITools({
       setActionOpen(true)
     },
     body: {
-      ...getConfigByModel(currentModel),
+      ...config,
       model: currentModel,
       useTools: false,
       // use word chunking for

components/remix-chat/chat.tsx

@@ -1,6 +1,6 @@
 "use client"
 
-import { useCallback, useEffect, useState } from "react"
+import { useCallback, useEffect, useMemo, useState } from "react"
 import type { Attachment, ChatRequestOptions, Message } from "ai"
 import { useChat } from "ai/react"
 import { AnimatePresence } from "framer-motion"
@@ -78,8 +78,27 @@ export function Chat({
     codingModel ?? findFirstAvailableModel()
   )
 
+  const textModelConfig = useMemo(() => {
+    const textModel = findAvailableModel(TaskType.Translation)
+    if (textModel) {
+      try {
+        return getConfigByModel(textModel)
+      } catch (error) {
+        return undefined
+      }
+    }
+    return undefined
+  }, [findAvailableModel, getConfigByModel])
+
+  const config = useMemo(() => {
+    try {
+      return getConfigByModel(aiModel)
+    } catch (error) {
+      return {}
+    }
+  }, [aiModel, getConfigByModel])
+
   const { space } = useCurrentPathInfo()
-  const textModel = findAvailableModel(TaskType.Translation)
   const {
     messages,
     setMessages,
@@ -93,16 +112,14 @@ export function Chat({
     reload,
   } = useChat({
     body: {
-      ...getConfigByModel(aiModel),
+      ...config,
       systemPrompt: remixPrompt,
       id,
       model: aiModel,
       space,
       projectId: scriptId,
       useTools: false,
-      textModel: textModel
-        ? getConfigByModel(findAvailableModel(TaskType.Translation))
-        : undefined,
+      textModel: textModelConfig,
     },
     initialMessages,
     onFinish: () => {