From 68cefc8e09a3f7544ce3da7343ccd4c3eebfd973 Mon Sep 17 00:00:00 2001
From: Joshen Lim <joshenlimek@gmail.com>
Date: Thu, 18 Sep 2025 21:53:38 +0800
Subject: [PATCH] asdasd

---
 apps/studio/lib/ai/tools/index.ts        | 5 ++++-
 apps/studio/lib/ai/tools/mcp-tools.ts    | 2 +-
 apps/studio/lib/ai/tools/schema-tools.ts | 8 +++++++-
 3 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/apps/studio/lib/ai/tools/index.ts b/apps/studio/lib/ai/tools/index.ts
index e657f1e5515..4a73a5c3818 100644
--- a/apps/studio/lib/ai/tools/index.ts
+++ b/apps/studio/lib/ai/tools/index.ts
@@ -1,10 +1,13 @@
 import { ToolSet } from 'ai'
+
 import { IS_PLATFORM } from 'common'
 import { AiOptInLevel } from '../org-ai-details'
 import { filterToolsByOptInLevel } from '../tool-filter'
-import { getFallbackTools } from './fallback-tools'
 import { getMcpTools } from './mcp-tools'
 import { getRenderingTools } from './rendering-tools'
+
+import { getFallbackTools } from './fallback-tools'
+
 import { getSchemaTools } from './schema-tools'
 
 export const getTools = async ({
diff --git a/apps/studio/lib/ai/tools/mcp-tools.ts b/apps/studio/lib/ai/tools/mcp-tools.ts
index 65c24779e3a..abbe1600cab 100644
--- a/apps/studio/lib/ai/tools/mcp-tools.ts
+++ b/apps/studio/lib/ai/tools/mcp-tools.ts
@@ -1,4 +1,4 @@
-import { AiOptInLevel } from 'hooks/misc/useOrgOptedIntoAi'
+import { AiOptInLevel } from '../org-ai-details'
 import { createSupabaseMCPClient } from '../supabase-mcp'
 import { filterToolsByOptInLevel, toolSetValidationSchema } from '../tool-filter'
 
diff --git a/apps/studio/lib/ai/tools/schema-tools.ts b/apps/studio/lib/ai/tools/schema-tools.ts
index 267dcc73b8d..f7ec642c21b 100644
--- a/apps/studio/lib/ai/tools/schema-tools.ts
+++ b/apps/studio/lib/ai/tools/schema-tools.ts
@@ -1,7 +1,13 @@
+// @ts-nocheck
+
 import { tool } from 'ai'
-import { getDatabasePolicies } from 'data/database-policies/database-policies-query'
+// import { getDatabasePolicies } from 'data/database-policies/database-policies-query'
 import { z } from 'zod'
 
+const getDatabasePolicies = async (val: any, val2: any, val3: any) => {
+  return { data: [] }
+}
+
 export const getSchemaTools = ({
   projectRef,
   connectionString,