LangChain

import os
from atlasent import AtlaSentClient
from atlasent_langchain import with_langchain_guard
from langchain_core.tools import tool

client = AtlaSentClient(api_key=os.environ["ATLASENT_API_KEY"])

def write_ehr(patient_id: str, note: str) -> dict:
    ...

guarded_write_ehr = with_langchain_guard(
    write_ehr,
    client,
    agent="service:clinical-agent",
    extra_context={"environment": "production"},
    # Action classes require a state snapshot by default (SNAPSHOT_REQUIRED otherwise)
    state_snapshot={"source": "clinical-agent", "complete": True},
)

write_ehr_tool = tool(guarded_write_ehr)

import { AtlaSentClient } from "@atlasent/sdk";
import { withLangChainGuard } from "@atlasent/langchain";
import { DynamicStructuredTool } from "@langchain/core/tools";
import { z } from "zod";

const client = new AtlaSentClient({ apiKey: process.env.ATLASENT_API_KEY! });

const [queryDb] = withLangChainGuard(
  [
    {
      name: "query_database",
      description: "Run a read-only SQL query and return results as JSON",
      execute: async ({ sql }) => JSON.stringify(await db.query(sql)),
    },
  ],
  client,
  {
    agent: "service:analytics-bot",
    extraContext: { environment: "production" },
    // Action classes require a state snapshot by default (SNAPSHOT_REQUIRED otherwise)
    stateSnapshot: { source: "analytics-bot", complete: true },
  },
);

const tool = new DynamicStructuredTool({
  name: queryDb.name,
  description: queryDb.description,
  schema: z.object({ sql: z.string() }),
  func: queryDb.execute,
});