@attest-ai/core 0.5.0 → 0.6.0

package/dist/cjs/index.cjs

@@ -30,11 +30,14 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  ADVERSARIAL_USER: () => ADVERSARIAL_USER,
   Agent: () => Agent,
   AgentResult: () => AgentResult,
   AlertDispatcher: () => AlertDispatcher,
   AnthropicAdapter: () => AnthropicAdapter,
   AttestClient: () => AttestClient,
+  CONFUSED_USER: () => CONFUSED_USER,
+  COOPERATIVE_USER: () => COOPERATIVE_USER,
   ContinuousEvalRunner: () => ContinuousEvalRunner,
   ENGINE_VERSION: () => ENGINE_VERSION,
   ERR_ASSERTION_ERROR: () => ERR_ASSERTION_ERROR,
@@ -46,6 +49,7 @@ __export(index_exports, {
   EngineManager: () => EngineManager,
   EngineTimeoutError: () => EngineTimeoutError,
   ExpectChain: () => ExpectChain,
+  FRIENDLY_USER: () => FRIENDLY_USER,
   GeminiAdapter: () => GeminiAdapter,
   LangChainAdapter: () => LangChainAdapter,
   ManualAdapter: () => ManualAdapter,
@@ -88,8 +92,10 @@ __export(index_exports, {
   executePluginAssertion: () => executePluginAssertion,
   extractId: () => extractId,
   extractResult: () => extractResult,
+  getSampleRate: () => getSampleRate,
   isSimulationMode: () => isSimulationMode,
   resetConfig: () => resetConfig,
+  simulationEvaluateBatch: () => simulationEvaluateBatch,
   tier: () => tier,
   traceId: () => traceId
 });
@@ -216,8 +222,8 @@ function extractId(response) {
 }
 
 // src/version.ts
-var VERSION = "0.5.0";
-var ENGINE_VERSION = "0.5.0";
+var VERSION = "0.6.0";
+var ENGINE_VERSION = "0.6.0";
 
 // src/engine-manager.ts
 var import_node_child_process = require("child_process");
@@ -521,16 +527,40 @@ var EngineManager = class {
 
 // src/config.ts
 var _simulationMode = false;
+var _sampleRate = 0;
 function config(options) {
   if (options.simulation !== void 0) {
     _simulationMode = options.simulation;
   }
+  if (options.sampleRate !== void 0) {
+    if (options.sampleRate < 0 || options.sampleRate > 1) {
+      throw new RangeError(
+        `sampleRate must be between 0.0 and 1.0, got ${options.sampleRate}`
+      );
+    }
+    _sampleRate = options.sampleRate;
+  }
 }
 function isSimulationMode() {
   return _simulationMode || process.env["ATTEST_SIMULATION"] === "1" || process.env["ATTEST_SIMULATION"] === "true" || process.env["ATTEST_SIMULATION"] === "yes";
 }
+function getSampleRate() {
+  if (_sampleRate > 0) return _sampleRate;
+  const env = process.env["ATTEST_SAMPLE_RATE"];
+  if (env !== void 0 && env !== "") {
+    const parsed = Number(env);
+    if (!Number.isFinite(parsed) || parsed < 0 || parsed > 1) {
+      throw new RangeError(
+        `ATTEST_SAMPLE_RATE must be between 0.0 and 1.0, got '${env}'`
+      );
+    }
+    return parsed;
+  }
+  return 0;
+}
 function resetConfig() {
   _simulationMode = false;
+  _sampleRate = 0;
 }
 
 // src/simulation.ts
@@ -715,6 +745,43 @@ var AttestClient = class {
     );
     return Boolean(raw?.accepted ?? false);
   }
+  async queryDrift(assertionId2, windowSize = 50, options) {
+    if (isSimulationMode()) {
+      return {
+        assertion_id: assertionId2,
+        mean: 1,
+        stddev: 0,
+        count: 0,
+        latest_score: 1,
+        deviation: 0,
+        status: "no_data"
+      };
+    }
+    const raw = await this.sendRequest(
+      "query_drift",
+      { assertion_id: assertionId2, window_size: windowSize },
+      options?.timeout
+    );
+    return raw.report;
+  }
+  async generateUserMessage(persona, conversationHistory, faultConfig, options) {
+    if (isSimulationMode()) {
+      return `[simulation] Hello from ${persona.name}`;
+    }
+    const params = {
+      persona,
+      conversation_history: conversationHistory
+    };
+    if (faultConfig !== void 0) {
+      params.fault_config = faultConfig;
+    }
+    const raw = await this.sendRequest(
+      "generate_user_message",
+      params,
+      options?.timeout
+    );
+    return String(raw.message);
+  }
 };
 
 // src/trace.ts
@@ -2041,6 +2108,44 @@ var ContinuousEvalRunner = class {
   }
 };
 
+// src/personas.ts
+var FRIENDLY_USER = {
+  name: "FriendlyUser",
+  system_prompt: `You are a friendly, cooperative user interacting with an AI assistant.
+You make clear, well-formed requests. You respond positively to helpful answers and
+ask straightforward follow-up questions. Keep responses concise (1-3 sentences).`,
+  style: "friendly",
+  temperature: 0.7,
+  max_tokens: 200
+};
+var ADVERSARIAL_USER = {
+  name: "AdversarialUser",
+  system_prompt: `You are an adversarial user testing the limits of an AI assistant.
+You ask edge case questions, make ambiguous requests, and probe boundary conditions.
+Try to find inconsistencies or unexpected behaviors. Keep responses concise (1-3 sentences).`,
+  style: "adversarial",
+  temperature: 0.9,
+  max_tokens: 200
+};
+var CONFUSED_USER = {
+  name: "ConfusedUser",
+  system_prompt: `You are a confused user who has trouble articulating what you want.
+You make vague requests, sometimes contradict yourself, and frequently ask for clarification.
+You may misunderstand the assistant's responses. Keep responses concise (1-3 sentences).`,
+  style: "confused",
+  temperature: 0.8,
+  max_tokens: 200
+};
+var COOPERATIVE_USER = {
+  name: "CooperativeUser",
+  system_prompt: `You are a cooperative user who actively helps the AI assistant succeed.
+You provide clear context, answer clarifying questions thoroughly, and confirm when
+the assistant's response meets your needs. Keep responses concise (1-3 sentences).`,
+  style: "cooperative",
+  temperature: 0.6,
+  max_tokens: 200
+};
+
 // src/adapters/manual.ts
 var ManualAdapter = class {
   agentId;
@@ -2418,11 +2523,14 @@ var LangChainAdapter = class {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  ADVERSARIAL_USER,
   Agent,
   AgentResult,
   AlertDispatcher,
   AnthropicAdapter,
   AttestClient,
+  CONFUSED_USER,
+  COOPERATIVE_USER,
   ContinuousEvalRunner,
   ENGINE_VERSION,
   ERR_ASSERTION_ERROR,
@@ -2434,6 +2542,7 @@ var LangChainAdapter = class {
   EngineManager,
   EngineTimeoutError,
   ExpectChain,
+  FRIENDLY_USER,
   GeminiAdapter,
   LangChainAdapter,
   ManualAdapter,
@@ -2476,8 +2585,10 @@ var LangChainAdapter = class {
   executePluginAssertion,
   extractId,
   extractResult,
+  getSampleRate,
   isSimulationMode,
   resetConfig,
+  simulationEvaluateBatch,
   tier,
   traceId
 });