agentid-sdk 0.1.17 → 0.1.18

package/dist/chunk-6YR4ECGB.mjs

@@ -0,0 +1,424 @@
+// src/langchain.ts
+import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
+function safeString(val) {
+  return typeof val === "string" ? val : "";
+}
+function callbackDebugEnabled() {
+  try {
+    return typeof process !== "undefined" && process?.env?.AGENTID_DEBUG_CALLBACK === "1";
+  } catch {
+    return false;
+  }
+}
+function logCallbackDebug(message, details) {
+  if (!callbackDebugEnabled()) return;
+  if (details) {
+    console.log(`[AgentID][LC] ${message}`, details);
+    return;
+  }
+  console.log(`[AgentID][LC] ${message}`);
+}
+function extractTextFromContent(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (Array.isArray(content)) {
+    const parts = content.map((item) => {
+      if (typeof item === "string") return item;
+      if (!item || typeof item !== "object") return "";
+      const record = item;
+      if (typeof record.text === "string") return record.text;
+      if (typeof record.content === "string") return record.content;
+      return "";
+    }).filter((part) => part.length > 0);
+    return parts.join("\n");
+  }
+  if (content && typeof content === "object") {
+    const record = content;
+    if (typeof record.text === "string") return record.text;
+    if (typeof record.content === "string") return record.content;
+  }
+  return "";
+}
+function getMessageRole(msg) {
+  if (!msg || typeof msg !== "object") return null;
+  const typed = msg;
+  if (typeof typed.role === "string") return typed.role;
+  if (typeof typed.type === "string") return typed.type;
+  if (typeof typed._getType === "function") {
+    try {
+      const role = typed._getType();
+      if (typeof role === "string") return role;
+    } catch {
+    }
+  }
+  if (typeof typed.getType === "function") {
+    try {
+      const role = typed.getType();
+      if (typeof role === "string") return role;
+    } catch {
+    }
+  }
+  return null;
+}
+function extractPromptFromPrompts(prompts) {
+  if (Array.isArray(prompts) && prompts.length > 0) {
+    return safeString(prompts[prompts.length - 1]);
+  }
+  return "";
+}
+function extractPromptFromMessages(messages) {
+  const flat = [];
+  if (Array.isArray(messages)) {
+    for (const item of messages) {
+      if (Array.isArray(item)) {
+        flat.push(...item);
+      } else {
+        flat.push(item);
+      }
+    }
+  }
+  let last = null;
+  for (const msg of flat) {
+    const typed = msg;
+    const role = getMessageRole(msg);
+    if (role === "user" || role === "human") {
+      last = typed;
+    }
+  }
+  if (!last || typeof last !== "object") {
+    return "";
+  }
+  const typedLast = last;
+  return extractTextFromContent(typedLast.content ?? typedLast.text);
+}
+function setPromptInPrompts(prompts, sanitizedInput) {
+  if (!Array.isArray(prompts) || prompts.length === 0) {
+    return false;
+  }
+  prompts[prompts.length - 1] = sanitizedInput;
+  return true;
+}
+function setPromptInMessages(messages, sanitizedInput) {
+  if (!Array.isArray(messages)) {
+    return false;
+  }
+  const flat = [];
+  for (const item of messages) {
+    if (Array.isArray(item)) {
+      flat.push(...item);
+    } else {
+      flat.push(item);
+    }
+  }
+  for (let i = flat.length - 1; i >= 0; i -= 1) {
+    const candidate = flat[i];
+    if (!candidate || typeof candidate !== "object") {
+      continue;
+    }
+    const typed = candidate;
+    const role = typed.role ?? typed.type;
+    if (role !== "user" && role !== "human") {
+      continue;
+    }
+    if ("content" in typed) {
+      typed.content = sanitizedInput;
+      return true;
+    }
+    if ("text" in typed) {
+      typed.text = sanitizedInput;
+      return true;
+    }
+    typed.content = sanitizedInput;
+    return true;
+  }
+  return false;
+}
+function extractModel(serialized, kwargs) {
+  const kw = (kwargs && typeof kwargs === "object" ? kwargs : null) ?? null;
+  const directModel = kw?.model ?? kw?.model_name ?? kw?.modelName;
+  if (typeof directModel === "string" && directModel) return directModel;
+  const invocationModel = kw?.invocation_params?.model ?? kw?.invocation_params?.model_name ?? kw?.invocation_params?.modelName;
+  if (typeof invocationModel === "string" && invocationModel) return invocationModel;
+  const nestedModel = kw?.options?.model ?? kw?.options?.model_name ?? kw?.options?.modelName ?? kw?.kwargs?.model ?? kw?.kwargs?.model_name ?? kw?.kwargs?.modelName;
+  if (typeof nestedModel === "string" && nestedModel) return nestedModel;
+  const ser = (serialized && typeof serialized === "object" ? serialized : null) ?? null;
+  const serKw = (ser?.kwargs && typeof ser.kwargs === "object" ? ser.kwargs : null) ?? null;
+  const serModel = serKw?.model ?? serKw?.model_name ?? serKw?.modelName;
+  if (typeof serModel === "string" && serModel) return serModel;
+  const name = ser?.name ?? ser?.id;
+  if (typeof name === "string" && name) return name;
+  return void 0;
+}
+function extractModelFromOutput(output) {
+  const llmOutput = output?.llmOutput ?? output?.llm_output;
+  const llmModel = llmOutput?.model ?? llmOutput?.model_name ?? llmOutput?.modelName;
+  if (typeof llmModel === "string" && llmModel) return llmModel;
+  const first = output?.generations?.[0]?.[0];
+  const responseMeta = first?.message?.response_metadata ?? first?.message?.responseMetadata;
+  const responseModel = responseMeta?.model_name ?? responseMeta?.model ?? responseMeta?.modelName;
+  if (typeof responseModel === "string" && responseModel) return responseModel;
+  const generationInfo = first?.generation_info ?? first?.generationInfo;
+  const genModel = generationInfo?.model_name ?? generationInfo?.model ?? generationInfo?.modelName;
+  if (typeof genModel === "string" && genModel) return genModel;
+  return void 0;
+}
+function extractOutputText(output) {
+  const gens = output?.generations;
+  const first = gens?.[0]?.[0];
+  const text = first?.text ?? first?.message?.content;
+  return typeof text === "string" ? text : "";
+}
+function extractTokenUsage(output) {
+  const llmOutput = output?.llmOutput ?? output?.llm_output;
+  const usage = llmOutput?.tokenUsage ?? llmOutput?.token_usage ?? llmOutput?.usage ?? void 0;
+  if (usage && typeof usage === "object") {
+    return usage;
+  }
+  const first = output?.generations?.[0]?.[0];
+  const usageMetadata = first?.message?.usage_metadata;
+  if (usageMetadata && typeof usageMetadata === "object") {
+    return usageMetadata;
+  }
+  const responseTokenUsage = first?.message?.response_metadata?.token_usage ?? first?.message?.response_metadata?.tokenUsage ?? void 0;
+  if (responseTokenUsage && typeof responseTokenUsage === "object") {
+    return responseTokenUsage;
+  }
+  const generationTokenUsage = first?.generation_info?.token_usage ?? first?.generation_info?.tokenUsage ?? void 0;
+  if (generationTokenUsage && typeof generationTokenUsage === "object") {
+    return generationTokenUsage;
+  }
+  return void 0;
+}
+function isUuidLike(value) {
+  return typeof value === "string" && /^[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$/i.test(value.trim());
+}
+function createClientEventId() {
+  if (typeof crypto !== "undefined" && typeof crypto.randomUUID === "function") {
+    return crypto.randomUUID();
+  }
+  const segment = () => Math.floor((1 + Math.random()) * 65536).toString(16).slice(1);
+  return `${segment()}${segment()}-${segment()}-4${segment().slice(1)}-a${segment().slice(1)}-${segment()}${segment()}${segment()}`;
+}
+function readBooleanField(value) {
+  return typeof value === "boolean" ? value : null;
+}
+function extractStreamFlag(serialized, extraParams) {
+  const extras = extraParams && typeof extraParams === "object" ? extraParams : null;
+  const direct = readBooleanField(extras?.stream) ?? readBooleanField(extras?.streaming);
+  if (direct !== null) {
+    return direct;
+  }
+  const invocation = extras?.invocation_params && typeof extras.invocation_params === "object" ? extras.invocation_params : null;
+  const invocationStream = readBooleanField(invocation?.stream) ?? readBooleanField(invocation?.streaming);
+  if (invocationStream !== null) {
+    return invocationStream;
+  }
+  const serializedRecord = serialized && typeof serialized === "object" ? serialized : null;
+  const kwargs = serializedRecord?.kwargs && typeof serializedRecord.kwargs === "object" ? serializedRecord.kwargs : null;
+  return readBooleanField(kwargs?.stream) ?? readBooleanField(kwargs?.streaming) ?? false;
+}
+var AgentIDCallbackHandler = class extends BaseCallbackHandler {
+  constructor(agent, options) {
+    super();
+    this.name = "agentid_callback_handler";
+    this.runs = /* @__PURE__ */ new Map();
+    this.agent = agent;
+    this.systemId = options.system_id;
+    this.apiKeyOverride = options.apiKey?.trim() || options.api_key?.trim() || void 0;
+  }
+  get requestOptions() {
+    return this.apiKeyOverride ? { apiKey: this.apiKeyOverride } : void 0;
+  }
+  getLangchainCapabilities() {
+    const piiMaskingEnabled = Boolean(
+      this.agent.piiMasking
+    );
+    return {
+      capabilities: {
+        has_feedback_handler: true,
+        pii_masking_enabled: piiMaskingEnabled,
+        framework: "langchain"
+      }
+    };
+  }
+  async preflight(input, stream) {
+    await this.agent.scanPromptInjection(input, this.requestOptions);
+    const prepared = await this.agent.prepareInputForDispatch({
+      input,
+      systemId: this.systemId,
+      stream,
+      skipInjectionScan: true
+    }, this.requestOptions);
+    return prepared.sanitizedInput;
+  }
+  async handleLLMStart(serialized, prompts, runId, _parentRunId, extraParams) {
+    const input = extractPromptFromPrompts(prompts);
+    const id = String(runId ?? "");
+    logCallbackDebug("handleLLMStart", { runId: id, hasInput: input.length > 0 });
+    if (!input) {
+      return;
+    }
+    const stream = extractStreamFlag(serialized, extraParams);
+    const sanitizedInput = await this.preflight(input, stream);
+    if (sanitizedInput !== input) {
+      const mutated = setPromptInPrompts(prompts, sanitizedInput);
+      if (!mutated) {
+        throw new Error(
+          "AgentID: Strict PII mode requires mutable LangChain prompt payload."
+        );
+      }
+    }
+    const requestedClientEventId = isUuidLike(id) ? id.trim() : createClientEventId();
+    const modelName = extractModel(serialized, extraParams);
+    const verdict = await this.agent.guard({
+      input: sanitizedInput,
+      system_id: this.systemId,
+      model: modelName,
+      client_event_id: requestedClientEventId,
+      client_capabilities: this.getLangchainCapabilities()
+    }, this.requestOptions);
+    if (!verdict.allowed) {
+      throw new Error(`AgentID: Security Blocked (${verdict.reason ?? "guard_denied"})`);
+    }
+    const canonicalClientEventId = isUuidLike(verdict.client_event_id) ? verdict.client_event_id.trim() : requestedClientEventId;
+    const guardEventId = typeof verdict.guard_event_id === "string" && verdict.guard_event_id.length > 0 ? verdict.guard_event_id : void 0;
+    let transformedInput = typeof verdict.transformed_input === "string" && verdict.transformed_input.length > 0 ? verdict.transformed_input : sanitizedInput;
+    if (transformedInput !== sanitizedInput) {
+      const mutated = setPromptInPrompts(prompts, transformedInput);
+      if (!mutated) {
+        transformedInput = sanitizedInput;
+      }
+    }
+    this.runs.set(id, {
+      input: transformedInput,
+      startedAtMs: Date.now(),
+      model: modelName,
+      clientEventId: canonicalClientEventId,
+      guardEventId
+    });
+    logCallbackDebug("handleLLMStart state_set", {
+      runId: id,
+      clientEventId: canonicalClientEventId,
+      guardEventId: guardEventId ?? null
+    });
+  }
+  async handleChatModelStart(serialized, messages, runId, _parentRunId, extraParams) {
+    const input = extractPromptFromMessages(messages);
+    const id = String(runId ?? "");
+    logCallbackDebug("handleChatModelStart", { runId: id, hasInput: input.length > 0 });
+    if (!input) {
+      return;
+    }
+    const stream = extractStreamFlag(serialized, extraParams);
+    const sanitizedInput = await this.preflight(input, stream);
+    if (sanitizedInput !== input) {
+      const mutated = setPromptInMessages(messages, sanitizedInput);
+      if (!mutated) {
+        throw new Error(
+          "AgentID: Strict PII mode requires mutable LangChain message payload."
+        );
+      }
+    }
+    const requestedClientEventId = isUuidLike(id) ? id.trim() : createClientEventId();
+    const modelName = extractModel(serialized, extraParams);
+    const verdict = await this.agent.guard({
+      input: sanitizedInput,
+      system_id: this.systemId,
+      model: modelName,
+      client_event_id: requestedClientEventId,
+      client_capabilities: this.getLangchainCapabilities()
+    }, this.requestOptions);
+    if (!verdict.allowed) {
+      throw new Error(`AgentID: Security Blocked (${verdict.reason ?? "guard_denied"})`);
+    }
+    const canonicalClientEventId = isUuidLike(verdict.client_event_id) ? verdict.client_event_id.trim() : requestedClientEventId;
+    const guardEventId = typeof verdict.guard_event_id === "string" && verdict.guard_event_id.length > 0 ? verdict.guard_event_id : void 0;
+    let transformedInput = typeof verdict.transformed_input === "string" && verdict.transformed_input.length > 0 ? verdict.transformed_input : sanitizedInput;
+    if (transformedInput !== sanitizedInput) {
+      const mutated = setPromptInMessages(messages, transformedInput);
+      if (!mutated) {
+        transformedInput = sanitizedInput;
+      }
+    }
+    this.runs.set(id, {
+      input: transformedInput,
+      startedAtMs: Date.now(),
+      model: modelName,
+      clientEventId: canonicalClientEventId,
+      guardEventId
+    });
+    logCallbackDebug("handleChatModelStart state_set", {
+      runId: id,
+      clientEventId: canonicalClientEventId,
+      guardEventId: guardEventId ?? null
+    });
+  }
+  async handleLLMEnd(output, runId) {
+    const id = String(runId ?? "");
+    logCallbackDebug("handleLLMEnd", { runId: id });
+    const state = this.runs.get(id);
+    if (!state) {
+      logCallbackDebug("handleLLMEnd missing_state", { runId: id });
+      return;
+    }
+    this.runs.delete(id);
+    const latency = Date.now() - state.startedAtMs;
+    const outText = extractOutputText(output);
+    const usage = extractTokenUsage(output);
+    const metadata = {};
+    if (state.clientEventId) {
+      metadata.client_event_id = state.clientEventId;
+    }
+    if (state.guardEventId) {
+      metadata.guard_event_id = state.guardEventId;
+    }
+    const resolvedModel = state.model ?? extractModelFromOutput(output) ?? "unknown";
+    await this.agent.log({
+      system_id: this.systemId,
+      input: state.input,
+      output: outText,
+      event_id: state.clientEventId,
+      model: resolvedModel,
+      usage,
+      latency,
+      metadata: Object.keys(metadata).length > 0 ? metadata : void 0,
+      client_capabilities: this.getLangchainCapabilities()
+    }, this.requestOptions);
+    logCallbackDebug("handleLLMEnd logged", {
+      runId: id,
+      clientEventId: state.clientEventId ?? null
+    });
+  }
+  async handleLLMError(err, runId) {
+    const id = String(runId ?? "");
+    logCallbackDebug("handleLLMError", { runId: id });
+    const state = this.runs.get(id);
+    if (state) this.runs.delete(id);
+    const message = err && typeof err === "object" && "message" in err ? String(err.message) : String(err ?? "");
+    const metadata = {
+      error_message: message
+    };
+    if (state?.clientEventId) {
+      metadata.client_event_id = state.clientEventId;
+    }
+    if (state?.guardEventId) {
+      metadata.guard_event_id = state.guardEventId;
+    }
+    await this.agent.log({
+      system_id: this.systemId,
+      input: state?.input ?? "",
+      output: "",
+      event_id: state?.clientEventId,
+      model: state?.model ?? "unknown",
+      event_type: "error",
+      severity: "error",
+      metadata,
+      client_capabilities: this.getLangchainCapabilities()
+    }, this.requestOptions);
+  }
+};
+
+export {
+  AgentIDCallbackHandler
+};

package/dist/index.d.mts

@@ -1,4 +1,5 @@
-export { A as AgentID, a as AgentIDCallbackHandler, G as GuardParams, b as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions } from './langchain-CWHTxef0.mjs';
+export { A as AgentID, a as AgentIDCallbackHandler, G as GuardParams, b as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions } from './langchain-BykeB2WB.mjs';
+import '@langchain/core/callbacks/base';
 
 type PIIMapping = Record<string, string>;
 declare class PIIManager {

package/dist/index.d.ts

@@ -1,4 +1,5 @@
-export { A as AgentID, a as AgentIDCallbackHandler, G as GuardParams, b as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions } from './langchain-CWHTxef0.js';
+export { A as AgentID, a as AgentIDCallbackHandler, G as GuardParams, b as GuardResponse, L as LogParams, P as PreparedInput, R as RequestOptions } from './langchain-BykeB2WB.js';
+import '@langchain/core/callbacks/base';
 
 type PIIMapping = Record<string, string>;
 declare class PIIManager {