Create main.ts

main.ts

@@ -0,0 +1,171 @@
+// deno run --allow-net --allow-env adapter.ts
+
+import { serve } from "https://deno.land/std@0.203.0/http/server.ts";
+
+// --- Hardcoded API Keys ---
+const CLIENT_API_KEYS = new Set([
+  "sk-client-1234567890abcdef",
+  // add more client keys here
+]);
+
+const CODEGEEX_TOKENS: {
+  token: string;
+  isValid: boolean;
+  lastUsed: number;
+  errorCount: number;
+}[] = [
+  { token: "cgx-token-AAAA", isValid: true, lastUsed: 0, errorCount: 0 },
+  { token: "cgx-token-BBBB", isValid: true, lastUsed: 0, errorCount: 0 },
+  // add more CodeGeeX tokens here
+];
+
+const MAX_ERROR_COUNT = 3;
+const ERROR_COOLDOWN = 300 * 1000; // ms
+
+// --- Utilities ---
+function now(): number {
+  return Date.now();
+}
+
+function rotateToken(): typeof CODEGEEX_TOKENS[0] | null {
+  const available = CODEGEEX_TOKENS.filter(t => {
+    if (!t.isValid) return false;
+    if (t.errorCount >= MAX_ERROR_COUNT && now() - t.lastUsed < ERROR_COOLDOWN) return false;
+    return true;
+  });
+  if (available.length === 0) return null;
+
+  // reset cooled-down tokens
+  for (const t of available) {
+    if (t.errorCount >= MAX_ERROR_COUNT && now() - t.lastUsed >= ERROR_COOLDOWN) {
+      t.errorCount = 0;
+    }
+  }
+
+  // pick the one least recently used, then lowest errorCount
+  available.sort((a, b) => a.lastUsed - b.lastUsed || a.errorCount - b.errorCount);
+  const tok = available[0];
+  tok.lastUsed = now();
+  return tok;
+}
+
+async function proxyChat(req: Request, params: { stream: boolean; model: string; messages: any[] }) {
+  const tokenObj = rotateToken();
+  if (!tokenObj) {
+    return new Response(JSON.stringify({ error: "No valid CodeGeeX tokens available" }), { status: 503 });
+  }
+
+  const payload = {
+    user_role: 0,
+    ide: "Deno",
+    prompt: params.messages.slice(-1)[0]?.content || "",
+    history: [], // Simplified: not preserving history in this example
+    model: params.model,
+  };
+
+  try {
+    const response = await fetch("https://codegeex.cn/prod/code/chatCodeSseV3/chat", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Accept": "text/event-stream",
+        "code-token": tokenObj.token,
+      },
+      body: JSON.stringify(payload),
+    });
+    if (!response.ok) {
+      if (response.status === 401 || response.status === 403) {
+        tokenObj.isValid = false;
+      } else {
+        tokenObj.errorCount++;
+      }
+      return new Response(JSON.stringify({ error: `Upstream error ${response.status}` }), { status: 502 });
+    }
+
+    if (params.stream) {
+      // proxy SSE
+      return new Response(response.body, {
+        status: 200,
+        headers: {
+          "Content-Type": "text/event-stream",
+          "Cache-Control": "no-cache",
+          Connection: "keep-alive",
+        },
+      });
+    } else {
+      // accumulate and return JSON
+      const text = await response.text();
+      return new Response(JSON.stringify({
+        id: `chatcmpl-${crypto.randomUUID()}`,
+        object: "chat.completion",
+        created: Math.floor(now() / 1000),
+        model: params.model,
+        choices: [{ message: { role: "assistant", content: text }, index: 0, finish_reason: "stop" }],
+      }), {
+        status: 200,
+        headers: { "Content-Type": "application/json" },
+      });
+    }
+  } catch (err) {
+    tokenObj.errorCount++;
+    return new Response(JSON.stringify({ error: err.message }), { status: 500 });
+  }
+}
+
+// --- Handlers ---
+async function handler(req: Request): Promise<Response> {
+  const url = new URL(req.url);
+
+  // /debug?enable=true|false
+  if (url.pathname === "/debug" && req.method === "GET") {
+    const en = url.searchParams.get("enable");
+    if (en !== null) {
+      Deno.env.set("DEBUG_MODE", en);
+    }
+    return new Response(JSON.stringify({ debug_mode: Deno.env.get("DEBUG_MODE") === "true" }), {
+      headers: { "Content-Type": "application/json" },
+    });
+  }
+
+  // Authentication middleware
+  const auth = req.headers.get("Authorization")?.replace(/^Bearer\s+/, "");
+  if (!CLIENT_API_KEYS.size) {
+    return new Response(JSON.stringify({ error: "Server misconfigured: no client keys" }), { status: 503 });
+  }
+  if (!auth || !CLIENT_API_KEYS.has(auth)) {
+    return new Response(JSON.stringify({ error: "Invalid or missing API key" }), {
+      status: 401,
+      headers: { "WWW-Authenticate": "Bearer" },
+    });
+  }
+
+  // GET /v1/models and GET /models
+  if ((url.pathname === "/v1/models" || url.pathname === "/models") && req.method === "GET") {
+    const data = CODEGEEX_TOKENS.map((_, i) => ({
+      id: [`claude-3-7-sonnet`, `claude-sonnet-4`][i % 2],
+      owned_by: "anthropic",
+      created: Math.floor(now() / 1000),
+      object: "model",
+    }));
+    return new Response(JSON.stringify({ object: "list", data }), {
+      headers: { "Content-Type": "application/json" },
+    });
+  }
+
+  // POST /v1/chat/completions
+  if (url.pathname === "/v1/chat/completions" && req.method === "POST") {
+    const body = await req.json();
+    const { model, messages, stream = true } = body;
+    if (!model || !Array.isArray(messages)) {
+      return new Response(JSON.stringify({ error: "Bad Request" }), { status: 400 });
+    }
+    return proxyChat(req, { model, messages, stream });
+  }
+
+  // Not found
+  return new Response(JSON.stringify({ error: "Not Found" }), { status: 404 });
+}
+
+// --- Start Server ---
+console.log("Starting Deno CodeGeeX Adapter on http://0.0.0.0:8000");
+serve(handler, { port: 8000 });