🐛 fix: fix ollama model download panel (#4070)

lobehub · Sep 22, 2024 · fc33336 · fc33336
1 parent 2d2231f
commit fc33336
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/features/Conversation/Error/OllamaBizError/InvalidOllamaModel/index.tsx b/src/features/Conversation/Error/OllamaBizError/InvalidOllamaModel/index.tsx
@@ -26,7 +26,7 @@ const OllamaModelForm = memo<OllamaModelFormProps>(({ id, model }) => {
   const [total, setTotal] = useState(0);
   const { remainingTime, downloadSpeed } = useDownloadMonitor(total, completed);
   const percent = useMemo(() => {
-    return total ? Number(((completed / total) * 100).toFixed(0)) : 0;
+    return total ? Number(((completed / total) * 100).toFixed(1)) : 0;
   }, [completed, total]);
 
   const [delAndRegenerateMessage, deleteMessage] = useChatStore((s) => [
@@ -114,8 +114,8 @@ const OllamaModelForm = memo<OllamaModelFormProps>(({ id, model }) => {
               !total
               ? t('ollama.unlock.starting')
               : t('ollama.unlock.downloaded', {
-                  completed: formatSize(completed),
-                  total: formatSize(total),
+                  completed: formatSize(completed, 2),
+                  total: formatSize(total, 2),
                 })}
         </Button>
         {isDownloading ? (

diff --git a/src/features/Conversation/Error/OllamaBizError/index.tsx b/src/features/Conversation/Error/OllamaBizError/index.tsx
@@ -23,7 +23,7 @@ interface OllamaErrorResponse {
   error: OllamaError;
 }
 
-const UNRESOLVED_MODEL_REGEXP = /model '([\w+,-_]+)' not found/;
+const UNRESOLVED_MODEL_REGEXP = /model "([\w+,-_]+)" not found/;
 
 const OllamaBizError = memo<ChatMessage>(({ error, id }) => {
   const errorBody: OllamaErrorResponse = (error as any)?.body;