@thanh01.pmt/curriculum-kit 1.0.14 → 1.0.16

package/dist/media/index.d.cts

@@ -1,4 +1,4 @@
-import { S as StreamRunnerOptions } from '../streamRunner-CcpmxOf1.cjs';
+import { S as StreamRunnerOptions } from '../streamRunner-DtQq0HV_.cjs';
 
 /**
  * Media Types — image generation, image search, video search cho học liệu.
@@ -57,6 +57,8 @@ interface GenerateImageOptions {
     provider?: MediaProvider;
     /** API key override (nếu không dùng env) */
     apiKey?: string;
+    /** Timeout tổng cho request sinh ảnh (mặc định 420_000ms = 7 phút) */
+    timeoutMs?: number;
 }
 interface SearchImageOptions {
     query: string;

package/dist/media/index.d.ts

@@ -1,4 +1,4 @@
-import { S as StreamRunnerOptions } from '../streamRunner-CcpmxOf1.js';
+import { S as StreamRunnerOptions } from '../streamRunner-DtQq0HV_.js';
 
 /**
  * Media Types — image generation, image search, video search cho học liệu.
@@ -57,6 +57,8 @@ interface GenerateImageOptions {
     provider?: MediaProvider;
     /** API key override (nếu không dùng env) */
     apiKey?: string;
+    /** Timeout tổng cho request sinh ảnh (mặc định 420_000ms = 7 phút) */
+    timeoutMs?: number;
 }
 interface SearchImageOptions {
     query: string;

package/dist/media/index.mjs

@@ -41,6 +41,10 @@ function resolveApiKey(options) {
   }
   return { provider, key };
 }
+function resolveImageTimeout(options) {
+  const envTimeout = Number(process.env.IMAGE_GEN_TIMEOUT_MS || process.env.ARTIFACT_STREAM_TOTAL_TIMEOUT_MS);
+  return Number.isFinite(options.timeoutMs) && options.timeoutMs > 0 ? options.timeoutMs : Number.isFinite(envTimeout) && envTimeout > 0 ? envTimeout : 42e4;
+}
 async function generateWithGemini(options, key) {
   const finalPrompt = buildImagePrompt(options);
   const model = "gemini-2.0-flash-exp";
@@ -51,7 +55,7 @@ async function generateWithGemini(options, key) {
       contents: [{ parts: [{ text: finalPrompt }] }],
       generationConfig: { responseModalities: ["TEXT", "IMAGE"] }
     }),
-    signal: AbortSignal.timeout(9e4)
+    signal: AbortSignal.timeout(resolveImageTimeout(options))
   });
   if (!res.ok) {
     const body = await res.text().catch(() => "");
@@ -84,7 +88,7 @@ async function generateWithOpenAI(options, key) {
       size: sizeMap[options.aspectRatio || "1:1"] || "1024x1024",
       n: 1
     }),
-    signal: AbortSignal.timeout(9e4)
+    signal: AbortSignal.timeout(resolveImageTimeout(options))
   });
   if (!res.ok) {
     const body = await res.text().catch(() => "");
@@ -495,26 +499,40 @@ function extractThoughtAndContent(rawText) {
     content: content.trim()
   };
 }
