wyxrouter 0.4.71

package/open-sse/executors/grok-web.js

@@ -0,0 +1,345 @@
+import { BaseExecutor } from "./base.js";
+import { PROVIDERS } from "../config/providers.js";
+
+const GROK_CHAT_API = PROVIDERS["grok-web"].baseUrl;
+const GROK_USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36";
+
+const MODEL_MAP = {
+  "grok-3": { grokModel: "grok-3", modelMode: "MODEL_MODE_GROK_3", isThinking: false },
+  "grok-3-mini": { grokModel: "grok-3", modelMode: "MODEL_MODE_GROK_3_MINI_THINKING", isThinking: true },
+  "grok-3-thinking": { grokModel: "grok-3", modelMode: "MODEL_MODE_GROK_3_THINKING", isThinking: true },
+  "grok-4": { grokModel: "grok-4", modelMode: "MODEL_MODE_GROK_4", isThinking: false },
+  "grok-4-mini": { grokModel: "grok-4-mini", modelMode: "MODEL_MODE_GROK_4_MINI_THINKING", isThinking: true },
+  "grok-4-thinking": { grokModel: "grok-4", modelMode: "MODEL_MODE_GROK_4_THINKING", isThinking: true },
+  "grok-4-heavy": { grokModel: "grok-4", modelMode: "MODEL_MODE_HEAVY", isThinking: true },
+  "grok-4.1-mini": { grokModel: "grok-4-1-thinking-1129", modelMode: "MODEL_MODE_GROK_4_1_MINI_THINKING", isThinking: true },
+  "grok-4.1-fast": { grokModel: "grok-4-1-thinking-1129", modelMode: "MODEL_MODE_FAST", isThinking: false },
+  "grok-4.1-expert": { grokModel: "grok-4-1-thinking-1129", modelMode: "MODEL_MODE_EXPERT", isThinking: true },
+  "grok-4.1-thinking": { grokModel: "grok-4-1-thinking-1129", modelMode: "MODEL_MODE_GROK_4_1_THINKING", isThinking: true },
+  "grok-4.2": { grokModel: "grok-420", modelMode: "MODEL_MODE_GROK_420", isThinking: false },
+  "grok-4.20": { grokModel: "grok-420", modelMode: "MODEL_MODE_GROK_420", isThinking: false },
+  "grok-4.20-beta": { grokModel: "grok-420", modelMode: "MODEL_MODE_GROK_420", isThinking: false },
+};
+
+function randomString(length, alphanumeric = false) {
+  const chars = alphanumeric ? "abcdefghijklmnopqrstuvwxyz0123456789" : "abcdefghijklmnopqrstuvwxyz";
+  let result = "";
+  for (let i = 0; i < length; i++) result += chars[Math.floor(Math.random() * chars.length)];
+  return result;
+}
+
+function generateStatsigId() {
+  const msg = Math.random() < 0.5
+    ? `e:TypeError: Cannot read properties of null (reading 'children["${randomString(5, true)}"]')`
+    : `e:TypeError: Cannot read properties of undefined (reading '${randomString(10)}')`;
+  return btoa(msg);
+}
+
+function randomHex(bytes) {
+  const arr = new Uint8Array(bytes);
+  crypto.getRandomValues(arr);
+  return Array.from(arr, (b) => b.toString(16).padStart(2, "0")).join("");
+}
+
+function parseOpenAIMessages(messages) {
+  const extracted = [];
+  for (const msg of messages) {
+    let role = String(msg.role || "user");
+    if (role === "developer") role = "system";
+    let content = "";
+    if (typeof msg.content === "string") {
+      content = msg.content;
+    } else if (Array.isArray(msg.content)) {
+      content = msg.content.filter((c) => c.type === "text").map((c) => String(c.text || "")).join(" ");
+    }
+    if (!content.trim()) continue;
+    extracted.push({ role, text: content });
+  }
+
+  let lastUserIdx = -1;
+  for (let i = extracted.length - 1; i >= 0; i--) {
+    if (extracted[i].role === "user") { lastUserIdx = i; break; }
+  }
+
+  const parts = [];
+  for (let i = 0; i < extracted.length; i++) {
+    const { role, text } = extracted[i];
+    parts.push(i === lastUserIdx ? text : `${role}: ${text}`);
+  }
+  return parts.join("\n\n");
+}
+
+async function* readGrokNdjsonEvents(body, signal) {
+  const reader = body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  try {
+    while (true) {
+      if (signal?.aborted) return;
+      const { value, done } = await reader.read();
+      if (done) break;
+      buffer += decoder.decode(value, { stream: true });
+      while (true) {
+        const idx = buffer.indexOf("\n");
+        if (idx < 0) break;
+        const line = buffer.slice(0, idx).trim();
+        buffer = buffer.slice(idx + 1);
+        if (!line) continue;
+        try { yield JSON.parse(line); } catch { /* skip */ }
+      }
+    }
+    buffer += decoder.decode();
+    const remaining = buffer.trim();
+    if (remaining) {
+      try { yield JSON.parse(remaining); } catch { /* skip */ }
+    }
+  } finally {
+    reader.releaseLock();
+  }
+}
+
+async function* extractContent(eventStream, isThinkingModel, signal) {
+  let fingerprint = "";
+  let responseId = "";
+  let thinkOpened = false;
+
+  for await (const event of readGrokNdjsonEvents(eventStream, signal)) {
+    if (event.error) {
+      yield { error: event.error.message || `Grok error: ${event.error.code}`, done: true };
+      return;
+    }
+    const resp = event.result?.response;
+    if (!resp) continue;
+
+    if (resp.llmInfo?.modelHash && !fingerprint) fingerprint = resp.llmInfo.modelHash;
+    if (resp.responseId) responseId = resp.responseId;
+
+    if (resp.modelResponse) {
+      const mr = resp.modelResponse;
+      if (thinkOpened && isThinkingModel) {
+        if (mr.message) yield { thinking: mr.message };
+        thinkOpened = false;
+      }
+      if (mr.message) yield { fullMessage: mr.message, fingerprint, responseId };
+      if (mr.metadata?.llm_info?.modelHash) fingerprint = mr.metadata.llm_info.modelHash;
+      continue;
+    }
+
+    if (resp.token != null) yield { delta: resp.token, fingerprint, responseId };
+  }
+  yield { done: true, fingerprint, responseId };
+}
+
+function sseChunk(data) {
+  return `data: ${JSON.stringify(data)}\n\n`;
+}
+
+function buildStreamingResponse(eventStream, model, cid, created, isThinkingModel, signal) {
+  const encoder = new TextEncoder();
+  return new ReadableStream({
+    async start(controller) {
+      try {
+        controller.enqueue(encoder.encode(sseChunk({
+          id: cid, object: "chat.completion.chunk", created, model, system_fingerprint: null,
+          choices: [{ index: 0, delta: { role: "assistant" }, finish_reason: null, logprobs: null }],
+        })));
+
+        let fp = "";
+        for await (const chunk of extractContent(eventStream, isThinkingModel, signal)) {
+          if (chunk.fingerprint) fp = chunk.fingerprint;
+
+          if (chunk.error) {
+            controller.enqueue(encoder.encode(sseChunk({
+              id: cid, object: "chat.completion.chunk", created, model, system_fingerprint: fp || null,
+              choices: [{ index: 0, delta: { content: `[Error: ${chunk.error}]` }, finish_reason: null, logprobs: null }],
+            })));
+            break;
+          }
+          if (chunk.thinking) {
+            controller.enqueue(encoder.encode(sseChunk({
+              id: cid, object: "chat.completion.chunk", created, model, system_fingerprint: fp || null,
+              choices: [{ index: 0, delta: { reasoning_content: chunk.thinking }, finish_reason: null, logprobs: null }],
+            })));
+            continue;
+          }
+          if (chunk.done) break;
+          if (chunk.delta) {
+            controller.enqueue(encoder.encode(sseChunk({
+              id: cid, object: "chat.completion.chunk", created, model, system_fingerprint: fp || null,
+              choices: [{ index: 0, delta: { content: chunk.delta }, finish_reason: null, logprobs: null }],
+            })));
+          }
+        }
+
+        controller.enqueue(encoder.encode(sseChunk({
+          id: cid, object: "chat.completion.chunk", created, model, system_fingerprint: fp || null,
+          choices: [{ index: 0, delta: {}, finish_reason: "stop", logprobs: null }],
+        })));
+        controller.enqueue(encoder.encode("data: [DONE]\n\n"));
+      } catch (err) {
+        controller.enqueue(encoder.encode(sseChunk({
+          id: cid, object: "chat.completion.chunk", created, model, system_fingerprint: null,
+          choices: [{ index: 0, delta: { content: `[Stream error: ${err.message || String(err)}]` }, finish_reason: "stop", logprobs: null }],
+        })));
+        controller.enqueue(encoder.encode("data: [DONE]\n\n"));
+      } finally {
+        controller.close();
+      }
+    },
+  });
+}
+
+async function buildNonStreamingResponse(eventStream, model, cid, created, isThinkingModel, signal) {
+  let fullContent = "";
+  let fingerprint = "";
+  const thinkingParts = [];
+
+  for await (const chunk of extractContent(eventStream, isThinkingModel, signal)) {
+    if (chunk.fingerprint) fingerprint = chunk.fingerprint;
+    if (chunk.error) {
+      return new Response(JSON.stringify({
+        error: { message: chunk.error, type: "upstream_error", code: "GROK_ERROR" },
+      }), { status: 502, headers: { "Content-Type": "application/json" } });
+    }
+    if (chunk.thinking) { thinkingParts.push(chunk.thinking); continue; }
+    if (chunk.done) break;
+    if (chunk.fullMessage) fullContent = chunk.fullMessage;
+    else if (chunk.delta) fullContent += chunk.delta;
+  }
+
+  const msg = { role: "assistant", content: fullContent };
+  if (thinkingParts.length > 0) msg.reasoning_content = thinkingParts.join("\n");
+
+  const promptTokens = Math.ceil(fullContent.length / 4);
+  const completionTokens = Math.ceil(fullContent.length / 4);
+
+  return new Response(JSON.stringify({
+    id: cid, object: "chat.completion", created, model, system_fingerprint: fingerprint || null,
+    choices: [{ index: 0, message: msg, finish_reason: "stop", logprobs: null }],
+    usage: { prompt_tokens: promptTokens, completion_tokens: completionTokens, total_tokens: promptTokens + completionTokens },
+  }), { status: 200, headers: { "Content-Type": "application/json" } });
+}
+
+export class GrokWebExecutor extends BaseExecutor {
+  constructor() {
+    super("grok-web", PROVIDERS["grok-web"]);
+  }
+
+  async execute({ model, body, stream, credentials, signal, log }) {
+    const messages = body?.messages;
+    if (!messages || !Array.isArray(messages) || messages.length === 0) {
+      const errResp = new Response(JSON.stringify({
+        error: { message: "Missing or empty messages array", type: "invalid_request" },
+      }), { status: 400, headers: { "Content-Type": "application/json" } });
+      return { response: errResp, url: GROK_CHAT_API, headers: {}, transformedBody: body };
+    }
+
+    const modelInfo = MODEL_MAP[model];
+    if (!modelInfo) log?.info?.("GROK-WEB", `Unmapped model ${model}, defaulting to grok-4.1-fast`);
+    const { grokModel, modelMode, isThinking } = modelInfo || MODEL_MAP["grok-4.1-fast"];
+
+    const message = parseOpenAIMessages(messages);
+    if (!message.trim()) {
+      const errResp = new Response(JSON.stringify({
+        error: { message: "Empty query after processing", type: "invalid_request" },
+      }), { status: 400, headers: { "Content-Type": "application/json" } });
+      return { response: errResp, url: GROK_CHAT_API, headers: {}, transformedBody: body };
+    }
+
+    const grokPayload = {
+      temporary: true, modelName: grokModel, modelMode, message,
+      fileAttachments: [], imageAttachments: [],
+      disableSearch: false, enableImageGeneration: false, returnImageBytes: false,
+      returnRawGrokInXaiRequest: false, enableImageStreaming: false, imageGenerationCount: 0,
+      forceConcise: false, toolOverrides: {}, enableSideBySide: true, sendFinalMetadata: true,
+      isReasoning: false, disableTextFollowUps: false, disableMemory: true,
+      forceSideBySide: false, isAsyncChat: false, disableSelfHarmShortCircuit: false,
+      deviceEnvInfo: {
+        darkModeEnabled: false, devicePixelRatio: 2,
+        screenWidth: 2056, screenHeight: 1329, viewportWidth: 2056, viewportHeight: 1083,
+      },
+    };
+
+    const traceId = randomHex(16);
+    const spanId = randomHex(8);
+    const headers = {
+      Accept: "*/*",
+      "Accept-Encoding": "gzip, deflate, br, zstd",
+      "Accept-Language": "en-US,en;q=0.9",
+      Baggage: "sentry-environment=production,sentry-release=d6add6fb0460641fd482d767a335ef72b9b6abb8,sentry-public_key=b311e0f2690c81f25e2c4cf6d4f7ce1c",
+      "Cache-Control": "no-cache",
+      "Content-Type": "application/json",
+      Origin: "https://grok.com",
+      Pragma: "no-cache",
+      Referer: "https://grok.com/",
+      "Sec-Ch-Ua": '"Google Chrome";v="136", "Chromium";v="136", "Not(A:Brand";v="24"',
+      "Sec-Ch-Ua-Mobile": "?0",
+      "Sec-Ch-Ua-Platform": '"macOS"',
+      "Sec-Fetch-Dest": "empty",
+      "Sec-Fetch-Mode": "cors",
+      "Sec-Fetch-Site": "same-origin",
+      "User-Agent": GROK_USER_AGENT,
+      "x-statsig-id": generateStatsigId(),
+      "x-xai-request-id": crypto.randomUUID(),
+      traceparent: `00-${traceId}-${spanId}-00`,
+    };
+
+    // Strip "sso=" prefix if user pasted it
+    if (credentials.apiKey) {
+      let token = credentials.apiKey;
+      if (token.startsWith("sso=")) token = token.slice(4);
+      headers["Cookie"] = `sso=${token}`;
+    }
+
+    log?.info?.("GROK-WEB", `Query to ${model} (grok=${grokModel}, mode=${modelMode}), len=${message.length}`);
+
+    let response;
+    try {
+      response = await fetch(GROK_CHAT_API, {
+        method: "POST", headers, body: JSON.stringify(grokPayload), signal,
+      });
+    } catch (err) {
+      log?.error?.("GROK-WEB", `Fetch failed: ${err.message || String(err)}`);
+      const errResp = new Response(JSON.stringify({
+        error: { message: `Grok connection failed: ${err.message || String(err)}`, type: "upstream_error" },
+      }), { status: 502, headers: { "Content-Type": "application/json" } });
+      return { response: errResp, url: GROK_CHAT_API, headers, transformedBody: grokPayload };
+    }
+
+    if (!response.ok) {
+      const status = response.status;
+      let errMsg = `Grok returned HTTP ${status}`;
+      if (status === 401 || status === 403) errMsg = "Grok auth failed — SSO cookie may be expired. Re-paste your sso cookie value from grok.com.";
+      else if (status === 429) errMsg = "Grok rate limited. Wait a moment and retry, or rotate cookies.";
+      log?.warn?.("GROK-WEB", errMsg);
+      const errResp = new Response(JSON.stringify({
+        error: { message: errMsg, type: "upstream_error", code: `HTTP_${status}` },
+      }), { status, headers: { "Content-Type": "application/json" } });
+      return { response: errResp, url: GROK_CHAT_API, headers, transformedBody: grokPayload };
+    }
+
+    if (!response.body) {
+      const errResp = new Response(JSON.stringify({
+        error: { message: "Grok returned empty response body", type: "upstream_error" },
+      }), { status: 502, headers: { "Content-Type": "application/json" } });
+      return { response: errResp, url: GROK_CHAT_API, headers, transformedBody: grokPayload };
+    }
+
+    const cid = `chatcmpl-grok-${crypto.randomUUID().slice(0, 12)}`;
+    const created = Math.floor(Date.now() / 1000);
+
+    let finalResponse;
+    if (stream) {
+      const sseStream = buildStreamingResponse(response.body, model, cid, created, isThinking, signal);
+      finalResponse = new Response(sseStream, {
+        status: 200,
+        headers: { "Content-Type": "text/event-stream", "Cache-Control": "no-cache", "X-Accel-Buffering": "no" },
+      });
+    } else {
+      finalResponse = await buildNonStreamingResponse(response.body, model, cid, created, isThinking, signal);
+    }
+    return { response: finalResponse, url: GROK_CHAT_API, headers, transformedBody: grokPayload };
+  }
+}
+
+export default GrokWebExecutor;

