amai 0.0.22 → 0.0.23

package/README.md

@@ -67,7 +67,7 @@ amai [command] [options]
 ### Environment Variables
 
 - `SERVER_URL` - WebSocket server URL to connect to (optional, has a default)
-  - Example: `ws://localhost:3000` or `wss://your-server.com`
+  - Example: `ws://localhost:8787` or `wss://your-server.com`
 
 ### Examples
 

package/dist/cli.cjs

@@ -29,8 +29,8 @@ var fs7__default = /*#__PURE__*/_interopDefault(fs7);
 var fs5__default = /*#__PURE__*/_interopDefault(fs5);
 var readline__default = /*#__PURE__*/_interopDefault(readline);
 
-var DEFAULT_SERVER_URL = "wss://bridge.ama.shujan.xyz";
-var CLIENT_ID = "client_01K4Y8A67H544Z6J8A47E5GJ9A";
+var DEFAULT_SERVER_URL = process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
+var CLIENT_ID = "client_01K4Y8A5Q3FYGXD362BJQ6AGYD";
 var AMA_DIR = path6__default.default.join(os3__default.default.homedir(), ".amai");
 var CODE_DIR = path6__default.default.join(AMA_DIR, "code");
 var STORAGE_DIR = path6__default.default.join(AMA_DIR, "storage");
@@ -750,6 +750,7 @@ var Snapshot;
 })(Snapshot || (Snapshot = {}));
 var CLIENT_ID2 = "app_EMoamEEZ73f0CkXaXp7hrann";
 var ISSUER = "https://auth.openai.com";
+var CODEX_API_ENDPOINT = "https://chatgpt.com/backend-api/codex/responses";
 var OAUTH_PORT = 1455;
 var CREDENTIALS_PATH2 = path6__default.default.join(AMA_DIR, "codex-credentials.json");
 var CALLBACK_PATH = "/auth/callback";
@@ -1141,6 +1142,30 @@ async function getCodexStatus() {
     return { authenticated: false };
   }
 }
+async function codexFetch(body) {
+  const { accessToken, accountId } = await getCodexTokens();
+  const response = await fetch(CODEX_API_ENDPOINT, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${accessToken}`,
+      "ChatGPT-Account-Id": accountId,
+      originator: "ama",
+      Accept: "text/event-stream"
+    },
+    body
+  });
+  const text = await response.text();
+  const headers = {};
+  response.headers.forEach((value, key) => {
+    headers[key] = value;
+  });
+  return {
+    status: response.status,
+    headers,
+    body: text
+  };
+}
 async function codexLogout() {
   pendingOAuth = void 0;
   stopOAuthServer();
@@ -1307,6 +1332,16 @@ var rpcHandlers = {
   "daemon:codex_get_tokens": async () => {
     return getCodexTokens();
   },
+  "daemon:codex_fetch": async ({ body }) => {
+    if (typeof body !== "string" || !body.trim()) {
+      const error = {
+        _tag: "ValidationError",
+        message: "body is required"
+      };
+      throw error;
+    }
+    return codexFetch(body);
+  },
   "daemon:codex_status": async () => {
     return getCodexStatus();
   },
@@ -1318,8 +1353,10 @@ var rpcHandlers = {
 var INITIAL_RECONNECT_DELAY = 1e3;
 var MAX_RECONNECT_DELAY = 6e4;
 var BACKOFF_MULTIPLIER = 2;
+var CLI_HEARTBEAT_INTERVAL_MS = 5e3;
 var reconnectTimeout = null;
 var reconnectAttempts = 0;
+var heartbeatInterval = null;
 function getReconnectDelay() {
   const delay = Math.min(
     INITIAL_RECONNECT_DELAY * Math.pow(BACKOFF_MULTIPLIER, reconnectAttempts),
@@ -1347,6 +1384,19 @@ var connectToUserStreams = async (serverUrl) => {
       Authorization: `Bearer ${tokens.access_token}`
     }
   });
+  const clearHeartbeat = () => {
+    if (heartbeatInterval) {
+      clearInterval(heartbeatInterval);
+      heartbeatInterval = null;
+    }
+  };
+  const sendHeartbeat = () => {
+    if (ws.readyState !== WebSocket__default.default.OPEN) return;
+    ws.send(JSON.stringify({
+      _tag: "cli_heartbeat",
+      timestamp: Date.now()
+    }));
+  };
   ws.on("open", () => {
     reconnectAttempts = 0;
     console.log(pc5__default.default.cyan("connected to user streams"));
@@ -1354,6 +1404,9 @@ var connectToUserStreams = async (serverUrl) => {
       clearTimeout(reconnectTimeout);
       reconnectTimeout = null;
     }
+    clearHeartbeat();
+    sendHeartbeat();
+    heartbeatInterval = setInterval(sendHeartbeat, CLI_HEARTBEAT_INTERVAL_MS);
   });
   ws.on("message", async (event) => {
     try {
@@ -1413,6 +1466,7 @@ var connectToUserStreams = async (serverUrl) => {
     }
   });
   ws.on("close", (code, reason) => {
+    clearHeartbeat();
     const delay = getReconnectDelay();
     reconnectAttempts++;
     console.log(pc5__default.default.gray(`user streams disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
@@ -1423,6 +1477,7 @@ var connectToUserStreams = async (serverUrl) => {
     }, delay);
   });
   ws.on("error", (error) => {
+    clearHeartbeat();
     console.error(pc5__default.default.red(`stream error: ${error.message}`));
   });
   return ws;