-function createIdleAbortController(idleMs) {
+var DEFAULT_ARTIFACT_STREAM_IDLE_MS = 45e3;
+var DEFAULT_ARTIFACT_STREAM_TOTAL_MS = 42e4;
+function createStreamAbortController(options) {
+  const envIdle = Number(process.env.ARTIFACT_STREAM_IDLE_TIMEOUT_MS || process.env.STREAM_IDLE_TIMEOUT_MS);
+  const envTotal = Number(process.env.ARTIFACT_STREAM_TOTAL_TIMEOUT_MS || process.env.STREAM_TOTAL_TIMEOUT_MS);
+  const idleMs = Number.isFinite(options?.idleMs) && options.idleMs > 0 ? options.idleMs : Number.isFinite(envIdle) && envIdle > 0 ? envIdle : DEFAULT_ARTIFACT_STREAM_IDLE_MS;
+  const totalMs = Number.isFinite(options?.totalMs) && options.totalMs > 0 ? options.totalMs : Number.isFinite(envTotal) && envTotal > 0 ? envTotal : DEFAULT_ARTIFACT_STREAM_TOTAL_MS;
   const controller = new AbortController();
-  let timer = null;
-  const arm = () => {
-    if (timer) clearTimeout(timer);
-    timer = setTimeout(
+  let idleTimer = null;
+  let totalTimer = null;
+  const armIdle = () => {
+    if (idleTimer) clearTimeout(idleTimer);
+    idleTimer = setTimeout(
       () => controller.abort(new Error(`Idle timeout: no data for ${Math.round(idleMs / 1e3)}s`)),
       idleMs
     );
-    timer?.unref?.();
+    idleTimer?.unref?.();
   };
-  arm();
+  armIdle();
+  if (totalMs > 0) {
+    totalTimer = setTimeout(
+      () => controller.abort(new Error(`Total stream timeout after ${Math.round(totalMs / 1e3)}s`)),
+      totalMs
+    );
+    totalTimer?.unref?.();
+  }
   return {
     signal: controller.signal,
-    /** Reset the idle window (call on every received chunk). */
-    kick: arm,
-    /** Clear the pending timer once the request lifecycle is over. */
+    kick: armIdle,
     dispose: () => {
-      if (timer) clearTimeout(timer);
-      timer = null;
+      if (idleTimer) clearTimeout(idleTimer);
+      if (totalTimer) clearTimeout(totalTimer);
+      idleTimer = null;
+      totalTimer = null;
     }
   };
 }
@@ -697,7 +715,7 @@ Guidelines:
     if (isNvidiaPreferred || uniqueNvidiaModels.length > 0) {
       for (const model of uniqueNvidiaModels) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch("https://integrate.api.nvidia.com/v1/chat/completions", {
             method: "POST",
             headers: {
@@ -741,7 +759,7 @@ Guidelines:
     const uniqueOrModels = Array.from(new Set(openrouterModels));
     for (const model of uniqueOrModels) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const res = await fetch("https://openrouter.ai/api/v1/chat/completions", {
           method: "POST",
           headers: {
@@ -780,7 +798,7 @@ Guidelines:
     const dsModel = options.model?.includes("deepseek-reasoner") ? "deepseek-reasoner" : "deepseek-chat";
     if (isModelAllowed(dsModel)) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const res = await fetch("https://api.deepseek.com/chat/completions", {
           method: "POST",
           headers: {
@@ -815,7 +833,7 @@ Guidelines:
     const baseUrl = alibabaKey.startsWith("sk-sp-") ? "https://token-plan.ap-southeast-1.maas.aliyuncs.com/compatible-mode/v1/chat/completions" : "https://dashscope-intl.aliyuncs.com/compatible-mode/v1/chat/completions";
     for (const qwenModel of qwenModels) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const payload = {
           model: qwenModel,
           messages: formattedMessages,
@@ -858,7 +876,7 @@ Guidelines:
     ];
     for (const gModel of geminiModels) {
       try {
-        const idle = createIdleAbortController(18e4);
+        const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
         const res = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/${gModel}:streamGenerateContent?alt=sse&key=${geminiKey}`, {
           method: "POST",
           headers: { "Content-Type": "application/json" },
@@ -894,7 +912,7 @@ Guidelines:
       const { provider, model } = target;
       if (provider === "nvidia" && nvidiaKey && isProviderEnabled("nvidia")) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch("https://integrate.api.nvidia.com/v1/chat/completions", {
             method: "POST",
             headers: {
@@ -926,7 +944,7 @@ Guidelines:
         }
       } else if (provider === "openrouter" && openrouterKey && isProviderEnabled("openrouter")) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch("https://openrouter.ai/api/v1/chat/completions", {
             method: "POST",
             headers: {
@@ -961,7 +979,7 @@ Guidelines:
       } else if ((provider === "alibaba" || provider === "dashscope") && alibabaKey && isProviderEnabled("alibaba")) {
         const baseUrl = alibabaKey.startsWith("sk-sp-") ? "https://token-plan.ap-southeast-1.maas.aliyuncs.com/compatible-mode/v1/chat/completions" : "https://dashscope-intl.aliyuncs.com/compatible-mode/v1/chat/completions";
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch(baseUrl, {
             method: "POST",
             headers: {
@@ -993,7 +1011,7 @@ Guidelines:
         }
       } else if ((provider === "google" || provider === "gemini") && geminiKey && (isProviderEnabled("gemini") || isProviderEnabled("google"))) {
         try {
-          const idle = createIdleAbortController(18e4);
+          const idle = createStreamAbortController({ idleMs: options.idleTimeoutMs, totalMs: options.timeoutMs });
           const res = await fetch(
             `https://generativelanguage.googleapis.com/v1beta/models/${model}:streamGenerateContent?alt=sse&key=${geminiKey}`,
             {