summaryrefslogtreecommitdiffhomepage
path: root/packages/console/core/src
diff options
context:
space:
mode:
Diffstat (limited to 'packages/console/core/src')
-rw-r--r--packages/console/core/src/model.ts6
-rw-r--r--packages/console/core/src/schema/ip.sql.ts12
2 files changed, 18 insertions, 0 deletions
diff --git a/packages/console/core/src/model.ts b/packages/console/core/src/model.ts
index fd6cd095e..8cc181b7c 100644
--- a/packages/console/core/src/model.ts
+++ b/packages/console/core/src/model.ts
@@ -24,6 +24,12 @@ export namespace ZenData {
cost: ModelCostSchema,
cost200K: ModelCostSchema.optional(),
allowAnonymous: z.boolean().optional(),
+ trial: z
+ .object({
+ limit: z.number(),
+ provider: z.string(),
+ })
+ .optional(),
rateLimit: z.number().optional(),
fallbackProvider: z.string().optional(),
providers: z.array(
diff --git a/packages/console/core/src/schema/ip.sql.ts b/packages/console/core/src/schema/ip.sql.ts
new file mode 100644
index 000000000..be5fb7fa2
--- /dev/null
+++ b/packages/console/core/src/schema/ip.sql.ts
@@ -0,0 +1,12 @@
+import { mysqlTable, int, primaryKey, varchar } from "drizzle-orm/mysql-core"
+import { timestamps } from "../drizzle/types"
+
+export const IpTable = mysqlTable(
+ "ip",
+ {
+ ip: varchar("ip", { length: 45 }).notNull(),
+ ...timestamps,
+ usage: int("usage"),
+ },
+ (table) => [primaryKey({ columns: [table.ip] })],
+)