🐛 fix: 模型加载错误

lobehub · Jun 19, 2024 · 25cf5f9 · 25cf5f9
1 parent a344d01
commit 25cf5f9
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 12 deletions.
diff --git a/src/features/AgentViewer/index.tsx b/src/features/AgentViewer/index.tsx
@@ -36,8 +36,12 @@ function AgentViewer(props: Props) {
         storage.getItem(modelPath).then((blob) => {
           if (!blob) {
             fetchModelBlob(agent.agentId, agent.meta.model!).then((res) => {
-              const modelUrl = URL.createObjectURL(res);
-              viewer.loadVrm(modelUrl);
+              if (res) {
+                const modelUrl = URL.createObjectURL(res);
+                viewer.loadVrm(modelUrl);
+              } else {
+                viewer.unloadVRM();
+              }
             });
           } else {
             const modelUrl = URL.createObjectURL(blob as Blob);

diff --git a/src/hooks/useLoadModel.tsx b/src/hooks/useLoadModel.tsx
@@ -11,17 +11,22 @@ export const useLoadModel = () => {
   const fetchModelBlob = async (agentId: string, modelUrl: string) => {
     setDownloading(true);
     setPercent(0);
+    try {
+      const blob = await fetchWithProgress(modelUrl, {
+        onProgress: (loaded, total) => {
+          setPercent(Math.ceil((loaded / total) * 100));
+        },
+      });
+      const modelPath = getModelPathByAgentId(agentId);
+      await storage.setItem(modelPath, blob);
 
-    const blob = await fetchWithProgress(modelUrl, {
-      onProgress: (loaded, total) => {
-        setPercent(Math.ceil((loaded / total) * 100));
-      },
-    });
-    setDownloading(false);
-
-    const modelPath = getModelPathByAgentId(agentId);
-    await storage.setItem(modelPath, blob);
-    return blob;
+      return blob;
+    } catch (e) {
+      console.error(e);
+      return null;
+    } finally {
+      setDownloading(false);
+    }
   };
 
   return {