@wix/evalforge-evaluator 0.221.0 → 0.222.0

package/build/index.js

@@ -6941,6 +6941,12 @@ function rethrowWithRequestId(err, action) {
   }
   throw err;
 }
+function requireEntity(entity, notFound) {
+  if (!entity) {
+    throw new Error(notFound);
+  }
+  return entity;
+}
 function resolveAmbassadorBaseUrl(serverUrl) {
   try {
     return new URL(serverUrl).origin;
@@ -6991,52 +6997,50 @@ function createApiClient(serverUrl, options = "") {
       const res = await httpClient.request(
         getEvalRun({ projectId: projectId2, evalRunId: id })
       );
-      const evalRun = res.data.evalRun;
-      if (!evalRun) {
-        throw new Error(`Eval run ${id} not found in project ${projectId2}`);
-      }
+      const evalRun = requireEntity(
+        res.data.evalRun,
+        `Eval run ${id} not found in project ${projectId2}`
+      );
       return evalRunFromProto(evalRun);
     },
     async getScenario(projectId2, id) {
       const res = await httpClient.request(
         getTestScenario({ projectId: projectId2, testScenarioId: id })
       );
-      const scenario = res.data.testScenario;
-      if (!scenario) {
-        throw new Error(
-          `Test scenario ${id} not found in project ${projectId2}`
-        );
-      }
+      const scenario = requireEntity(
+        res.data.testScenario,
+        `Test scenario ${id} not found in project ${projectId2}`
+      );
       return testScenarioFromProto(scenario);
     },
     async getAgent(projectId2, id) {
       const res = await httpClient.request(
         getAgent({ projectId: projectId2, agentId: id })
       );
-      const agent = res.data.agent;
-      if (!agent) {
-        throw new Error(`Agent ${id} not found in project ${projectId2}`);
-      }
+      const agent = requireEntity(
+        res.data.agent,
+        `Agent ${id} not found in project ${projectId2}`
+      );
       return agentFromProto(agent);
     },
     async getTemplate(projectId2, id) {
       const res = await httpClient.request(
         getTemplate({ projectId: projectId2, templateId: id })
       );
-      const template = res.data.template;
-      if (!template) {
-        throw new Error(`Template ${id} not found in project ${projectId2}`);
-      }
+      const template = requireEntity(
+        res.data.template,
+        `Template ${id} not found in project ${projectId2}`
+      );
       return templateFromProto(template);
     },
     async getPreset(projectId2, id) {
       const res = await httpClient.request(
         getPreset({ projectId: projectId2, presetId: id })
       );
-      const preset = res.data.preset;
-      if (!preset) {
-        throw new Error(`Preset ${id} not found in project ${projectId2}`);
-      }
+      const preset = requireEntity(
+        res.data.preset,
+        `Preset ${id} not found in project ${projectId2}`
+      );
       return presetFromProto(preset);
     },
     // The legacy REST endpoint enriched the capability with its latest version
@@ -7057,10 +7061,10 @@ function createApiClient(serverUrl, options = "") {
       if (capResult.status === "rejected") {
         throw capResult.reason;
       }
-      const capability = capResult.value.data.capability;
-      if (!capability) {
-        throw new Error(`Capability ${id} not found in project ${projectId2}`);
-      }
+      const capability = requireEntity(
+        capResult.value.data.capability,
+        `Capability ${id} not found in project ${projectId2}`
+      );
       let latestVersion;
       if (versionResult.status === "fulfilled" && versionResult.value.data.capabilityVersion) {
         latestVersion = capabilityVersionFromProto(
@@ -7083,12 +7087,10 @@ function createApiClient(serverUrl, options = "") {
           capabilityVersionId: versionId
         })
       );
-      const version = res.data.capabilityVersion;
-      if (!version) {
-        throw new Error(
-          `Capability version ${versionId} not found for capability ${capabilityId}`
-        );
-      }
+      const version = requireEntity(
+        res.data.capabilityVersion,
+        `Capability version ${versionId} not found for capability ${capabilityId}`
+      );
       return capabilityVersionFromProto(version, projectId2);
     },
     async addResult(projectId2, evalRunId2, result) {