@beeuni/agent-tunnel 0.1.1 → 0.1.2

package/dist/agent-cli.js

@@ -4036,6 +4036,10 @@ function verifyMessageSignature(signingKey, payload, nonce, signature) {
 
 // src/agent/agent.ts
 var AGENT_VERSION = agentTunnelVersion();
+function openRelaySocket(url) {
+  const RelaySocket = WebSocket;
+  return new RelaySocket(url, { headers: { "User-Agent": `beeuni-agent-tunnel/${AGENT_VERSION}` } });
+}
 var AUTH_REJECTED_CLOSE_CODES = [4001, 4003];
 var RELAY_FAULT_4001_REASONS = new Set(["auth timeout", "authentication error", "authentication response failed"]);
 function isCredentialRejection(code, reason = "") {
@@ -4137,7 +4141,7 @@ class TunnelAgent {
     this.connectStartedAt = this.now();
     this.setStatus("connecting");
     try {
-      this.ws = new WebSocket(new URL(wsUrl));
+      this.ws = openRelaySocket(new URL(wsUrl).href);
       this.setupWsHandlers();
     } catch (err) {
       log(`${c.red}✗${c.reset}`, `Connection failed`);
@@ -5677,17 +5681,17 @@ function probeCredentials(config, options = {}) {
   return new Promise((resolve3) => {
     let socket;
     try {
-      socket = new WebSocket(new URL(buildTunnelWsUrl(config)));
+      socket = openRelaySocket(new URL(buildTunnelWsUrl(config)).href);
     } catch {
       resolve3("unreachable");
       return;
     }
     const settle = (result) => {
       clearTimeout(timer);
+      resolve3(result);
       try {
         socket.close(1000, "probe complete");
       } catch {}
-      resolve3(result);
     };
     const timer = setTimeout(() => settle("unreachable"), options.timeoutMs ?? DEFAULT_PROBE_TIMEOUT_MS);
     socket.addEventListener("open", () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@beeuni/agent-tunnel",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "private": false,
   "type": "module",
   "description": "Tunnel relay between cloud AI agents and local machines — server relay, local agent, client SDK, JSON-RPC, HMAC signing",