Simplify table quickstart API endpoint

Remove unnecessary abstraction layers and dead code:
- Combine handler and handlePost into single function
- Remove shouldStream function and non-streaming path
- Always stream responses (only path client uses)
- Clean up imports in TableEditor

apps/studio/components/interfaces/TableGridEditor/SidePanelEditor/TableEditor/TableEditor.tsx

@@ -1,9 +1,10 @@
 import type { PostgresTable } from '@supabase/postgres-meta'
 import dayjs from 'dayjs'
 import { isEmpty, isUndefined, noop } from 'lodash'
-import { useCallback, useEffect, useMemo, useState } from 'react'
+import { useEffect, useMemo, useState } from 'react'
 import { toast } from 'sonner'
 
+import { LOCAL_STORAGE_KEYS } from 'common'
 import { DocsButton } from 'components/ui/DocsButton'
 import { useDatabasePublicationsQuery } from 'data/database-publications/database-publications-query'
 import {
@@ -19,13 +20,14 @@ import { useEnumeratedTypesQuery } from 'data/enumerated-types/enumerated-types-
 import { useSendEventMutation } from 'data/telemetry/send-event-mutation'
 import { useCustomContent } from 'hooks/custom-content/useCustomContent'
 import { useIsFeatureEnabled } from 'hooks/misc/useIsFeatureEnabled'
+import { useLocalStorage } from 'hooks/misc/useLocalStorage'
 import { useQuerySchemaState } from 'hooks/misc/useSchemaQueryState'
 import { useSelectedOrganizationQuery } from 'hooks/misc/useSelectedOrganization'
 import { useSelectedProjectQuery } from 'hooks/misc/useSelectedProject'
+import { usePHFlag } from 'hooks/ui/useFlag'
 import { useUrlState } from 'hooks/ui/useUrlState'
 import { useProtectedSchemas } from 'hooks/useProtectedSchemas'
 import { DOCS_URL } from 'lib/constants'
-import { usePHFlag } from 'hooks/ui/useFlag'
 import { useTableEditorStateSnapshot } from 'state/table-editor'
 import { Badge, Checkbox, Input, SidePanel } from 'ui'
 import { Admonition } from 'ui-patterns'
@@ -49,8 +51,6 @@ import {
 } from './TableEditor.utils'
 import { TableTemplateSelector } from './TableQuickstart/TableTemplateSelector'
 import { QuickstartVariant } from './TableQuickstart/types'
-import { LOCAL_STORAGE_KEYS } from 'common'
-import { useLocalStorage } from 'hooks/misc/useLocalStorage'
 
 const NEW_PROJECT_THRESHOLD_DAYS = 7
 

apps/studio/pages/api/ai/table-quickstart/generate-schemas.ts

@@ -35,23 +35,10 @@ const ResponseSchema = z.object({
 type SchemaStreamResult = ReturnType<typeof streamObject<typeof ResponseSchema>>
 
 const handler = async (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` } })
+  if (req.method !== 'POST') {
+    return res.status(405).json({ error: { message: `Method ${req.method} Not Allowed` } })
   }
-}
 
-const wrapper = (req: NextApiRequest, res: NextApiResponse) =>
-  apiWrapper(req, res, handler, { withAuth: true })
-
-export default wrapper
-
-const handlePost = async (req: NextApiRequest, res: NextApiResponse) => {
   const {
     model,
     error: modelError,
@@ -70,7 +57,7 @@ const handlePost = async (req: NextApiRequest, res: NextApiResponse) => {
   const { prompt } = req.body
 
   if (!prompt) {
-    return res.status(400).json({ error: 'Please provide a description of your application.' })
+    return res.status(400).json({ error: 'Please provide a description of your app.' })
   }
 
   if (typeof prompt !== 'string' || prompt.length > LIMITS.MAX_PROMPT_LENGTH) {
@@ -80,8 +67,6 @@ const handlePost = async (req: NextApiRequest, res: NextApiResponse) => {
   }
 
   try {
-    const wantsStream = shouldStream(req)
-
     const abortController = new AbortController()
     req.on('close', () => abortController.abort())
     req.on('aborted', () => abortController.abort())
@@ -116,15 +101,7 @@ const handlePost = async (req: NextApiRequest, res: NextApiResponse) => {
         prompt: userPrompt,
       })
 
-    if (wantsStream) {
-      await streamSchemaResponse(res, generateStream)
-      return
-    }
-
-    const result = generateStream()
-    const object = await result.object
-
-    return res.status(200).json(object)
+    await streamSchemaResponse(res, generateStream)
   } catch (error) {
     if (error instanceof Error) {
       if (error.message.includes('context_length') || error.message.includes('too long')) {
@@ -140,11 +117,10 @@ const handlePost = async (req: NextApiRequest, res: NextApiResponse) => {
   }
 }
 
-const shouldStream = (req: NextApiRequest) => {
-  if (req.headers['accept']?.includes('text/event-stream')) return true
-  if (req.headers['x-ai-stream'] === 'true') return true
-  return false
-}
+const wrapper = (req: NextApiRequest, res: NextApiResponse) =>
+  apiWrapper(req, res, handler, { withAuth: true })
+
+export default wrapper
 
 const streamSchemaResponse = async (
   res: NextApiResponse,