chore: Add max duration for the response

getzep · Jan 30, 2025 · 9cce91b · 9cce91b
1 parent 809bab2
commit 9cce91b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/vercel_ai/scripts/seed_zep.ts b/examples/vercel_ai/scripts/seed_zep.ts
@@ -151,7 +151,7 @@ async function seedZepData() {
     };
 
     try {
-        console.log("🌱 Seeding Zep with initial data...");
+        console.log("🤖 Seeding Zep with initial data...");
 
         // Users in Zep may have one or more chat sessions. These are threads of messages between the user and an agent.
         // TIP: Include the user's full name and email address when creating a user.

diff --git a/examples/vercel_ai/src/app/api/chat/route.ts b/examples/vercel_ai/src/app/api/chat/route.ts
@@ -3,6 +3,9 @@ import { streamText, tool } from "ai";
 import { z } from "zod";
 import { ZepClient, Zep } from "@getzep/zep-cloud";
 
+// Allow streaming responses up to 30 seconds
+export const maxDuration = 30;
+
 export async function POST(req: Request) {
     const { messages, sessionId, userId } = await req.json();
     const zep = new ZepClient({ apiKey: process.env.ZEP_API_KEY });
@@ -95,7 +98,7 @@ export async function POST(req: Request) {
             await zep.memory.add(sessionId, {
                 messages: [
                     {
-                        role: "assistant",
+                        role: "SupportBot",
                         roleType: "assistant",
                         content: result.text,
                     },