77 lines
3.0 KiB
PowerShell
77 lines
3.0 KiB
PowerShell
param(
|
|
[Parameter(Mandatory=$true)]
|
|
[string]$Prompt,
|
|
|
|
[ValidateSet("Worker", "QA", "Utility")]
|
|
[string]$Role = "Utility",
|
|
|
|
[string]$Model = "flash",
|
|
|
|
[switch]$ShowContext
|
|
)
|
|
|
|
# Ensure the session has the API key loaded
|
|
if (Test-Path "C:\projects\misc\setup_gemini.ps1") {
|
|
. C:\projects\misc\setup_gemini.ps1
|
|
}
|
|
|
|
$SystemPrompts = @{
|
|
"Worker" = "STRICT SYSTEM DIRECTIVE: You are a stateless Tier 3 Worker (Contributor). Your goal is to generate high-quality code or diffs based on the provided ticket. DO NOT USE ANY TOOLS (no write_file, no run_shell_command, etc.). ONLY output the clean code or the requested diff inside XML-style tags if requested, otherwise just the code. No pleasantries."
|
|
"QA" = "STRICT SYSTEM DIRECTIVE: You are a stateless Tier 4 QA Agent. Your goal is to analyze the provided error trace and compress it into a surgical, 20-word fix. DO NOT USE ANY TOOLS. ONLY output the compressed fix. No explanations."
|
|
"Utility" = "STRICT SYSTEM DIRECTIVE: You are a stateless utility function. DO NOT USE ANY TOOLS. ONLY output the exact requested text, code, or JSON."
|
|
}
|
|
|
|
$SelectedPrompt = $SystemPrompts[$Role]
|
|
$SafePrompt = "$SelectedPrompt`n`nUSER PROMPT:`n$Prompt"
|
|
|
|
# Ensure log directory exists
|
|
if (-not (Test-Path "logs")) { New-Item -ItemType Directory -Path "logs" | Out-Null }
|
|
$LogFile = "logs/mma_delegation.log"
|
|
$Timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
|
|
|
|
$LogEntry = @"
|
|
--------------------------------------------------
|
|
TIMESTAMP: $Timestamp
|
|
TIER: $Role
|
|
SYSTEM PROMPT: $SelectedPrompt
|
|
USER PROMPT: $Prompt
|
|
--------------------------------------------------
|
|
"@
|
|
$LogEntry | Out-File -FilePath $LogFile -Append
|
|
|
|
if ($ShowContext) {
|
|
Write-Host "`n[MMA ORCHESTRATOR] Spawning Tier: $Role" -ForegroundColor Cyan
|
|
Write-Host "[MMA SYSTEM PROMPT]:`n$SelectedPrompt" -ForegroundColor Gray
|
|
Write-Host "[USER PROMPT]:`n$Prompt" -ForegroundColor White
|
|
Write-Host "--------------------------------------------------"
|
|
}
|
|
|
|
# Execute headless Gemini using -p, suppressing stderr noise
|
|
$jsonOutput = gemini -p $SafePrompt --model $Model --output-format json 2>$null
|
|
|
|
try {
|
|
# Extract only the JSON part
|
|
$fullString = $jsonOutput -join "`n"
|
|
$jsonStartIndex = $fullString.IndexOf("{")
|
|
|
|
if ($jsonStartIndex -ge 0) {
|
|
$cleanJsonString = $fullString.Substring($jsonStartIndex)
|
|
$parsed = $cleanJsonString | ConvertFrom-Json
|
|
|
|
# Log response
|
|
"RESPONSE:`n$($parsed.response)" | Out-File -FilePath $LogFile -Append
|
|
|
|
# Output only the clean response text
|
|
Write-Output $parsed.response
|
|
} else {
|
|
"ERROR: No JSON found in output.`n$fullString" | Out-File -FilePath $LogFile -Append
|
|
Write-Warning "No JSON object found in output."
|
|
Write-Output $fullString
|
|
}
|
|
} catch {
|
|
"FATAL ERROR: Parsing failed.`n$_" | Out-File -FilePath $LogFile -Append
|
|
# Fallback if parsing fails
|
|
Write-Warning "Failed to parse JSON from sub-agent. Raw output:"
|
|
Write-Output $jsonOutput
|
|
}
|