97 lines
3.9 KiB
PowerShell
97 lines
3.9 KiB
PowerShell
|
|
# run_eval.ps1 - Siemens RAGAS Evaluation Runner
|
||
|
|
# Usage:
|
||
|
|
# .\run_eval.ps1 # online eval (default)
|
||
|
|
# .\run_eval.ps1 offline # offline smoke
|
||
|
|
# .\run_eval.ps1 path\to\scenario.yaml # custom scenario
|
||
|
|
# .\run_eval.ps1 online DEBUG # second arg = log level (DEBUG/INFO/WARNING)
|
||
|
|
# Or: powershell -ExecutionPolicy Bypass -File run_eval.ps1 [scenario] [log-level]
|
||
|
|
|
||
|
|
param(
|
||
|
|
[string]$Scenario = "online",
|
||
|
|
[string]$LogLevel = "INFO"
|
||
|
|
)
|
||
|
|
|
||
|
|
$ErrorActionPreference = "Stop"
|
||
|
|
Set-Location $PSScriptRoot
|
||
|
|
|
||
|
|
Write-Host ""
|
||
|
|
Write-Host "============================================================" -ForegroundColor Cyan
|
||
|
|
Write-Host " Siemens RAGAS - Evaluation Runner" -ForegroundColor Cyan
|
||
|
|
Write-Host "============================================================" -ForegroundColor Cyan
|
||
|
|
Write-Host ""
|
||
|
|
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
# 1. Resolve scenario path
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
$scenarioMap = @{
|
||
|
|
"online" = "scenarios\online\siemens-pdf-question-bank-online.yaml"
|
||
|
|
"offline" = "scenarios\offline\siemens-pdf-offline-smoke.yaml"
|
||
|
|
}
|
||
|
|
if ($scenarioMap.ContainsKey($Scenario.ToLower())) {
|
||
|
|
$Scenario = $scenarioMap[$Scenario.ToLower()]
|
||
|
|
}
|
||
|
|
if (-not (Test-Path $Scenario)) {
|
||
|
|
Write-Host "[ERROR] Scenario file not found: $Scenario" -ForegroundColor Red
|
||
|
|
Write-Host ""
|
||
|
|
Write-Host "Usage examples:"
|
||
|
|
Write-Host " .\run_eval.ps1 - online eval (default)"
|
||
|
|
Write-Host " .\run_eval.ps1 offline - offline smoke"
|
||
|
|
Write-Host " .\run_eval.ps1 path\to\file.yaml - custom scenario"
|
||
|
|
Read-Host "Press Enter to exit"
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
Write-Host "[OK] Scenario : $Scenario" -ForegroundColor Green
|
||
|
|
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
# 2. Validate log level
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
$validLevels = @("DEBUG", "INFO", "WARNING", "ERROR")
|
||
|
|
if ($validLevels -notcontains $LogLevel.ToUpper()) {
|
||
|
|
Write-Host "[WARN] Unknown log level '$LogLevel', defaulting to INFO" -ForegroundColor Yellow
|
||
|
|
$LogLevel = "INFO"
|
||
|
|
}
|
||
|
|
Write-Host "[OK] Log level: $LogLevel" -ForegroundColor Green
|
||
|
|
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
# 3. Create logs dir with timestamped filename
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
if (-not (Test-Path "logs")) { New-Item -ItemType Directory "logs" | Out-Null }
|
||
|
|
$timestamp = Get-Date -Format "yyyy-MM-dd_HHmmss"
|
||
|
|
$logFile = "logs\eval_$timestamp.log"
|
||
|
|
Write-Host "[OK] Log file : $logFile" -ForegroundColor Green
|
||
|
|
|
||
|
|
Write-Host ""
|
||
|
|
Write-Host "============================================================" -ForegroundColor Cyan
|
||
|
|
Write-Host " Starting evaluation..." -ForegroundColor Cyan
|
||
|
|
Write-Host " Logs also written to: $logFile" -ForegroundColor Cyan
|
||
|
|
Write-Host " Press Ctrl+C to abort" -ForegroundColor Yellow
|
||
|
|
Write-Host "============================================================" -ForegroundColor Cyan
|
||
|
|
Write-Host ""
|
||
|
|
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
# 4. Run evaluation
|
||
|
|
# ----------------------------------------------------------------
|
||
|
|
$env:PYTHONIOENCODING = "utf-8"
|
||
|
|
$env:PYTHONPATH = "."
|
||
|
|
|
||
|
|
& python main.py `
|
||
|
|
--scenario $Scenario `
|
||
|
|
--log-file $logFile `
|
||
|
|
--log-level $LogLevel.ToUpper()
|
||
|
|
|
||
|
|
if ($LASTEXITCODE -ne 0) {
|
||
|
|
Write-Host ""
|
||
|
|
Write-Host "[ERROR] Evaluation failed. Check log: $logFile" -ForegroundColor Red
|
||
|
|
Read-Host "Press Enter to exit"
|
||
|
|
exit 1
|
||
|
|
}
|
||
|
|
|
||
|
|
Write-Host ""
|
||
|
|
Write-Host "============================================================" -ForegroundColor Green
|
||
|
|
Write-Host " Evaluation complete!" -ForegroundColor Green
|
||
|
|
Write-Host " Log saved to: $logFile" -ForegroundColor Green
|
||
|
|
Write-Host " Open the web console to view results: start.bat" -ForegroundColor Cyan
|
||
|
|
Write-Host "============================================================" -ForegroundColor Green
|
||
|
|
Write-Host ""
|
||
|
|
Read-Host "Press Enter to exit"
|