From a9871f3a8da2fe60e50f334f45e8a95754c4f2eb Mon Sep 17 00:00:00 2001
From: Yash Kumar <hello@yash.run>
Date: Mon, 21 Sep 2026 02:48:26 +0530
Subject: [PATCH] refactor(ai): DRY saveAiSettings and enforce strict NOT_FOUND
 validation

---
 packages/server/src/services/ai.ts | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

diff --git a/packages/server/src/services/ai.ts b/packages/server/src/services/ai.ts
index 6157484e2..6f6b56a04 100644
--- a/packages/server/src/services/ai.ts
+++ b/packages/server/src/services/ai.ts
@@ -138,15 +138,7 @@ export const saveAiSettings = async (organizationId: string, settings: any) => {
 	const aiId = settings.aiId;
 
 	if (aiId) {
-		const existing = await db.query.ai.findFirst({
-			where: eq(ai.aiId, aiId),
-		});
-		if (existing && existing.organizationId !== organizationId) {
-			throw new TRPCError({
-				code: "FORBIDDEN",
-				message: "You do not have permission to modify this AI configuration",
-			});
-		}
+		await getAiSettingById(aiId, organizationId);
 	}
 
 	if (settings.apiUrl) {