chore(deps): update ai sdk to latest version and related deps

Author: bxxf

app/api/chat/route.ts

@@ -5,7 +5,7 @@ import { toPrompt } from '@/lib/prompt'
 import ratelimit from '@/lib/ratelimit'
 import { fragmentSchema as schema } from '@/lib/schema'
 import { Templates } from '@/lib/templates'
-import { streamObject, LanguageModel, CoreMessage } from 'ai'
+import { streamText, Output, LanguageModel, type ModelMessage } from 'ai'
 
 export const maxDuration = 300
 
@@ -25,7 +25,7 @@ export async function POST(req: Request) {
     model,
     config,
   }: {
-    messages: CoreMessage[]
+    messages: ModelMessage[]
     userID: string | undefined
     teamID: string | undefined
     template: Templates
@@ -54,18 +54,29 @@ export async function POST(req: Request) {
   const { model: modelNameString, apiKey: modelApiKey, ...modelParams } = config
   const modelClient = getModelClient(model, config)
 
-  try {
-    const stream = await streamObject({
-      model: modelClient as LanguageModel,
-      schema,
-      system: toPrompt(template),
-      messages,
-      maxRetries: 0, // do not retry on errors
-      ...modelParams,
-    })
+  let apiError: any = null
+
+  const result = streamText({
+    model: modelClient as LanguageModel,
+    output: Output.object({ schema }),
+    system: toPrompt(template),
+    messages,
+    maxRetries: 0,
+    onError: ({ error }) => {
+      apiError = error
+    },
+    ...modelParams,
+  })
 
-    return stream.toTextStreamResponse()
-  } catch (error: any) {
-    return handleAPIError(error, { hasOwnApiKey: !!config.apiKey })
+  // Check if API call succeeds by awaiting first chunk
+  try {
+    await result.response
+  } catch {
+    // apiError is set by onError callback with the actual API error
+    if (apiError) {
+      return handleAPIError(apiError, { hasOwnApiKey: !!config.apiKey })
+    }
   }
+
+  return result.toTextStreamResponse()
 }

app/api/morph-chat/route.ts

@@ -3,8 +3,8 @@ import { Duration } from '@/lib/duration'
 import { getModelClient, LLMModel, LLMModelConfig } from '@/lib/models'
 import { applyPatch } from '@/lib/morph'
 import ratelimit from '@/lib/ratelimit'
-import { FragmentSchema, morphEditSchema, MorphEditSchema } from '@/lib/schema'
-import { generateObject, LanguageModel, CoreMessage } from 'ai'
+import { FragmentSchema, morphEditSchema } from '@/lib/schema'
+import { generateText, Output, LanguageModel, type ModelMessage } from 'ai'
 
 export const maxDuration = 300
 
@@ -24,7 +24,7 @@ export async function POST(req: Request) {
     config,
     currentFragment,
   }: {
-    messages: CoreMessage[]
+    messages: ModelMessage[]
     model: LLMModel
     config: LLMModelConfig
     currentFragment: FragmentSchema
@@ -64,16 +64,16 @@ ${currentFragment.code}
 
 `
 
-    const result = await generateObject({
+    const result = await generateText({
       model: modelClient as LanguageModel,
       system: contextualSystemPrompt,
       messages,
-      schema: morphEditSchema,
+      output: Output.object({ schema: morphEditSchema }),
       maxRetries: 0,
       ...modelParams,
     })
 
-    const editInstructions = result.object
+    const editInstructions = result.output!
 
     // Apply edits using Morph
     const morphResult = await applyPatch({
@@ -105,7 +105,7 @@ ${currentFragment.code}
         'Content-Type': 'text/plain; charset=utf-8',
       },
     })
-  } catch (error: any) {
+  } catch (error: unknown) {
     return handleAPIError(error, { hasOwnApiKey: !!config.apiKey })
   }
 }

app/page.tsx

@@ -17,7 +17,7 @@ import { supabase } from '@/lib/supabase'
 import templates from '@/lib/templates'
 import { ExecutionResult } from '@/lib/types'
 import { DeepPartial } from 'ai'
-import { experimental_useObject as useObject } from 'ai/react'
+import { experimental_useObject as useObject } from '@ai-sdk/react'
 import { usePostHog } from 'posthog-js/react'
 import { SetStateAction, useEffect, useState } from 'react'
 import { useLocalStorage } from 'usehooks-ts'

app/providers.tsx

@@ -1,7 +1,6 @@
 'use client'
 
-import { ThemeProvider as NextThemesProvider } from 'next-themes'
-import { type ThemeProviderProps } from 'next-themes/dist/types'
+import { ThemeProvider as NextThemesProvider, type ThemeProviderProps } from 'next-themes'
 import posthog from 'posthog-js'
 import { PostHogProvider as PostHogProviderJS } from 'posthog-js/react'
 

components/auth.tsx

@@ -559,7 +559,7 @@ function Auth({
   magicLink = false,
   onSignUpValidate,
   metadata,
-}: AuthProps): JSX.Element | null {
+}: AuthProps): React.ReactNode {
   const [authView, setAuthView] = useState<ViewType>(view)
   const {
     loading,

components/chat-settings.tsx

@@ -124,15 +124,15 @@ export function ChatSettings({
             </span>
             <Input
               type="number"
-              defaultValue={languageModel.maxTokens}
+              defaultValue={languageModel.maxOutputTokens}
               min={50}
               max={10000}
               step={1}
               className="h-6 rounded-sm w-[84px] text-xs text-center tabular-nums"
               placeholder="Auto"
               onChange={(e) =>
                 onLanguageModelChange({
-                  maxTokens: parseFloat(e.target.value) || undefined,
+                  maxOutputTokens: parseFloat(e.target.value) || undefined,
                 })
               }
             />

lib/api-errors.ts

@@ -7,30 +7,41 @@ export interface APIError {
   message: string
 }
 
+/**
+ * Gets the status code from an error
+ */
+function getStatusCode(error: any): number | undefined {
+  return error?.statusCode ?? error?.status
+}
+
 /**
  * Checks if an error is a rate limit error
  */
 export function isRateLimitError(error: any): boolean {
+  const status = getStatusCode(error)
   return (
     error &&
-    (error.statusCode === 429 ||
-      error.message.toLowerCase().includes('limit') ||
-      error.message.toLowerCase().includes('billing'))
+    (status === 429 ||
+      error.message?.toLowerCase().includes('limit') ||
+      error.message?.toLowerCase().includes('billing'))
   )
 }
 
 /**
  * Checks if an error is an overloaded/unavailable error
  */
 export function isOverloadedError(error: any): boolean {
-  return error && (error.statusCode === 529 || error.statusCode === 503)
+  const status = getStatusCode(error)
+  return error && (status === 529 || status === 503)
 }
 
 /**
  * Checks if an error is an access denied/unauthorized error
  */
 export function isAccessDeniedError(error: any): boolean {
-  return error && (error.statusCode === 403 || error.statusCode === 401)
+  const status = getStatusCode(error)
+  const message = error?.message?.toLowerCase() || ''
+  return error && (status === 403 || status === 401 || message.includes('api key') || message.includes('x-api-key') || message.includes('unauthorized'))
 }
 
 /**
@@ -40,8 +51,7 @@ export function handleAPIError(
   error: any,
   context?: { hasOwnApiKey?: boolean },
 ): Response {
-  // Log the error for debugging
-  console.error('API Error:', error)
+  console.error('API Error:', error?.message || error)
 
   if (isRateLimitError(error)) {
     const message = context?.hasOwnApiKey

lib/models.ts

@@ -22,7 +22,7 @@ export type LLMModelConfig = {
   topK?: number
   frequencyPenalty?: number
   presencePenalty?: number
-  maxTokens?: number
+  maxOutputTokens?: number
 }
 
 export function getModelClient(model: LLMModel, config: LLMModelConfig) {

lib/morph.ts

@@ -1,4 +1,4 @@
-import { createOpenAI } from '@ai-sdk/openai'
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible'
 import { generateText, LanguageModel } from 'ai'
 
 export async function applyPatch({
@@ -23,17 +23,29 @@ export async function applyPatch({
     )
   }
 
-  const openai = createOpenAI({
+  const morph = createOpenAICompatible({
+    name: 'morph',
     apiKey: morphApiKey,
     baseURL: 'https://api.morphllm.com/v1',
   })
 
   try {
+    console.log('[Morph] Starting applyPatch', {
+      targetFile,
+      instructionsLength: instructions.length,
+      initialCodeLength: initialCode.length,
+      codeEditLength: codeEdit.length,
+      hasApiKey: !!morphApiKey
+    })
+
     const { text: mergedCode } = await generateText({
-      model: openai('morph-v3-large') as LanguageModel,
+      model: morph('morph-v3-large') as LanguageModel,
       prompt: `<instruction>${instructions}</instruction>\n<code>${initialCode}</code>\n<update>${codeEdit}</update>`,
+      maxRetries: 0,
     })
 
+    console.log('[Morph] Success', { mergedCodeLength: mergedCode?.length })
+
     if (!mergedCode) {
       throw new Error('Morph Apply returned empty content')
     }
@@ -43,6 +55,13 @@ export async function applyPatch({
       code: mergedCode,
     }
   } catch (error: any) {
+    console.error('[Morph] Error', {
+      message: error.message,
+      status: error.status,
+      cause: error.cause,
+      name: error.name,
+      stack: error.stack?.substring(0, 500),
+    })
     if (error.message.includes('Invalid API key') || error.status === 401) {
       throw new Error('Invalid Morph API key. Please check your settings.')
     }