PowerShell 技能连载 - AI 智能脚本生成引擎优化

function Invoke-AIScriptGeneration {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory=$true)]
        [string]$NaturalLanguageQuery,
        
        [ValidateRange(1,5)]
        [int]$MaxAttempts = 3
    )

    $codeReport = [PSCustomObject]@{
        Timestamp = Get-Date -Format 'yyyy-MM-dd HH:mm:ss'
        GeneratedScript = $null
        ValidationErrors = @()
        OptimizationLevel = 0
    }

    try {
        $prompt = @"
作为PowerShell专家，请将以下运维需求转换为安全可靠的代码：
需求：$NaturalLanguageQuery
要求：
1. 包含try/catch错误处理
2. 支持WhatIf预执行模式
3. 输出结构化对象
4. 符合PowerShell最佳实践
"@

        # 调用OpenAI API
        $response = Invoke-RestMethod -Uri 'https://api.openai.com/v1/chat/completions' \
            -Headers @{ Authorization = "Bearer $env:OPENAI_API_KEY" } \
            -Body (@{
                model = "gpt-4-turbo"
                messages = @(@{ role = "user"; content = $prompt })
                temperature = 0.2
                max_tokens = 1500
            } | ConvertTo-Json)

        # 代码安全验证
        $validationResults = $response.choices[0].message.content | 
            Where-Object { $_ -notmatch 'Remove-Item|Format-Table' } |
            Test-ScriptAnalyzer -Severity Error

        $codeReport.GeneratedScript = $response.choices[0].message.content
        $codeReport.ValidationErrors = $validationResults
        $codeReport.OptimizationLevel = (100 - ($validationResults.Count * 20))
    }
    catch {
        Write-Error "AI脚本生成失败: $_"
        if ($MaxAttempts -gt 1) {
            return Invoke-AIScriptGeneration -NaturalLanguageQuery $NaturalLanguageQuery -MaxAttempts ($MaxAttempts - 1)
        }
    }

    # 生成智能编码报告
    $codeReport | Export-Csv -Path "$env:TEMP/AIScriptReport_$(Get-Date -Format yyyyMMdd).csv" -NoTypeInformation
    return $codeReport
}

核心功能：

1. 自然语言到代码的智能转换
2. 生成代码的安全验证
3. 多轮重试机制
4. 代码优化评分系统

应用场景：

• 运维需求快速原型开发
• 新手脚本编写辅助
• 跨团队需求标准化
• 自动化脚本知识库构建