1124 lines
40 KiB
TypeScript
1124 lines
40 KiB
TypeScript
import { BaseExecutor, type ExecuteInput } from "./base.ts";
|
|
import { solveDeepSeekPowAsync } from "../lib/deepseek-pow.ts";
|
|
import { type OpenAIToolCall } from "../translator/webTools.ts";
|
|
import {
|
|
serializeDeepSeekToolPrompt,
|
|
parseDeepSeekToolCalls,
|
|
buildToolConversationPrompt,
|
|
} from "../translator/deepseekWebTools.ts";
|
|
import { sanitizeErrorMessage } from "../utils/error.ts";
|
|
import {
|
|
isThinkingModel,
|
|
isSearchModel,
|
|
formatStreamContent,
|
|
appendSearchCitations,
|
|
type DeepSeekSearchResult,
|
|
} from "./deepseek-web/stream-format.ts";
|
|
|
|
export const DEEPSEEK_WEB_BASE = "https://chat.deepseek.com";
|
|
const DEEPSEEK_API_BASE = `${DEEPSEEK_WEB_BASE}/api`;
|
|
const COMPLETION_URL = `${DEEPSEEK_API_BASE}/v0/chat/completion`;
|
|
|
|
// Fingerprint headers the chat.deepseek.com web client sends on every /api/v0/*
|
|
// request. Kept in sync with a real captured completion request (client v2.0.0):
|
|
// the 2.0.0 web build DROPPED the legacy `X-App-Version` build stamp and ADDED
|
|
// `X-Client-Bundle-Id`. Sending the stale `X-App-Version` (and the old 1.8.0
|
|
// version) is itself a bot-detection signal, so match the current client exactly.
|
|
// NOTE: the live client also sends `x-hif-leim`, a signed client-attestation
|
|
// token generated by obfuscated JS. It is intentionally omitted — reproducing it
|
|
// requires porting that JS, and it is not currently enforced by the completion
|
|
// endpoint. Revisit if requests start failing a client-attestation check.
|
|
const FAKE_HEADERS: Record<string, string> = {
|
|
Accept: "*/*",
|
|
"Accept-Encoding": "gzip, deflate, br, zstd",
|
|
"Accept-Language": "en-US,en;q=0.9",
|
|
Origin: DEEPSEEK_WEB_BASE,
|
|
Referer: `${DEEPSEEK_WEB_BASE}/`,
|
|
"User-Agent":
|
|
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/149.0.0.0 Safari/537.36",
|
|
"X-Client-Bundle-Id": "com.deepseek.chat",
|
|
"X-Client-Locale": "en-US",
|
|
"X-Client-Platform": "web",
|
|
"X-Client-Version": "2.0.0",
|
|
};
|
|
|
|
// ── Types ────────────────────────────────────────────────────────────────
|
|
|
|
interface PowChallenge {
|
|
algorithm: string;
|
|
challenge: string;
|
|
salt: string;
|
|
signature: string;
|
|
difficulty: number;
|
|
expire_at: number;
|
|
expire_after: number;
|
|
target_path: string;
|
|
}
|
|
|
|
interface TokenInfo {
|
|
accessToken: string;
|
|
expiresAt: number;
|
|
}
|
|
|
|
// ── Token cache (keyed by userToken → short-lived access token) ─────────
|
|
|
|
const tokenCache = new Map<string, TokenInfo>();
|
|
const sessionCache = new Map<string, { sessionId: string; createdAt: number }>();
|
|
|
|
const CACHE_MAX_SIZE = 100;
|
|
|
|
function evictOldest(cache: Map<string, unknown>): void {
|
|
if (cache.size >= CACHE_MAX_SIZE) {
|
|
const first = cache.keys().next().value;
|
|
if (first) cache.delete(first);
|
|
}
|
|
}
|
|
|
|
// ── Helpers ──────────────────────────────────────────────────────────────
|
|
|
|
export function extractUserToken(credentials: Record<string, unknown>): string | null {
|
|
const raw = credentials?.apiKey || credentials?.accessToken;
|
|
if (typeof raw !== "string" || raw.length === 0) return null;
|
|
// Handle JSON-wrapped tokens (DeepSeek stores token as {"value":"..."})
|
|
try {
|
|
const parsed = JSON.parse(raw);
|
|
if (typeof parsed?.value === "string") return parsed.value;
|
|
} catch {
|
|
// not JSON, use raw
|
|
}
|
|
return raw;
|
|
}
|
|
|
|
function errorResponse(status: number, message: string, dsCode?: number): Response {
|
|
return new Response(
|
|
JSON.stringify({
|
|
error: { message, type: "upstream_error", code: dsCode ?? `HTTP_${status}` },
|
|
}),
|
|
{ status, headers: { "Content-Type": "application/json" } }
|
|
);
|
|
}
|
|
|
|
function resolveModelOptions(
|
|
model?: string,
|
|
bodyObj?: Record<string, unknown>
|
|
): {
|
|
modelType: string;
|
|
thinkingEnabled: boolean;
|
|
searchEnabled: boolean;
|
|
} {
|
|
const m = (model || "").toLowerCase();
|
|
const modelType = m.includes("pro") || m.includes("expert") ? "expert" : "default";
|
|
const thinkingEnabled =
|
|
m.includes("r1") ||
|
|
m.includes("think") ||
|
|
m.includes("reason") ||
|
|
bodyObj?.thinking_enabled === true ||
|
|
bodyObj?.thinking === true ||
|
|
!!bodyObj?.reasoning_effort;
|
|
const searchEnabled =
|
|
m.includes("search") ||
|
|
bodyObj?.search_enabled === true ||
|
|
bodyObj?.search === true ||
|
|
bodyObj?.web_search === true;
|
|
return { modelType, thinkingEnabled, searchEnabled };
|
|
}
|
|
|
|
function generateFakeCookie(): string {
|
|
const ts = Date.now();
|
|
const hex = (n: number) =>
|
|
Array.from({ length: n }, () => Math.floor(Math.random() * 16).toString(16)).join("");
|
|
const uid = () =>
|
|
"xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx".replace(/[xy]/g, (c) => {
|
|
const r = (Math.random() * 16) | 0;
|
|
return (c === "x" ? r : (r & 0x3) | 0x8).toString(16);
|
|
});
|
|
return `intercom-HWWAFSESTIME=${ts}; HWWAFSESID=${hex(18)}; Hm_lvt_${uid()}=${Math.floor(ts / 1000)}; _frid=${uid()}`;
|
|
}
|
|
|
|
// ── PoW Solver (DeepSeekHashV1) ─────────────────────────────────────────
|
|
|
|
async function solvePow(challenge: PowChallenge): Promise<string> {
|
|
const answer = await solveDeepSeekPowAsync(
|
|
challenge.algorithm,
|
|
challenge.challenge,
|
|
challenge.salt,
|
|
challenge.difficulty,
|
|
challenge.expire_at
|
|
);
|
|
if (answer < 0) throw new Error("PoW solver failed");
|
|
return Buffer.from(
|
|
JSON.stringify({
|
|
algorithm: challenge.algorithm,
|
|
challenge: challenge.challenge,
|
|
salt: challenge.salt,
|
|
answer,
|
|
signature: challenge.signature,
|
|
target_path: challenge.target_path,
|
|
})
|
|
).toString("base64");
|
|
}
|
|
|
|
// ── SSE Transform (DeepSeek → OpenAI) ───────────────────────────────────
|
|
|
|
function transformSSE(deepseekStream: ReadableStream, model: string): ReadableStream {
|
|
const encoder = new TextEncoder();
|
|
const decoder = new TextDecoder();
|
|
const streamModel = model || "deepseek-web";
|
|
const id = `chatcmpl-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
const created = Math.floor(Date.now() / 1000);
|
|
let emittedRole = false;
|
|
let currentPath: "thinking" | "content" | "" = "";
|
|
const thinkingModel = isThinkingModel(streamModel);
|
|
const searchResults: DeepSeekSearchResult[] = [];
|
|
|
|
return new ReadableStream(
|
|
{
|
|
async start(controller) {
|
|
const reader = deepseekStream.getReader();
|
|
let buffer = "";
|
|
|
|
const emit = (obj: object) => {
|
|
controller.enqueue(encoder.encode(`data: ${JSON.stringify(obj)}\n\n`));
|
|
};
|
|
|
|
const chunk = (delta: object, finish?: string) => {
|
|
emit({
|
|
id,
|
|
object: "chat.completion.chunk",
|
|
created,
|
|
model: streamModel,
|
|
choices: [{ index: 0, delta, finish_reason: finish ?? null }],
|
|
});
|
|
};
|
|
|
|
const ensureRole = () => {
|
|
if (!emittedRole) {
|
|
emittedRole = true;
|
|
chunk({ role: "assistant", content: "" });
|
|
}
|
|
};
|
|
|
|
const finishStream = () => {
|
|
const citations = appendSearchCitations(searchResults, streamModel);
|
|
if (citations) {
|
|
ensureRole();
|
|
chunk({ content: `\n\n${citations}` });
|
|
}
|
|
ensureRole();
|
|
chunk({}, "stop");
|
|
controller.enqueue(encoder.encode("data: [DONE]\n\n"));
|
|
controller.close();
|
|
};
|
|
|
|
const sendByPath = (raw: string) => {
|
|
const text = formatStreamContent(raw, streamModel);
|
|
if (!text) return;
|
|
ensureRole();
|
|
let path = currentPath;
|
|
if (!path && thinkingModel) path = "thinking";
|
|
else if (!path && isSearchModel(streamModel)) path = "content";
|
|
if (path === "thinking") {
|
|
chunk({ reasoning_content: text });
|
|
} else {
|
|
chunk({ content: text });
|
|
}
|
|
};
|
|
|
|
const applyFragmentType = (frag: any) => {
|
|
const type = String(frag?.type || "").toUpperCase();
|
|
if (type === "THINK") currentPath = "thinking";
|
|
else if (type === "ANSWER" || type === "RESPONSE") currentPath = "content";
|
|
};
|
|
|
|
const handleFragment = (frag: any, setPathFromType = false) => {
|
|
if (setPathFromType) applyFragmentType(frag);
|
|
if (typeof frag?.content !== "string" || frag.content.length === 0) return;
|
|
if (!setPathFromType) {
|
|
const type = String(frag?.type || "").toUpperCase();
|
|
if (type === "THINK") currentPath = "thinking";
|
|
else if (type === "ANSWER" || type === "RESPONSE") currentPath = "content";
|
|
}
|
|
sendByPath(frag.content);
|
|
};
|
|
|
|
try {
|
|
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: ") && !line.startsWith("data:")) continue;
|
|
const payload = line.replace(/^data:\s*/, "").trim();
|
|
|
|
if (payload === "[DONE]") {
|
|
finishStream();
|
|
return;
|
|
}
|
|
|
|
let data: Record<string, unknown>;
|
|
try {
|
|
data = JSON.parse(payload);
|
|
} catch {
|
|
continue;
|
|
}
|
|
|
|
const p = (data as any)?.p;
|
|
const o = (data as any)?.o;
|
|
const v = (data as any)?.v;
|
|
|
|
if (v && typeof v === "object" && v.response) {
|
|
if (v.response.thinking_enabled === true) currentPath = "thinking";
|
|
else if (v.response.thinking_enabled === false) currentPath = "content";
|
|
const fragments = v.response.fragments;
|
|
if (Array.isArray(fragments)) {
|
|
for (const frag of fragments) handleFragment(frag, false);
|
|
}
|
|
}
|
|
|
|
if (p === "response/fragments") {
|
|
if (Array.isArray(v)) {
|
|
for (const frag of v) handleFragment(frag, true);
|
|
} else if (v && typeof v === "object") {
|
|
handleFragment(v, true);
|
|
}
|
|
}
|
|
|
|
if (p === "response" && Array.isArray(v)) {
|
|
for (const entry of v) {
|
|
if (entry?.p === "response" && entry?.v?.thinking_enabled === true) {
|
|
currentPath = "thinking";
|
|
}
|
|
}
|
|
}
|
|
|
|
if (p === "response/search_status") continue;
|
|
|
|
if (p === "response/search_results" && Array.isArray(v)) {
|
|
if (o !== "BATCH") {
|
|
searchResults.length = 0;
|
|
searchResults.push(...v);
|
|
} else {
|
|
for (const op of v) {
|
|
const match = String(op?.p || "").match(/^(\d+)\/cite_index$/);
|
|
if (match) {
|
|
const index = parseInt(match[1], 10);
|
|
if (searchResults[index]) searchResults[index].cite_index = op.v;
|
|
}
|
|
}
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (typeof v === "string") {
|
|
sendByPath(v);
|
|
} else if (Array.isArray(v) && p === "response") {
|
|
for (const entry of v) {
|
|
if (Array.isArray(entry?.v)) {
|
|
const joined = entry.v.map((item: any) => item?.content || "").join("");
|
|
if (joined) sendByPath(joined);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Do not close on FINISHED — DeepSeek may still send search_results afterward.
|
|
if (p === "response/status" && v === "FINISHED") {
|
|
continue;
|
|
}
|
|
}
|
|
}
|
|
} catch (err) {
|
|
controller.error(err);
|
|
return;
|
|
}
|
|
|
|
finishStream();
|
|
},
|
|
},
|
|
{ highWaterMark: 16384 }
|
|
);
|
|
}
|
|
|
|
async function collectSSEContent(
|
|
deepseekStream: ReadableStream,
|
|
model: string
|
|
): Promise<{ content: string; reasoningContent: string }> {
|
|
const decoder = new TextDecoder();
|
|
const reader = deepseekStream.getReader();
|
|
let buffer = "";
|
|
let content = "";
|
|
let reasoningContent = "";
|
|
let currentPath: "thinking" | "content" | "" = "";
|
|
const streamModel = model || "deepseek-web";
|
|
const thinkingModel = isThinkingModel(streamModel);
|
|
const searchResults: DeepSeekSearchResult[] = [];
|
|
|
|
const appendByPath = (raw: string) => {
|
|
const text = formatStreamContent(raw, streamModel);
|
|
if (!text) return;
|
|
let path = currentPath;
|
|
if (!path && thinkingModel) path = "thinking";
|
|
else if (!path && isSearchModel(streamModel)) path = "content";
|
|
if (path === "thinking") reasoningContent += text;
|
|
else content += text;
|
|
};
|
|
|
|
const applyFragmentType = (frag: any) => {
|
|
const type = String(frag?.type || "").toUpperCase();
|
|
if (type === "THINK") currentPath = "thinking";
|
|
else if (type === "ANSWER" || type === "RESPONSE") currentPath = "content";
|
|
};
|
|
|
|
const handleFragment = (frag: any, setPathFromType = false) => {
|
|
if (setPathFromType) applyFragmentType(frag);
|
|
if (typeof frag?.content !== "string" || frag.content.length === 0) return;
|
|
if (!setPathFromType) {
|
|
const type = String(frag?.type || "").toUpperCase();
|
|
if (type === "THINK") currentPath = "thinking";
|
|
else if (type === "ANSWER" || type === "RESPONSE") currentPath = "content";
|
|
}
|
|
appendByPath(frag.content);
|
|
};
|
|
|
|
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: ") && !line.startsWith("data:")) continue;
|
|
const payload = line.replace(/^data:\s*/, "").trim();
|
|
try {
|
|
const data = JSON.parse(payload);
|
|
const p = data?.p;
|
|
const v = data?.v;
|
|
|
|
if (v && typeof v === "object" && v.response) {
|
|
if (v.response.thinking_enabled === true) currentPath = "thinking";
|
|
else if (v.response.thinking_enabled === false) currentPath = "content";
|
|
if (Array.isArray(v.response.fragments)) {
|
|
for (const frag of v.response.fragments) handleFragment(frag, false);
|
|
}
|
|
}
|
|
|
|
if (p === "response/fragments") {
|
|
if (Array.isArray(v)) {
|
|
for (const frag of v) handleFragment(frag, true);
|
|
} else if (v && typeof v === "object") {
|
|
handleFragment(v, true);
|
|
}
|
|
}
|
|
|
|
if (p === "response" && Array.isArray(v)) {
|
|
for (const entry of v) {
|
|
if (entry?.p === "response" && entry?.v?.thinking_enabled === true) {
|
|
currentPath = "thinking";
|
|
}
|
|
}
|
|
}
|
|
|
|
if (p === "response/search_status") continue;
|
|
|
|
if (p === "response/search_results" && Array.isArray(v)) {
|
|
if (data?.o !== "BATCH") {
|
|
searchResults.length = 0;
|
|
searchResults.push(...v);
|
|
} else {
|
|
for (const op of v) {
|
|
const match = String(op?.p || "").match(/^(\d+)\/cite_index$/);
|
|
if (match) {
|
|
const index = parseInt(match[1], 10);
|
|
if (searchResults[index]) searchResults[index].cite_index = op.v;
|
|
}
|
|
}
|
|
}
|
|
continue;
|
|
}
|
|
|
|
if (typeof v === "string") {
|
|
appendByPath(v);
|
|
} else if (Array.isArray(v) && p === "response") {
|
|
for (const entry of v) {
|
|
if (Array.isArray(entry?.v)) {
|
|
const joined = entry.v.map((item: any) => item?.content || "").join("");
|
|
if (joined) appendByPath(joined);
|
|
}
|
|
}
|
|
}
|
|
} catch {
|
|
// skip
|
|
}
|
|
}
|
|
}
|
|
|
|
const citations = appendSearchCitations(searchResults, streamModel);
|
|
if (citations) content += `\n\n${citations}`;
|
|
|
|
return { content, reasoningContent };
|
|
}
|
|
|
|
// ── Prompt builder (DeepSeek native format, matches Chat2API) ────────────
|
|
|
|
function extractMessageText(content: unknown): string {
|
|
if (Array.isArray(content)) {
|
|
return (content as any[])
|
|
.filter((item: any) => item.type === "text")
|
|
.map((item: any) => item.text)
|
|
.join("\n");
|
|
}
|
|
return String(content || "");
|
|
}
|
|
|
|
/**
|
|
* Build the single prompt string the DeepSeek web API accepts.
|
|
*
|
|
* The web endpoint (`/api/v0/chat/completion`) takes only a `prompt` string, not a
|
|
* `messages` array. With `historyWindow <= 0` (default) we keep the legacy behavior —
|
|
* system prompt(s) + the last user message only — which is fine for plain chat.
|
|
*
|
|
* With `historyWindow > 0` we stitch the last N non-system messages into a role-tagged
|
|
* transcript so agentic multi-turn clients keep context across turns (rolling-window
|
|
* memory, #2942). The system prompt(s) still lead the prompt and the newest user turn
|
|
* is the last line of the transcript.
|
|
*/
|
|
export function messagesToPrompt(
|
|
messages: Array<{ role: string; content: string; tool_call_id?: string; name?: string }>,
|
|
historyWindow = 0
|
|
): string {
|
|
if (messages.length === 0) return "";
|
|
|
|
const systemParts: string[] = [];
|
|
const conversation: Array<{ role: string; text: string }> = [];
|
|
const callNameById = new Map<string, string>();
|
|
let lastUserContent = "";
|
|
for (const m of messages) {
|
|
const text = extractMessageText(m.content).trim();
|
|
if (m.role === "system") {
|
|
if (text) systemParts.push(text);
|
|
} else if (m.role === "user" || m.role === "assistant") {
|
|
if (text) conversation.push({ role: m.role, text });
|
|
if (m.role === "user") lastUserContent = text;
|
|
const calls = Array.isArray((m as { tool_calls?: unknown }).tool_calls)
|
|
? (m as { tool_calls: Array<{ id?: string; function?: { name?: string } }> }).tool_calls
|
|
: [];
|
|
for (const c of calls) {
|
|
if (c?.id && typeof c.function?.name === "string") callNameById.set(c.id, c.function.name);
|
|
}
|
|
} else if (m.role === "tool") {
|
|
// Tool results have no native slot in deepseek-web's single-prompt format. Without
|
|
// this branch they were silently dropped (#4712) — the model never saw the tool
|
|
// output and either re-called the tool endlessly or answered "I don't have that
|
|
// information". Fold them into the transcript as plain text, mirroring the agentic
|
|
// buildToolConversationPrompt() path.
|
|
if (text) {
|
|
const name = (m.tool_call_id && callNameById.get(m.tool_call_id)) || m.name || "tool";
|
|
conversation.push({ role: "tool", text: `(${name}) ${text}` });
|
|
}
|
|
}
|
|
}
|
|
|
|
const parts: string[] = [];
|
|
if (systemParts.length > 0) {
|
|
parts.push(systemParts.join("\n\n"));
|
|
}
|
|
|
|
if (historyWindow > 0 && conversation.length > 1) {
|
|
// Rolling-window transcript of the most recent turns (#2942).
|
|
const recent = conversation.slice(-historyWindow);
|
|
const transcript = recent
|
|
.map((turn) =>
|
|
turn.role === "assistant"
|
|
? `Assistant: ${turn.text}`
|
|
: turn.role === "tool"
|
|
? `Tool result ${turn.text}`
|
|
: `User: ${turn.text}`
|
|
)
|
|
.join("\n\n");
|
|
parts.push(transcript);
|
|
} else if (lastUserContent) {
|
|
parts.push(lastUserContent);
|
|
}
|
|
|
|
return parts.join("\n\n").replace(/!\[.*?\]\(.*?\)/g, "");
|
|
}
|
|
|
|
// ── DeepSeek API calls (Bearer token auth, like Chat2API) ───────────────
|
|
|
|
async function acquireAccessToken(
|
|
userToken: string,
|
|
signal?: AbortSignal | null,
|
|
log?: ExecuteInput["log"]
|
|
): Promise<string> {
|
|
const cached = tokenCache.get(userToken);
|
|
if (cached && cached.expiresAt > Math.floor(Date.now() / 1000)) {
|
|
return cached.accessToken;
|
|
}
|
|
|
|
log?.info?.("DEEPSEEK-WEB", "Acquiring access token from /users/current...");
|
|
const resp = await fetch(`${DEEPSEEK_API_BASE}/v0/users/current`, {
|
|
headers: {
|
|
Authorization: `Bearer ${userToken}`,
|
|
...FAKE_HEADERS,
|
|
},
|
|
signal: signal ?? undefined,
|
|
});
|
|
|
|
if (resp.status === 401 || resp.status === 403) {
|
|
throw new Error("Token invalid or expired — get a new userToken from DeepSeek localStorage");
|
|
}
|
|
if (!resp.ok) {
|
|
throw new Error(`users/current HTTP ${resp.status}`);
|
|
}
|
|
|
|
const json = await resp.json();
|
|
if (json?.code && json.code !== 0) {
|
|
const errMsg = json.msg || json?.data?.biz_msg || `error code ${json.code}`;
|
|
tokenCache.delete(userToken);
|
|
throw new Error(`DeepSeek rejected token: ${errMsg}`);
|
|
}
|
|
const bizData = json?.data?.biz_data || json?.biz_data;
|
|
if (!bizData?.token) {
|
|
const errMsg = json?.msg || json?.data?.biz_msg || "Unknown error";
|
|
throw new Error(`Failed to acquire token: ${errMsg}`);
|
|
}
|
|
|
|
const accessToken = bizData.token;
|
|
evictOldest(tokenCache);
|
|
tokenCache.set(userToken, {
|
|
accessToken,
|
|
expiresAt: Math.floor(Date.now() / 1000) + 3600,
|
|
});
|
|
|
|
log?.info?.("DEEPSEEK-WEB", `Access token acquired (${accessToken.length} chars)`);
|
|
return accessToken;
|
|
}
|
|
|
|
function parseDeepSeekErrorPayload(payload: unknown): { code?: number; message: string } | null {
|
|
if (!payload || typeof payload !== "object") return null;
|
|
const record = payload as Record<string, unknown>;
|
|
const codeRaw = record.code;
|
|
const code = typeof codeRaw === "number" ? codeRaw : undefined;
|
|
const msg = record.msg;
|
|
const data = record.data as Record<string, unknown> | undefined;
|
|
const bizMsg = data?.biz_msg;
|
|
const messageRaw = typeof msg === "string" ? msg : typeof bizMsg === "string" ? bizMsg : "";
|
|
if (code !== undefined && code !== 0) {
|
|
return { code, message: messageRaw || `DeepSeek error ${code}` };
|
|
}
|
|
return null;
|
|
}
|
|
|
|
async function createSession(accessToken: string, signal?: AbortSignal | null): Promise<string> {
|
|
const resp = await fetch(`${DEEPSEEK_API_BASE}/v0/chat_session/create`, {
|
|
method: "POST",
|
|
headers: {
|
|
...FAKE_HEADERS,
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${accessToken}`,
|
|
Cookie: generateFakeCookie(),
|
|
},
|
|
body: JSON.stringify({}),
|
|
signal: signal ?? undefined,
|
|
});
|
|
|
|
if (!resp.ok) throw new Error(`chat_session/create HTTP ${resp.status}`);
|
|
const json = await resp.json();
|
|
const bizData = json?.data?.biz_data || json?.biz_data;
|
|
const id = bizData?.chat_session?.id;
|
|
if (!id) throw new Error(`No session id: code=${json?.code}`);
|
|
return id;
|
|
}
|
|
|
|
async function deleteSessionOnDeepSeek(accessToken: string, sessionId: string): Promise<void> {
|
|
try {
|
|
await fetch(`${DEEPSEEK_API_BASE}/v0/chat_session/delete`, {
|
|
method: "POST",
|
|
headers: {
|
|
...FAKE_HEADERS,
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${accessToken}`,
|
|
},
|
|
body: JSON.stringify({ chat_session_id: sessionId }),
|
|
});
|
|
} catch {
|
|
// best-effort cleanup
|
|
}
|
|
}
|
|
|
|
function wrapStreamWithCleanup(
|
|
responseStream: ReadableStream,
|
|
cleanup: () => Promise<void>
|
|
): ReadableStream {
|
|
const reader = responseStream.getReader();
|
|
return new ReadableStream({
|
|
async pull(controller) {
|
|
const { done, value } = await reader.read();
|
|
if (done) {
|
|
controller.close();
|
|
cleanup().catch(() => {});
|
|
return;
|
|
}
|
|
controller.enqueue(value);
|
|
},
|
|
cancel() {
|
|
reader.cancel();
|
|
cleanup().catch(() => {});
|
|
},
|
|
});
|
|
}
|
|
|
|
async function getPowChallenge(
|
|
accessToken: string,
|
|
signal?: AbortSignal | null
|
|
): Promise<PowChallenge> {
|
|
const resp = await fetch(`${DEEPSEEK_API_BASE}/v0/chat/create_pow_challenge`, {
|
|
method: "POST",
|
|
headers: {
|
|
...FAKE_HEADERS,
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${accessToken}`,
|
|
},
|
|
body: JSON.stringify({ target_path: "/api/v0/chat/completion" }),
|
|
signal: signal ?? undefined,
|
|
});
|
|
if (!resp.ok) throw new Error(`create_pow_challenge HTTP ${resp.status}`);
|
|
const json = await resp.json();
|
|
const bizData = json?.data?.biz_data || json?.biz_data;
|
|
if (!bizData?.challenge?.challenge) throw new Error(`No PoW challenge: code=${json?.code}`);
|
|
return bizData.challenge as PowChallenge;
|
|
}
|
|
|
|
// ── Tool-call response builder (#2820) ──────────────────────────────────
|
|
|
|
/**
|
|
* Build the executor result for a tool-translated reply. Emits OpenAI `tool_calls`
|
|
* with `finish_reason: "tool_calls"` when tool calls were parsed, otherwise plain
|
|
* content. Supports both streaming (synthetic SSE) and non-streaming clients.
|
|
*/
|
|
function buildToolAwareResult(opts: {
|
|
stream: boolean;
|
|
clientModel: string;
|
|
content: string;
|
|
reasoningContent?: string;
|
|
toolCalls: OpenAIToolCall[] | null;
|
|
reqHeaders: Record<string, string>;
|
|
requestPayload: unknown;
|
|
}) {
|
|
const { stream, clientModel, content, reasoningContent, toolCalls, reqHeaders, requestPayload } =
|
|
opts;
|
|
const hasCalls = !!toolCalls && toolCalls.length > 0;
|
|
const finishReason = hasCalls ? "tool_calls" : "stop";
|
|
const id = `chatcmpl-${Date.now()}`;
|
|
const created = Math.floor(Date.now() / 1000);
|
|
|
|
if (stream) {
|
|
const encoder = new TextEncoder();
|
|
const emit = (
|
|
controller: ReadableStreamDefaultController,
|
|
delta: object,
|
|
finish: string | null
|
|
) => {
|
|
controller.enqueue(
|
|
encoder.encode(
|
|
`data: ${JSON.stringify({
|
|
id,
|
|
object: "chat.completion.chunk",
|
|
created,
|
|
model: clientModel,
|
|
choices: [{ index: 0, delta, finish_reason: finish }],
|
|
})}\n\n`
|
|
)
|
|
);
|
|
};
|
|
const sse = new ReadableStream({
|
|
start(controller) {
|
|
emit(controller, { role: "assistant", content: "" }, null);
|
|
// Surrounding natural-language text (and reasoning) is emitted before the tool_calls
|
|
// so a model reply that interleaves a plan with a call still reaches the client (#7).
|
|
if (reasoningContent) emit(controller, { reasoning_content: reasoningContent }, null);
|
|
if (content) emit(controller, { content }, null);
|
|
if (hasCalls) {
|
|
emit(
|
|
controller,
|
|
{
|
|
tool_calls: toolCalls!.map((tc, i) => ({
|
|
index: i,
|
|
id: tc.id,
|
|
type: "function",
|
|
function: { name: tc.function.name, arguments: tc.function.arguments },
|
|
})),
|
|
},
|
|
null
|
|
);
|
|
}
|
|
emit(controller, {}, finishReason);
|
|
controller.enqueue(encoder.encode("data: [DONE]\n\n"));
|
|
controller.close();
|
|
},
|
|
});
|
|
return {
|
|
response: new Response(sse, {
|
|
status: 200,
|
|
headers: { "Content-Type": "text/event-stream", "Cache-Control": "no-cache" },
|
|
}),
|
|
url: COMPLETION_URL,
|
|
headers: reqHeaders,
|
|
transformedBody: requestPayload,
|
|
};
|
|
}
|
|
|
|
const message: Record<string, unknown> = { role: "assistant", content: content || "" };
|
|
if (reasoningContent) message.reasoning_content = reasoningContent;
|
|
if (hasCalls) {
|
|
message.tool_calls = toolCalls;
|
|
if (!content) message.content = null;
|
|
}
|
|
const openaiResponse = {
|
|
id,
|
|
object: "chat.completion",
|
|
created,
|
|
model: clientModel,
|
|
choices: [{ index: 0, message, finish_reason: finishReason }],
|
|
usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
|
|
};
|
|
return {
|
|
response: new Response(JSON.stringify(openaiResponse), {
|
|
status: 200,
|
|
headers: { "Content-Type": "application/json" },
|
|
}),
|
|
url: COMPLETION_URL,
|
|
headers: reqHeaders,
|
|
transformedBody: requestPayload,
|
|
};
|
|
}
|
|
|
|
// ── Executor ─────────────────────────────────────────────────────────────
|
|
|
|
export class DeepSeekWebExecutor extends BaseExecutor {
|
|
constructor() {
|
|
super("deepseek-web", { baseUrl: DEEPSEEK_WEB_BASE });
|
|
}
|
|
|
|
async testConnection(
|
|
credentials: Record<string, unknown>,
|
|
signal?: AbortSignal
|
|
): Promise<boolean> {
|
|
try {
|
|
const userToken = extractUserToken(credentials);
|
|
if (!userToken) return false;
|
|
const accessToken = await acquireAccessToken(userToken, signal);
|
|
return !!accessToken;
|
|
} catch {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
async execute({ model, body, stream, credentials, signal, log }: ExecuteInput) {
|
|
const bodyObj = (body || {}) as Record<string, unknown>;
|
|
|
|
// chat.deepseek.com's web API only accepts {prompt, ref_file_ids,
|
|
// thinking_enabled, search_enabled} - no native tools field. Instead of failing
|
|
// tool-using requests, translate them (#2820): serialize the OpenAI tools[] into a
|
|
// <tool>...</tool> prompt contract on the way in, and parse the model's text reply
|
|
// back into OpenAI tool_calls on the way out.
|
|
const requestedTools = bodyObj.tools;
|
|
const hasTools = Array.isArray(requestedTools) && requestedTools.length > 0;
|
|
const toolSystemPrompt = hasTools ? serializeDeepSeekToolPrompt(requestedTools) : "";
|
|
|
|
const messages = (Array.isArray(bodyObj.messages) ? bodyObj.messages : []) as Array<{
|
|
role: string;
|
|
content: string;
|
|
}>;
|
|
const promptMessages = toolSystemPrompt
|
|
? [{ role: "system", content: toolSystemPrompt }, ...messages]
|
|
: messages;
|
|
const rawCreds = credentials as unknown as Record<string, unknown>;
|
|
|
|
const userToken = extractUserToken(rawCreds);
|
|
if (!userToken) {
|
|
return {
|
|
response: errorResponse(
|
|
400,
|
|
"Invalid credentials: paste your userToken from DeepSeek localStorage " +
|
|
"(DevTools → Application → Local Storage → chat.deepseek.com → userToken)"
|
|
),
|
|
url: COMPLETION_URL,
|
|
headers: {},
|
|
transformedBody: body,
|
|
};
|
|
}
|
|
|
|
const { modelType, thinkingEnabled, searchEnabled } = resolveModelOptions(
|
|
model as string,
|
|
bodyObj
|
|
);
|
|
|
|
// Per-connection memory config (#2942). Defaults preserve the legacy
|
|
// fresh-session-per-request, last-user-message-only behavior.
|
|
const psd = (rawCreds.providerSpecificData ?? {}) as Record<string, unknown>;
|
|
const persistSession = psd.persistSession === true;
|
|
const historyWindow =
|
|
typeof psd.historyWindow === "number" && psd.historyWindow > 0 ? psd.historyWindow : 0;
|
|
|
|
try {
|
|
let t0 = Date.now();
|
|
const accessToken = await acquireAccessToken(userToken, signal, log);
|
|
log?.info?.("DEEPSEEK-WEB", `Token acquired in ${Date.now() - t0}ms`);
|
|
|
|
// Tool (agentic) requests replay the whole trajectory — prior tool calls and their
|
|
// results — so the model keeps context across turns instead of restarting each time.
|
|
// Plain chat keeps the legacy last-user-message / rolling-window behavior.
|
|
const prompt = hasTools
|
|
? buildToolConversationPrompt(messages, toolSystemPrompt)
|
|
: messagesToPrompt(promptMessages, historyWindow);
|
|
const refFileIds = Array.isArray(bodyObj.ref_file_ids) ? bodyObj.ref_file_ids : [];
|
|
log?.info?.(
|
|
"DEEPSEEK-WEB",
|
|
`model_type=${modelType}, thinking=${thinkingEnabled}, search=${searchEnabled}, files=${refFileIds.length}, stream=${stream !== false}, persist=${persistSession}, window=${historyWindow}`
|
|
);
|
|
|
|
// One completion attempt against a given session id (fresh PoW per attempt).
|
|
const performCompletion = async (sid: string) => {
|
|
const powChallenge = await getPowChallenge(accessToken, signal);
|
|
const powAnswer = await solvePow(powChallenge);
|
|
const reqHeaders: Record<string, string> = {
|
|
...FAKE_HEADERS,
|
|
"Content-Type": "application/json",
|
|
Authorization: `Bearer ${accessToken}`,
|
|
"X-Ds-Pow-Response": powAnswer,
|
|
"X-Client-Timezone-Offset": String(new Date().getTimezoneOffset() * -60),
|
|
Cookie: generateFakeCookie(),
|
|
};
|
|
const requestPayload = {
|
|
chat_session_id: sid,
|
|
parent_message_id: null,
|
|
model_type: modelType,
|
|
prompt,
|
|
ref_file_ids: refFileIds,
|
|
thinking_enabled: thinkingEnabled,
|
|
search_enabled: searchEnabled,
|
|
preempt: false,
|
|
};
|
|
const resp = await fetch(COMPLETION_URL, {
|
|
method: "POST",
|
|
headers: reqHeaders,
|
|
body: JSON.stringify(requestPayload),
|
|
signal: signal ?? undefined,
|
|
});
|
|
return { resp, reqHeaders, requestPayload };
|
|
};
|
|
|
|
// Acquire a session. With persistSession we reuse one upstream session per
|
|
// userToken (rolling-window memory); otherwise we create a fresh one per
|
|
// request (legacy behavior — dodges stale sessions when the user deletes
|
|
// chats in the DeepSeek UI). (#2942)
|
|
const acquireSession = async (): Promise<{ sessionId: string; reused: boolean }> => {
|
|
if (persistSession) {
|
|
const cached = sessionCache.get(userToken);
|
|
if (cached) return { sessionId: cached.sessionId, reused: true };
|
|
const created = await createSession(accessToken, signal);
|
|
evictOldest(sessionCache);
|
|
sessionCache.set(userToken, { sessionId: created, createdAt: Date.now() });
|
|
return { sessionId: created, reused: false };
|
|
}
|
|
return { sessionId: await createSession(accessToken, signal), reused: false };
|
|
};
|
|
|
|
t0 = Date.now();
|
|
let { sessionId, reused: reusedSession } = await acquireSession();
|
|
log?.info?.(
|
|
"DEEPSEEK-WEB",
|
|
`Session ${reusedSession ? "reused" : "created"} in ${Date.now() - t0}ms`
|
|
);
|
|
|
|
t0 = Date.now();
|
|
log?.info?.("DEEPSEEK-WEB", `POST ${COMPLETION_URL}`);
|
|
let { resp, reqHeaders, requestPayload } = await performCompletion(sessionId);
|
|
log?.info?.(
|
|
"DEEPSEEK-WEB",
|
|
`Completion response in ${Date.now() - t0}ms, status=${resp.status}`
|
|
);
|
|
|
|
// A reused session that fails is likely stale (user deleted the chat in the
|
|
// DeepSeek UI). Drop it, create a fresh session, and retry once. (#2942)
|
|
if (!resp.ok && persistSession && reusedSession) {
|
|
log?.warn?.("DEEPSEEK-WEB", "Reused session failed — retrying with a fresh session");
|
|
sessionCache.delete(userToken);
|
|
sessionId = await createSession(accessToken, signal);
|
|
evictOldest(sessionCache);
|
|
sessionCache.set(userToken, { sessionId, createdAt: Date.now() });
|
|
reusedSession = false;
|
|
({ resp, reqHeaders, requestPayload } = await performCompletion(sessionId));
|
|
}
|
|
|
|
if (!resp.ok) {
|
|
const status = resp.status;
|
|
let errMsg = `DeepSeek API error (${status})`;
|
|
if (status === 401 || status === 403) {
|
|
tokenCache.delete(userToken);
|
|
errMsg = "DeepSeek token expired — get a fresh userToken from localStorage.";
|
|
} else if (status === 429) {
|
|
errMsg = "DeepSeek rate limited. Wait and retry.";
|
|
}
|
|
log?.warn?.("DEEPSEEK-WEB", errMsg);
|
|
|
|
try {
|
|
const errBody = await resp.json();
|
|
if (errBody?.code && errBody.code !== 0) {
|
|
errMsg = `DeepSeek error ${errBody.code}: ${errBody.msg}`;
|
|
}
|
|
} catch {
|
|
/* ignore */
|
|
}
|
|
|
|
if (persistSession) sessionCache.delete(userToken);
|
|
deleteSessionOnDeepSeek(accessToken, sessionId).catch(() => {});
|
|
return {
|
|
response: errorResponse(status, errMsg),
|
|
url: COMPLETION_URL,
|
|
headers: reqHeaders,
|
|
transformedBody: requestPayload,
|
|
};
|
|
}
|
|
|
|
// Check for HTTP 200 with DeepSeek error JSON
|
|
const ct = resp.headers.get("content-type") || "";
|
|
if (ct.includes("application/json")) {
|
|
try {
|
|
const json = await resp.json();
|
|
const parsed = parseDeepSeekErrorPayload(json);
|
|
if (parsed) {
|
|
const errMsg = `DeepSeek error ${parsed.code}: ${parsed.message}`;
|
|
log?.warn?.("DEEPSEEK-WEB", errMsg);
|
|
const status = parsed.code === 40003 ? 401 : parsed.code === 40002 ? 429 : 502;
|
|
if (parsed.code === 40003) {
|
|
tokenCache.delete(userToken);
|
|
}
|
|
if (persistSession) sessionCache.delete(userToken);
|
|
deleteSessionOnDeepSeek(accessToken, sessionId).catch(() => {});
|
|
return {
|
|
response: errorResponse(status, errMsg, parsed.code),
|
|
url: COMPLETION_URL,
|
|
headers: reqHeaders,
|
|
transformedBody: requestPayload,
|
|
};
|
|
}
|
|
if (!persistSession) deleteSessionOnDeepSeek(accessToken, sessionId).catch(() => {});
|
|
return {
|
|
response: new Response(JSON.stringify(json), {
|
|
status: 200,
|
|
headers: { "Content-Type": "application/json" },
|
|
}),
|
|
url: COMPLETION_URL,
|
|
headers: reqHeaders,
|
|
transformedBody: requestPayload,
|
|
};
|
|
} catch {
|
|
/* not JSON, continue */
|
|
}
|
|
}
|
|
|
|
// Persistent sessions are kept across requests for reuse; only delete the
|
|
// upstream chat session when persistence is off (legacy behavior). (#2942)
|
|
const cleanupFn = persistSession
|
|
? async () => {}
|
|
: () => deleteSessionOnDeepSeek(accessToken, sessionId);
|
|
|
|
const clientModel = typeof model === "string" && model.trim() ? model.trim() : "deepseek-web";
|
|
|
|
// Tool-call translation: buffer the full reply and parse <tool> blocks into
|
|
// OpenAI tool_calls. Buffering (even for stream clients) is acceptable because
|
|
// tool invocations are short and need the complete block to parse. (#2820)
|
|
if (hasTools) {
|
|
const { content, reasoningContent } = await collectSSEContent(resp.body!, clientModel);
|
|
await cleanupFn();
|
|
const { content: cleanedContent, toolCalls } = parseDeepSeekToolCalls(
|
|
content,
|
|
`call-${Date.now()}`,
|
|
requestedTools
|
|
);
|
|
return buildToolAwareResult({
|
|
stream: stream !== false,
|
|
clientModel,
|
|
content: cleanedContent,
|
|
reasoningContent,
|
|
toolCalls,
|
|
reqHeaders,
|
|
requestPayload,
|
|
});
|
|
}
|
|
|
|
if (stream !== false) {
|
|
const openaiStream = transformSSE(resp.body!, clientModel);
|
|
const wrappedStream = wrapStreamWithCleanup(openaiStream, cleanupFn);
|
|
return {
|
|
response: new Response(wrappedStream, {
|
|
status: 200,
|
|
headers: { "Content-Type": "text/event-stream", "Cache-Control": "no-cache" },
|
|
}),
|
|
url: COMPLETION_URL,
|
|
headers: reqHeaders,
|
|
transformedBody: requestPayload,
|
|
};
|
|
}
|
|
|
|
const { content, reasoningContent } = await collectSSEContent(resp.body!, clientModel);
|
|
await cleanupFn();
|
|
const message: Record<string, string> = { role: "assistant", content };
|
|
if (reasoningContent) message.reasoning_content = reasoningContent;
|
|
const openaiResponse = {
|
|
id: `chatcmpl-${Date.now()}`,
|
|
object: "chat.completion",
|
|
created: Math.floor(Date.now() / 1000),
|
|
model: model || modelType,
|
|
choices: [
|
|
{
|
|
index: 0,
|
|
message,
|
|
finish_reason: "stop",
|
|
},
|
|
],
|
|
usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
|
|
};
|
|
return {
|
|
response: new Response(JSON.stringify(openaiResponse), {
|
|
status: 200,
|
|
headers: { "Content-Type": "application/json" },
|
|
}),
|
|
url: COMPLETION_URL,
|
|
headers: reqHeaders,
|
|
transformedBody: requestPayload,
|
|
};
|
|
} catch (err) {
|
|
const msg = err instanceof Error ? err.message : String(err);
|
|
log?.error?.("DEEPSEEK-WEB", `Execute failed: ${msg}`);
|
|
|
|
if (err instanceof DOMException && err.name === "AbortError") {
|
|
return {
|
|
response: errorResponse(499, "Request cancelled"),
|
|
url: COMPLETION_URL,
|
|
headers: {},
|
|
transformedBody: body,
|
|
};
|
|
}
|
|
|
|
return {
|
|
response: errorResponse(502, `DeepSeek error: ${sanitizeErrorMessage(msg)}`),
|
|
url: COMPLETION_URL,
|
|
headers: {},
|
|
transformedBody: body,
|
|
};
|
|
}
|
|
}
|
|
}
|
|
|
|
export const deepseekWebExecutor = new DeepSeekWebExecutor();
|
|
|
|
// Re-export for auto-refresh executor and tests
|
|
export { acquireAccessToken, tokenCache, sessionCache };
|