Feat: Memory add and search, reasoning, fixed AI thinking UI

Author: Omm2005

app/(chat)/api/chat/route.ts

@@ -44,6 +44,10 @@ import { getUsage } from 'tokenlens/helpers';
 import type { ModelCatalog } from 'tokenlens/core';
 import type { AppUsage } from '@/lib/usage';
 import getWeather from '@/lib/ai/tools/get-weather';
+import {
+  addMemoryTool,
+  searchMemoriesTool,
+} from '@/lib/ai/tools/supermemory-tools';
 
 export const maxDuration = 60;
 
@@ -184,21 +188,47 @@ export async function POST(request: Request) {
           system: systemPrompt({ selectedChatModel, requestHints }),
           messages: convertToModelMessages(uiMessages),
           stopWhen: stepCountIs(5),
-          experimental_activeTools:
-            selectedChatModel === 'chat-model-reasoning'
-              ? []
-              : [
-                  'getWeather',
-                  'createDocument',
-                  'updateDocument',
-                  'requestSuggestions',
-                ],
+          experimental_activeTools: (() => {
+            if (selectedChatModel === 'chat-model-reasoning') {
+              const base: ('addMemory' | 'searchMemories')[] = [];
+              if (userType !== 'guest')
+                base.push('addMemory', 'searchMemories');
+              return base;
+            }
+            const base: (
+              | 'getWeather'
+              | 'createDocument'
+              | 'updateDocument'
+              | 'requestSuggestions'
+              | 'addMemory'
+              | 'searchMemories'
+            )[] = [
+              'getWeather',
+              'createDocument',
+              'updateDocument',
+              'requestSuggestions',
+            ];
+            // Only enable memory tools for non-guest users
+            if (userType !== 'guest') {
+              base.push('addMemory', 'searchMemories');
+            }
+            return base;
+          })(),
           experimental_transform: smoothStream({ chunking: 'word' }),
+          //@ts-ignore
           tools: {
             getWeather,
             createDocument: createDocument({ session, dataStream }),
             updateDocument: updateDocument({ session, dataStream }),
             requestSuggestions: requestSuggestions({ session, dataStream }),
+
+            addMemory: addMemoryTool(process.env.SUPERMEMORY_API_KEY || '', {
+              projectId: session.user.id,
+            }),
+            searchMemories: searchMemoriesTool(
+              process.env.SUPERMEMORY_API_KEY || '',
+              { projectId: session.user.id },
+            ),
           },
           experimental_telemetry: {
             isEnabled: isProductionEnvironment,
@@ -239,7 +269,6 @@ export async function POST(request: Request) {
         });
 
         result.consumeStream();
-        // console.log(result);
 
         dataStream.merge(
           result.toUIMessageStream({

app/(chat)/api/memory/route.ts

@@ -0,0 +1,30 @@
+import { auth } from '@/app/(auth)/auth';
+import { ChatSDKError } from '@/lib/errors';
+import Supermemory from 'supermemory';
+
+export async function POST(request: Request) {
+  try {
+    const session = await auth();
+    if (!session?.user) return new ChatSDKError('unauthorized:api').toResponse();
+    if (session.user.type === 'guest')
+      return new ChatSDKError('forbidden:auth').toResponse();
+
+    const { memory } = (await request.json()) as { memory?: string };
+    if (!memory || typeof memory !== 'string' || memory.trim().length === 0)
+      return new ChatSDKError('bad_request:api').toResponse();
+
+    if (!process.env.SUPERMEMORY_API_KEY)
+      return new ChatSDKError('offline:api', 'Missing Supermemory API key').toResponse();
+
+    const client = new Supermemory({ apiKey: process.env.SUPERMEMORY_API_KEY });
+    const containerTags = [`sm_project_${session.user.id}`];
+
+    const result = await client.memories.add({ content: memory, containerTags });
+
+    return Response.json({ success: true, memory: result }, { status: 200 });
+  } catch (err: any) {
+    console.error('Memory add failed', err);
+    return new ChatSDKError('offline:api', err?.message || 'Memory add failed').toResponse();
+  }
+}
+