fix: avoid too long chat title for chat-engine test API

pingcap · Jun 4, 2024 · 6d4ae17 · 6d4ae17
1 parent 78ee0ae
commit 6d4ae17
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/app/api/v1/chat_engines/[id]/test/route.ts b/src/app/api/v1/chat_engines/[id]/test/route.ts
@@ -49,14 +49,15 @@ export const POST = defineHandler({
   await eachOfLimit(dataset.items, concurrency, async (item: any) => {
     const userMessage = item.input;
     const userId = 'tester';
+    const title = item.input.length > 255 ? item.input.substring(0, 255) : item.input;
     const chat = await createChat({
       engine: engine.engine,
       engine_id: engine.id,
       engine_name: engine.name,
       engine_options: JSON.stringify(engine.engine_options),
       created_at: new Date(),
       created_by: userId,
-      title: userMessage,
+      title: title,
     });
 
     console.log(`[Testing] Testing with question (chat engine: ${id}): ${userMessage}`)
@@ -72,4 +73,4 @@ export const POST = defineHandler({
 
 export const dynamic = 'force-dynamic';
 
-export const maxDuration = 150;
+export const maxDuration = 300;