billing.sql.ts 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. import { bigint, boolean, integer, pgTable, varchar } from "drizzle-orm/pg-core"
  2. import { timestamps, workspaceColumns } from "../drizzle/types"
  3. import { workspaceIndexes } from "./workspace.sql"
  4. export const BillingTable = pgTable(
  5. "billing",
  6. {
  7. ...workspaceColumns,
  8. ...timestamps,
  9. customerID: varchar("customer_id", { length: 255 }),
  10. paymentMethodID: varchar("payment_method_id", { length: 255 }),
  11. paymentMethodLast4: varchar("payment_method_last4", { length: 4 }),
  12. balance: bigint("balance", { mode: "number" }).notNull(),
  13. reload: boolean("reload"),
  14. },
  15. (table) => [...workspaceIndexes(table)],
  16. )
  17. export const PaymentTable = pgTable(
  18. "payment",
  19. {
  20. ...workspaceColumns,
  21. ...timestamps,
  22. customerID: varchar("customer_id", { length: 255 }),
  23. paymentID: varchar("payment_id", { length: 255 }),
  24. amount: bigint("amount", { mode: "number" }).notNull(),
  25. },
  26. (table) => [...workspaceIndexes(table)],
  27. )
  28. export const UsageTable = pgTable(
  29. "usage",
  30. {
  31. ...workspaceColumns,
  32. ...timestamps,
  33. model: varchar("model", { length: 255 }).notNull(),
  34. inputTokens: integer("input_tokens").notNull(),
  35. outputTokens: integer("output_tokens").notNull(),
  36. reasoningTokens: integer("reasoning_tokens"),
  37. cacheReadTokens: integer("cache_read_tokens"),
  38. cacheWriteTokens: integer("cache_write_tokens"),
  39. cost: bigint("cost", { mode: "number" }).notNull(),
  40. },
  41. (table) => [...workspaceIndexes(table)],
  42. )