@@ -1909,7 +1964,7 @@ function getDaemonPid() {
     return null;
   }
 }
-var VERSION = "0.0.22";
+var VERSION = "0.0.23";
 var PROJECT_DIR = process.cwd();
 var LOGO = `
    __ _ _ __ ___   __ _ 

package/dist/cli.js

@@ -16,8 +16,8 @@ import { batchTool, bashTool, apply_patch, read_file, list, globTool, grepTool,
 import { fileURLToPath } from 'url';
 import readline from 'readline';
 
-var DEFAULT_SERVER_URL = "wss://bridge.ama.shujan.xyz";
-var CLIENT_ID = "client_01K4Y8A67H544Z6J8A47E5GJ9A";
+var DEFAULT_SERVER_URL = process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
+var CLIENT_ID = "client_01K4Y8A5Q3FYGXD362BJQ6AGYD";
 var AMA_DIR = path6.join(os3.homedir(), ".amai");
 var CODE_DIR = path6.join(AMA_DIR, "code");
 var STORAGE_DIR = path6.join(AMA_DIR, "storage");
@@ -737,6 +737,7 @@ var Snapshot;
 })(Snapshot || (Snapshot = {}));
 var CLIENT_ID2 = "app_EMoamEEZ73f0CkXaXp7hrann";
 var ISSUER = "https://auth.openai.com";
+var CODEX_API_ENDPOINT = "https://chatgpt.com/backend-api/codex/responses";
 var OAUTH_PORT = 1455;
 var CREDENTIALS_PATH2 = path6.join(AMA_DIR, "codex-credentials.json");
 var CALLBACK_PATH = "/auth/callback";
@@ -1128,6 +1129,30 @@ async function getCodexStatus() {
     return { authenticated: false };
   }
 }
+async function codexFetch(body) {
+  const { accessToken, accountId } = await getCodexTokens();
+  const response = await fetch(CODEX_API_ENDPOINT, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${accessToken}`,
+      "ChatGPT-Account-Id": accountId,
+      originator: "ama",
+      Accept: "text/event-stream"
+    },
+    body
+  });
+  const text = await response.text();
+  const headers = {};
+  response.headers.forEach((value, key) => {
+    headers[key] = value;
+  });
+  return {
+    status: response.status,
+    headers,
+    body: text
+  };
+}
 async function codexLogout() {
   pendingOAuth = void 0;
   stopOAuthServer();
@@ -1294,6 +1319,16 @@ var rpcHandlers = {
   "daemon:codex_get_tokens": async () => {
     return getCodexTokens();
   },
+  "daemon:codex_fetch": async ({ body }) => {
+    if (typeof body !== "string" || !body.trim()) {
+      const error = {
+        _tag: "ValidationError",
+        message: "body is required"
+      };
+      throw error;
+    }
+    return codexFetch(body);
+  },
   "daemon:codex_status": async () => {
     return getCodexStatus();
   },
@@ -1305,8 +1340,10 @@ var rpcHandlers = {
 var INITIAL_RECONNECT_DELAY = 1e3;
 var MAX_RECONNECT_DELAY = 6e4;
 var BACKOFF_MULTIPLIER = 2;
+var CLI_HEARTBEAT_INTERVAL_MS = 5e3;
 var reconnectTimeout = null;
 var reconnectAttempts = 0;
+var heartbeatInterval = null;
 function getReconnectDelay() {
   const delay = Math.min(
     INITIAL_RECONNECT_DELAY * Math.pow(BACKOFF_MULTIPLIER, reconnectAttempts),
@@ -1334,6 +1371,19 @@ var connectToUserStreams = async (serverUrl) => {
       Authorization: `Bearer ${tokens.access_token}`
     }
   });
+  const clearHeartbeat = () => {
+    if (heartbeatInterval) {
+      clearInterval(heartbeatInterval);
+      heartbeatInterval = null;
+    }
+  };
+  const sendHeartbeat = () => {
+    if (ws.readyState !== WebSocket.OPEN) return;
+    ws.send(JSON.stringify({
+      _tag: "cli_heartbeat",
+      timestamp: Date.now()
+    }));
+  };
   ws.on("open", () => {
     reconnectAttempts = 0;
     console.log(pc5.cyan("connected to user streams"));
@@ -1341,6 +1391,9 @@ var connectToUserStreams = async (serverUrl) => {
       clearTimeout(reconnectTimeout);
       reconnectTimeout = null;
     }
+    clearHeartbeat();
+    sendHeartbeat();
+    heartbeatInterval = setInterval(sendHeartbeat, CLI_HEARTBEAT_INTERVAL_MS);
   });
   ws.on("message", async (event) => {
     try {
@@ -1400,6 +1453,7 @@ var connectToUserStreams = async (serverUrl) => {
     }
   });
   ws.on("close", (code, reason) => {
+    clearHeartbeat();
     const delay = getReconnectDelay();
     reconnectAttempts++;
     console.log(pc5.gray(`user streams disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
@@ -1410,6 +1464,7 @@ var connectToUserStreams = async (serverUrl) => {
     }, delay);
   });
   ws.on("error", (error) => {
+    clearHeartbeat();
     console.error(pc5.red(`stream error: ${error.message}`));
   });
   return ws;
@@ -1896,7 +1951,7 @@ function getDaemonPid() {
     return null;
   }
 }
-var VERSION = "0.0.22";
+var VERSION = "0.0.23";
 var PROJECT_DIR = process.cwd();
 var LOGO = `
    __ _ _ __ ___   __ _ 

package/dist/lib/code-server.cjs

@@ -15,6 +15,7 @@ var path2__default = /*#__PURE__*/_interopDefault(path2);
 var pc__default = /*#__PURE__*/_interopDefault(pc);
 var os__default = /*#__PURE__*/_interopDefault(os);
 
+process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
 var AMA_DIR = path2__default.default.join(os__default.default.homedir(), ".amai");
 var CODE_DIR = path2__default.default.join(AMA_DIR, "code");
 var STORAGE_DIR = path2__default.default.join(AMA_DIR, "storage");

package/dist/lib/code-server.js

@@ -6,6 +6,7 @@ import path2 from 'path';
 import pc from 'picocolors';
 import os from 'os';
 
+process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
 var AMA_DIR = path2.join(os.homedir(), ".amai");
 var CODE_DIR = path2.join(AMA_DIR, "code");
 var STORAGE_DIR = path2.join(AMA_DIR, "storage");

package/dist/lib/daemon-entry.cjs

@@ -25,7 +25,7 @@ var pc3__default = /*#__PURE__*/_interopDefault(pc3);
 var fs7__default = /*#__PURE__*/_interopDefault(fs7);
 var fs5__default = /*#__PURE__*/_interopDefault(fs5);
 
-var DEFAULT_SERVER_URL = "wss://bridge.ama.shujan.xyz";
+var DEFAULT_SERVER_URL = process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
 var AMA_DIR = path6__default.default.join(os3__default.default.homedir(), ".amai");
 var CODE_DIR = path6__default.default.join(AMA_DIR, "code");
 var STORAGE_DIR = path6__default.default.join(AMA_DIR, "storage");
@@ -616,6 +616,7 @@ var Snapshot;
 })(Snapshot || (Snapshot = {}));
 var CLIENT_ID2 = "app_EMoamEEZ73f0CkXaXp7hrann";
 var ISSUER = "https://auth.openai.com";
+var CODEX_API_ENDPOINT = "https://chatgpt.com/backend-api/codex/responses";
 var OAUTH_PORT = 1455;
 var CREDENTIALS_PATH2 = path6__default.default.join(AMA_DIR, "codex-credentials.json");
 var CALLBACK_PATH = "/auth/callback";
@@ -1007,6 +1008,30 @@ async function getCodexStatus() {
     return { authenticated: false };
   }
 }
+async function codexFetch(body) {
+  const { accessToken, accountId } = await getCodexTokens();
+  const response = await fetch(CODEX_API_ENDPOINT, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${accessToken}`,
+      "ChatGPT-Account-Id": accountId,
+      originator: "ama",
+      Accept: "text/event-stream"
+    },
+    body
+  });
+  const text = await response.text();
+  const headers = {};
+  response.headers.forEach((value, key) => {
+    headers[key] = value;
+  });
+  return {
+    status: response.status,
+    headers,
+    body: text
+  };
+}
 async function codexLogout() {
   pendingOAuth = void 0;
   stopOAuthServer();
@@ -1173,6 +1198,16 @@ var rpcHandlers = {
   "daemon:codex_get_tokens": async () => {
     return getCodexTokens();
   },
+  "daemon:codex_fetch": async ({ body }) => {
+    if (typeof body !== "string" || !body.trim()) {
+      const error = {
+        _tag: "ValidationError",
+        message: "body is required"
+      };
+      throw error;
+    }
+    return codexFetch(body);
+  },
   "daemon:codex_status": async () => {
     return getCodexStatus();
   },
@@ -1184,8 +1219,10 @@ var rpcHandlers = {
 var INITIAL_RECONNECT_DELAY = 1e3;
 var MAX_RECONNECT_DELAY = 6e4;
 var BACKOFF_MULTIPLIER = 2;
+var CLI_HEARTBEAT_INTERVAL_MS = 5e3;
 var reconnectTimeout = null;
 var reconnectAttempts = 0;
+var heartbeatInterval = null;
 function getReconnectDelay() {
   const delay = Math.min(
     INITIAL_RECONNECT_DELAY * Math.pow(BACKOFF_MULTIPLIER, reconnectAttempts),
@@ -1213,6 +1250,19 @@ var connectToUserStreams = async (serverUrl) => {
       Authorization: `Bearer ${tokens.access_token}`
     }
   });
+  const clearHeartbeat = () => {
+    if (heartbeatInterval) {
+      clearInterval(heartbeatInterval);
+      heartbeatInterval = null;
+    }
+  };
+  const sendHeartbeat = () => {
+    if (ws.readyState !== WebSocket__default.default.OPEN) return;
+    ws.send(JSON.stringify({
+      _tag: "cli_heartbeat",
+      timestamp: Date.now()
+    }));
+  };
   ws.on("open", () => {
     reconnectAttempts = 0;
     console.log(pc3__default.default.cyan("connected to user streams"));
@@ -1220,6 +1270,9 @@ var connectToUserStreams = async (serverUrl) => {
       clearTimeout(reconnectTimeout);
       reconnectTimeout = null;
     }
+    clearHeartbeat();
+    sendHeartbeat();
+    heartbeatInterval = setInterval(sendHeartbeat, CLI_HEARTBEAT_INTERVAL_MS);
   });
   ws.on("message", async (event) => {
     try {
@@ -1279,6 +1332,7 @@ var connectToUserStreams = async (serverUrl) => {
     }
   });
   ws.on("close", (code, reason) => {
+    clearHeartbeat();
     const delay = getReconnectDelay();
     reconnectAttempts++;
     console.log(pc3__default.default.gray(`user streams disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
@@ -1289,6 +1343,7 @@ var connectToUserStreams = async (serverUrl) => {
     }, delay);
   });
   ws.on("error", (error) => {
+    clearHeartbeat();
     console.error(pc3__default.default.red(`stream error: ${error.message}`));
   });
   return ws;

package/dist/lib/daemon-entry.js

@@ -14,7 +14,7 @@ import { z } from 'zod';
 import { createServer } from 'http';
 import { batchTool, bashTool, apply_patch, read_file, list, globTool, grepTool, deleteFile, editFiles } from '@ama/agent';
 
-var DEFAULT_SERVER_URL = "wss://bridge.ama.shujan.xyz";
+var DEFAULT_SERVER_URL = process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
 var AMA_DIR = path6.join(os3.homedir(), ".amai");
 var CODE_DIR = path6.join(AMA_DIR, "code");
 var STORAGE_DIR = path6.join(AMA_DIR, "storage");
@@ -605,6 +605,7 @@ var Snapshot;
 })(Snapshot || (Snapshot = {}));
 var CLIENT_ID2 = "app_EMoamEEZ73f0CkXaXp7hrann";
 var ISSUER = "https://auth.openai.com";
+var CODEX_API_ENDPOINT = "https://chatgpt.com/backend-api/codex/responses";
 var OAUTH_PORT = 1455;
 var CREDENTIALS_PATH2 = path6.join(AMA_DIR, "codex-credentials.json");
 var CALLBACK_PATH = "/auth/callback";
@@ -996,6 +997,30 @@ async function getCodexStatus() {
     return { authenticated: false };
   }
 }
+async function codexFetch(body) {
+  const { accessToken, accountId } = await getCodexTokens();
+  const response = await fetch(CODEX_API_ENDPOINT, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${accessToken}`,
+      "ChatGPT-Account-Id": accountId,
+      originator: "ama",
+      Accept: "text/event-stream"
+    },
+    body
+  });
+  const text = await response.text();
+  const headers = {};
+  response.headers.forEach((value, key) => {
+    headers[key] = value;
+  });
+  return {
+    status: response.status,
+    headers,
+    body: text
+  };
+}
 async function codexLogout() {
   pendingOAuth = void 0;
   stopOAuthServer();
@@ -1162,6 +1187,16 @@ var rpcHandlers = {
   "daemon:codex_get_tokens": async () => {
     return getCodexTokens();
   },
+  "daemon:codex_fetch": async ({ body }) => {
+    if (typeof body !== "string" || !body.trim()) {
+      const error = {
+        _tag: "ValidationError",
+        message: "body is required"
+      };
+      throw error;
+    }
+    return codexFetch(body);
+  },
   "daemon:codex_status": async () => {
     return getCodexStatus();
   },
@@ -1173,8 +1208,10 @@ var rpcHandlers = {
 var INITIAL_RECONNECT_DELAY = 1e3;
 var MAX_RECONNECT_DELAY = 6e4;
 var BACKOFF_MULTIPLIER = 2;
+var CLI_HEARTBEAT_INTERVAL_MS = 5e3;
 var reconnectTimeout = null;
 var reconnectAttempts = 0;
+var heartbeatInterval = null;
 function getReconnectDelay() {
   const delay = Math.min(
     INITIAL_RECONNECT_DELAY * Math.pow(BACKOFF_MULTIPLIER, reconnectAttempts),
@@ -1202,6 +1239,19 @@ var connectToUserStreams = async (serverUrl) => {
       Authorization: `Bearer ${tokens.access_token}`
     }
   });
+  const clearHeartbeat = () => {
+    if (heartbeatInterval) {
+      clearInterval(heartbeatInterval);
+      heartbeatInterval = null;
+    }
+  };
+  const sendHeartbeat = () => {
+    if (ws.readyState !== WebSocket.OPEN) return;
+    ws.send(JSON.stringify({
+      _tag: "cli_heartbeat",
+      timestamp: Date.now()
+    }));
+  };
   ws.on("open", () => {
     reconnectAttempts = 0;
     console.log(pc3.cyan("connected to user streams"));
@@ -1209,6 +1259,9 @@ var connectToUserStreams = async (serverUrl) => {
       clearTimeout(reconnectTimeout);
       reconnectTimeout = null;
     }
+    clearHeartbeat();
+    sendHeartbeat();
+    heartbeatInterval = setInterval(sendHeartbeat, CLI_HEARTBEAT_INTERVAL_MS);
   });
   ws.on("message", async (event) => {
     try {
@@ -1268,6 +1321,7 @@ var connectToUserStreams = async (serverUrl) => {
     }
   });
   ws.on("close", (code, reason) => {
+    clearHeartbeat();
     const delay = getReconnectDelay();
     reconnectAttempts++;
     console.log(pc3.gray(`user streams disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
@@ -1278,6 +1332,7 @@ var connectToUserStreams = async (serverUrl) => {
     }, delay);
   });
   ws.on("error", (error) => {
+    clearHeartbeat();
     console.error(pc3.red(`stream error: ${error.message}`));
   });
   return ws;

