Function downloadModel

src/hooks/useModelDownload.ts:84–117

(modelName: string, preferredQuantization?: string)

  }, [onComplete]);

  const downloadModel = async (modelName: string, preferredQuantization?: string) => {
    setDownloads((prev) => {
      const newMap = new Map(prev);
      newMap.set(modelName, {
        status: 'starting',
        percentage: 0,
        isDownloading: true,
        error: null,
        downloadedBytes: 0,
        totalBytes: 0,
        speedMbps: 0,
      });
      return newMap;
    });

    try {
      await pullModel(modelName, preferredQuantization);
    } catch (err) {
      const errorMessage = err instanceof Error ? err.message : 'Download failed';
      setDownloads((prev) => {
        const newMap = new Map(prev);
        newMap.set(modelName, {
          status: 'error',
          percentage: 0,
          isDownloading: false,
          error: errorMessage,
          downloadedBytes: 0,
          totalBytes: 0,
          speedMbps: 0,
        });
        return newMap;
      });
    }
  };

  return {
    downloads,