neuronpedia/dist/index.js

Neuronpedia (Official)

"use strict";
var __create = Object.create;
var __defProp = Object.defineProperty;
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
var __getOwnPropNames = Object.getOwnPropertyNames;
var __getProtoOf = Object.getPrototypeOf;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __export = (target, all) => {
  for (var name in all)
    __defProp(target, name, { get: all[name], enumerable: true });
};
var __copyProps = (to, from, except, desc) => {
  if (from && typeof from === "object" || typeof from === "function") {
    for (let key of __getOwnPropNames(from))
      if (!__hasOwnProp.call(to, key) && key !== except)
        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
  }
  return to;
};
var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
  // If the importer is in node compatibility mode or this is not an ESM
  // file that has been converted to a CommonJS file using a Babel-
  // compatible transform (i.e. "__esModule" has not been set), then set
  // "default" to the CommonJS "module.exports" for node compatibility.
  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
  mod
));
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
var __async = (__this, __arguments, generator) => {
  return new Promise((resolve, reject) => {
    var fulfilled = (value) => {
      try {
        step(generator.next(value));
      } catch (e) {
        reject(e);
      }
    };
    var rejected = (value) => {
      try {
        step(generator.throw(value));
      } catch (e) {
        reject(e);
      }
    };
    var step = (x) => x.done ? resolve(x.value) : Promise.resolve(x.value).then(fulfilled, rejected);
    step((generator = generator.apply(__this, __arguments)).next());
  });
};

// src/index.ts
var index_exports = {};
__export(index_exports, {
  NPScoreFuzzDetectionType: () => NPScoreFuzzDetectionType,
  getEleutherExplanationDefault: () => getEleutherExplanationDefault,
  getEleutherScoreEmbedding: () => getEleutherScoreEmbedding,
  getEleutherScoreFuzzDetection: () => getEleutherScoreFuzzDetection,
  postApiSaeEval: () => postApiSaeEval,
  postExplainDefault: () => postExplainDefault,
  postScoreEmbedding: () => postScoreEmbedding,
  postScoreFuzzDetection: () => postScoreFuzzDetection
});
module.exports = __toCommonJS(index_exports);

// src/autointerp/score.ts
var import_axios3 = __toESM(require("axios"));

// src/types/autointerp/score/embedding.ts
var import_axios = __toESM(require("axios"));
var postScoreEmbedding = (nPScoreEmbeddingRequest, options) => {
  return import_axios.default.post(
    `/score/embedding`,
    nPScoreEmbeddingRequest,
    options
  );
};

// src/types/autointerp/score/fuzz_detection.ts
var import_axios2 = __toESM(require("axios"));
var NPScoreFuzzDetectionType = {
  fuzz: "fuzz",
  detection: "detection"
};
var postScoreFuzzDetection = (nPScoreFuzzDetectionRequest, options) => {
  return import_axios2.default.post(
    `/score/fuzz_detection`,
    nPScoreFuzzDetectionRequest,
    options
  );
};

// src/const.ts
var USE_LOCAL_AUTOINTERP = process.env.USE_LOCAL_AUTOINTERP === "true";
var AUTOINTERP_SERVER = USE_LOCAL_AUTOINTERP ? "http://127.0.0.1:5003" : process.env.AUTOINTERP_SERVER;

// src/autointerp/score.ts
function getEleutherScoreFuzzDetection(type, activations, explanation, openRouterKey, openRouterModel, inferenceServerSecret) {
  return __async(this, null, function* () {
    import_axios3.default.defaults.baseURL = AUTOINTERP_SERVER;
    const result = yield postScoreFuzzDetection({
      type,
      activations,
      explanation,
      openrouter_key: openRouterKey,
      model: openRouterModel,
      secret: inferenceServerSecret
    });
    if (result.status !== 200 || !result.data) {
      throw new Error(
        "Failed to generate score: " + JSON.stringify(result.statusText)
      );
    }
    return result.data;
  });
}
function getEleutherScoreEmbedding(activations, explanation, inferenceServerSecret) {
  return __async(this, null, function* () {
    import_axios3.default.defaults.baseURL = AUTOINTERP_SERVER;
    const result = yield postScoreEmbedding({
      activations,
      explanation,
      secret: inferenceServerSecret
    });
    if (result.status !== 200 || !result.data) {
      throw new Error(
        "Failed to generate score: " + JSON.stringify(result.statusText)
      );
    }
    return result.data;
  });
}

// src/autointerp/explain.ts
var import_axios5 = __toESM(require("axios"));

// src/types/autointerp/explain/default.ts
var import_axios4 = __toESM(require("axios"));
var postExplainDefault = (nPExplainDefaultRequest, options) => {
  return import_axios4.default.post(
    `/explain/default`,
    nPExplainDefaultRequest,
    options
  );
};

// src/autointerp/explain.ts
function getEleutherExplanationDefault(activations, openRouterKey, openRouterModel, inferenceServerSecret) {
  return __async(this, null, function* () {
    import_axios5.default.defaults.baseURL = AUTOINTERP_SERVER;
    const result = yield postExplainDefault({
      activations,
      openrouter_key: openRouterKey,
      model: openRouterModel,
      secret: inferenceServerSecret
    });
    if (result.status !== 200 || !result.data) {
      throw new Error(
        "Failed to generate explanation: " + JSON.stringify(result.statusText)
      );
    }
    return result.data;
  });
}

// src/types/webapp/api/sae_eval.ts
var import_axios6 = __toESM(require("axios"));
var postApiSaeEval = (nPSaeEvalListRequest, options) => {
  return import_axios6.default.post(
    `/api/sae-eval`,
    nPSaeEvalListRequest,
    options
  );
};
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
  NPScoreFuzzDetectionType,
  getEleutherExplanationDefault,
  getEleutherScoreEmbedding,
  getEleutherScoreFuzzDetection,
  postApiSaeEval,
  postExplainDefault,
  postScoreEmbedding,
  postScoreFuzzDetection
});
//# sourceMappingURL=index.js.map