SDK quickstart

import atlan_ai

client = atlan_ai.init(
    service_name="support-rover",  # required; names your agent
)

import * as atlan from "@atlanai/sdk";

const client = atlan.init({
  serviceName: "support-rover", // required; names your agent
});

visitor = client.identify(
    identity_source="slack",
    external_id="U0123ABC",
    traits={"name": "Jordan Lee", "email": "jordan@example.com"},
)

const visitor = await client.identify({
  identitySource: "slack",
  externalId: "U0123ABC",
  traits: { name: "Jordan Lee", email: "jordan@example.com" },
});

with atlan_ai.propagate_attributes(
    session_id="ticket-4821",
    visitor_id=visitor.id if visitor else None,
):
    with client.start_as_current_span("handle-ticket", as_type="task") as root:
        with client.start_as_current_span("chat", as_type="llm") as generation:
            generation.update(
                model="claude-sonnet-5",
                provider="anthropic",
                usage={"input_tokens": 1200, "output_tokens": 340},
                cost={"input": 0.0036, "output": 0.0051},
            )
        root.score_trace("resolved", value=True, data_type="BOOLEAN")

await atlan.propagateAttributes(
  { sessionId: "ticket-4821", visitorId: visitor?.id },
  async () => {
    await client.startAsCurrentSpan("handle-ticket", { asType: "task" }, async (root) => {
      await client.startAsCurrentSpan("chat", { asType: "llm" }, async (generation) => {
        generation.update({
          model: "claude-sonnet-5",
          provider: "anthropic",
          usage: { input_tokens: 1200, output_tokens: 340 },
          cost: { input: 0.0036, output: 0.0051 },
        });
      });
      root.scoreTrace("resolved", true, { dataType: "BOOLEAN" });
    });
  },
);

client.flush()

await client.flush();