package/dist/server.cjs

@@ -25,7 +25,7 @@ var pc3__default = /*#__PURE__*/_interopDefault(pc3);
 var fs4__default = /*#__PURE__*/_interopDefault(fs4);
 var fs5__default = /*#__PURE__*/_interopDefault(fs5);
 
-var DEFAULT_SERVER_URL = "wss://bridge.ama.shujan.xyz";
+var DEFAULT_SERVER_URL = process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
 var AMA_DIR = path6__default.default.join(os3__default.default.homedir(), ".amai");
 path6__default.default.join(AMA_DIR, "code");
 path6__default.default.join(AMA_DIR, "storage");
@@ -616,6 +616,7 @@ var Snapshot;
 })(Snapshot || (Snapshot = {}));
 var CLIENT_ID2 = "app_EMoamEEZ73f0CkXaXp7hrann";
 var ISSUER = "https://auth.openai.com";
+var CODEX_API_ENDPOINT = "https://chatgpt.com/backend-api/codex/responses";
 var OAUTH_PORT = 1455;
 var CREDENTIALS_PATH2 = path6__default.default.join(AMA_DIR, "codex-credentials.json");
 var CALLBACK_PATH = "/auth/callback";
@@ -1007,6 +1008,30 @@ async function getCodexStatus() {
     return { authenticated: false };
   }
 }
