netra-sdk 1.1.0 → 1.2.0

package/dist/index.cjs

@@ -24997,7 +24997,130 @@ async function uninstrumentAll() {
     Logger.debug("Failed to uninstrument undici:", e);
   }
 }
+
+// src/simulation/task.ts
+var BaseTask = class {
+};
+
+// src/simulation/utils.ts
 var LOG_PREFIX = "netra.simulation";
+var DEFAULT_FILE_DOWNLOAD_TIMEOUT_S = 30;
+var MAX_FILE_DOWNLOAD_WORKERS = 8;
+var _cachedFileDownloadTimeoutMs = null;
+function validateSimulationInputs(datasetId, task2) {
+  if (!datasetId) {
+    Logger.error(`${LOG_PREFIX}: dataset_id is required`);
+    return false;
+  }
+  if (!(task2 instanceof BaseTask)) {
+    Logger.error(`${LOG_PREFIX}: task must be a BaseTask instance`);
+    return false;
+  }
+  return true;
+}
+function _getFileDownloadTimeout() {
+  if (_cachedFileDownloadTimeoutMs !== null) {
+    return _cachedFileDownloadTimeoutMs;
+  }
+  const envVal = process.env.NETRA_SIMULATION_FILE_DOWNLOAD_TIMEOUT;
+  if (envVal) {
+    const parsed = parseFloat(envVal);
+    if (!isNaN(parsed) && parsed > 0) {
+      _cachedFileDownloadTimeoutMs = parsed * 1e3;
+      return _cachedFileDownloadTimeoutMs;
+    }
+    Logger.warn(
+      `${LOG_PREFIX}: Invalid file download timeout '${envVal}', using default ${DEFAULT_FILE_DOWNLOAD_TIMEOUT_S}s`
+    );
+  }
+  _cachedFileDownloadTimeoutMs = DEFAULT_FILE_DOWNLOAD_TIMEOUT_S * 1e3;
+  return _cachedFileDownloadTimeoutMs;
+}
+async function _downloadSingleFile(fileData, timeoutMs, signal) {
+  try {
+    const response = await axios__default.default.get(fileData.downloadUrl, {
+      responseType: "arraybuffer",
+      timeout: timeoutMs,
+      signal
+    });
+    const encoded = Buffer.from(response.data).toString("base64");
+    return {
+      fileName: fileData.fileName,
+      contentType: fileData.contentType,
+      description: fileData.description,
+      data: encoded
+    };
+  } catch (error) {
+    if (axios__default.default.isCancel(error)) {
+      throw new Error(
+        `Download of '${fileData.fileName}' was cancelled`
+      );
+    }
+    const status = axios__default.default.isAxiosError(error) ? error.response?.status : void 0;
+    const reason = status ? `HTTP ${status}` : error instanceof Error ? error.message : String(error);
+    Logger.error(
+      `${LOG_PREFIX}: Failed to download file '${fileData.fileName}': ${reason}`
+    );
+    throw new Error(
+      `Failed to download file '${fileData.fileName}': ${reason}`
+    );
+  }
+}
+function parseFiles(rawFiles) {
+  if (!rawFiles) {
+    return [];
+  }
+  const parsed = [];
+  for (const entry of rawFiles) {
+    const fileName = entry.fileName || "";
+    const downloadUrl = entry.downloadUrl || "";
+    if (!fileName || !downloadUrl) {
+      Logger.warn(
+        `${LOG_PREFIX}: Skipping malformed file attachment (missing fileName or downloadUrl)`
+      );
+      continue;
+    }
+    parsed.push({
+      fileName,
+      contentType: entry.contentType || "",
+      description: entry.description || void 0,
+      downloadUrl
+    });
+  }
+  return parsed;
+}
+async function processFiles(files) {
+  if (!files || files.length === 0) {
+    return null;
+  }
+  const timeoutMs = _getFileDownloadTimeout();
+  const limit = pLimit__default.default(MAX_FILE_DOWNLOAD_WORKERS);
+  const controller = new AbortController();
+  const downloadPromises = files.map(
+    (file) => limit(() => _downloadSingleFile(file, timeoutMs, controller.signal))
+  );
+  try {
+    return await Promise.all(downloadPromises);
+  } catch (error) {
+    controller.abort();
+    limit.clearQueue();
+    throw error;
+  }
+}
+async function executeTask2(task2, message, sessionId, rawFiles) {
+  const processedFiles = rawFiles && rawFiles.length > 0 ? await processFiles(rawFiles) : null;
+  const result = task2.run(message, sessionId, processedFiles);
+  const resolvedResult = result instanceof Promise ? await result : result;
+  if (typeof resolvedResult === "object" && resolvedResult !== null && "message" in resolvedResult && "sessionId" in resolvedResult) {
+    return [resolvedResult.message, resolvedResult.sessionId];
+  }
+  throw new Error(
+    `Task must return TaskResult, got ${typeof resolvedResult}`
+  );
+}
+
+// src/simulation/client.ts
+var LOG_PREFIX2 = "netra.simulation";
 var DEFAULT_TIMEOUT = 1e4;
 var SimulationHttpClient = class {
   constructor(config2) {
@@ -25010,7 +25133,7 @@ var SimulationHttpClient = class {
   _createClient(config2) {
     const endpoint = (config2.otlpEndpoint || "").trim();
     if (!endpoint) {
-      Logger.error(`${LOG_PREFIX}: NETRA_OTLP_ENDPOINT is required`);
+      Logger.error(`${LOG_PREFIX2}: NETRA_OTLP_ENDPOINT is required`);
       return null;
     }
     const baseURL = this._resolveBaseUrl(endpoint);
@@ -25032,7 +25155,7 @@ var SimulationHttpClient = class {
       );
       return instance;
     } catch (error) {
-      Logger.error(`${LOG_PREFIX}: Failed to create HTTP client:`, error);
+      Logger.error(`${LOG_PREFIX2}: Failed to create HTTP client:`, error);
       return null;
     }
   }
@@ -25067,7 +25190,7 @@ var SimulationHttpClient = class {
     const timeout = parseFloat(timeoutStr);
     if (isNaN(timeout)) {
       Logger.warn(
-        `${LOG_PREFIX}: Invalid timeout '${timeoutStr}', using default ${DEFAULT_TIMEOUT}ms`
+        `${LOG_PREFIX2}: Invalid timeout '${timeoutStr}', using default ${DEFAULT_TIMEOUT}ms`
       );
       return DEFAULT_TIMEOUT;
     }
@@ -25078,7 +25201,7 @@ var SimulationHttpClient = class {
    */
   async createRun(name, datasetId, context17) {
     if (!this.client) {
-      Logger.error(`${LOG_PREFIX}: Client not initialized`);
+      Logger.error(`${LOG_PREFIX2}: Client not initialized`);
       return null;
     }
     try {
@@ -25093,7 +25216,7 @@ var SimulationHttpClient = class {
       const responseData = data.data || {};
       const userMessages = responseData.userMessages || [];
       if (userMessages.length === 0) {
-        Logger.warn(`${LOG_PREFIX}: No user messages returned from create_run`);
+        Logger.warn(`${LOG_PREFIX2}: No user messages returned from create_run`);
         return null;
       }
       const runId = responseData.id || "";
@@ -25101,7 +25224,8 @@ var SimulationHttpClient = class {
         (msg) => ({
           runItemId: msg.testRunItemId || "",
           message: msg.userMessage || "",
-          turnId: msg.turnId || ""
+          turnId: msg.turnId || "",
+          files: parseFiles(msg.attachments)
         })
       );
       return {
@@ -25110,7 +25234,7 @@ var SimulationHttpClient = class {
       };
     } catch (error) {
       const errorMsg = this._extractErrorMessage(error);
-      Logger.error(`${LOG_PREFIX}: Failed to create simulation run:`, errorMsg);
+      Logger.error(`${LOG_PREFIX2}: Failed to create simulation run:`, errorMsg);
       return null;
     }
   }
@@ -25119,7 +25243,7 @@ var SimulationHttpClient = class {
    */
   async triggerConversation(message, turnId, sessionId, traceId) {
     if (!this.client) {
-      Logger.error(`${LOG_PREFIX}: Client not initialized`);
+      Logger.error(`${LOG_PREFIX2}: Client not initialized`);
       return null;
     }
     try {
@@ -25142,7 +25266,7 @@ var SimulationHttpClient = class {
       }
       const userMessages = responseData.userMessages || [];
       if (userMessages.length === 0) {
-        Logger.warn(`${LOG_PREFIX}: No user messages in continue response`);
+        Logger.warn(`${LOG_PREFIX2}: No user messages in continue response`);
         return null;
       }
       const nextMsg = userMessages[0];
@@ -25150,11 +25274,12 @@ var SimulationHttpClient = class {
         decision,
         nextTurnId: nextMsg.turnId || "",
         nextUserMessage: nextMsg.userMessage || "",
-        nextRunItemId: nextMsg.testRunItemId || ""
+        nextRunItemId: nextMsg.testRunItemId || "",
+        nextFiles: parseFiles(nextMsg.attachments)
       };
     } catch (error) {
       const errorMsg = this._extractErrorMessage(error);
-      Logger.error(`${LOG_PREFIX}: Failed to trigger conversation:`, errorMsg);
+      Logger.error(`${LOG_PREFIX2}: Failed to trigger conversation:`, errorMsg);
       return null;
     }
   }
@@ -25163,17 +25288,17 @@ var SimulationHttpClient = class {
    */
   async reportFailure(runId, runItemId, error) {
     if (!this.client) {
-      Logger.error(`${LOG_PREFIX}: Client not initialized`);
+      Logger.error(`${LOG_PREFIX2}: Client not initialized`);
       return;
     }
     try {
       const url = `/evaluations/run/${runId}/item/${runItemId}/status`;
       const payload = { status: "failed", failureReason: error };
       await this.client.patch(url, payload);
-      Logger.info(`${LOG_PREFIX}: Reported failure - ${error}`);
+      Logger.info(`${LOG_PREFIX2}: Reported failure - ${error}`);
     } catch (err) {
       const errorMsg = this._extractErrorMessage(err);
-      Logger.error(`${LOG_PREFIX}: Failed to report failure:`, errorMsg);
+      Logger.error(`${LOG_PREFIX2}: Failed to report failure:`, errorMsg);
     }
   }
   /**
@@ -25182,7 +25307,7 @@ var SimulationHttpClient = class {
   async postRunStatus(runId, status) {
     if (!this.client) {
       Logger.error(
-        `${LOG_PREFIX}: Client not initialized; cannot post run status`
+        `${LOG_PREFIX2}: Client not initialized; cannot post run status`
       );
       return { success: false };
     }
@@ -25192,14 +25317,14 @@ var SimulationHttpClient = class {
       const response = await this.client.post(url, payload);
       const data = response.data;
       if (data && typeof data === "object" && "data" in data) {
-        Logger.info(`${LOG_PREFIX}: Completed test run successfully`);
+        Logger.info(`${LOG_PREFIX2}: Completed test run successfully`);
         return data.data || {};
       }
       return data;
     } catch (error) {
       const errorMsg = this._extractErrorMessage(error);
       Logger.error(
-        `${LOG_PREFIX}: Failed to post run status for run '${runId}':`,
+        `${LOG_PREFIX2}: Failed to post run status for run '${runId}':`,
         errorMsg
       );
       return { success: false };
@@ -25223,34 +25348,6 @@ var SimulationHttpClient = class {
   }
 };
 
-// src/simulation/task.ts
-var BaseTask = class {
-};
-
-// src/simulation/utils.ts
-var LOG_PREFIX2 = "netra.simulation";
-function validateSimulationInputs(datasetId, task2) {
-  if (!datasetId) {
-    Logger.error(`${LOG_PREFIX2}: dataset_id is required`);
-    return false;
-  }
-  if (!(task2 instanceof BaseTask)) {
-    Logger.error(`${LOG_PREFIX2}: task must be a BaseTask instance`);
-    return false;
-  }
-  return true;
-}
-async function executeTask2(task2, message, sessionId) {
-  const result = task2.run(message, sessionId);
-  const resolvedResult = result instanceof Promise ? await result : result;
-  if (typeof resolvedResult === "object" && resolvedResult !== null && "message" in resolvedResult && "sessionId" in resolvedResult) {
-    return [resolvedResult.message, resolvedResult.sessionId];
-  }
-  throw new Error(
-    `Task must return TaskResult, got ${typeof resolvedResult}`
-  );
-}
-
 // src/simulation/api.ts
 var LOG_PREFIX3 = "netra.simulation";
 var SPAN_NAME = "Netra.Simulation.TestRun";
@@ -25384,17 +25481,18 @@ var Simulation = class {
    * Execute a multi-turn conversation for a single simulation item.
    */
   async _executeConversation(runId, runItem, task2) {
-    const { runItemId, message: initialMessage, turnId: initialTurnId } = runItem;
+    const { runItemId, message: initialMessage, turnId: initialTurnId, files: initialFiles } = runItem;
     let message = initialMessage;
     let turnId = initialTurnId;
     let sessionId = null;
+    let rawFiles = initialFiles ?? [];
     while (true) {
       const span2 = new SpanWrapper(SPAN_NAME, {}, LOG_PREFIX3);
       span2.start();
       try {
         const traceId = span2.getCurrentSpan()?.spanContext().traceId ?? "";
         const [responseMessage, taskSessionId] = await span2.withActive(
-          () => executeTask2(task2, message, sessionId)
+          () => executeTask2(task2, message, sessionId, rawFiles)
         );
         if (taskSessionId) {
           sessionId = taskSessionId;
@@ -25428,6 +25526,7 @@ var Simulation = class {
         }
         message = response.nextUserMessage;
         turnId = response.nextTurnId;
+        rawFiles = response.nextFiles || [];
       } catch (error) {
         const errorMsg = error instanceof Error ? error.message : String(error);
         Logger.error(