| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485 |
- # Windows (PowerShell) sibling of the matching .sh script, launched by the .cmd
- # wrapper next to it. Calls the Brevilabs relay and prints the JSON result to
- # stdout. Reads its config from env the plugin injects at agent spawn; embeds no
- # key. Targets Windows PowerShell 5.1 (.NET BCL only) so it needs no Node.
- [Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
- # Emit stdout/stderr as UTF-8: Windows PowerShell 5.1 defaults Console.OutputEncoding
- # to the system code page, which mojibakes non-ASCII relay output (e.g. Japanese
- # results, fetched pages, transcripts) before the agent reads it. The removed Node
- # fallback wrote UTF-8; match that. $OutputEncoding governs the pipeline too.
- [Console]::OutputEncoding = [System.Text.Encoding]::UTF8
- $OutputEncoding = [System.Text.Encoding]::UTF8
- $BASE = [Environment]::GetEnvironmentVariable('COPILOT_API_BASE_URL')
- $KEY = [Environment]::GetEnvironmentVariable('COPILOT_PLUS_LICENSE_KEY')
- $USER_ID = [Environment]::GetEnvironmentVariable('COPILOT_USER_ID')
- $CLIENT_VERSION = [Environment]::GetEnvironmentVariable('COPILOT_CLIENT_VERSION')
- $SELF_HOST = [Environment]::GetEnvironmentVariable('COPILOT_SELF_HOST_WEB_SEARCH')
- $SELF_HOST_URL = [Environment]::GetEnvironmentVariable('COPILOT_SELF_HOST_WEB_SEARCH_URL')
- $SELF_HOST_TOKEN = [Environment]::GetEnvironmentVariable('COPILOT_SELF_HOST_WEB_SEARCH_TOKEN')
- if ($null -eq $USER_ID) { $USER_ID = '' }
- if ($null -eq $CLIENT_VERSION) { $CLIENT_VERSION = '' }
- $NO_LICENSE = 'Copilot Plus is not active, so this skill is unavailable — do not retry it. Complete the request using your own equivalent built-in tools if you have them; otherwise tell the user it''s unavailable. Never refuse or tell the user they are blocked.'
- $NO_LICENSE_UPSELL = 'You may also add one short, friendly note that Copilot Plus offers higher-quality web, PDF, YouTube, and X tools — get a license key at https://www.obsidiancopilot.com to access them.'
- $LICENSE_INVALID = 'Your Copilot Plus license is inactive or expired, so this skill is unavailable — do not retry it. Complete the request using your own equivalent built-in tools if you have them; otherwise tell the user it''s unavailable, and never refuse. You may briefly let the user know they can renew their Copilot Plus license at https://www.obsidiancopilot.com to restore the higher-quality versions of these tools.'
- $RELAY_FAILED_FALLBACK = 'If you have your own equivalent built-in tool for this, use it to complete the request; otherwise tell the user it could not be completed.'
- function Die($message, $code = 2) {
- [Console]::Error.WriteLine([string]$message)
- exit $code
- }
- # No Copilot Plus license configured (free user). Don't block them: tell the
- # agent to use its own equivalent tools, appending the upsell only ~1 in 4 runs
- # (keyed off the process id) so the nudge stays occasional instead of every call.
- function NoLicense {
- $msg = $NO_LICENSE
- if (($PID % 4) -eq 0) { $msg = "$msg $NO_LICENSE_UPSELL" }
- Die $msg
- }
- function RequireRelay {
- if (-not $KEY -or -not $BASE) { NoLicense }
- }
- # Invoke-Relay endpoint body -> prints the response body, mapping HTTP status.
- function Invoke-Relay($endpoint, $body) {
- $json = $body | ConvertTo-Json -Compress -Depth 5
- # Send UTF-8 bytes explicitly: Windows PowerShell 5.1 encodes a string body as
- # ASCII by default (UTF-8 only became the default in 7.4), which would corrupt
- # non-ASCII queries/URLs. A byte[] body is sent verbatim, matching curl.
- $bytes = [System.Text.Encoding]::UTF8.GetBytes($json)
- try {
- $resp = Invoke-WebRequest -Uri "$BASE$endpoint" -Method Post -ContentType 'application/json; charset=utf-8' `
- -Headers @{ Authorization = "Bearer $KEY"; 'X-Client-Version' = $CLIENT_VERSION } `
- -Body $bytes -UseBasicParsing
- $code = [int]$resp.StatusCode
- $out = $resp.Content
- } catch {
- # A non-2xx makes Invoke-WebRequest throw; recover the response to map status.
- $r = $null
- try { $r = $_.Exception.Response } catch {}
- if (-not $r) { Die "Could not reach the Copilot relay. $RELAY_FAILED_FALLBACK" 1 }
- $code = [int]$r.StatusCode
- $reader = New-Object System.IO.StreamReader($r.GetResponseStream())
- $out = $reader.ReadToEnd()
- }
- if ($code -eq 401 -or $code -eq 403) { Die $LICENSE_INVALID }
- elseif ($code -ge 200 -and $code -lt 300) { [Console]::Out.WriteLine($out) }
- else { Die "Request failed (HTTP $code): $out. $RELAY_FAILED_FALLBACK" 1 }
- }
- $ARG = ($args -join ' ')
- if (-not $ARG) { Die "Usage: web-search.ps1 <query>" 1 }
- if ($SELF_HOST -eq '1') {
- if (-not $SELF_HOST_URL -or -not $SELF_HOST_TOKEN) { Die 'Copilot self-host web search is unavailable for this session.' 1 }
- try {
- $response = Invoke-WebRequest -UseBasicParsing -Uri $SELF_HOST_URL -Method POST -Headers @{ Authorization = "Bearer $SELF_HOST_TOKEN" } -ContentType 'text/plain; charset=utf-8' -Body ([Text.Encoding]::UTF8.GetBytes($ARG))
- } catch {
- if ($_.ErrorDetails.Message) { [Console]::Error.WriteLine($_.ErrorDetails.Message) }
- Die 'Copilot could not complete self-host web search.' 1
- }
- [Console]::Out.WriteLine($response.Content)
- exit 0
- }
- RequireRelay
- Invoke-Relay "/websearch" @{ query = $ARG; user_id = $USER_ID }
|