package/open-sse/executors/iflow.js

@@ -0,0 +1,108 @@
+import crypto from "crypto";
+import { BaseExecutor } from "./base.js";
+import { PROVIDERS } from "../config/providers.js";
+
+/**
+ * IFlowExecutor - Executor for iFlow API with HMAC-SHA256 signature
+ */
+export class IFlowExecutor extends BaseExecutor {
+  constructor() {
+    super("iflow", PROVIDERS.iflow);
+  }
+
+  /**
+   * Generate UUID v4
+   * @returns {string} UUID v4 string
+   */
+  generateUUID() {
+    return crypto.randomUUID();
+  }
+
+  /**
+   * Create iFlow signature using HMAC-SHA256
+   * @param {string} userAgent - User agent string
+   * @param {string} sessionID - Session ID
+   * @param {number} timestamp - Unix timestamp in milliseconds
+   * @param {string} apiKey - API key for signing
+   * @returns {string} Hex-encoded signature
+   */
+  createIFlowSignature(userAgent, sessionID, timestamp, apiKey) {
+    if (!apiKey) return "";
+    const payload = `${userAgent}:${sessionID}:${timestamp}`;
+    const hmac = crypto.createHmac("sha256", apiKey);
+    hmac.update(payload);
+    return hmac.digest("hex");
+  }
+
+  /**
+   * Build headers with iFlow-specific signature
+   * @param {object} credentials - Provider credentials
+   * @param {boolean} stream - Whether streaming is enabled
+   * @returns {object} Headers object
+   */
+  buildHeaders(credentials, stream = true) {
+    // Generate session ID and timestamp
+    const sessionID = `session-${this.generateUUID()}`;
+    const timestamp = Date.now();
+
+    // Get user agent from config
+    const userAgent = this.config.headers["User-Agent"] || "iFlow-Cli";
+
+    // Get API key (prefer apiKey, fallback to accessToken)
+    const apiKey = credentials.apiKey || credentials.accessToken || "";
+
+    // Create signature
+    const signature = this.createIFlowSignature(userAgent, sessionID, timestamp, apiKey);
+
+    // Build headers
+    const headers = {
+      "Content-Type": "application/json",
+      ...this.config.headers,
+      "session-id": sessionID,
+      "x-iflow-timestamp": timestamp.toString(),
+      "x-iflow-signature": signature
+    };
+
+    // Add authorization
+    if (credentials.apiKey) {
+      headers["Authorization"] = `Bearer ${credentials.apiKey}`;
+    }
+
+    // Add streaming header
+    if (stream) {
+      headers["Accept"] = "text/event-stream";
+    }
+
+    return headers;
+  }
+
+  /**
+   * Build URL for iFlow API
+   * @param {string} model - Model name
+   * @param {boolean} stream - Whether streaming is enabled
+   * @param {number} urlIndex - URL index for fallback
+   * @param {object} credentials - Provider credentials
+   * @returns {string} API URL
+   */
+  buildUrl(model, stream, urlIndex = 0, credentials = null) {
+    return this.config.baseUrl;
+  }
+
+  /**
+   * Transform request body - inject stream_options for usage data
+   * @param {string} model - Model name
+   * @param {object} body - Request body
+   * @param {boolean} stream - Whether streaming is enabled
+   * @param {object} credentials - Provider credentials
+   * @returns {object} Transformed body
+   */
+  transformRequest(model, body, stream, credentials) {
+    // Inject stream_options for streaming requests to get usage data
+    if (stream && body.messages && !body.stream_options) {
+      body.stream_options = { include_usage: true };
+    }
+    return body;
+  }
+}
+
+export default IFlowExecutor;