+async function codexFetch(body) {
+  const { accessToken, accountId } = await getCodexTokens();
+  const response = await fetch(CODEX_API_ENDPOINT, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${accessToken}`,
+      "ChatGPT-Account-Id": accountId,
+      originator: "ama",
+      Accept: "text/event-stream"
+    },
+    body
+  });
+  const text = await response.text();
+  const headers = {};
+  response.headers.forEach((value, key) => {
+    headers[key] = value;
+  });
+  return {
+    status: response.status,
+    headers,
+    body: text
+  };
+}
 async function codexLogout() {
   pendingOAuth = void 0;
   stopOAuthServer();
@@ -1173,6 +1198,16 @@ var rpcHandlers = {
   "daemon:codex_get_tokens": async () => {
     return getCodexTokens();
   },
+  "daemon:codex_fetch": async ({ body }) => {
+    if (typeof body !== "string" || !body.trim()) {
+      const error = {
+        _tag: "ValidationError",
+        message: "body is required"
+      };
+      throw error;
+    }
+    return codexFetch(body);
+  },
   "daemon:codex_status": async () => {
     return getCodexStatus();
   },
@@ -1184,8 +1219,10 @@ var rpcHandlers = {
 var INITIAL_RECONNECT_DELAY = 1e3;
 var MAX_RECONNECT_DELAY = 6e4;
 var BACKOFF_MULTIPLIER = 2;
+var CLI_HEARTBEAT_INTERVAL_MS = 5e3;
 var reconnectTimeout = null;
 var reconnectAttempts = 0;
+var heartbeatInterval = null;
 function getReconnectDelay() {
   const delay = Math.min(
     INITIAL_RECONNECT_DELAY * Math.pow(BACKOFF_MULTIPLIER, reconnectAttempts),
@@ -1213,6 +1250,19 @@ var connectToUserStreams = async (serverUrl) => {
       Authorization: `Bearer ${tokens.access_token}`
     }
   });
+  const clearHeartbeat = () => {
+    if (heartbeatInterval) {
+      clearInterval(heartbeatInterval);
+      heartbeatInterval = null;
+    }
+  };
+  const sendHeartbeat = () => {
+    if (ws.readyState !== WebSocket2__default.default.OPEN) return;
+    ws.send(JSON.stringify({
+      _tag: "cli_heartbeat",
+      timestamp: Date.now()
+    }));
+  };
   ws.on("open", () => {
     reconnectAttempts = 0;
     console.log(pc3__default.default.cyan("connected to user streams"));
@@ -1220,6 +1270,9 @@ var connectToUserStreams = async (serverUrl) => {
       clearTimeout(reconnectTimeout);
       reconnectTimeout = null;
     }
+    clearHeartbeat();
+    sendHeartbeat();
+    heartbeatInterval = setInterval(sendHeartbeat, CLI_HEARTBEAT_INTERVAL_MS);
   });
   ws.on("message", async (event) => {
     try {
@@ -1279,6 +1332,7 @@ var connectToUserStreams = async (serverUrl) => {
     }
   });
   ws.on("close", (code, reason) => {
+    clearHeartbeat();
     const delay = getReconnectDelay();
     reconnectAttempts++;
     console.log(pc3__default.default.gray(`user streams disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
@@ -1289,6 +1343,7 @@ var connectToUserStreams = async (serverUrl) => {
     }, delay);
   });
   ws.on("error", (error) => {
+    clearHeartbeat();
     console.error(pc3__default.default.red(`stream error: ${error.message}`));
   });
   return ws;

