agent-foundation/scripts/local_media_worker.ps1
2026-06-18 13:50:03 +08:00

72 lines
1.6 KiB
PowerShell

param(
[string]$BaseUrl = "http://localhost:8880",
[string]$WorkerId = "local-worker-01",
[string]$WorkerToken = "local-worker-dev-token"
)
$headers = @{
"X-Worker-Token" = $WorkerToken
}
$pollBody = @{
workerId = $WorkerId
platforms = @("douyin", "xhs", "wxvideo", "kuaishou", "bilibili", "weibo")
maxTasks = 5
} | ConvertTo-Json
$tasks = Invoke-RestMethod `
-Uri "$BaseUrl/v1/media/workers/poll" `
-Method Post `
-Headers $headers `
-ContentType "application/json" `
-Body $pollBody
foreach ($task in $tasks) {
if ($task.taskType -eq "SYNC_COMMENTS") {
$result = @{
status = "SUCCESS"
result = @{
comments = @(
@{
platform = $task.platform
accountId = $task.accountId
userName = "测试用户"
content = "这个设备多少钱?"
likedCount = 12
intent = "sales_lead"
}
)
}
}
} elseif ($task.taskType -eq "SYNC_METRICS") {
$result = @{
status = "SUCCESS"
result = @{
metrics = @{
views = 12000
likes = 860
comments = 56
collects = 128
shares = 43
}
}
}
} else {
$result = @{
status = "SUCCESS"
result = @{
message = "local worker mock completed"
}
}
}
Invoke-RestMethod `
-Uri "$BaseUrl/v1/media/workers/tasks/$($task.id)/result" `
-Method Post `
-Headers $headers `
-ContentType "application/json" `
-Body ($result | ConvertTo-Json -Depth 10) | Out-Null
Write-Host "completed $($task.id) $($task.taskType) $($task.platform)"
}