opsveritas-sdk 0.3.3 → 0.4.0

package/dist/index.d.mts

@@ -13,6 +13,12 @@ interface TraceOptions {
 }
 declare function trace<T>(agentName: string, fn: () => Promise<T>, options?: TraceOptions): Promise<T>;
 
+declare class OpsVeritasKilledError extends Error {
+    agentName: string;
+    reason?: string | undefined;
+    constructor(agentName: string, reason?: string | undefined);
+}
+
 interface WrapOptions {
     agentName: string;
     platform?: string;
@@ -39,6 +45,28 @@ interface LangChainOptions {
     userId?: string;
 }
 declare function langchain(agentName: string, opts?: LangChainOptions): Record<string, unknown>;
+interface LangChainModel {
+    invoke?: (...args: unknown[]) => unknown;
+    ainvoke?: (...args: unknown[]) => Promise<unknown>;
+    [key: string]: unknown;
+}
+/**
+ * Kill-switch enforcement for LangChain models — ADDITIVE, does not replace
+ * `langchain()` above. `langchain()` is telemetry-only (a callback handler;
+ * LangChain's callback manager swallows exceptions raised inside callbacks by
+ * default, so it can never actually block a call). This patches the model's
+ * own `invoke`/`ainvoke` directly — the same technique wrap() uses on raw
+ * provider clients — so a kill-switch check genuinely runs before the real
+ * API call happens.
+ *
+ *   const model = new ChatOpenAI({ callbacks: [opsveritas.langchain('My Agent')] }); // telemetry
+ *   opsveritas.wrapLangChain(model, { agentName: 'My Agent' });                      // enforcement
+ *
+ * Scoped to invoke/ainvoke only — stream/astream/batch are out of scope for
+ * this pass (streaming needs mid-stream cancellation semantics; most LangChain
+ * usage goes through invoke/ainvoke).
+ */
+declare function wrapLangChain<T extends LangChainModel>(model: T, opts: WrapOptions): T;
 
 interface ExecutionPayload {
     platform: string;
@@ -66,6 +94,7 @@ declare const OpsVeritas: {
     trace: typeof trace;
     wrap: typeof wrap;
     langchain: typeof langchain;
+    wrapLangChain: typeof wrapLangChain;
 };
 
-export { type ExecutionPayload, type LangChainOptions, OpsVeritas, type TraceOptions, type WrapOptions, OpsVeritas as default, init, langchain, run, trace, wrap };
+export { type ExecutionPayload, type LangChainOptions, OpsVeritas, OpsVeritasKilledError, type TraceOptions, type WrapOptions, OpsVeritas as default, init, langchain, run, trace, wrap, wrapLangChain };

package/dist/index.d.ts

@@ -13,6 +13,12 @@ interface TraceOptions {
 }
 declare function trace<T>(agentName: string, fn: () => Promise<T>, options?: TraceOptions): Promise<T>;
 
+declare class OpsVeritasKilledError extends Error {
+    agentName: string;
+    reason?: string | undefined;
+    constructor(agentName: string, reason?: string | undefined);
+}
+
 interface WrapOptions {
     agentName: string;
     platform?: string;
@@ -39,6 +45,28 @@ interface LangChainOptions {
     userId?: string;
 }
 declare function langchain(agentName: string, opts?: LangChainOptions): Record<string, unknown>;
+interface LangChainModel {
+    invoke?: (...args: unknown[]) => unknown;
+    ainvoke?: (...args: unknown[]) => Promise<unknown>;
+    [key: string]: unknown;
+}
+/**
+ * Kill-switch enforcement for LangChain models — ADDITIVE, does not replace
+ * `langchain()` above. `langchain()` is telemetry-only (a callback handler;
+ * LangChain's callback manager swallows exceptions raised inside callbacks by
+ * default, so it can never actually block a call). This patches the model's
+ * own `invoke`/`ainvoke` directly — the same technique wrap() uses on raw
+ * provider clients — so a kill-switch check genuinely runs before the real
+ * API call happens.
+ *
+ *   const model = new ChatOpenAI({ callbacks: [opsveritas.langchain('My Agent')] }); // telemetry
+ *   opsveritas.wrapLangChain(model, { agentName: 'My Agent' });                      // enforcement
+ *
+ * Scoped to invoke/ainvoke only — stream/astream/batch are out of scope for
+ * this pass (streaming needs mid-stream cancellation semantics; most LangChain
+ * usage goes through invoke/ainvoke).
+ */
+declare function wrapLangChain<T extends LangChainModel>(model: T, opts: WrapOptions): T;
 
 interface ExecutionPayload {
     platform: string;
@@ -66,6 +94,7 @@ declare const OpsVeritas: {
     trace: typeof trace;
     wrap: typeof wrap;
     langchain: typeof langchain;
+    wrapLangChain: typeof wrapLangChain;
 };
 
-export { type ExecutionPayload, type LangChainOptions, OpsVeritas, type TraceOptions, type WrapOptions, OpsVeritas as default, init, langchain, run, trace, wrap };
+export { type ExecutionPayload, type LangChainOptions, OpsVeritas, OpsVeritasKilledError, type TraceOptions, type WrapOptions, OpsVeritas as default, init, langchain, run, trace, wrap, wrapLangChain };

package/dist/index.js

@@ -21,12 +21,14 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var index_exports = {};
 __export(index_exports, {
   OpsVeritas: () => OpsVeritas,
+  OpsVeritasKilledError: () => OpsVeritasKilledError,
   default: () => index_default,
   init: () => init,
   langchain: () => langchain,
   run: () => run,
   trace: () => trace,
-  wrap: () => wrap
+  wrap: () => wrap,
+  wrapLangChain: () => wrapLangChain
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -355,9 +357,70 @@ async function trace(agentName, fn, options) {
   return result;
 }
 
+// src/killSwitch.ts
+var POLL_INTERVAL_MS = 18e4;
+var OpsVeritasKilledError = class extends Error {
+  constructor(agentName, reason) {
+    super(`[OpsVeritas] Agent "${agentName}" is paused (kill switch: ${reason ?? "budget exceeded"}). Call opsveritas restart from the dashboard to resume.`);
+    this.agentName = agentName;
+    this.reason = reason;
+    this.name = "OpsVeritasKilledError";
+  }
+};
+var cache = /* @__PURE__ */ new Map();
+var polling = /* @__PURE__ */ new Set();
+async function poll(agentName, framework) {
+  let cfg;
+  try {
+    cfg = getConfig();
+  } catch {
+    return void 0;
+  }
+  const params = new URLSearchParams({ agent_name: agentName });
+  if (framework) params.set("framework", framework);
+  try {
+    const res = await fetch(`${cfg.endpoint}/webhooks/agent-budget-status?${params}`, {
+      headers: { "x-opsveritas-key": cfg.apiKey }
+    });
+    if (!res.ok) return void 0;
+    const data = await res.json();
+    cache.set(agentName, { killed: !!data.killed, reason: data.reason });
+    return data.kill_switch_enabled;
+  } catch {
+    return void 0;
+  }
+}
+function startPolling(agentName, framework) {
+  if (polling.has(agentName)) return;
+  polling.add(agentName);
+  void poll(agentName, framework).then((enabled) => {
+    if (enabled === false) return;
+    const timer = setInterval(() => void poll(agentName, framework), POLL_INTERVAL_MS);
+    const t = timer;
+    if (typeof t.unref === "function") t.unref();
+  });
+}
+function isKilled(agentName) {
+  return cache.get(agentName) ?? { killed: false };
+}
+
 // src/wrap.ts
 var PATCHED = /* @__PURE__ */ Symbol("opsveritas.patched");
 var OUTPUT_SUMMARY_MAX = 3e3;
+var TOOL_INPUT_TEXT_FIELDS = ["body", "content", "text", "summary", "message", "output"];
+function extractToolInputText(input) {
+  if (!input || typeof input !== "object") return void 0;
+  const obj = input;
+  for (const field of TOOL_INPUT_TEXT_FIELDS) {
+    const v = obj[field];
+    if (typeof v === "string" && v.trim()) return v;
+  }
+  try {
+    return JSON.stringify(obj);
+  } catch {
+    return void 0;
+  }
+}
 function extractOutput(resp) {
   if (!resp) return void 0;
   const choices = resp.choices;
@@ -369,6 +432,11 @@ function extractOutput(resp) {
   if (Array.isArray(blocks)) {
     const tb = blocks.find((b) => b.type === "text");
     if (typeof tb?.text === "string" && tb.text.trim()) return tb.text.slice(0, OUTPUT_SUMMARY_MAX);
+    const toolBlock = blocks.find((b) => b.type === "tool_use");
+    if (toolBlock) {
+      const s = extractToolInputText(toolBlock.input);
+      if (s) return s.slice(0, OUTPUT_SUMMARY_MAX);
+    }
   }
   return void 0;
 }
@@ -376,7 +444,10 @@ function patchOpenAI(client, opts) {
   const completions = client.chat?.completions;
   if (!completions || completions[PATCHED]) return;
   const orig = completions.create.bind(completions);
+  startPolling(opts.agentName, "sdk");
   completions.create = async function(...args) {
+    const killState = isKilled(opts.agentName);
+    if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
     const executedAt = (/* @__PURE__ */ new Date()).toISOString();
     const t0 = Date.now();
     let status = "success";
@@ -427,7 +498,10 @@ function patchAnthropic(client, opts) {
   const messages = client.messages;
   if (!messages || messages[PATCHED]) return;
   const orig = messages.create.bind(messages);
+  startPolling(opts.agentName, "sdk");
   messages.create = async function(...args) {
+    const killState = isKilled(opts.agentName);
+    if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
     const executedAt = (/* @__PURE__ */ new Date()).toISOString();
     const t0 = Date.now();
     let status = "success";
@@ -476,7 +550,10 @@ function patchGemini(client, opts) {
   if (client[PATCHED]) return;
   const modelName = typeof client.model === "string" ? client.model.replace(/^models\//, "") : void 0;
   const orig = client.generateContent.bind(client);
+  startPolling(opts.agentName, "sdk");
   client.generateContent = async function(...args) {
+    const killState = isKilled(opts.agentName);
+    if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
     const executedAt = (/* @__PURE__ */ new Date()).toISOString();
     const t0 = Date.now();
     let status = "success";
@@ -536,6 +613,7 @@ function wrap(client, opts) {
 }
 
 // src/langchain.ts
+var LC_PATCHED = /* @__PURE__ */ Symbol("opsveritas.langchain.patched");
 function extractUsage2(output) {
   let model;
   let inputTokens;
@@ -639,16 +717,41 @@ function langchain(agentName, opts = {}) {
     }
   };
 }
+function wrapLangChain(model, opts) {
+  const m = model;
+  if (m[LC_PATCHED]) return model;
+  startPolling(opts.agentName, "sdk");
+  if (typeof m.invoke === "function") {
+    const origInvoke = m.invoke.bind(model);
+    m.invoke = function(...args) {
+      const killState = isKilled(opts.agentName);
+      if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
+      return origInvoke(...args);
+    };
+  }
+  if (typeof m.ainvoke === "function") {
+    const origAinvoke = m.ainvoke.bind(model);
+    m.ainvoke = async function(...args) {
+      const killState = isKilled(opts.agentName);
+      if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
+      return origAinvoke(...args);
+    };
+  }
+  m[LC_PATCHED] = true;
+  return model;
+}
 
 // src/index.ts
-var OpsVeritas = { init, run, trace, wrap, langchain };
+var OpsVeritas = { init, run, trace, wrap, langchain, wrapLangChain };
 var index_default = OpsVeritas;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   OpsVeritas,
+  OpsVeritasKilledError,
   init,
   langchain,
   run,
   trace,
-  wrap
+  wrap,
+  wrapLangChain
 });

package/dist/index.mjs

@@ -323,9 +323,70 @@ async function trace(agentName, fn, options) {
   return result;
 }
 
+// src/killSwitch.ts
+var POLL_INTERVAL_MS = 18e4;
+var OpsVeritasKilledError = class extends Error {
+  constructor(agentName, reason) {
+    super(`[OpsVeritas] Agent "${agentName}" is paused (kill switch: ${reason ?? "budget exceeded"}). Call opsveritas restart from the dashboard to resume.`);
+    this.agentName = agentName;
+    this.reason = reason;
+    this.name = "OpsVeritasKilledError";
+  }
+};
+var cache = /* @__PURE__ */ new Map();
+var polling = /* @__PURE__ */ new Set();
+async function poll(agentName, framework) {
+  let cfg;
+  try {
+    cfg = getConfig();
+  } catch {
+    return void 0;
+  }
+  const params = new URLSearchParams({ agent_name: agentName });
+  if (framework) params.set("framework", framework);
+  try {
+    const res = await fetch(`${cfg.endpoint}/webhooks/agent-budget-status?${params}`, {
+      headers: { "x-opsveritas-key": cfg.apiKey }
+    });
+    if (!res.ok) return void 0;
+    const data = await res.json();
+    cache.set(agentName, { killed: !!data.killed, reason: data.reason });
+    return data.kill_switch_enabled;
+  } catch {
+    return void 0;
+  }
+}
+function startPolling(agentName, framework) {
+  if (polling.has(agentName)) return;
+  polling.add(agentName);
+  void poll(agentName, framework).then((enabled) => {
+    if (enabled === false) return;
+    const timer = setInterval(() => void poll(agentName, framework), POLL_INTERVAL_MS);
+    const t = timer;
+    if (typeof t.unref === "function") t.unref();
+  });
+}
+function isKilled(agentName) {
+  return cache.get(agentName) ?? { killed: false };
+}
+
 // src/wrap.ts
 var PATCHED = /* @__PURE__ */ Symbol("opsveritas.patched");
 var OUTPUT_SUMMARY_MAX = 3e3;
+var TOOL_INPUT_TEXT_FIELDS = ["body", "content", "text", "summary", "message", "output"];
+function extractToolInputText(input) {
+  if (!input || typeof input !== "object") return void 0;
+  const obj = input;
+  for (const field of TOOL_INPUT_TEXT_FIELDS) {
+    const v = obj[field];
+    if (typeof v === "string" && v.trim()) return v;
+  }
+  try {
+    return JSON.stringify(obj);
+  } catch {
+    return void 0;
+  }
+}
 function extractOutput(resp) {
   if (!resp) return void 0;
   const choices = resp.choices;
@@ -337,6 +398,11 @@ function extractOutput(resp) {
   if (Array.isArray(blocks)) {
     const tb = blocks.find((b) => b.type === "text");
     if (typeof tb?.text === "string" && tb.text.trim()) return tb.text.slice(0, OUTPUT_SUMMARY_MAX);
+    const toolBlock = blocks.find((b) => b.type === "tool_use");
+    if (toolBlock) {
+      const s = extractToolInputText(toolBlock.input);
+      if (s) return s.slice(0, OUTPUT_SUMMARY_MAX);
+    }
   }
   return void 0;
 }
@@ -344,7 +410,10 @@ function patchOpenAI(client, opts) {
   const completions = client.chat?.completions;
   if (!completions || completions[PATCHED]) return;
   const orig = completions.create.bind(completions);
+  startPolling(opts.agentName, "sdk");
   completions.create = async function(...args) {
+    const killState = isKilled(opts.agentName);
+    if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
     const executedAt = (/* @__PURE__ */ new Date()).toISOString();
     const t0 = Date.now();
     let status = "success";
@@ -395,7 +464,10 @@ function patchAnthropic(client, opts) {
   const messages = client.messages;
   if (!messages || messages[PATCHED]) return;
   const orig = messages.create.bind(messages);
+  startPolling(opts.agentName, "sdk");
   messages.create = async function(...args) {
+    const killState = isKilled(opts.agentName);
+    if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
     const executedAt = (/* @__PURE__ */ new Date()).toISOString();
     const t0 = Date.now();
     let status = "success";
@@ -444,7 +516,10 @@ function patchGemini(client, opts) {
   if (client[PATCHED]) return;
   const modelName = typeof client.model === "string" ? client.model.replace(/^models\//, "") : void 0;
   const orig = client.generateContent.bind(client);
+  startPolling(opts.agentName, "sdk");
   client.generateContent = async function(...args) {
+    const killState = isKilled(opts.agentName);
+    if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
     const executedAt = (/* @__PURE__ */ new Date()).toISOString();
     const t0 = Date.now();
     let status = "success";
@@ -504,6 +579,7 @@ function wrap(client, opts) {
 }
 
 // src/langchain.ts
+var LC_PATCHED = /* @__PURE__ */ Symbol("opsveritas.langchain.patched");
 function extractUsage2(output) {
   let model;
   let inputTokens;
@@ -607,16 +683,41 @@ function langchain(agentName, opts = {}) {
     }
   };
 }
+function wrapLangChain(model, opts) {
+  const m = model;
+  if (m[LC_PATCHED]) return model;
+  startPolling(opts.agentName, "sdk");
+  if (typeof m.invoke === "function") {
+    const origInvoke = m.invoke.bind(model);
+    m.invoke = function(...args) {
+      const killState = isKilled(opts.agentName);
+      if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
+      return origInvoke(...args);
+    };
+  }
+  if (typeof m.ainvoke === "function") {
+    const origAinvoke = m.ainvoke.bind(model);
+    m.ainvoke = async function(...args) {
+      const killState = isKilled(opts.agentName);
+      if (killState.killed) throw new OpsVeritasKilledError(opts.agentName, killState.reason);
+      return origAinvoke(...args);
+    };
+  }
+  m[LC_PATCHED] = true;
+  return model;
+}
 
 // src/index.ts
-var OpsVeritas = { init, run, trace, wrap, langchain };
+var OpsVeritas = { init, run, trace, wrap, langchain, wrapLangChain };
 var index_default = OpsVeritas;
 export {
   OpsVeritas,
+  OpsVeritasKilledError,
   index_default as default,
   init,
   langchain,
   run,
   trace,
-  wrap
+  wrap,
+  wrapLangChain
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opsveritas-sdk",
-  "version": "0.3.3",
+  "version": "0.4.0",
   "description": "Monitor your AI agents in 3 lines of code — tokens, cost, latency, and silent failures",
   "main": "dist/index.js",
   "module": "dist/index.mjs",