fix Custom Agents not working

This commit is contained in:
被遗忘的记忆
2025-03-10 00:33:43 +08:00
committed by GitHub
parent a2a52df40d
commit b9bf36b0ad
+35 -2
View File
@@ -14,6 +14,7 @@ function getGitRevision() {
} }
} }
// 创建目录
function createDirectoryIfNotExists(dirPath) { function createDirectoryIfNotExists(dirPath) {
if (!fs.existsSync(dirPath)) { if (!fs.existsSync(dirPath)) {
fs.mkdirSync(dirPath, {recursive: true}); fs.mkdirSync(dirPath, {recursive: true});
@@ -25,17 +26,21 @@ function extractCookie(cookies) {
let jwtToken = null; let jwtToken = null;
let ds = null; let ds = null;
let dsr = null; let dsr = null;
let you_subscription = null;
let youpro_subscription = null;
cookies = cookie.parse(cookies); cookies = cookie.parse(cookies);
if (cookies["stytch_session"]) jwtSession = cookies["stytch_session"]; if (cookies["stytch_session"]) jwtSession = cookies["stytch_session"];
if (cookies["stytch_session_jwt"]) jwtToken = cookies["stytch_session_jwt"]; if (cookies["stytch_session_jwt"]) jwtToken = cookies["stytch_session_jwt"];
if (cookies["DS"]) ds = cookies["DS"]; if (cookies["DS"]) ds = cookies["DS"];
if (cookies["DSR"]) dsr = cookies["DSR"]; if (cookies["DSR"]) dsr = cookies["DSR"];
if (cookies["you_subscription"]) you_subscription = cookies["you_subscription"];
if (cookies["youpro_subscription"]) youpro_subscription = cookies["youpro_subscription"];
return { jwtSession, jwtToken, ds, dsr }; return {jwtSession, jwtToken, ds, dsr, you_subscription, youpro_subscription};
} }
function getSessionCookie(jwtSession, jwtToken, ds, dsr) { function getSessionCookie(jwtSession, jwtToken, ds, dsr, you_subscription, youpro_subscription) {
let sessionCookie = []; let sessionCookie = [];
// 处理旧版 cookie // 处理旧版 cookie
@@ -110,6 +115,32 @@ function getSessionCookie(jwtSession, jwtToken, ds, dsr) {
}); });
} }
if (you_subscription) {
sessionCookie.push({
name: "you_subscription",
value: you_subscription,
domain: "you.com",
path: "/",
expires: 1800000000,
httpOnly: false,
secure: true,
sameSite: "Lax",
});
}
if (youpro_subscription) {
sessionCookie.push({
name: "youpro_subscription",
value: youpro_subscription,
domain: "you.com",
path: "/",
expires: 1800000000,
httpOnly: false,
secure: true,
sameSite: "Lax",
});
}
// 添加隐身模式 cookie(如果启用) // 添加隐身模式 cookie(如果启用)
if (process.env.INCOGNITO_MODE === "true") { if (process.env.INCOGNITO_MODE === "true") {
sessionCookie.push({ sessionCookie.push({
@@ -127,6 +158,7 @@ function getSessionCookie(jwtSession, jwtToken, ds, dsr) {
function sleep(ms) { function sleep(ms) {
return new Promise((resolve) => setTimeout(resolve, ms)); return new Promise((resolve) => setTimeout(resolve, ms));
} }
function createDocx(content) { function createDocx(content) {
let paragraphs = []; let paragraphs = [];
content.split("\n").forEach((line) => { content.split("\n").forEach((line) => {
@@ -146,6 +178,7 @@ function createDocx(content) {
}); });
return docx.Packer.toBuffer(doc).then((buffer) => buffer); return docx.Packer.toBuffer(doc).then((buffer) => buffer);
} }
// eventStream util // eventStream util
function createEvent(event, data) { function createEvent(event, data) {
// if data is object, stringify it // if data is object, stringify it