Setup and Teardown Hooks

import { connectAgent } from "langwatch/agent";

const shopAssistant = connectAgent(
  {
    name: "shop-assistant",
    parameters: {
      loyalty_tier: { default: "none" },
    },
  },
  async ({ messages, state }) => {
    // `state` is what setup returned
    return await runShopAssistant({ messages, customerId: state?.customerId });
  },
);

shopAssistant.scenario.setup(async (run, params) => {
  const customer = await createTestCustomer({
    tier: params.loyalty_tier,
    label: run.scenarioRunId,
  });
  return { customerId: customer.id };
});

shopAssistant.scenario.teardown(async (run, state, params) => {
  // Setup may have failed before it made the customer
  if (!state?.customerId) return;
  await deleteTestCustomer(state.customerId as string);
  return { deletedCustomer: state.customerId };
});

import langwatch


@langwatch.connect_agent(name="shop-assistant")
async def shop_assistant(messages: list[dict], state: dict | None = None) -> str:
    # `state` is what setup returned
    return await run_shop_assistant(messages, customer_id=(state or {}).get("customerId"))


@shop_assistant.scenario.setup
async def setup(run: langwatch.agent.ScenarioRun, loyalty_tier: str = "none") -> dict:
    customer = await create_test_customer(tier=loyalty_tier, label=run.scenario_run_id)
    return {"customerId": customer.id}


@shop_assistant.scenario.teardown
async def teardown(run: langwatch.agent.ScenarioRun, state: dict | None, loyalty_tier: str = "none") -> dict | None:
    # Setup may have failed before it made the customer
    customer_id = (state or {}).get("customerId")
    if not customer_id:
        return None
    await delete_test_customer(customer_id)
    return {"deletedCustomer": customer_id}