Score production sessions

import os
from atlanai import AtlanClient
from atlanai.tracing import init_logger, propagate_attributes

client = AtlanClient(os.environ.get("ATLAN_BASE_URL", "https://api.atlan.com"),
                     bearer_token=os.environ["ATLAN_API_KEY"], workspace=os.environ["ATLAN_WORKSPACE_ID"])
logger = init_logger(project_name="support-agent")

conversation_id = "chat-0001"                       # your own conversation or thread ID
with propagate_attributes(session_id=conversation_id):
    with logger.client.start_as_current_span("handle-request", as_type="task") as root:
        answer = your_agent("Where is my refund for order 1042?")
        root.update(input="Where is my refund for order 1042?", output=answer)
logger.flush()

# When the conversation ends:
session = client.sessions.create_record({
    "workspace_id": os.environ["ATLAN_WORKSPACE_ID"],
    "name": conversation_id,
    "subject_kind": "agent", "subject_id": "agent_01example",
    "external_session_id": conversation_id,          # must equal the session_id on the spans
    "session_status": "completed",
})
print(session.id)                                     # pass this to session_ids

import { AtlanClient, rawAgent } from "@atlanai/sdk";
import { initLogger, propagateAttributes } from "@atlanai/sdk/tracing";

const workspace = process.env.ATLAN_WORKSPACE_ID!;
const client = new AtlanClient({
  gatewayOrigin: process.env.ATLAN_BASE_URL ?? "https://api.atlan.com",
  bearerToken: process.env.ATLAN_API_KEY!,
  workspace,
});
const logger = initLogger({ projectName: "support-agent" });

const conversationId = "chat-0001";                 // your own conversation or thread ID
await propagateAttributes({ sessionId: conversationId }, () =>
  logger.client.startAsCurrentSpan("handle-request", { asType: "task" }, async (root) => {
    const answer = await yourAgent("Where is my refund for order 1042?");
    root.update({ input: "Where is my refund for order 1042?", output: answer });
  }));
await logger.flush();

// When the conversation ends:
const session = await client.sessions.createRecord({
  workspaceId: workspace,
  name: conversationId,
  subjectKind: rawAgent.AgentSubjectKind.Agent, subjectId: "agent_01example",
  externalSessionId: conversationId,                  // must equal the sessionId on the spans
  sessionStatus: rawAgent.AgentSessionStatus.Completed,
});
console.log(session.id);                              // pass this to sessionIds