agentid-sdk 0.1.16 → 0.1.18

package/dist/chunk-DXUA5DKG.mjs

@@ -1,266 +0,0 @@
-// src/langchain.ts
-function safeString(val) {
-  return typeof val === "string" ? val : "";
-}
-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 = typed?.role ?? typed?.type;
-    if (role === "user" || role === "human") {
-      last = typed;
-    }
-  }
-  if (!last || typeof last !== "object") {
-    return "";
-  }
-  const typedLast = last;
-  return safeString(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 model = kw?.model ?? kw?.model_name ?? kw?.modelName;
-  if (typeof model === "string" && model) return model;
-  const ser = (serialized && typeof serialized === "object" ? serialized : null) ?? null;
-  const name = ser?.name ?? ser?.id;
-  if (typeof name === "string" && name) return name;
-  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;
-  return usage && typeof usage === "object" ? usage : void 0;
-}
-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 {
-  constructor(agent, options) {
-    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 ?? "");
-    if (!input) {
-      throw new Error("AgentID: No prompt found. Security guard requires string input.");
-    }
-    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 verdict = await this.agent.guard({
-      input: sanitizedInput,
-      system_id: this.systemId,
-      client_capabilities: this.getLangchainCapabilities()
-    }, this.requestOptions);
-    if (!verdict.allowed) {
-      throw new Error(`AgentID: Security Blocked (${verdict.reason ?? "guard_denied"})`);
-    }
-    const 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) {
-        throw new Error(
-          "AgentID: Guard transformed input could not be applied to LangChain prompt payload."
-        );
-      }
-    }
-    this.runs.set(id, {
-      input: transformedInput,
-      startedAtMs: Date.now(),
-      model: extractModel(serialized, extraParams)
-    });
-  }
-  async handleChatModelStart(serialized, messages, runId, _parentRunId, extraParams) {
-    const input = extractPromptFromMessages(messages);
-    const id = String(runId ?? "");
-    if (!input) {
-      throw new Error("AgentID: No user message found. Security guard requires string input.");
-    }
-    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 verdict = await this.agent.guard({
-      input: sanitizedInput,
-      system_id: this.systemId,
-      client_capabilities: this.getLangchainCapabilities()
-    }, this.requestOptions);
-    if (!verdict.allowed) {
-      throw new Error(`AgentID: Security Blocked (${verdict.reason ?? "guard_denied"})`);
-    }
-    const 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) {
-        throw new Error(
-          "AgentID: Guard transformed input could not be applied to LangChain message payload."
-        );
-      }
-    }
-    this.runs.set(id, {
-      input: transformedInput,
-      startedAtMs: Date.now(),
-      model: extractModel(serialized, extraParams)
-    });
-  }
-  async handleLLMEnd(output, runId) {
-    const id = String(runId ?? "");
-    const state = this.runs.get(id);
-    if (!state) return;
-    this.runs.delete(id);
-    const latency = Date.now() - state.startedAtMs;
-    const outText = extractOutputText(output);
-    const usage = extractTokenUsage(output);
-    this.agent.log({
-      system_id: this.systemId,
-      input: state.input,
-      output: outText,
-      model: state.model ?? "unknown",
-      usage,
-      latency,
-      client_capabilities: this.getLangchainCapabilities()
-    }, this.requestOptions);
-  }
-  async handleLLMError(err, runId) {
-    const id = String(runId ?? "");
-    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 ?? "");
-    this.agent.log({
-      system_id: this.systemId,
-      input: state?.input ?? "",
-      output: "",
-      model: state?.model ?? "unknown",
-      event_type: "error",
-      severity: "error",
-      metadata: {
-        error_message: message
-      },
-      client_capabilities: this.getLangchainCapabilities()
-    }, this.requestOptions);
-  }
-};
-
-export {
-  AgentIDCallbackHandler
-};