refactor: add many AI providers & improve prompt

packages/server/src/db/schema/admin.ts

@@ -9,111 +9,113 @@ import { registry } from "./registry";
 import { certificateType } from "./shared";
 import { sshKeys } from "./ssh-key";
 import { users } from "./user";
+import { ai } from "./ai";
 
 export const admins = pgTable("admin", {
-	adminId: text("adminId")
-		.notNull()
-		.primaryKey()
-		.$defaultFn(() => nanoid()),
-	serverIp: text("serverIp"),
-	certificateType: certificateType("certificateType").notNull().default("none"),
-	host: text("host"),
-	letsEncryptEmail: text("letsEncryptEmail"),
-	sshPrivateKey: text("sshPrivateKey"),
-	enableDockerCleanup: boolean("enableDockerCleanup").notNull().default(false),
-	enableLogRotation: boolean("enableLogRotation").notNull().default(false),
-	authId: text("authId")
-		.notNull()
-		.references(() => auth.id, { onDelete: "cascade" }),
-	createdAt: text("createdAt")
-		.notNull()
-		.$defaultFn(() => new Date().toISOString()),
-	stripeCustomerId: text("stripeCustomerId"),
-	stripeSubscriptionId: text("stripeSubscriptionId"),
-	serversQuantity: integer("serversQuantity").notNull().default(0),
+  adminId: text("adminId")
+    .notNull()
+    .primaryKey()
+    .$defaultFn(() => nanoid()),
+  serverIp: text("serverIp"),
+  certificateType: certificateType("certificateType").notNull().default("none"),
+  host: text("host"),
+  letsEncryptEmail: text("letsEncryptEmail"),
+  sshPrivateKey: text("sshPrivateKey"),
+  enableDockerCleanup: boolean("enableDockerCleanup").notNull().default(false),
+  enableLogRotation: boolean("enableLogRotation").notNull().default(false),
+  authId: text("authId")
+    .notNull()
+    .references(() => auth.id, { onDelete: "cascade" }),
+  createdAt: text("createdAt")
+    .notNull()
+    .$defaultFn(() => new Date().toISOString()),
+  stripeCustomerId: text("stripeCustomerId"),
+  stripeSubscriptionId: text("stripeSubscriptionId"),
+  serversQuantity: integer("serversQuantity").notNull().default(0),
 });
 
 export const adminsRelations = relations(admins, ({ one, many }) => ({
-	auth: one(auth, {
-		fields: [admins.authId],
-		references: [auth.id],
-	}),
-	users: many(users),
-	registry: many(registry),
-	sshKeys: many(sshKeys),
-	certificates: many(certificates),
+  auth: one(auth, {
+    fields: [admins.authId],
+    references: [auth.id],
+  }),
+  users: many(users),
+  registry: many(registry),
+  sshKeys: many(sshKeys),
+  certificates: many(certificates),
+  ai: many(ai),
 }));
 
 const createSchema = createInsertSchema(admins, {
-	adminId: z.string(),
-	enableDockerCleanup: z.boolean().optional(),
-	sshPrivateKey: z.string().optional(),
-	certificateType: z.enum(["letsencrypt", "none"]).default("none"),
-	serverIp: z.string().optional(),
-	letsEncryptEmail: z.string().optional(),
+  adminId: z.string(),
+  enableDockerCleanup: z.boolean().optional(),
+  sshPrivateKey: z.string().optional(),
+  certificateType: z.enum(["letsencrypt", "none"]).default("none"),
+  serverIp: z.string().optional(),
+  letsEncryptEmail: z.string().optional(),
 });
 
 export const apiUpdateAdmin = createSchema.partial();
 
 export const apiSaveSSHKey = createSchema
-	.pick({
-		sshPrivateKey: true,
-	})
-	.required();
+  .pick({
+    sshPrivateKey: true,
+  })
+  .required();
 
 export const apiAssignDomain = createSchema
-	.pick({
-		host: true,
-		certificateType: true,
-		letsEncryptEmail: true,
-	})
-	.required()
-	.partial({
-		letsEncryptEmail: true,
-	});
+  .pick({
+    host: true,
+    certificateType: true,
+    letsEncryptEmail: true,
+  })
+  .required()
+  .partial({
+    letsEncryptEmail: true,
+  });
 
 export const apiUpdateDockerCleanup = createSchema
