@corsair-dev/studio 0.1.2 → 0.1.4

package/dist/server/index.js

@@ -341,6 +341,594 @@ var exchangeOAuth = async (ctx) => {
   return { ok: true };
 };
 
+// src/server/handlers/chat.ts
+import { buildCorsairToolDefs } from "@corsair-dev/mcp";
+import { streamText, tool } from "ai";
+import { z } from "zod";
+
+// src/server/chat-store.ts
+import Database from "better-sqlite3";
+import { Kysely, SqliteDialect } from "kysely";
+
+// src/server/model-pricing.ts
+var MODEL_PRICES = {
+  "gpt-4o": { input: 2.5, output: 10 },
+  "gpt-4o-2024-11-20": { input: 2.5, output: 10 },
+  "gpt-4o-2024-08-06": { input: 2.5, output: 10 },
+  "gpt-4o-mini": { input: 0.15, output: 0.6 },
+  "gpt-4-turbo": { input: 10, output: 30 },
+  "gpt-4": { input: 30, output: 60 },
+  "gpt-3.5-turbo": { input: 0.5, output: 1.5 },
+  "gpt-4.1": { input: 2, output: 8 },
+  "gpt-4.1-mini": { input: 0.4, output: 1.6 },
+  "gpt-4.1-nano": { input: 0.1, output: 0.4 },
+  "gpt-5": { input: 1.25, output: 10 },
+  "gpt-5-mini": { input: 0.25, output: 2 },
+  "gpt-5-nano": { input: 0.05, output: 0.4 },
+  o1: { input: 15, output: 60 },
+  "o1-mini": { input: 3, output: 12 },
+  "o1-preview": { input: 15, output: 60 },
+  "o3-mini": { input: 1.1, output: 4.4 },
+  o3: { input: 2, output: 8 },
+  "o4-mini": { input: 1.1, output: 4.4 },
+  "claude-opus-4-5": { input: 15, output: 75 },
+  "claude-opus-4-7": { input: 15, output: 75 },
+  "claude-opus-4-1": { input: 15, output: 75 },
+  "claude-sonnet-4-5": { input: 3, output: 15 },
+  "claude-sonnet-4-6": { input: 3, output: 15 },
+  "claude-sonnet-4": { input: 3, output: 15 },
+  "claude-haiku-4-5": { input: 1, output: 5 },
+  "claude-3-7-sonnet-latest": { input: 3, output: 15 },
+  "claude-3-5-sonnet-latest": { input: 3, output: 15 },
+  "claude-3-5-sonnet-20241022": { input: 3, output: 15 },
+  "claude-3-5-haiku-latest": { input: 0.8, output: 4 },
+  "claude-3-5-haiku-20241022": { input: 0.8, output: 4 },
+  "claude-3-opus-latest": { input: 15, output: 75 },
+  "claude-3-haiku-20240307": { input: 0.25, output: 1.25 },
+  "gemini-2.5-pro": { input: 1.25, output: 10 },
+  "gemini-2.5-flash": { input: 0.3, output: 2.5 },
+  "gemini-2.5-flash-lite": { input: 0.1, output: 0.4 },
+  "gemini-2.0-flash": { input: 0.1, output: 0.4 },
+  "gemini-2.0-flash-001": { input: 0.1, output: 0.4 },
+  "gemini-2.0-flash-lite": { input: 0.075, output: 0.3 },
+  "gemini-2.0-pro-exp": { input: 1.25, output: 5 },
+  "gemini-1.5-pro": { input: 1.25, output: 5 },
+  "gemini-1.5-pro-latest": { input: 1.25, output: 5 },
+  "gemini-1.5-flash": { input: 0.075, output: 0.3 },
+  "gemini-1.5-flash-latest": { input: 0.075, output: 0.3 },
+  "gemini-1.5-flash-8b": { input: 0.0375, output: 0.15 },
+  "llama-3.3-70b-versatile": { input: 0.59, output: 0.79 },
+  "llama-3.1-70b-versatile": { input: 0.59, output: 0.79 },
+  "llama-3.1-8b-instant": { input: 0.05, output: 0.08 },
+  "llama-3.2-1b-preview": { input: 0.04, output: 0.04 },
+  "llama-3.2-3b-preview": { input: 0.06, output: 0.06 },
+  "llama-3.2-11b-vision-preview": { input: 0.18, output: 0.18 },
+  "llama-3.2-90b-vision-preview": { input: 0.9, output: 0.9 },
+  "mixtral-8x7b-32768": { input: 0.24, output: 0.24 },
+  "gemma2-9b-it": { input: 0.2, output: 0.2 },
+  "deepseek-r1-distill-llama-70b": { input: 0.75, output: 0.99 },
+  "deepseek-chat": { input: 0.27, output: 1.1 },
+  "deepseek-reasoner": { input: 0.55, output: 2.19 },
+  "mistral-large-latest": { input: 2, output: 6 },
+  "mistral-small-latest": { input: 0.2, output: 0.6 },
+  "codestral-latest": { input: 0.3, output: 0.9 },
+  "grok-2": { input: 2, output: 10 },
+  "grok-2-mini": { input: 0.3, output: 0.5 },
+  "grok-3": { input: 3, output: 15 },
+  "grok-3-mini": { input: 0.3, output: 0.5 },
+  "grok-4": { input: 5, output: 25 }
+};
+function normalizeId(id) {
+  return id.toLowerCase().replace(/^models\//, "").replace(/^anthropic\//, "").replace(/^openai\//, "").replace(/^google\//, "").replace(/^groq\//, "").trim();
+}
+function candidateIds(modelId) {
+  const norm = normalizeId(modelId);
+  const variants = /* @__PURE__ */ new Set([modelId, norm]);
+  const stripDate = norm.replace(/-\d{8}$/, "").replace(/-\d{4}-\d{2}-\d{2}$/, "");
+  if (stripDate !== norm) variants.add(stripDate);
+  const stripVersion = norm.replace(/-(?:00\d|0?\d{2,3})$/, "");
+  if (stripVersion !== norm) variants.add(stripVersion);
+  const stripLatest = norm.replace(/-latest$/, "");
+  if (stripLatest !== norm) variants.add(stripLatest);
+  const stripPreview = norm.replace(/-preview$/, "").replace(/-exp$/, "");
+  if (stripPreview !== norm) variants.add(stripPreview);
+  return Array.from(variants);
+}
+var PRICES_NORMALIZED = (() => {
+  const map = {};
+  for (const [k, v] of Object.entries(MODEL_PRICES)) {
+    map[normalizeId(k)] = v;
+  }
+  return map;
+})();
+function getModelPrice(modelId) {
+  if (MODEL_PRICES[modelId]) return MODEL_PRICES[modelId];
+  for (const candidate of candidateIds(modelId)) {
+    const direct = PRICES_NORMALIZED[candidate];
+    if (direct) return direct;
+  }
+  const norm = normalizeId(modelId);
+  let bestMatch = null;
+  for (const [key, price] of Object.entries(PRICES_NORMALIZED)) {
+    if (norm.startsWith(key) || key.startsWith(norm)) {
+      if (!bestMatch || key.length > bestMatch.key.length) {
+        bestMatch = { key, price };
+      }
+    }
+  }
+  return bestMatch?.price ?? null;
+}
+function computeCost(modelId, usage) {
+  const price = getModelPrice(modelId);
+  if (!price) {
+    console.warn(`[corsair:pricing] no price entry for model "${modelId}"`);
+    return null;
+  }
+  const input = usage.inputTokens / 1e6 * price.input;
+  const output = usage.outputTokens / 1e6 * price.output;
+  return input + output;
+}
+
+// src/server/chat-store.ts
+var sqlite = new Database(":memory:");
+var db = new Kysely({
+  dialect: new SqliteDialect({ database: sqlite })
+});
+var schemaReady = initSchema();
+async function initSchema() {
+  await db.schema.createTable("chats").ifNotExists().addColumn("id", "text", (col) => col.primaryKey()).addColumn("title", "text", (col) => col.notNull()).addColumn("created_at", "integer", (col) => col.notNull()).execute();
+  await db.schema.createTable("chat_messages").ifNotExists().addColumn("id", "text", (col) => col.primaryKey()).addColumn("chat_id", "text", (col) => col.notNull()).addColumn("role", "text", (col) => col.notNull()).addColumn("blocks", "text", (col) => col.notNull()).addColumn("error", "text").addColumn("seq", "integer", (col) => col.notNull()).addColumn("usage", "text").execute();
+}
+var _seq = 0;
+function getTextFromBlocks(blocks) {
+  return blocks.filter(
+    (block) => block.type === "text"
+  ).map((block) => block.content).join("");
+}
+function parseRawUsage(raw) {
+  if (!raw) return null;
+  try {
+    const parsed = JSON.parse(raw);
+    if (typeof parsed.model !== "string" || typeof parsed.inputTokens !== "number" || typeof parsed.outputTokens !== "number" || typeof parsed.totalTokens !== "number") {
+      return null;
+    }
+    return {
+      model: parsed.model,
+      inputTokens: parsed.inputTokens,
+      outputTokens: parsed.outputTokens,
+      totalTokens: parsed.totalTokens
+    };
+  } catch {
+    return null;
+  }
+}
+function withCost(raw) {
+  return {
+    ...raw,
+    cost: computeCost(raw.model, raw)
+  };
+}
+function aggregateUsage(rows) {
+  const total = {
+    inputTokens: 0,
+    outputTokens: 0,
+    totalTokens: 0,
+    cost: 0,
+    hasUnknownCost: false
+  };
+  for (const u of rows) {
+    if (!u) continue;
+    total.inputTokens += u.inputTokens;
+    total.outputTokens += u.outputTokens;
+    total.totalTokens += u.totalTokens;
+    if (u.cost === null) {
+      total.hasUnknownCost = true;
+    } else {
+      total.cost += u.cost;
+    }
+  }
+  return total;
+}
+async function createChat() {
+  await schemaReady;
+  const id = crypto.randomUUID();
+  const created_at = Date.now();
+  const title = "New chat";
+  await db.insertInto("chats").values({ id, title, created_at }).execute();
+  return {
+    id,
+    title,
+    created_at,
+    usage_total: aggregateUsage([])
+  };
+}
+async function listChats() {
+  await schemaReady;
+  const chats = await db.selectFrom("chats").select(["id", "title", "created_at"]).orderBy("created_at", "desc").execute();
+  const messageRows = await db.selectFrom("chat_messages").select(["chat_id", "usage"]).execute();
+  const usageByChat = /* @__PURE__ */ new Map();
+  for (const row of messageRows) {
+    const raw = parseRawUsage(row.usage);
+    if (!raw) continue;
+    const list = usageByChat.get(row.chat_id) ?? [];
+    list.push(withCost(raw));
+    usageByChat.set(row.chat_id, list);
+  }
+  return chats.map((chat) => ({
+    id: chat.id,
+    title: chat.title,
+    created_at: chat.created_at,
+    usage_total: aggregateUsage(usageByChat.get(chat.id) ?? [])
+  }));
+}
+async function chatExists(chatId) {
+  await schemaReady;
+  const row = await db.selectFrom("chats").select("id").where("id", "=", chatId).executeTakeFirst();
+  return !!row;
+}
+async function getMessages(chatId) {
+  await schemaReady;
+  const rows = await db.selectFrom("chat_messages").select(["id", "chat_id", "role", "blocks", "error", "usage"]).where("chat_id", "=", chatId).orderBy("seq", "asc").execute();
+  return rows.map((row) => {
+    const raw = parseRawUsage(row.usage);
+    return {
+      id: row.id,
+      chat_id: row.chat_id,
+      role: row.role,
+      blocks: JSON.parse(row.blocks),
+      error: row.error,
+      usage: raw ? withCost(raw) : null
+    };
+  });
+}
+async function appendMessage(chatId, id, role, blocks, options = {}) {
+  await schemaReady;
+  await db.insertInto("chat_messages").values({
+    id,
+    chat_id: chatId,
+    role,
+    blocks: JSON.stringify(blocks),
+    error: options.error ?? null,
+    seq: ++_seq,
+    usage: options.usage ? JSON.stringify(options.usage) : null
+  }).execute();
+  if (role === "user") {
+    const row = await db.selectFrom("chats").select("title").where("id", "=", chatId).executeTakeFirst();
+    if (row?.title === "New chat") {
+      const text = getTextFromBlocks(blocks);
+      const newTitle = text.slice(0, 60).trim();
+      if (newTitle) {
+        await db.updateTable("chats").set({ title: newTitle }).where("id", "=", chatId).execute();
+      }
+    }
+  }
+}
+
+// src/server/handlers/chat.ts
+function isTextBlock(block) {
+  return block.type === "text";
+}
+function isCompletedToolBlock(block) {
+  return block.type === "tool" && block.result !== void 0;
+}
+function isPendingToolBlock(block) {
+  return block.type === "tool" && block.result === void 0;
+}
+function joinTextBlocks(blocks) {
+  return blocks.map((block) => block.content).join("");
+}
+function toToolArgs(args) {
+  if (args && typeof args === "object") {
+    const toolArgs = {};
+    for (const [key, value] of Object.entries(args)) {
+      toolArgs[key] = value;
+    }
+    return toolArgs;
+  }
+  return {};
+}
+function makeTextPart(block) {
+  return { type: "text", text: block.content };
+}
+function makeToolCallPart(toolCallId, block) {
+  return {
+    type: "tool-call",
+    toolCallId,
+    toolName: block.name,
+    args: toToolArgs(block.args)
+  };
+}
+function makeToolResultPart(toolCallId, block) {
+  return {
+    type: "tool-result",
+    toolCallId,
+    toolName: block.name,
+    result: block.result
+  };
+}
+function findPendingToolBlockIndex(blocks, toolName) {
+  return blocks.findLastIndex(
+    (block) => isPendingToolBlock(block) && block.name === toolName
+  );
+}
+function toAiMessages(stored) {
+  const result = [];
+  for (const msg of stored) {
+    const textBlocks = msg.blocks.filter(isTextBlock);
+    const text = joinTextBlocks(textBlocks);
+    if (msg.role === "user") {
+      result.push({ role: "user", content: text });
+      continue;
+    }
+    const toolBlocks = msg.blocks.filter(isCompletedToolBlock);
+    if (toolBlocks.length === 0) {
+      result.push({ role: "assistant", content: text });
+      continue;
+    }
+    const toolCallIds = toolBlocks.map(() => crypto.randomUUID());
+    const assistantParts = textBlocks.map(makeTextPart);
+    const toolParts = [];
+    for (const [index, block] of toolBlocks.entries()) {
+      const toolCallId = toolCallIds[index];
+      if (!toolCallId) {
+        continue;
+      }
+      assistantParts.push(makeToolCallPart(toolCallId, block));
+      toolParts.push(makeToolResultPart(toolCallId, block));
+    }
+    result.push({ role: "assistant", content: assistantParts });
+    result.push({ role: "tool", content: toolParts });
+  }
+  return result;
+}
+function buildAiTools(corsairClient) {
+  const defs = buildCorsairToolDefs({ corsair: corsairClient, setup: false });
+  const tools = {};
+  for (const def of defs) {
+    tools[def.name] = tool({
+      description: def.description,
+      parameters: z.object(def.shape),
+      execute: async (args) => {
+        const result = await def.handler(args);
+        const texts = result.content.filter((c) => c.type === "text");
+        if (result.isError) {
+          throw new Error(texts.map((c) => c.text).join("\n"));
+        }
+        const text = texts[0]?.text ?? "";
+        try {
+          return JSON.parse(text);
+        } catch {
+          return text;
+        }
+      }
+    });
+  }
+  return tools;
+}
+async function resolveModel() {
+  const override = process.env.CORSAIR_CHAT_MODEL;
+  console.log(
+    "[corsair:chat] resolving model \u2014 OPENAI_API_KEY:",
+    !!process.env.OPENAI_API_KEY,
+    "| ANTHROPIC_API_KEY:",
+    !!process.env.ANTHROPIC_API_KEY,
+    "| CORSAIR_CHAT_MODEL:",
+    override ?? "(default)"
+  );
+  if (process.env.OPENAI_API_KEY) {
+    const { openai } = await import("@ai-sdk/openai");
+    const modelId = override ?? "gpt-4o-mini";
+    console.log("[corsair:chat] using openai:", modelId);
+    return { model: openai(modelId), modelId };
+  }
+  if (process.env.ANTHROPIC_API_KEY) {
+    const { anthropic } = await import("@ai-sdk/anthropic");
+    const modelId = override ?? "claude-sonnet-4-6";
+    console.log("[corsair:chat] using anthropic:", modelId);
+    return { model: anthropic(modelId), modelId };
+  }
+  if (process.env.GOOGLE_GENERATIVE_AI_API_KEY) {
+    const { google } = await import("@ai-sdk/google");
+    const modelId = override ?? "gemini-2.0-flash";
+    return { model: google(modelId), modelId };
+  }
+  if (process.env.GROQ_API_KEY) {
+    const { createGroq } = await import("@ai-sdk/groq");
+    const modelId = override ?? "llama-3.3-70b-versatile";
+    return { model: createGroq()(modelId), modelId };
+  }
+  throw new Error(
+    "No AI provider configured. Set one of: OPENAI_API_KEY, ANTHROPIC_API_KEY, GOOGLE_GENERATIVE_AI_API_KEY, or GROQ_API_KEY."
+  );
+}
+function errorMessage(err) {
+  return err instanceof Error ? err.message : String(err);
+}
+function isObject(value) {
+  return typeof value === "object" && value !== null;
+}
+function parseRawUsage2(raw) {
+  if (!isObject(raw)) return void 0;
+  const usage = {};
+  if (typeof raw.promptTokens === "number")
+    usage.promptTokens = raw.promptTokens;
+  if (typeof raw.completionTokens === "number")
+    usage.completionTokens = raw.completionTokens;
+  if (typeof raw.totalTokens === "number") usage.totalTokens = raw.totalTokens;
+  return usage;
+}
+function parseStreamPart(raw) {
+  if (!isObject(raw) || typeof raw.type !== "string") {
+    return null;
+  }
+  const textDelta = typeof raw.textDelta === "string" ? raw.textDelta : void 0;
+  const toolName = typeof raw.toolName === "string" ? raw.toolName : void 0;
+  let responseModelId;
+  if (isObject(raw.response) && typeof raw.response.modelId === "string") {
+    responseModelId = raw.response.modelId;
+  }
+  return {
+    type: raw.type,
+    textDelta,
+    toolName,
+    args: raw.args,
+    result: raw.result,
+    usage: parseRawUsage2(raw.usage),
+    responseModelId
+  };
+}
+function buildRawMessageUsage(modelId, raw) {
+  if (!raw) return null;
+  const inputTokens = raw.promptTokens ?? 0;
+  const outputTokens = raw.completionTokens ?? 0;
+  const totalTokens = raw.totalTokens ?? inputTokens + outputTokens;
+  if (inputTokens === 0 && outputTokens === 0 && totalTokens === 0) return null;
+  return { model: modelId, inputTokens, outputTokens, totalTokens };
+}
+var chatHandler = async (ctx) => {
+  console.log("[corsair:chat] request received");
+  const body = await readJsonBody(ctx.req);
+  const tenant = body.tenant ? String(body.tenant) : void 0;
+  const chatId = body.chatId ? String(body.chatId) : void 0;
+  const newUserText = body.message ? String(body.message) : void 0;
+  const storedHistory = chatId && await chatExists(chatId) ? await getMessages(chatId) : [];
+  const messages = toAiMessages(storedHistory);
+  if (newUserText) {
+    messages.push({ role: "user", content: newUserText });
+  }
+  console.log(
+    "[corsair:chat] history:",
+    storedHistory.length,
+    "stored msgs \u2192",
+    messages.length,
+    "AI msgs | tenant:",
+    tenant ?? "(none)",
+    "| chatId:",
+    chatId ?? "(none)"
+  );
+  let resolved;
+  try {
+    resolved = await resolveModel();
+  } catch (err) {
+    const message = errorMessage(err);
+    console.error("[corsair:chat] model resolution failed:", message);
+    ctx.res.writeHead(400, { "content-type": "application/json" });
+    ctx.res.end(JSON.stringify({ error: message }));
+    return;
+  }
+  const handle = await ctx.getCorsair();
+  const client = handle.resolveClient(tenant);
+  const tools = buildAiTools(client);
+  const userMsgId = crypto.randomUUID();
+  if (chatId && await chatExists(chatId) && newUserText) {
+    try {
+      await appendMessage(chatId, userMsgId, "user", [
+        { type: "text", content: newUserText }
+      ]);
+    } catch (err) {
+      console.error("[corsair:chat] failed to save user message:", err);
+    }
+  }
+  ctx.res.writeHead(200, {
+    "content-type": "text/event-stream",
+    "cache-control": "no-cache",
+    connection: "keep-alive"
+  });
+  const send = (data) => {
+    ctx.res.write(`data: ${JSON.stringify(data)}
+
+`);
+  };
+  const assistantBlocks = [];
+  let finalRawUsage = null;
+  console.log("[corsair:chat] starting stream");
+  try {
+    const result = streamText({
+      model: resolved.model,
+      system: "You are a helpful assistant with access to Corsair tools. Use the tools to answer questions about the user's integrations and data. Use list_operations to get the exact endpoint names.",
+      messages,
+      tools,
+      maxSteps: 10
+    });
+    for await (const raw of result.fullStream) {
+      const part = parseStreamPart(raw);
+      if (!part) {
+        continue;
+      }
+      if (part.type === "text-delta") {
+        send({ type: "text", text: part.textDelta });
+        const last = assistantBlocks[assistantBlocks.length - 1];
+        if (last?.type === "text") {
+          last.content += part.textDelta ?? "";
+        } else {
+          assistantBlocks.push({ type: "text", content: part.textDelta ?? "" });
+        }
+      } else if (part.type === "tool-call") {
+        send({ type: "tool-start", name: part.toolName, args: part.args });
+        assistantBlocks.push({
+          type: "tool",
+          name: part.toolName ?? "",
+          args: part.args
+        });
+      } else if (part.type === "tool-result") {
+        send({ type: "tool-end", name: part.toolName, result: part.result });
+        const idx = findPendingToolBlockIndex(assistantBlocks, part.toolName);
+        const block = idx >= 0 ? assistantBlocks[idx] : void 0;
+        if (block && isPendingToolBlock(block)) {
+          block.result = part.result;
+        }
+      } else if (part.type === "finish") {
+        const effectiveModelId = part.responseModelId ?? resolved.modelId;
+        const raw2 = buildRawMessageUsage(effectiveModelId, part.usage);
+        console.log(
+          "[corsair:chat] finish \u2014 model:",
+          effectiveModelId,
+          "| usage:",
+          part.usage
+        );
+        if (raw2) {
+          finalRawUsage = raw2;
+          send({ type: "usage", usage: withCost(raw2) });
+        }
+        send({ type: "done" });
+      }
+    }
+  } catch (err) {
+    const message = errorMessage(err);
+    console.error("[corsair:chat] stream error:", message);
+    send({ type: "error", message });
+  } finally {
+    if (chatId && await chatExists(chatId) && assistantBlocks.length > 0) {
+      try {
+        await appendMessage(
+          chatId,
+          crypto.randomUUID(),
+          "assistant",
+          assistantBlocks,
+          { usage: finalRawUsage ?? void 0 }
+        );
+      } catch (err) {
+        console.error("[corsair:chat] failed to save assistant message:", err);
+      }
+    }
+    console.log("[corsair:chat] done");
+    ctx.res.end();
+  }
+};
+
+// src/server/handlers/chats.ts
+var listChatsHandler = async () => {
+  return { chats: await listChats() };
+};
+var createChatHandler = async () => {
+  return { chat: await createChat() };
+};
+var getChatMessagesHandler = async (ctx) => {
+  const chatId = ctx.url.searchParams.get("chatId");
+  if (!chatId) throw new Error("chatId is required");
+  return { messages: await getMessages(chatId) };
+};
+
 // src/server/handlers/credentials-internal.ts
 var BASE_FIELDS = {
   oauth_2: {
@@ -533,9 +1121,9 @@ function getKyselyDb(database) {
 var listDbTables = async (ctx) => {
   const { internal } = await ctx.getCorsair();
   if (!internal.database) throw new Error("No database configured.");
-  const db = getKyselyDb(internal.database);
-  if (!db) throw new Error("Could not access kysely db handle.");
-  const existing = await db.introspection.getTables();
+  const db2 = getKyselyDb(internal.database);
+  if (!db2) throw new Error("Could not access kysely db handle.");
+  const existing = await db2.introspection.getTables();
   const existingNames = new Set(existing.map((t) => t.name));
   return {
     core: CORE_TABLES.filter((t) => existingNames.has(t)),
@@ -551,14 +1139,14 @@ var listDbRows = async (ctx) => {
   if (!table) throw new Error("Missing table.");
   const { internal } = await ctx.getCorsair();
   if (!internal.database) throw new Error("No database configured.");
-  const db = getKyselyDb(internal.database);
-  if (!db) throw new Error("Could not access kysely db handle.");
+  const db2 = getKyselyDb(internal.database);
+  if (!db2) throw new Error("Could not access kysely db handle.");
   let rows;
   try {
-    const q = db.selectFrom(table).selectAll().limit(limit).offset(offset);
+    const q = db2.selectFrom(table).selectAll().limit(limit).offset(offset);
     rows = await q.orderBy("created_at", "desc").execute();
   } catch {
-    rows = await db.selectFrom(table).selectAll().limit(limit).offset(offset).execute();
+    rows = await db2.selectFrom(table).selectAll().limit(limit).offset(offset).execute();
   }
   const safeRows = rows.map(redactSensitive);
   return { rows: safeRows, limit, offset };
@@ -580,9 +1168,9 @@ var queryEntityData = async (ctx) => {
   if (!entity) throw new Error("Missing entity.");
   const { internal } = await ctx.getCorsair();
   if (!internal.database) throw new Error("No database configured.");
-  const db = getKyselyDb(internal.database);
-  if (!db) throw new Error("Could not access kysely db handle.");
-  const base = db.selectFrom("corsair_entities as e").innerJoin("corsair_accounts as a", "a.id", "e.account_id").innerJoin("corsair_integrations as i", "i.id", "a.integration_id").where("a.tenant_id", "=", tenant).where("i.name", "=", integration).where("e.entity_type", "=", entity);
+  const db2 = getKyselyDb(internal.database);
+  if (!db2) throw new Error("Could not access kysely db handle.");
+  const base = db2.selectFrom("corsair_entities as e").innerJoin("corsair_accounts as a", "a.id", "e.account_id").innerJoin("corsair_integrations as i", "i.id", "a.integration_id").where("a.tenant_id", "=", tenant).where("i.name", "=", integration).where("e.entity_type", "=", entity);
   let rowsRaw = [];
   let hasMore = false;
   let total = 0;
@@ -682,14 +1270,14 @@ function collectPrimitiveValues(value) {
 var listPermissions = async (ctx) => {
   const { internal } = await ctx.getCorsair();
   if (!internal.database) throw new Error("No database configured.");
-  const db = getKyselyDb(internal.database);
-  if (!db) throw new Error("Could not access kysely db handle.");
+  const db2 = getKyselyDb(internal.database);
+  if (!db2) throw new Error("Could not access kysely db handle.");
   const limit = Math.min(
     Math.max(Number(ctx.url.searchParams.get("limit") ?? 100), 1),
     500
   );
   try {
-    const rows = await db.selectFrom("corsair_permissions").selectAll().limit(limit).offset(0).orderBy("created_at", "desc").execute();
+    const rows = await db2.selectFrom("corsair_permissions").selectAll().limit(limit).offset(0).orderBy("created_at", "desc").execute();
     return { rows };
   } catch (err) {
     return { rows: [], note: err.message };
@@ -809,19 +1397,19 @@ var setupPlugin = async (ctx) => {
   if (!hasAuthConfig(internal, pluginId)) {
     throw new Error(`Plugin '${pluginId}' has no credential setup.`);
   }
-  const db = asDb(internal.database);
+  const db2 = asDb(internal.database);
   const now = /* @__PURE__ */ new Date();
-  let integration = await db.selectFrom("corsair_integrations").selectAll().where("name", "=", pluginId).executeTakeFirst();
+  let integration = await db2.selectFrom("corsair_integrations").selectAll().where("name", "=", pluginId).executeTakeFirst();
   if (!integration) {
     const id = randomUUID();
-    await db.insertInto("corsair_integrations").values({
+    await db2.insertInto("corsair_integrations").values({
       id,
       name: pluginId,
       config: {},
       created_at: now,
       updated_at: now
     }).execute();
-    integration = await db.selectFrom("corsair_integrations").selectAll().where("id", "=", id).executeTakeFirst();
+    integration = await db2.selectFrom("corsair_integrations").selectAll().where("id", "=", id).executeTakeFirst();
   }
   if (!integration) {
     throw new Error(`Failed to create integration for '${pluginId}'.`);
@@ -830,9 +1418,9 @@ var setupPlugin = async (ctx) => {
   if (typeof integrationId !== "string") {
     throw new Error(`Invalid integration row for '${pluginId}'.`);
   }
-  let account = await db.selectFrom("corsair_accounts").selectAll().where("tenant_id", "=", tenantId).where("integration_id", "=", integrationId).executeTakeFirst();
+  let account = await db2.selectFrom("corsair_accounts").selectAll().where("tenant_id", "=", tenantId).where("integration_id", "=", integrationId).executeTakeFirst();
   if (!account) {
-    await db.insertInto("corsair_accounts").values({
+    await db2.insertInto("corsair_accounts").values({
       id: randomUUID(),
       tenant_id: tenantId,
       integration_id: integrationId,
@@ -840,7 +1428,7 @@ var setupPlugin = async (ctx) => {
       created_at: now,
       updated_at: now
     }).execute();
-    account = await db.selectFrom("corsair_accounts").selectAll().where("tenant_id", "=", tenantId).where("integration_id", "=", integrationId).executeTakeFirst();
+    account = await db2.selectFrom("corsair_accounts").selectAll().where("tenant_id", "=", tenantId).where("integration_id", "=", integrationId).executeTakeFirst();
   }
   const rootKeys = instance.keys ?? null;
   const integrationNamespace = rootKeys?.[pluginId] ?? null;
@@ -948,8 +1536,8 @@ var listTenants = async (ctx) => {
     return { tenants: ["default"] };
   }
   if (!internal.database) throw new Error("No database configured.");
-  const db = asDb2(internal.database);
-  const rows = await db.selectFrom("corsair_accounts").selectAll().execute();
+  const db2 = asDb2(internal.database);
+  const rows = await db2.selectFrom("corsair_accounts").selectAll().execute();
   const ids = /* @__PURE__ */ new Set(["default"]);
   for (const row of rows) {
     const tenantId = row.tenant_id;
@@ -973,13 +1561,13 @@ var createTenant = async (ctx) => {
     throw new Error("Multi-tenancy is not enabled for this Corsair instance.");
   }
   if (!internal.database) throw new Error("No database configured.");
-  const db = asDb2(internal.database);
+  const db2 = asDb2(internal.database);
   const now = /* @__PURE__ */ new Date();
   const authPluginIds = getAuthPluginIds(internal);
   if (authPluginIds.length === 0) {
     return { ok: true, created: false };
   }
-  const integrations = await db.selectFrom("corsair_integrations").selectAll().execute();
+  const integrations = await db2.selectFrom("corsair_integrations").selectAll().execute();
   const integrationByName = /* @__PURE__ */ new Map();
   for (const row of integrations) {
     const name = row.name;
@@ -995,10 +1583,10 @@ var createTenant = async (ctx) => {
       created_at: now,
       updated_at: now
     };
-    await db.insertInto("corsair_integrations").values(row).execute();
+    await db2.insertInto("corsair_integrations").values(row).execute();
     integrationByName.set(pluginId, row);
   }
-  const existingAccounts = await db.selectFrom("corsair_accounts").selectAll().where("tenant_id", "=", tenantId).execute();
+  const existingAccounts = await db2.selectFrom("corsair_accounts").selectAll().where("tenant_id", "=", tenantId).execute();
   const accountByIntegrationId = /* @__PURE__ */ new Map();
   for (const row of existingAccounts) {
     const integrationId = row.integration_id;
@@ -1016,7 +1604,7 @@ var createTenant = async (ctx) => {
     if (!existing) {
       createdAny = true;
       pluginsMissingDek.add(pluginId);
-      await db.insertInto("corsair_accounts").values({
+      await db2.insertInto("corsair_accounts").values({
         id: randomUUID2(),
         tenant_id: tenantId,
         integration_id: integrationId,
@@ -1066,7 +1654,15 @@ var routes = [
   { method: "GET", path: "/api/db/tables", handler: listDbTables },
   { method: "POST", path: "/api/db/rows", handler: listDbRows },
   { method: "POST", path: "/api/db/entities/query", handler: queryEntityData },
-  { method: "GET", path: "/api/db/permissions", handler: listPermissions }
+  { method: "GET", path: "/api/db/permissions", handler: listPermissions },
+  { method: "GET", path: "/api/chats", handler: listChatsHandler },
+  { method: "POST", path: "/api/chats", handler: createChatHandler },
+  {
+    method: "GET",
+    path: "/api/chats/messages",
+    handler: getChatMessagesHandler
+  },
+  { method: "POST", path: "/api/chat", handler: chatHandler }
 ];
 async function handleApi(req, res, baseCtx) {
   const url = new URL(req.url ?? "/", "http://localhost");