function Invoke-AIOpsAutomation {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory=$true)]
        [string]$OperationContext,
        
        [ValidateRange(1,100)]
        [int]$MaxTokens = 60
    )

    $apiKey = $env:OPENAI_API_KEY
    $prompt = @"
作为资深PowerShell运维专家，请根据以下运维场景生成可执行的解决方案：
场景：$OperationContext
要求：
1. 使用标准PowerShell命令
2. 包含错误处理机制
3. 输出结构化数据
4. 确保跨平台兼容性
"@

    $body = @{
        model = "gpt-3.5-turbo"
        messages = @(@{role="user"; content=$prompt})
        max_tokens = $MaxTokens
    } | ConvertTo-Json

    $response = Invoke-RestMethod -Uri 'https://api.openai.com/v1/chat/completions' \
        -Method Post \
        -Headers @{ Authorization = "Bearer $apiKey" } \
        -ContentType 'application/json' \
        -Body $body

    $codeBlock = $response.choices[0].message.content -replace '```powershell','' -replace '```',''
    [scriptblock]::Create($codeBlock).Invoke()
}

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
}

function Invoke-AIScriptGeneration {
    param(
        [Parameter(Mandatory=$true)]
        [string]$Prompt,
        [ValidateSet('AWS','Azure','Windows','Linux')]
        [string]$Environment = 'Windows'
    )

    $apiKey = $env:OPENAI_API_KEY
    $headers = @{
        'Authorization' = "Bearer $apiKey"
        'Content-Type' = 'application/json'
    }

    $body = @{
        model = 'gpt-4-turbo'
        messages = @(
            @{
                role = 'system'
                content = "你是一名资深PowerShell专家，根据用户需求生成可直接执行的运维脚本。当前环境：$Environment"
            },
            @{
                role = 'user'
                content = $Prompt
            }
        )
        temperature = 0.3
        max_tokens = 1024
    } | ConvertTo-Json -Depth 5

    try {
        $response = Invoke-RestMethod -Uri 'https://api.openai.com/v1/chat/completions' \
            -Method Post \
            -Headers $headers \
            -Body $body

        $scriptBlock = [scriptblock]::Create($response.choices[0].message.content)
        Write-Host "生成脚本验证通过：" -ForegroundColor Green
        $scriptBlock.Invoke()
    }
    catch {
        Write-Error "AI脚本生成失败: $($_.Exception.Message)"
    }
}

function Invoke-AIScriptGeneration {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory=$true)]
        [string]$NaturalLanguageQuery,
        
        [ValidateRange(1,4096)]
        [int]$MaxTokens = 1024
    )

    $apiKey = Read-Host -Prompt '输入OpenAI API密钥' -AsSecureString
    $cred = New-Object System.Management.Automation.PSCredential ('api', $apiKey)

    $body = @{
        model = 'gpt-4-turbo-preview'
        messages = @(
            @{
                role = 'system'
                content = '你是一个PowerShell专家，将自然语言查询转换为可执行的PowerShell脚本。确保代码符合最佳安全实践，包含错误处理，并添加中文注释。'
            },
            @{
                role = 'user'
                content = $NaturalLanguageQuery
            }
        )
        temperature = 0.2
        max_tokens = $MaxTokens
    } | ConvertTo-Json -Depth 5

    try {
        $response = Invoke-RestMethod -Uri 'https://api.openai.com/v1/chat/completions' \
            -Method POST \
            -Headers @{ Authorization = 'Bearer ' + $cred.GetNetworkCredential().Password } \
            -ContentType 'application/json' \
            -Body $body

        $scriptBlock = [scriptblock]::Create($response.choices[0].message.content)
        $transcript = [PSCustomObject]@{
            Timestamp = Get-Date -Format 'yyyy-MM-dd HH:mm:ss'
            OriginalQuery = $NaturalLanguageQuery
            GeneratedScript = $response.choices[0].message.content
            TokenUsage = $response.usage
        }

        $transcript | Export-Clixml -Path "$env:TEMP/AIScriptTranscript_$(Get-Date -Format yyyyMMdd).xml"
        return $scriptBlock
    }
    catch {
        Write-Error "AI脚本生成失败: $_"
    }
}