-	.pick({
-		enableDockerCleanup: true,
-	})
-	.required()
-	.extend({
-		serverId: z.string().optional(),
-	});
+  .pick({
+    enableDockerCleanup: true,
+  })
+  .required()
+  .extend({
+    serverId: z.string().optional(),
+  });
 
 export const apiTraefikConfig = z.object({
-	traefikConfig: z.string().min(1),
+  traefikConfig: z.string().min(1),
 });
 
 export const apiModifyTraefikConfig = z.object({
-	path: z.string().min(1),
-	traefikConfig: z.string().min(1),
-	serverId: z.string().optional(),
+  path: z.string().min(1),
+  traefikConfig: z.string().min(1),
+  serverId: z.string().optional(),
 });
 export const apiReadTraefikConfig = z.object({
-	path: z.string().min(1),
-	serverId: z.string().optional(),
+  path: z.string().min(1),
+  serverId: z.string().optional(),
 });
 
 export const apiEnableDashboard = z.object({
-	enableDashboard: z.boolean().optional(),
-	serverId: z.string().optional(),
+  enableDashboard: z.boolean().optional(),
+  serverId: z.string().optional(),
 });
 
 export const apiServerSchema = z
-	.object({
-		serverId: z.string().optional(),
-	})
-	.optional();
+  .object({
+    serverId: z.string().optional(),
+  })
+  .optional();
 
 export const apiReadStatsLogs = z.object({
-	page: z
-		.object({
-			pageIndex: z.number(),
-			pageSize: z.number(),
-		})
-		.optional(),
-	status: z.string().array().optional(),
-	search: z.string().optional(),
-	sort: z.object({ id: z.string(), desc: z.boolean() }).optional(),
+  page: z
+    .object({
+      pageIndex: z.number(),
+      pageSize: z.number(),
+    })
+    .optional(),
+  status: z.string().array().optional(),
+  search: z.string().optional(),
+  sort: z.object({ id: z.string(), desc: z.boolean() }).optional(),
 });

packages/server/src/db/schema/ai.ts

@@ -1,37 +1,66 @@
 import { boolean, pgTable, text } from "drizzle-orm/pg-core";
 import { createInsertSchema } from "drizzle-zod";
 import { z } from "zod";
+import { admins } from "./admin";
+import { relations } from "drizzle-orm";
+import { nanoid } from "nanoid";
 
 export const ai = pgTable("ai", {
-	authId: text("authId").notNull().primaryKey(),
-	apiUrl: text("apiUrl").notNull(),
-	apiKey: text("apiKey").notNull(),
-	model: text("model").notNull(),
-	isEnabled: boolean("isEnabled").notNull().default(true),
+  aiId: text("aiId")
+    .notNull()
+    .primaryKey()
+    .$defaultFn(() => nanoid()),
+  name: text("name").notNull(),
+  apiUrl: text("apiUrl").notNull(),
+  apiKey: text("apiKey").notNull(),
+  model: text("model").notNull(),
+  isEnabled: boolean("isEnabled").notNull().default(true),
+  adminId: text("adminId")
+    .notNull()
+    .references(() => admins.adminId, { onDelete: "cascade" }), // Admin ID who created the AI settings
+  createdAt: text("createdAt")
+    .notNull()
+    .$defaultFn(() => new Date().toISOString()),
 });
 
+export const aiRelations = relations(ai, ({ one }) => ({
+  admin: one(admins, {
+    fields: [ai.adminId],
+    references: [admins.adminId],
+  }),
+}));
+
 const createSchema = createInsertSchema(ai, {
-	apiUrl: z.string().url({ message: "Please enter a valid URL" }),
-	apiKey: z.string().min(1, { message: "API Key is required" }),
-	model: z.string().min(1, { message: "Model is required" }),
-	isEnabled: z.boolean().optional(),
+  name: z.string().min(1, { message: "Name is required" }),
+  apiUrl: z.string().url({ message: "Please enter a valid URL" }),
+  apiKey: z.string().min(1, { message: "API Key is required" }),
+  model: z.string().min(1, { message: "Model is required" }),
+  isEnabled: z.boolean().optional(),
 });
 
-export const apiAiSettingsSchema = createSchema
-	.pick({
-		apiUrl: true,
-		apiKey: true,
-		model: true,
-		isEnabled: true,
-	})
-	.required();
+export const apiCreateAi = createSchema
+  .pick({
+    name: true,
+    apiUrl: true,
+    apiKey: true,
+    model: true,
+    isEnabled: true,
+  })
+  .required();
+
+export const apiUpdateAi = createSchema
+  .partial()
+  .extend({
+    aiId: z.string().min(1),
+  })
+  .omit({ adminId: true });
 
 export const deploySuggestionSchema = z.object({
-	projectId: z.string().min(1),
-	id: z.string().min(1),
-	dockerCompose: z.string().min(1),
-	envVariables: z.string(),
-	serverId: z.string().optional(),
-	name: z.string().min(1),
-	description: z.string(),
+  projectId: z.string().min(1),
+  id: z.string().min(1),
+  dockerCompose: z.string().min(1),
+  envVariables: z.string(),
+  serverId: z.string().optional(),
+  name: z.string().min(1),
+  description: z.string(),
 });