chatgpt-websocket 1.0.0

package/bin/llmbackend.js

@@ -0,0 +1,153 @@
+#!/usr/bin/env node
+
+const { WebSocketServer } = require('ws');
+const crypto = require('crypto');
+
+// Parse args: llmbackend token=<token> [port=<port>]
+const args = {};
+for (const arg of process.argv.slice(2)) {
+  const eq = arg.indexOf('=');
+  if (eq !== -1) {
+    args[arg.slice(0, eq)] = arg.slice(eq + 1);
+  }
+}
+
+const token = args.token;
+if (!token) {
+  console.error('Usage: llmbackend token=<your_access_token> [port=8080]');
+  process.exit(1);
+}
+
+const PORT = parseInt(args.port, 10) || 8080;
+
+// Decode JWT to extract chatgpt_account_id
+let accountId;
+try {
+  const payload = JSON.parse(
+    Buffer.from(token.split('.')[1], 'base64url').toString()
+  );
+  accountId = payload['https://api.openai.com/auth']?.chatgpt_account_id;
+} catch (e) {
+  // ignore
+}
+if (!accountId) {
+  console.error('Could not extract chatgpt_account_id from token');
+  process.exit(1);
+}
+
+const INSTRUCTIONS = `You are a coding assistant. You help the user with software engineering tasks including writing code, debugging, and explaining code. Be concise and direct.`;
+
+async function streamChat(message, history, ws) {
+  // Add user message to history
+  history.push({
+    type: 'message',
+    role: 'user',
+    content: [{ type: 'input_text', text: message }]
+  });
+
+  const body = JSON.stringify({
+    model: 'gpt-5',
+    instructions: INSTRUCTIONS,
+    input: history,
+    store: false,
+    stream: true,
+    include: ['reasoning.encrypted_content'],
+    tools: [],
+    tool_choice: 'none'
+  });
+
+  const res = await fetch('https://chatgpt.com/backend-api/codex/responses', {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+      'Authorization': `Bearer ${token}`,
+      'Accept': 'text/event-stream',
+      'OpenAI-Beta': 'responses=experimental',
+      'session_id': crypto.randomUUID(),
+      'originator': 'codex_cli_rs',
+      'chatgpt-account-id': accountId
+    },
+    body
+  });
+
+  if (!res.ok) {
+    const errText = await res.text();
+    ws.send(JSON.stringify({ type: 'error', error: `HTTP ${res.status}: ${errText}` }));
+    return;
+  }
+
+  const reader = res.body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = '';
+  let assistantText = '';
+
+  while (true) {
+    const { done, value } = await reader.read();
+    if (done) break;
+
+    buffer += decoder.decode(value, { stream: true });
+    const lines = buffer.split('\n');
+    buffer = lines.pop();
+
+    for (const line of lines) {
+      if (!line.startsWith('data: ')) continue;
+      const data = line.slice(6).trim();
+      if (data === '[DONE]') {
+        history.push({ type: 'message', role: 'assistant', content: [{ type: 'output_text', text: assistantText }] });
+        ws.send(JSON.stringify({ type: 'done' }));
+        return;
+      }
+
+      try {
+        const event = JSON.parse(data);
+        if (event.type === 'response.output_text.delta' && event.delta) {
+          assistantText += event.delta;
+          ws.send(JSON.stringify({ type: 'chunk', content: event.delta }));
+        }
+        if (event.type === 'response.completed') {
+          history.push({ type: 'message', role: 'assistant', content: [{ type: 'output_text', text: assistantText }] });
+          ws.send(JSON.stringify({ type: 'done' }));
+          return;
+        }
+      } catch (e) {
+        // skip non-JSON lines
+      }
+    }
+  }
+
+  history.push({ type: 'message', role: 'assistant', content: [{ type: 'output_text', text: assistantText }] });
+  ws.send(JSON.stringify({ type: 'done' }));
+}
+
+const wss = new WebSocketServer({ port: PORT });
+
+wss.on('connection', (ws) => {
+  console.log('Client connected');
+  const history = [];
+
+  ws.on('message', async (raw) => {
+    let msg;
+    try {
+      msg = JSON.parse(raw.toString());
+    } catch (e) {
+      ws.send(JSON.stringify({ type: 'error', error: 'Invalid JSON' }));
+      return;
+    }
+
+    if (msg.type !== 'chat' || !msg.message) {
+      ws.send(JSON.stringify({ type: 'error', error: 'Expected {"type":"chat","message":"..."}' }));
+      return;
+    }
+
+    try {
+      await streamChat(msg.message, history, ws);
+    } catch (err) {
+      console.error('Stream error:', err);
+      ws.send(JSON.stringify({ type: 'error', error: err.message }));
+    }
+  });
+
+  ws.on('close', () => console.log('Client disconnected'));
+});
+
+console.log(`llmbackend running on ws://localhost:${PORT}`);

package/package.json

@@ -0,0 +1,13 @@
+{
+  "name": "chatgpt-websocket",
+  "version": "1.0.0",
+  "description": "WebSocket server wrapping ChatGPT backend API for GPT-5 access",
+  "bin": {
+    "chatgpt-websocket": "bin/llmbackend.js"
+  },
+  "keywords": ["chatgpt", "gpt-5", "websocket", "openai", "streaming"],
+  "license": "MIT",
+  "dependencies": {
+    "ws": "^8.18.0"
+  }
+}