feat: temp test api endpoint for bedrock (supabase#36668)

* feat: temp test api endpoint for bedrock

* refactor: consolidate ai imports

Author: gregnr

apps/studio/middleware.ts

@@ -14,6 +14,7 @@ const HOSTED_SUPPORTED_API_URLS = [
   '/ai/sql/title',
   '/ai/sql/cron',
   '/ai/feedback/classify',
+  '/ai/temp/bedrock-test',
   '/get-ip-address',
   '/get-utc-time',
   '/edge-functions/test',

apps/studio/package.json

@@ -25,6 +25,7 @@
     "build:graphql-types:watch": "pnpm graphql-codegen --config scripts/codegen.ts --watch"
   },
   "dependencies": {
+    "@ai-sdk/amazon-bedrock": "^2.2.10",
     "@ai-sdk/openai": "^0.0.72",
     "@dagrejs/dagre": "^1.0.4",
     "@deno/eszip": "0.83.0",
@@ -60,9 +61,11 @@
     "@tanstack/react-query-devtools": "4.35.7",
     "@tanstack/react-table": "^8.21.3",
     "@uidotdev/usehooks": "^2.4.1",
+    "@vercel/functions": "^2.2.2",
     "@vitejs/plugin-react": "^4.3.4",
     "@zip.js/zip.js": "^2.7.29",
     "ai": "^3.4.33",
+    "ai-4": "npm:ai@4",
     "ai-commands": "workspace:*",
     "awesome-debounce-promise": "^2.1.0",
     "common": "workspace:*",

apps/studio/pages/api/ai/temp/bedrock-test.ts

@@ -0,0 +1,70 @@
+import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock'
+import { awsCredentialsProvider } from '@vercel/functions/oidc'
+import { CoreMessage, streamText } from 'ai-4'
+import apiWrapper from 'lib/api/apiWrapper'
+import { NextApiRequest, NextApiResponse } from 'next'
+import { z } from 'zod'
+
+const bedrock = createAmazonBedrock({
+  credentialProvider: awsCredentialsProvider({
+    roleArn: process.env.AWS_BEDROCK_ROLE_ARN!,
+  }),
+})
+
+const messageSchema = z.object({
+  role: z.enum(['user', 'assistant']),
+  content: z.string(),
+})
+
+const requestBodySchema = z.object({
+  messages: z.array(messageSchema),
+})
+
+async function handler(req: NextApiRequest, res: NextApiResponse) {
+  const { method } = req
+
+  switch (method) {
+    case 'POST':
+      return handlePost(req, res)
+    default:
+      res.setHeader('Allow', ['POST'])
+      res.status(405).json({ data: null, error: { message: `Method ${method} Not Allowed` } })
+  }
+}
+
+async function handlePost(req: NextApiRequest, res: NextApiResponse) {
+  const { data, error: parseError } = requestBodySchema.safeParse(req.body)
+
+  if (parseError) {
+    return res.status(400).json({
+      error: 'Invalid request body',
+      issues: parseError.issues,
+    })
+  }
+
+  const { messages: clientMessages } = data
+
+  const model = bedrock('us.anthropic.claude-sonnet-4-20250514-v1:0')
+  const messages: CoreMessage[] = [
+    {
+      role: 'system',
+      content: 'You are a helpful assistant.',
+    },
+    ...clientMessages,
+  ]
+
+  const result = streamText({
+    model,
+    messages,
+    onError: ({ error }) => {
+      console.error(error)
+    },
+  })
+
+  return result.pipeDataStreamToResponse(res)
+}
+
+const wrapper = (req: NextApiRequest, res: NextApiResponse) =>
+  apiWrapper(req, res, handler, { withAuth: false })
+
+export default wrapper