package/dist/server.js

@@ -14,7 +14,7 @@ import { z } from 'zod';
 import { createServer } from 'http';
 import { batchTool, bashTool, apply_patch, read_file, list, globTool, grepTool, deleteFile, editFiles } from '@ama/agent';
 
-var DEFAULT_SERVER_URL = "wss://bridge.ama.shujan.xyz";
+var DEFAULT_SERVER_URL = process.env.SERVER_URL || "wss://ama-bridge.shujanshaikh05.workers.dev";
 var AMA_DIR = path6.join(os3.homedir(), ".amai");
 path6.join(AMA_DIR, "code");
 path6.join(AMA_DIR, "storage");
@@ -605,6 +605,7 @@ var Snapshot;
 })(Snapshot || (Snapshot = {}));
 var CLIENT_ID2 = "app_EMoamEEZ73f0CkXaXp7hrann";
 var ISSUER = "https://auth.openai.com";
+var CODEX_API_ENDPOINT = "https://chatgpt.com/backend-api/codex/responses";
 var OAUTH_PORT = 1455;
 var CREDENTIALS_PATH2 = path6.join(AMA_DIR, "codex-credentials.json");
 var CALLBACK_PATH = "/auth/callback";
@@ -996,6 +997,30 @@ async function getCodexStatus() {
     return { authenticated: false };
   }
 }
