import { SqliteStorage } from "@agentium/core";

const storage = new SqliteStorage("./data/agentium.db");

// No initialize() required—table is created on first use
await storage.set("sessions", "user-123", { lastActive: Date.now() });
const session = await storage.get("sessions", "user-123");

// Persistent file
const storage = new SqliteStorage("./data/agentium.db");

// In-memory (useful for tests)
const memoryStorage = new SqliteStorage(":memory:");

import { Agent, Memory, SqliteStorage, openai } from "@agentium/core";

const storage = new SqliteStorage("./data/agent.db");

const memory = new Memory({
  storage,
  maxShortTermMessages: 50,
  enableLongTerm: true,
});

const agent = new Agent({
  name: "Assistant",
  model: openai("gpt-4o"),
  storage,
  memory,
});

const result = await agent.run("Remember: my favorite color is blue.", {
  sessionId: "user-456",
});

// Data persists across restarts
await agent.run("What's my favorite color?", { sessionId: "user-456" });

// List all keys in namespace
const all = await storage.list("sessions");

// List keys with prefix
const userSessions = await storage.list("sessions", "user-");

process.on("SIGTERM", async () => {
  await storage.close();
  process.exit(0);
});