feat: Implement Calendar Notepad with AI integration and collaborator tagging

app/actions.tsx

@@ -103,23 +103,24 @@ async function submit(formData?: FormData, skip?: boolean) {
       try {
         const streamResult = await resolutionSearch(messages, timezone, drawnFeatures, location);
 
+        let partialObject: any;
         let fullSummary = '';
-        for await (const partialObject of streamResult.partialObjectStream) {
+        for await (partialObject of streamResult.partialObjectStream) {
           if (partialObject.summary) {
             fullSummary = partialObject.summary;
             summaryStream.update(fullSummary);
           }
         }
 
-        const analysisResult = await streamResult.object;
+        const analysisResult: any = await streamResult.object;
         summaryStream.done(analysisResult.summary || 'Analysis complete.');
 
         // Reconstruct standard GeoJSON from flattened schema if present
         let geoJson: FeatureCollection | null = null;
         if (analysisResult.geoJson && analysisResult.geoJson.features) {
           geoJson = {
             type: 'FeatureCollection',
-            features: analysisResult.geoJson.features.map(f => ({
+            features: analysisResult.geoJson.features.map( (f: any) => ({
               type: 'Feature',
               geometry: {
                 type: f.geometryType as any,
@@ -334,6 +335,9 @@ async function submit(formData?: FormData, skip?: boolean) {
     }
   }
 
+  const { getChatNotes } = await import('@/lib/actions/calendar')
+  const calendarNotes = await getChatNotes(aiState.get().chatId)
+
   let filteredImagesCount = 0
   let retainedImagesCount = 0
   const messages: CoreMessage[] = [...(aiState.get().messages as any[])]
@@ -525,14 +529,34 @@ async function submit(formData?: FormData, skip?: boolean) {
         messages,
         mapProvider,
         useSpecificAPI,
-        drawnFeatures
+        drawnFeatures,
+        calendarNotes
       )
       answer = fullResponse
       toolOutputs = toolResponses
       errorOccurred = hasError
 
       if (toolOutputs.length > 0) {
-        toolOutputs.map(output => {
+        await Promise.all(toolOutputs.map(async output => {
+          const result = output.result as any
+
+          if (output.toolName === 'calendarTool' && result.type === 'CALENDAR_ACTION') {
+             const { saveNote, getChatNotes } = await import('@/lib/actions/calendar')
+             if (result.action === 'create') {
+                const saved = await saveNote({
+                  ...result.note,
+                  chatId: aiState.get().chatId,
+                  userId: '', // set in saveNote
+                  userTags: null,
+                  mapFeatureId: null
+                })
+                output.result = { success: true, note: saved }
+             } else if (result.action === 'list') {
+                const notes = await getChatNotes(aiState.get().chatId)
+                output.result = { success: true, notes }
+             }
+          }
+
           aiState.update({
             ...aiState.get(),
             messages: [
@@ -546,7 +570,7 @@ async function submit(formData?: FormData, skip?: boolean) {
               }
             ]
           })
-        })
+        }))
       }
     }
 

components/calendar-toggle-context.tsx

@@ -22,9 +22,7 @@ export const CalendarToggleProvider = ({ children }: { children: ReactNode }) =>
   const [isCalendarOpen, setIsCalendarOpen] = useState(false)
 
   const toggleCalendar = () => {
-    startTransition(() => {
-      setIsCalendarOpen(prevState => !prevState)
-    })
+    setIsCalendarOpen(prevState => !prevState)
   }
 
   return (

components/map/mapbox-map.tsx

@@ -246,7 +246,7 @@ export const Mapbox: React.FC<{ position?: { latitude: number; longitude: number
         map.current.off('draw.create', updateMeasurementLabels)
         map.current.off('draw.delete', updateMeasurementLabels)
         map.current.off('draw.update', updateMeasurementLabels)
-        map.current.removeControl(drawRef.current)
+        map.current.removeControl(drawRef.current as any)
         drawRef.current = null
 
         // Clean up any existing labels
@@ -282,7 +282,7 @@ export const Mapbox: React.FC<{ position?: { latitude: number; longitude: number
     })
 
     // Add control to map
-    map.current.addControl(drawRef.current, 'top-right')
+    map.current.addControl(drawRef.current as any, 'top-right')
 
     // Add navigation control only on desktop
     if (window.innerWidth > 768) {
@@ -509,7 +509,7 @@ export const Mapbox: React.FC<{ position?: { latitude: number; longitude: number
             map.current.off('draw.create', updateMeasurementLabels)
             map.current.off('draw.delete', updateMeasurementLabels)
             map.current.off('draw.update', updateMeasurementLabels)
-            map.current.removeControl(drawRef.current)
+            map.current.removeControl(drawRef.current as any)
             drawRef.current = null
 
             // Clean up any existing labels

components/message.tsx

@@ -19,13 +19,14 @@ export function BotMessage({ content }: { content: StreamableValue<string> }) {
 
   return (
     <div className="overflow-x-auto">
+      <div className="prose-sm prose-neutral prose-a:text-accent-foreground/50">
       <MemoizedReactMarkdown
         rehypePlugins={[[rehypeExternalLinks, { target: '_blank' }], rehypeKatex]}
         remarkPlugins={[remarkGfm, remarkMath]}
-        className="prose-sm prose-neutral prose-a:text-accent-foreground/50"
       >
         {processedData}
       </MemoizedReactMarkdown>
+      </div>
     </div>
   )
 }

components/theme-provider.tsx

@@ -2,7 +2,7 @@
 
 import * as React from 'react'
 import { ThemeProvider as NextThemesProvider } from 'next-themes'
-import { type ThemeProviderProps } from 'next-themes/dist/types'
+import { type ThemeProviderProps } from 'next-themes'
 
 export function ThemeProvider({ children, ...props }: ThemeProviderProps) {
   return <NextThemesProvider {...props}>{children}</NextThemesProvider>

components/ui/markdown.tsx

@@ -4,6 +4,5 @@ import ReactMarkdown, { Options } from 'react-markdown'
 export const MemoizedReactMarkdown: FC<Options> = memo(
   ReactMarkdown,
   (prevProps, nextProps) =>
-    prevProps.children === nextProps.children &&
-    prevProps.className === nextProps.className
+    prevProps.children === nextProps.children
 )

lib/actions/calendar.ts

@@ -56,6 +56,33 @@ export async function getNotes(date: Date, chatId: string | null): Promise<Calen
   }
 }
 
+/**
+ * Retrieves all notes for a specific chat session.
+ * @param chatId - The ID of the chat session.
+ * @returns A promise that resolves to an array of notes.
+ */
+export async function getChatNotes(chatId: string): Promise<CalendarNote[]> {
+  const userId = await getCurrentUserIdOnServer()
+  if (!userId) {
+    console.error('getChatNotes: User not authenticated')
+    return []
+  }
+
+  try {
+    const notes = await db
+      .select()
+      .from(calendarNotes)
+      .where(and(eq(calendarNotes.userId, userId), eq(calendarNotes.chatId, chatId)))
+      .orderBy(desc(calendarNotes.date))
+      .execute()
+
+    return notes
+  } catch (error) {
+    console.error('Error fetching chat notes:', error)
+    return []
+  }
+}
+
 /**
  * Saves a new note or updates an existing one.
  * @param noteData - The note data to save.
@@ -84,9 +111,16 @@ export async function saveNote(noteData: NewCalendarNote | CalendarNote): Promis
     } else {
         // Create new note
         try {
+            // Parse @mentions from content
+            const mentions = noteData.content.match(/@(\w+)/g)?.map(m => m.substring(1)) || null;
+            const updatedNoteData = {
+                ...noteData,
+                userTags: mentions
+            };
+
             const [newNote] = await db
                 .insert(calendarNotes)
-                .values({ ...noteData, userId })
+                .values({ ...updatedNoteData, userId })
                 .returning();
 
             if (newNote && newNote.chatId) {

lib/actions/suggest.ts

@@ -28,7 +28,7 @@ export async function getSuggestions(
 
   ;(async () => {
     const result = await streamObject({
-      model: (await getModel()) as LanguageModel,
+      model: (await getModel()) as any,
       system: systemPrompt,
       messages: [{ role: 'user', content: query }],
       schema: relatedSchema

lib/agents/inquire.tsx

@@ -24,7 +24,7 @@ export async function inquire(
 
   let finalInquiry: PartialInquiry = {};
   const result = await streamObject({
-    model: (await getModel()) as LanguageModel,
+    model: (await getModel()) as any,
     system: `You are a helpful assistant that gathers clarifying information from the user. 
     Generate a structured inquiry with a clear question, multiple choice options, and optionally allow free-text input.
     Ensure the inquiry is concise and helps narrow down the user's intent.`,

lib/agents/query-suggestor.tsx

@@ -56,7 +56,7 @@ export async function querySuggestor(
 
   // OPTIMIZATION: Use a more concise system prompt to reduce token usage
   const result = await streamObject({
-    model: (await getModel()) as LanguageModel,
+    model: (await getModel()) as any,
     system: `Generate 3 follow-up queries that explore the subject matter deeper. Format as JSON with an "items" array containing objects with "query" fields. Keep queries concise and relevant.`,
     messages,
     schema: relatedSchema,

lib/agents/researcher.tsx

@@ -13,11 +13,12 @@ import { getTools } from './tools'
 import { getModel } from '../utils'
 import { MapProvider } from '@/lib/store/settings'
 import { DrawnFeature } from './resolution-search'
+import { CalendarNote } from '@/lib/types'
 
 // This magic tag lets us write raw multi-line strings with backticks, arrows, etc.
 const raw = String.raw
 
-const getDefaultSystemPrompt = (date: string, drawnFeatures?: DrawnFeature[]) => raw`
+const getDefaultSystemPrompt = (date: string, drawnFeatures?: DrawnFeature[], calendarNotes?: CalendarNote[]) => raw`
 As a comprehensive AI assistant, your primary directive is **Exploration Efficiency**. You must use the provided tools judiciously to gather information and formulate a response.
 
 Current date and time: ${date}.
@@ -26,6 +27,11 @@ ${drawnFeatures && drawnFeatures.length > 0 ? `The user has drawn the following
 ${drawnFeatures.map(f => `- ${f.type} with measurement ${f.measurement}`).join('\n')}
 Use these user-drawn areas/lines as primary areas of interest for your analysis if applicable to the query.` : ''}
 
+${calendarNotes && calendarNotes.length > 0 ? `### **Calendar Context**
+The following notes are relevant to this session:
+${calendarNotes.map(n => `- [${new Date(n.date).toLocaleDateString()}]: ${n.content}`).join('\n')}
+` : ''}
+
 **Exploration Efficiency Directives:**
 1. **Tool First:** Always check if a tool can directly or partially answer the user's query. Use the most specific tool available.
 2. **Geospatial Priority:** For any query involving locations, places, addresses, geographical features, finding businesses, distances, or directions → you **MUST** use the 'geospatialQueryTool'.
@@ -86,7 +92,8 @@ export async function researcher(
   messages: CoreMessage[],
   mapProvider: MapProvider,
   useSpecificModel?: boolean,
-  drawnFeatures?: DrawnFeature[]
+  drawnFeatures?: DrawnFeature[],
+  calendarNotes?: CalendarNote[]
 ) {
   let fullResponse = ''
   let hasError = false
@@ -102,7 +109,7 @@ export async function researcher(
   const systemPromptToUse =
     dynamicSystemPrompt?.trim()
       ? dynamicSystemPrompt
-      : getDefaultSystemPrompt(currentDate, drawnFeatures)
+      : getDefaultSystemPrompt(currentDate, drawnFeatures, calendarNotes)
 
   // Check if any message contains an image
   const hasImage = messages.some(message =>
@@ -133,7 +140,7 @@ export async function researcher(
   })
 
   const result = await nonexperimental_streamText({
-    model: (await getModel(hasImage)) as LanguageModel,
+    model: (await getModel(hasImage)) as any,
     maxTokens: 4096,
     system: systemPromptToUse,
     messages,
@@ -185,5 +192,9 @@ export async function researcher(
     messages.push({ role: 'tool', content: toolResponses })
   }
 
+  // Log token usage
+  const usage = await result.usage;
+  console.log('AI Generation Usage:', usage);
+
   return { result, fullResponse, hasError, toolResponses }
 }

lib/agents/resolution-search.tsx

@@ -144,7 +144,7 @@ Analyze the user's prompt and the image to provide a holistic understanding of t
 
   // Use streamObject to get partial results.
   return streamObject({
-    model: await getModel(hasImage),
+    model: (await getModel(hasImage)) as any,
     system: systemPrompt,
     messages: filteredMessages,
     schema: resolutionSearchSchema,

lib/agents/task-manager.tsx

@@ -16,7 +16,7 @@ export async function taskManager(messages: CoreMessage[]) {
     }
 
     const result = await generateObject({
-      model: (await getModel()) as LanguageModel,
+      model: (await getModel()) as any,
       system: `As a planet computer, your primary objective is to act as an efficient **Task Manager** for the user's query. Your goal is to minimize unnecessary steps and maximize the efficiency of the subsequent exploration phase (researcher agent).
 
 	    You must first analyze the user's input and determine the optimal course of action. You have two options at your disposal:

lib/agents/tools/calendar.tsx

@@ -0,0 +1,44 @@
+import { tool } from 'ai'
+import { calendarSchema } from '@/lib/schema/calendar'
+import { saveNote, getChatNotes } from '@/lib/actions/calendar'
+import { ToolProps } from '.'
+import { Section } from '@/components/section'
+import { BotMessage } from '@/components/message'
+import { nanoid } from '@/lib/utils'
+
+export const calendarTool = ({ uiStream }: ToolProps) =>
+  tool({
+    description: 'Create reminders, add notes to the calendar, or list existing notes for the current chat.',
+    parameters: calendarSchema,
+    execute: async ({ action, content, date, location }) => {
+      // In a real execution, we need the chatId.
+      // Since tools are called within the submit action context,
+      // we might need to find a way to pass chatId here if it's not readily available.
+      // For this implementation, we'll assume the AI state provides it or we handle it in app/actions.
+
+      if (action === 'create_note' && content) {
+        // We'll return a result that the caller (submit action) can use to actually save
+        return {
+          type: 'CALENDAR_ACTION',
+          action: 'create',
+          note: {
+            content,
+            date: date ? new Date(date) : new Date(),
+            locationTags: location ? {
+              type: 'Point',
+              coordinates: [location.longitude, location.latitude]
+            } : null
+          }
+        }
+      }
+
+      if (action === 'list_notes') {
+        return {
+          type: 'CALENDAR_ACTION',
+          action: 'list'
+        }
+      }
+
+      return { error: 'Invalid calendar action' }
+    }
+  })

lib/agents/tools/index.tsx

@@ -3,6 +3,7 @@ import { retrieveTool } from './retrieve'
 import { searchTool } from './search'
 import { videoSearchTool } from './video-search'
 import { geospatialTool } from './geospatial' // Removed useGeospatialToolMcp import
+import { calendarTool } from './calendar'
 
 import { MapProvider } from '@/lib/store/settings'
 
@@ -21,6 +22,10 @@ export const getTools = ({ uiStream, fullResponse, mapProvider }: ToolProps) =>
     geospatialQueryTool: geospatialTool({
       uiStream,
       mapProvider
+    }),
+    calendarTool: calendarTool({
+      uiStream,
+      fullResponse
     })
   }
 

lib/agents/writer.tsx

@@ -32,7 +32,7 @@ export async function writer(
   const systemToUse = dynamicSystemPrompt && dynamicSystemPrompt.trim() !== '' ? dynamicSystemPrompt : default_system_prompt;
 
   const result = await nonexperimental_streamText({
-    model: (await getModel()) as LanguageModel,
+    model: (await getModel()) as any,
     maxTokens: 2500,
     system: systemToUse, // Use the dynamic or default system prompt
     messages