ollama-download.ps1

# .\ollama-download.ps1 -model deepseek-coder-v2:16b

param (
    [string]$model
)

if ([string]::IsNullOrEmpty($model)) {
    Write-Host "Ошибка: Не передан аргумент модели. Пожалуйста, укажите модель при запуске скрипта." -ForegroundColor Red
    exit 1
}

$attempt = 0
$timeout = 10
while ($true) {
    $attempt++
    
    $startTime = Get-Date
    
    Write-Host "Attempting to download model '$model'... Attempt $attempt"
    $process = Start-Process -FilePath "ollama" -ArgumentList "pull $model" -PassThru -NoNewWindow
    try {
        $process | Wait-Process -Timeout $timeout -ErrorAction Stop
        if ($null -ne $process -and $process.ExitCode -eq 0) {
            Write-Host "Model downloaded successfully!"
            break
        } else {
            Write-Host "Download failed (Exit code: $($process.ExitCode)). Retrying..."
        }
    } catch {
        Write-Host "Timeout occurred, restarting download..."
        $process | Stop-Process -Force -ErrorAction SilentlyContinue
    }

    $endTime = Get-Date
    Write-Host "Elapsed time for attempt: $(($endTime - $startTime).TotalSeconds) seconds"

    Start-Sleep -Seconds 2
}