fix Custom Agents not working

utils.mjs

@@ -1,22 +1,23 @@
 import * as docx from "docx";
 import cookie from "cookie";
 import fs from "fs";
-import { execSync } from "child_process";
+import {execSync} from "child_process";
 
 function getGitRevision() {
     // get git revision and branch
     try {
-		const revision = execSync("git rev-parse --short HEAD", { stdio: "pipe" }).toString().trim();
+        const revision = execSync("git rev-parse --short HEAD", {stdio: "pipe"}).toString().trim();
-		const branch = execSync("git rev-parse --abbrev-ref HEAD", { stdio: "pipe" }).toString().trim();
+        const branch = execSync("git rev-parse --abbrev-ref HEAD", {stdio: "pipe"}).toString().trim();
-		return { revision, branch };
+        return {revision, branch};
     } catch (e) {
-		return { revision: "unknown", branch: "unknown" };
+        return {revision: "unknown", branch: "unknown"};
     }
 }
 
+// 创建目录
 function createDirectoryIfNotExists(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 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,8 +115,34 @@ 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") {
+    if (process.env.INCOGNITO_MODE === "true") {
         sessionCookie.push({
             name: "incognito",
             value: "true",
@@ -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