Merge pull request #456 from miurla/fix/model-config-loading

refactor: update getModels to use fetch

Author: miurla

app/page.tsx

@@ -2,8 +2,8 @@ import { Chat } from '@/components/chat'
 import { getModels } from '@/lib/config/models'
 import { generateId } from 'ai'
 
-export default function Page() {
+export default async function Page() {
   const id = generateId()
-  const models = getModels()
+  const models = await getModels()
   return <Chat id={id} models={models} />
 }

app/search/[id]/page.tsx

@@ -34,6 +34,6 @@ export default async function SearchPage(props: {
     notFound()
   }
 
-  const models = getModels()
+  const models = await getModels()
   return <Chat id={id} savedMessages={messages} models={models} />
 }

app/search/page.tsx

@@ -14,6 +14,6 @@ export default async function SearchPage(props: {
   }
 
   const id = generateId()
-  const models = getModels()
+  const models = await getModels()
   return <Chat id={id} query={q} models={models} />
 }

app/share/[id]/page.tsx

@@ -29,7 +29,7 @@ export default async function SharePage(props: {
     return notFound()
   }
 
-  const models = getModels()
+  const models = await getModels()
   return (
     <Chat
       id={chat.id}

lib/config/models.ts

@@ -1,6 +1,5 @@
 import { Model } from '@/lib/types/models'
-import { existsSync, readFileSync } from 'fs'
-import { join } from 'path'
+import { headers } from 'next/headers'
 
 export function validateModel(model: any): model is Model {
   return (
@@ -14,11 +13,16 @@ export function validateModel(model: any): model is Model {
   )
 }
 
-export function getModels(): Model[] {
-  const configPath = join(process.cwd(), 'public', 'config', 'models.json')
-  
+export async function getModels(): Promise<Model[]> {
   try {
-    const config = JSON.parse(readFileSync(configPath, 'utf-8'))
+    const headersList = await headers()
+    const baseUrl = new URL(headersList.get('x-url') || 'http://localhost:3000')
+    const modelUrl = new URL('/config/models.json', baseUrl.origin)
+
+    const response = await fetch(modelUrl, {
+      cache: 'no-store'
+    })
+    const config = await response.json()
     if (Array.isArray(config.models) && config.models.every(validateModel)) {
       return config.models
     }

middleware.ts

@@ -0,0 +1,14 @@
+import { NextResponse } from 'next/server'
+import type { NextRequest } from 'next/server'
+
+export function middleware(request: NextRequest) {
+  // Create a response
+  const response = NextResponse.next()
+
+  // Add request information to response headers
+  response.headers.set('x-url', request.url)
+  response.headers.set('x-host', request.headers.get('host') || '')
+  response.headers.set('x-protocol', request.nextUrl.protocol)
+
+  return response
+}