axiom 0.19.0 → 0.20.0

package/dist/bin.cjs

@@ -349,8 +349,8 @@ var loadPullCommand = (program2) => {
   ).option("--version <version>", "The version to pull, default: latest", "latest").option("--output <path>", "Output file path (optional, defaults to <slug>.prompt.ts)").action(async (slug, options) => {
     try {
       console.log(`Pulling prompt: ${slug} (version: ${options.version})`);
-      const url2 = `${process.env.AXIOM_URL}/v1/prompts/${slug}`;
-      const response = await fetch(url2, {
+      const url = `${process.env.AXIOM_URL}/v1/prompts/${slug}`;
+      const response = await fetch(url, {
         method: "GET",
         headers: {
           Authorization: `Bearer ${process.env.AXIOM_TOKEN}`,
@@ -479,11 +479,16 @@ var r = process.env.FORCE_TTY !== void 0 || (0, import_tty.isatty)(1);
 var u = p(r);
 
 // src/evals/eval.service.ts
-var datasetName = process.env.AXIOM_DATASET ?? "";
-var url = process.env.AXIOM_URL ?? "https://api.axiom.co";
-var token = process.env.AXIOM_TOKEN;
+var getEnvVars = () => {
+  return {
+    datasetName: process.env.AXIOM_DATASET ?? "",
+    url: process.env.AXIOM_URL ?? "https://api.axiom.co",
+    token: process.env.AXIOM_TOKEN
+  };
+};
 var findEvaluationCases = async (evalId) => {
   try {
+    const { datasetName, url, token } = getEnvVars();
     const apl = `['${datasetName}'] | where trace_id == "${evalId}" | order by _time`;
     const headers = new Headers({
       Authorization: `Bearer ${token}`,
@@ -689,7 +694,7 @@ var AxiomReporter = class {
     console.log(" ", u.dim("Start at"), new Date(this.startTime).toTimeString());
     console.log(" ", u.dim("Duration"), `${duration}s`);
     const meta = testSuite.meta();
-    const url2 = `https://app.axiom.co/evaluations/${meta.evaluation.name}/${meta.evaluation.id}`;
+    const url = `https://app.axiom.co/evaluations/${meta.evaluation.name}/${meta.evaluation.id}`;
     for (const test of testSuite.children) {
       if (test.type !== "test") return;
       this.printCaseResult(test);
@@ -697,7 +702,7 @@ var AxiomReporter = class {
     console.log("");
     console.log(
       " ",
-      `see results for ${meta.evaluation.name}-${meta.evaluation.version} at ${url2}`
+      `see results for ${meta.evaluation.name}-${meta.evaluation.version} at ${url}`
     );
     console.log(
       " ",
@@ -772,7 +777,7 @@ var import_api2 = require("@opentelemetry/api");
 // package.json
 var package_default = {
   name: "axiom",
-  version: "0.19.0",
+  version: "0.20.0",
   type: "module",
   author: "Axiom, Inc.",
   contributors: [
@@ -934,12 +939,12 @@ var processor = new import_sdk_trace_node.BatchSpanProcessor(exporter, {
 var provider = new import_sdk_trace_node.NodeTracerProvider({
   resource: (0, import_resources.resourceFromAttributes)({
     ["service.name"]: "axiom",
-    ["service.version"]: "0.19.0"
+    ["service.version"]: "0.20.0"
   }),
   spanProcessors: [processor]
 });
 provider.register();
-var tracer = import_api3.trace.getTracer("axiom", "0.19.0");
+var tracer = import_api3.trace.getTracer("axiom", "0.20.0");
 var flush = async () => {
   await provider.forceFlush();
 };
@@ -985,36 +990,28 @@ var import_node_fs = require("fs");
 var import_api10 = require("@opentelemetry/api");
 
 // src/evals/context/manager.ts
+var import_node_module = require("module");
 var import_meta = {};
 var CONTEXT_MANAGER_SYMBOL = Symbol.for("axiom.context_manager");
 function getGlobalContextManager() {
   return globalThis[CONTEXT_MANAGER_SYMBOL];
 }
-function setGlobalContextManager(manager2) {
-  globalThis[CONTEXT_MANAGER_SYMBOL] = manager2;
+function setGlobalContextManager(manager) {
+  globalThis[CONTEXT_MANAGER_SYMBOL] = manager;
 }
-var isNodeJS = typeof process !== "undefined" && process.versions && process.versions.node;
+var isNodeJS = typeof process !== "undefined" && !!process.versions?.node;
 function getContextManager() {
   const existing = getGlobalContextManager();
-  if (existing) {
-    return existing;
-  }
+  if (existing) return existing;
   let manager;
   if (isNodeJS) {
     try {
       let AsyncLocalStorage;
+      const req = (0, import_node_module.createRequire)(import_meta.url);
       try {
-        const requireFn = eval("require");
-        AsyncLocalStorage = requireFn("async_hooks").AsyncLocalStorage;
-      } catch (directError) {
-        try {
-          const requireFn = eval("require");
-          const { createRequire } = requireFn("module");
-          const dynamicRequire = createRequire(import_meta.url);
-          AsyncLocalStorage = dynamicRequire("async_hooks").AsyncLocalStorage;
-        } catch (_createRequireError) {
-          throw directError;
-        }
+        AsyncLocalStorage = req("node:async_hooks").AsyncLocalStorage;
+      } catch {
+        AsyncLocalStorage = req("async_hooks").AsyncLocalStorage;
       }
       manager = new AsyncLocalStorage();
     } catch (error) {
@@ -1022,6 +1019,7 @@ function getContextManager() {
       manager = createFallbackManager();
     }
   } else {
+    console.warn("AsyncLocalStorage not available, using fallback context manager");
     manager = createFallbackManager();
   }
   setGlobalContextManager(manager);
@@ -1045,21 +1043,18 @@ function createFallbackManager() {
 function createAsyncHook(_name) {
   return {
     get() {
-      const manager2 = getContextManager();
-      if (manager2.getStore) {
-        return manager2.getStore();
+      const manager = getContextManager();
+      if (manager.getStore) {
+        return manager.getStore();
       }
       return void 0;
     },
     run(value, fn) {
-      const manager2 = getContextManager();
-      return manager2.run(value, fn);
+      const manager = getContextManager();
+      return manager.run(value, fn);
     }
   };
 }
-function __resetContextManagerForTests() {
-  delete globalThis[CONTEXT_MANAGER_SYMBOL];
-}
 
 // src/evals/context/global-flags.ts
 var GLOBAL_OVERRIDES_SYMBOL = Symbol.for("axiom.global_flag_overrides");
@@ -1277,9 +1272,9 @@ function extractOverrides(argv) {
   let hasCliFlags = false;
   let configPathCount = 0;
   for (let i = 0; i < argv.length; i++) {
-    const token2 = argv[i];
-    const configMatch = token2.match(CONFIG_RE);
-    const flagMatch = token2.match(FLAG_RE);
+    const token = argv[i];
+    const configMatch = token.match(CONFIG_RE);
+    const flagMatch = token.match(FLAG_RE);
     if (configMatch) {
       configPathCount++;
       if (configPathCount > 1) {
@@ -1304,7 +1299,7 @@ function extractOverrides(argv) {
       const finalValue = value === void 0 ? "true" : value;
       overrides2[key] = coerceValue(finalValue);
     } else {
-      cleanedArgv2.push(token2);
+      cleanedArgv2.push(token);
     }
   }
   if (configPath && hasCliFlags) {
@@ -1329,7 +1324,7 @@ var import_commander4 = require("commander");
 var loadVersionCommand = (program2) => {
   return program2.addCommand(
     new import_commander4.Command("version").description("cli version").action(() => {
-      console.log("0.19.0");
+      console.log("0.20.0");
     })
   );
 };
@@ -1339,7 +1334,7 @@ var { loadEnvConfig } = import_env.default;
 loadEnvConfig(process.cwd());
 var { cleanedArgv, overrides } = extractOverrides(process.argv.slice(2));
 var program = new import_commander5.Command();
-program.name("axiom").description("Axiom's CLI to manage your objects and run evals").version("0.19.0");
+program.name("axiom").description("Axiom's CLI to manage your objects and run evals").version("0.20.0");
 loadPushCommand(program);
 loadPullCommand(program);
 loadEvalCommand(program, overrides);