function Invoke-AIOpsAssistant {
    param(
        [Parameter(Mandatory=$true)]
        [string]$Prompt,
        [int]$MaxTokens = 200
    )

    $apiKey = 'sk-xxxxxxxxxxxx'
    $headers = @{
        'Authorization' = "Bearer $apiKey"
        'Content-Type' = 'application/json'
    }

    $body = @{
        model = 'gpt-3.5-turbo'
        messages = @(
            @{
                role = 'system'
                content = '你是一个PowerShell专家，根据用户需求生成可直接运行的脚本。要求：1) 使用原生命令 2) 添加详细注释 3) 包含错误处理'
            },
            @{
                role = 'user'
                content = $Prompt
            }
        )
        max_tokens = $MaxTokens
    } | ConvertTo-Json -Depth 5

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

        $generatedCode = $response.choices[0].message.content
        $tempScript = [System.IO.Path]::GetTempFileName() + '.ps1'
        $generatedCode | Out-File -FilePath $tempScript
        & $tempScript
    }
    catch {
        Write-Error "AI脚本生成失败：$_"
    }
}

# 配置 OpenAI API 密钥
$openAIKey = 'your-api-key'

function Get-AIScript {
    param(
        [string]$Prompt
    )

    $headers = @{
        'Authorization' = "Bearer $openAIKey"
        'Content-Type' = 'application/json'
    }

    $body = @{
        model = 'gpt-4'
        messages = @(
            @{
                role = 'system'
                content = '你是一个 PowerShell 专家，请生成符合最佳实践的脚本。要求：1. 包含错误处理 2. 支持verbose输出 3. 包含帮助文档'
            },
            @{
                role = 'user'
                content = $Prompt
            }
        )
    } | ConvertTo-Json

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

    $response.choices[0].message.content
}

# 示例：生成 AD 用户创建脚本
$prompt = @"
创建 PowerShell 函数 New-ADUserWithValidation，要求：
1. 验证输入的邮箱格式
2. 检查用户名唯一性
3. 自动生成随机初始密码
4. 支持WhatIf参数
"@

Get-AIScript -Prompt $prompt