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) {
if (!fs.existsSync(dirPath)) {
fs.mkdirSync(dirPath, {recursive: true});
@@ -25,17 +26,21 @@ function extractCookie(cookies) {
let jwtToken = null;
let ds = null;
let dsr = null;
let you_subscription = null;
let youpro_subscription = null;
cookies = cookie.parse(cookies);
if (cookies["stytch_session"]) jwtSession = cookies["stytch_session"];
if (cookies["stytch_session_jwt"]) jwtToken = cookies["stytch_session_jwt"];
if (cookies["DS"]) ds = cookies["DS"];
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 = [];
// 处理旧版 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(如果启用)
if (process.env.INCOGNITO_MODE === "true") {
sessionCookie.push({
@@ -127,6 +158,7 @@ function getSessionCookie(jwtSession, jwtToken, ds, dsr) {
function sleep(ms) {
return new Promise((resolve) => setTimeout(resolve, ms));
}
function createDocx(content) {
let paragraphs = [];
content.split("\n").forEach((line) => {
@@ -146,6 +178,7 @@ function createDocx(content) {
});
return docx.Packer.toBuffer(doc).then((buffer) => buffer);
}
// eventStream util
function createEvent(event, data) {
// if data is object, stringify it