package/open-sse/executors/index.js

@@ -0,0 +1,75 @@
+import { AntigravityExecutor } from "./antigravity.js";
+import { AzureExecutor } from "./azure.js";
+import { GeminiCLIExecutor } from "./gemini-cli.js";
+import { GithubExecutor } from "./github.js";
+import { IFlowExecutor } from "./iflow.js";
+import { QoderExecutor } from "./qoder.js";
+import { KiroExecutor } from "./kiro.js";
+import { CodexExecutor } from "./codex.js";
+import { CursorExecutor } from "./cursor.js";
+import { VertexExecutor } from "./vertex.js";
+import { QwenExecutor } from "./qwen.js";
+import { OpenCodeExecutor } from "./opencode.js";
+import { OpenCodeGoExecutor } from "./opencode-go.js";
+import { GrokWebExecutor } from "./grok-web.js";
+import { PerplexityWebExecutor } from "./perplexity-web.js";
+import { OllamaLocalExecutor } from "./ollama-local.js";
+import { CommandCodeExecutor } from "./commandcode.js";
+import { XiaomiTokenplanExecutor } from "./xiaomi-tokenplan.js";
+import { DefaultExecutor } from "./default.js";
+
+const executors = {
+  antigravity: new AntigravityExecutor(),
+  azure: new AzureExecutor(),
+  "gemini-cli": new GeminiCLIExecutor(),
+  github: new GithubExecutor(),
+  iflow: new IFlowExecutor(),
+  qoder: new QoderExecutor(),
+  kiro: new KiroExecutor(),
+  codex: new CodexExecutor(),
+  cursor: new CursorExecutor(),
+  cu: new CursorExecutor(), // Alias for cursor
+  vertex: new VertexExecutor("vertex"),
+  "vertex-partner": new VertexExecutor("vertex-partner"),
+  qwen: new QwenExecutor(),
+  opencode: new OpenCodeExecutor(),
+  "opencode-go": new OpenCodeGoExecutor(),
+  "grok-web": new GrokWebExecutor(),
+  "perplexity-web": new PerplexityWebExecutor(),
+  "ollama-local": new OllamaLocalExecutor(),
+  commandcode: new CommandCodeExecutor(),
+  "xiaomi-tokenplan": new XiaomiTokenplanExecutor(),
+};
+
+const defaultCache = new Map();
+
+export function getExecutor(provider) {
+  if (executors[provider]) return executors[provider];
+  if (!defaultCache.has(provider)) defaultCache.set(provider, new DefaultExecutor(provider));
+  return defaultCache.get(provider);
+}
+
+export function hasSpecializedExecutor(provider) {
+  return !!executors[provider];
+}
+
+export { BaseExecutor } from "./base.js";
+export { AntigravityExecutor } from "./antigravity.js";
+export { AzureExecutor } from "./azure.js";
+export { GeminiCLIExecutor } from "./gemini-cli.js";
+export { GithubExecutor } from "./github.js";
+export { IFlowExecutor } from "./iflow.js";
+export { QoderExecutor } from "./qoder.js";
+export { KiroExecutor } from "./kiro.js";
+export { CodexExecutor } from "./codex.js";
+export { CursorExecutor } from "./cursor.js";
+export { VertexExecutor } from "./vertex.js";
+export { DefaultExecutor } from "./default.js";
+export { QwenExecutor } from "./qwen.js";
+export { OpenCodeExecutor } from "./opencode.js";
+export { OpenCodeGoExecutor } from "./opencode-go.js";
+export { GrokWebExecutor } from "./grok-web.js";
+export { PerplexityWebExecutor } from "./perplexity-web.js";
+export { OllamaLocalExecutor } from "./ollama-local.js";
+export { CommandCodeExecutor } from "./commandcode.js";
+export { XiaomiTokenplanExecutor } from "./xiaomi-tokenplan.js";