+async function codexFetch(body) {
+  const { accessToken, accountId } = await getCodexTokens();
+  const response = await fetch(CODEX_API_ENDPOINT, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${accessToken}`,
+      "ChatGPT-Account-Id": accountId,
+      originator: "ama",
+      Accept: "text/event-stream"
+    },
+    body
+  });
+  const text = await response.text();
+  const headers = {};
+  response.headers.forEach((value, key) => {
+    headers[key] = value;
+  });
+  return {
+    status: response.status,
+    headers,
+    body: text
+  };
+}
 async function codexLogout() {
   pendingOAuth = void 0;
   stopOAuthServer();
@@ -1162,6 +1187,16 @@ var rpcHandlers = {
   "daemon:codex_get_tokens": async () => {
     return getCodexTokens();
   },
+  "daemon:codex_fetch": async ({ body }) => {
+    if (typeof body !== "string" || !body.trim()) {
+      const error = {
+        _tag: "ValidationError",
+        message: "body is required"
+      };
+      throw error;
+    }
+    return codexFetch(body);
+  },
   "daemon:codex_status": async () => {
     return getCodexStatus();
   },
@@ -1173,8 +1208,10 @@ var rpcHandlers = {
 var INITIAL_RECONNECT_DELAY = 1e3;
 var MAX_RECONNECT_DELAY = 6e4;
 var BACKOFF_MULTIPLIER = 2;
+var CLI_HEARTBEAT_INTERVAL_MS = 5e3;
 var reconnectTimeout = null;
 var reconnectAttempts = 0;
+var heartbeatInterval = null;
 function getReconnectDelay() {
   const delay = Math.min(
     INITIAL_RECONNECT_DELAY * Math.pow(BACKOFF_MULTIPLIER, reconnectAttempts),
@@ -1202,6 +1239,19 @@ var connectToUserStreams = async (serverUrl) => {
       Authorization: `Bearer ${tokens.access_token}`
     }
   });
+  const clearHeartbeat = () => {
+    if (heartbeatInterval) {
+      clearInterval(heartbeatInterval);
+      heartbeatInterval = null;
+    }
+  };
+  const sendHeartbeat = () => {
+    if (ws.readyState !== WebSocket2.OPEN) return;
+    ws.send(JSON.stringify({
+      _tag: "cli_heartbeat",
+      timestamp: Date.now()
+    }));
+  };
   ws.on("open", () => {
     reconnectAttempts = 0;
     console.log(pc3.cyan("connected to user streams"));
@@ -1209,6 +1259,9 @@ var connectToUserStreams = async (serverUrl) => {
       clearTimeout(reconnectTimeout);
       reconnectTimeout = null;
     }
+    clearHeartbeat();
+    sendHeartbeat();
+    heartbeatInterval = setInterval(sendHeartbeat, CLI_HEARTBEAT_INTERVAL_MS);
   });
   ws.on("message", async (event) => {
     try {
@@ -1268,6 +1321,7 @@ var connectToUserStreams = async (serverUrl) => {
     }
   });
   ws.on("close", (code, reason) => {
+    clearHeartbeat();
     const delay = getReconnectDelay();
     reconnectAttempts++;
     console.log(pc3.gray(`user streams disconnected, reconnecting in ${Math.round(delay / 1e3)}s...`));
@@ -1278,6 +1332,7 @@ var connectToUserStreams = async (serverUrl) => {
     }, delay);
   });
   ws.on("error", (error) => {
+    clearHeartbeat();
     console.error(pc3.red(`stream error: ${error.message}`));
   });
   return ws;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "amai",
   "type": "module",
-  "version": "0.0.22",
+  "version": "0.0.23",
   "description": "ama cli",
   "keywords": [
     "ama",