Sessions

import { Agent } from "cognitio-agent-sdk"

const apiKey = process.env.OPENAI_API_KEY
if (!apiKey) throw new Error("Set OPENAI_API_KEY before running this example.")

const agent = new Agent({
  model: "openai/gpt-4.1-mini",
  instructions: "Remember conversation facts and answer briefly.",
  spawn: { env: { OPENAI_API_KEY: apiKey } },
  disallowedTools: ["*"],
  maxTurns: 1,
  maxBudgetUsd: 0.03,
})
try {
  const session = await agent.createSession({ title: "Database choice" })
  const saved = await session.send("Remember that this project uses PostgreSQL. Reply with Saved.")
  if (saved.subtype !== "success") throw new Error(`First turn ended: ${saved.subtype}`)

  const sessionId = session.id
  await session.close()

  const resumed = await agent.resume(sessionId)
  const result = await resumed.send("Which database did I name? Reply only with its name.")
  if (result.subtype !== "success") throw new Error(`Follow-up ended: ${result.subtype}`)
  if (!result.text?.toLowerCase().includes("postgresql")) throw new Error("The expected fact was not recalled.")
  console.log({ sessionId: resumed.id, answer: result.text })
  await resumed.close()
} finally {
  await agent.close()
}

const branch = await agent.fork(session.id)
try {
  const alternative = await branch.send("Given the earlier context, suggest an alternative database.")
  if (alternative.subtype !== "success") throw new Error(`Branch ended: ${alternative.subtype}`)
  console.log({ original: session.id, branch: branch.id, answer: alternative.text })
} finally {
  await branch.close()
}