[tools] use calender list tools and add date to prompt

deepu105 · deepu105 · commit e9ab9edc369e · 2025-08-28T14:42:36.000+02:00
diff --git a/py-langchain/backend/app/agents/assistant0.py b/py-langchain/backend/app/agents/assistant0.py
@@ -5,14 +5,23 @@
 from app.agents.tools.google_calendar import list_upcoming_events
 from app.agents.tools.user_info import get_user_info
 from app.agents.tools.context_docs import get_context_docs
+from datetime import date
 
 
 tools = [get_user_info, list_upcoming_events, shop_online, get_context_docs]
 
 llm = ChatOpenAI(model="gpt-4.1-mini")
 
+def get_prompt():
+    today_str = date.today().strftime('%Y-%m-%d')
+    return (
+        f"You are a personal assistant named Assistant0. You are a helpful assistant that can answer questions and help with tasks. "
+        f"Today's date is {today_str}. You have access to a set of tools, use the tools as needed to answer the user's question. "
+        f"Render the email body as a markdown block, do not wrap it in code blocks."
+    )
+
 agent = create_react_agent(
     llm,
     tools=ToolNode(tools, handle_tool_errors=False),
-    prompt="You are a personal assistant named Assistant0. You are a helpful assistant that can answer questions and help with tasks. You have access to a set of tools, use the tools as needed to answer the user's question. Render the email body as a markdown block, do not wrap it in code blocks.",
+    prompt=get_prompt(),
 )
diff --git a/ts-langchain/src/lib/agent.ts b/ts-langchain/src/lib/agent.ts
@@ -10,7 +10,7 @@ import { getAccessToken, withGoogleConnection, withAsyncAuthorization } from './
 import { getUserInfoTool } from './tools/user-info';
 import { shopOnlineTool } from './tools/shop-online';
 import { getContextDocumentsTool } from './tools/context-docs';
-import { checkUsersCalendarTool } from './tools/google-calender';
+import { getCalendarEventsTool } from './tools/google-calender';
 
 const date = new Date().toISOString();
 
@@ -40,7 +40,7 @@ const tools = [
   withGoogleConnection(new GmailCreateDraft(gmailParams)),
   withGoogleConnection(new GoogleCalendarCreateTool(googleCalendarParams)),
   withGoogleConnection(new GoogleCalendarViewTool(googleCalendarParams)),
-  withGoogleConnection(checkUsersCalendarTool),
+  withGoogleConnection(getCalendarEventsTool),
   getUserInfoTool,
   withAsyncAuthorization(shopOnlineTool),
   getContextDocumentsTool,
diff --git a/ts-langchain/src/lib/auth0-ai.ts b/ts-langchain/src/lib/auth0-ai.ts
@@ -13,7 +13,6 @@ export const withGoogleConnection = auth0AI.withTokenForConnection({
     'https://www.googleapis.com/auth/gmail.readonly',
     'https://www.googleapis.com/auth/gmail.compose',
     'https://www.googleapis.com/auth/calendar.events',
-    'https://www.googleapis.com/auth/calendar.freebusy',
   ],
 });
 
diff --git a/ts-langchain/src/lib/tools/google-calender.ts b/ts-langchain/src/lib/tools/google-calender.ts
@@ -1,13 +1,13 @@
 import { tool } from '@langchain/core/tools';
-import { addHours, formatISO } from 'date-fns';
+import { endOfDay, formatISO, startOfDay } from 'date-fns';
 import { GaxiosError } from 'gaxios';
 import { google } from 'googleapis';
 import { z } from 'zod';
 import { FederatedConnectionError } from '@auth0/ai/interrupts';
 
 import { getAccessToken } from '../auth0-ai';
 
-export const checkUsersCalendarTool = tool(
+export const getCalendarEventsTool = tool(
   async ({ date }) => {
     // Get the access token from Auth0 AI
     const accessToken = await getAccessToken();
@@ -21,18 +21,38 @@ export const checkUsersCalendarTool = tool(
         access_token: accessToken,
       });
 
-      const response = await calendar.freebusy.query({
+      // Get events for the entire day
+      const response = await calendar.events.list({
         auth,
-        requestBody: {
-          timeMin: formatISO(date),
-          timeMax: addHours(date, 1).toISOString(),
-          timeZone: 'UTC',
-          items: [{ id: 'primary' }],
-        },
+        calendarId: 'primary',
+        timeMin: formatISO(startOfDay(date)),
+        timeMax: formatISO(endOfDay(date)),
+        singleEvents: true,
+        orderBy: 'startTime',
+        maxResults: 50,
       });
 
+      const events = response.data.items || [];
+
       return {
-        available: response.data?.calendars?.primary?.busy?.length === 0,
+        date: formatISO(date, { representation: 'date' }),
+        eventsCount: events.length,
+        events: events.map((event) => ({
+          id: event.id,
+          summary: event.summary || 'No title',
+          description: event.description,
+          startTime: event.start?.dateTime || event.start?.date,
+          endTime: event.end?.dateTime || event.end?.date,
+          location: event.location,
+          attendees:
+            event.attendees?.map((attendee) => ({
+              email: attendee.email,
+              name: attendee.displayName,
+              responseStatus: attendee.responseStatus,
+            })) || [],
+          status: event.status,
+          htmlLink: event.htmlLink,
+        })),
       };
     } catch (error) {
       if (error instanceof GaxiosError) {
@@ -45,8 +65,8 @@ export const checkUsersCalendarTool = tool(
     }
   },
   {
-    name: 'check_users_calendar_availability',
-    description: 'Check user availability on a given date time on their calendar',
+    name: 'get_calendar_events',
+    description: `Get calendar events for a given date from the user's Google Calendar`,
     schema: z.object({
       date: z.coerce.date(),
     }),
diff --git a/ts-llamaindex/src/app/api/chat/route.ts b/ts-llamaindex/src/app/api/chat/route.ts
@@ -9,7 +9,7 @@ import { errorSerializer } from '@auth0/ai-vercel/interrupts';
 import { serpApiTool } from '@/lib/tools/serpapi';
 import { getUserInfoTool } from '@/lib/tools/user-info';
 import { gmailDraftTool, gmailSearchTool } from '@/lib/tools/gmail';
-import { checkUsersCalendarTool } from '@/lib/tools/google-calender';
+import { getCalendarEventsTool } from '@/lib/tools/google-calender';
 import { shopOnlineTool } from '@/lib/tools/shop-online';
 import { getContextDocumentsTool } from '@/lib/tools/context-docs';
 
@@ -29,7 +29,7 @@ async function initializeAgent(sanitizedMessages: Message[] = []) {
       getUserInfoTool,
       gmailSearchTool,
       gmailDraftTool,
-      checkUsersCalendarTool,
+      getCalendarEventsTool,
       shopOnlineTool,
       getContextDocumentsTool,
     ];
diff --git a/ts-llamaindex/src/lib/auth0-ai.ts b/ts-llamaindex/src/lib/auth0-ai.ts
@@ -14,7 +14,7 @@ export const withGoogleConnection = auth0AI.withTokenForConnection({
   scopes: [
     'https://www.googleapis.com/auth/gmail.readonly',
     'https://www.googleapis.com/auth/gmail.compose',
-    'https://www.googleapis.com/auth/calendar.freebusy',
+    'https://www.googleapis.com/auth/calendar.events',
   ],
   refreshToken: getRefreshToken,
   credentialsContext: 'tool-call',
diff --git a/ts-llamaindex/src/lib/tools/google-calender.ts b/ts-llamaindex/src/lib/tools/google-calender.ts
@@ -1,16 +1,16 @@
 import { tool } from 'llamaindex';
-import { addHours, formatISO } from 'date-fns';
+import { endOfDay, formatISO, startOfDay } from 'date-fns';
 import { GaxiosError } from 'gaxios';
 import { google } from 'googleapis';
 import { z } from 'zod';
 import { FederatedConnectionError } from '@auth0/ai/interrupts';
 
 import { getAccessToken, withGoogleConnection } from '../auth0-ai';
 
-export const checkUsersCalendarTool = withGoogleConnection(
+export const getCalendarEventsTool = withGoogleConnection(
   tool({
-    name: 'checkUsersCalendar',
-    description: 'Check user availability on a given date time on their calendar',
+    name: 'getCalendarEvents',
+    description: `Get calendar events for a given date from the user's Google Calendar`,
     parameters: z.object({
       date: z.coerce.date(),
     }),
@@ -27,18 +27,38 @@ export const checkUsersCalendarTool = withGoogleConnection(
           access_token: accessToken,
         });
 
-        const response = await calendar.freebusy.query({
+        // Get events for the entire day
+        const response = await calendar.events.list({
           auth,
-          requestBody: {
-            timeMin: formatISO(date),
-            timeMax: addHours(date, 1).toISOString(),
-            timeZone: 'UTC',
-            items: [{ id: 'primary' }],
-          },
+          calendarId: 'primary',
+          timeMin: formatISO(startOfDay(date)),
+          timeMax: formatISO(endOfDay(date)),
+          singleEvents: true,
+          orderBy: 'startTime',
+          maxResults: 50,
         });
 
+        const events = response.data.items || [];
+
         return {
-          available: response.data?.calendars?.primary?.busy?.length === 0,
+          date: formatISO(date, { representation: 'date' }),
+          eventsCount: events.length,
+          events: events.map((event) => ({
+            id: event.id || '',
+            summary: event.summary || 'No title',
+            description: event.description || '',
+            startTime: event.start?.dateTime || event.start?.date || '',
+            endTime: event.end?.dateTime || event.end?.date || '',
+            location: event.location || '',
+            attendees:
+              event.attendees?.map((attendee) => ({
+                email: attendee.email || '',
+                name: attendee.displayName || '',
+                responseStatus: attendee.responseStatus || '',
+              })) || [],
+            status: event.status || '',
+            htmlLink: event.htmlLink || '',
+          })),
         };
       } catch (error) {
         if (error instanceof GaxiosError) {
diff --git a/ts-vercel-ai/src/app/api/chat/route.ts b/ts-vercel-ai/src/app/api/chat/route.ts
@@ -7,7 +7,7 @@ import { errorSerializer, withInterruptions } from '@auth0/ai-vercel/interrupts'
 import { serpApiTool } from '@/lib/tools/serpapi';
 import { getUserInfoTool } from '@/lib/tools/user-info';
 import { gmailDraftTool, gmailSearchTool } from '@/lib/tools/gmail';
-import { checkUsersCalendarTool } from '@/lib/tools/google-calender';
+import { getCalendarEventsTool } from '@/lib/tools/google-calender';
 import { shopOnlineTool } from '@/lib/tools/shop-online';
 import { getContextDocumentsTool } from '@/lib/tools/context-docs';
 
@@ -30,7 +30,7 @@ export async function POST(req: NextRequest) {
     getUserInfoTool,
     gmailSearchTool,
     gmailDraftTool,
-    checkUsersCalendarTool,
+    getCalendarEventsTool,
     shopOnlineTool,
     getContextDocumentsTool,
   };
diff --git a/ts-vercel-ai/src/lib/tools/google-calender.ts b/ts-vercel-ai/src/lib/tools/google-calender.ts
@@ -1,15 +1,15 @@
 import { tool } from 'ai';
-import { addHours, formatISO } from 'date-fns';
+import { endOfDay, formatISO, startOfDay } from 'date-fns';
 import { GaxiosError } from 'gaxios';
 import { google } from 'googleapis';
 import { z } from 'zod';
 import { FederatedConnectionError } from '@auth0/ai/interrupts';
 
 import { getAccessToken, withGoogleConnection } from '../auth0-ai';
 
-export const checkUsersCalendarTool = withGoogleConnection(
+export const getCalendarEventsTool = withGoogleConnection(
   tool({
-    description: 'Check user availability on a given date time on their calendar',
+    description: `Get calendar events for a given date from the user's Google Calendar`,
     parameters: z.object({
       date: z.coerce.date(),
     }),
@@ -26,18 +26,38 @@ export const checkUsersCalendarTool = withGoogleConnection(
           access_token: accessToken,
         });
 
-        const response = await calendar.freebusy.query({
+        // Get events for the entire day
+        const response = await calendar.events.list({
           auth,
-          requestBody: {
-            timeMin: formatISO(date),
-            timeMax: addHours(date, 1).toISOString(),
-            timeZone: 'UTC',
-            items: [{ id: 'primary' }],
-          },
+          calendarId: 'primary',
+          timeMin: formatISO(startOfDay(date)),
+          timeMax: formatISO(endOfDay(date)),
+          singleEvents: true,
+          orderBy: 'startTime',
+          maxResults: 50,
         });
 
+        const events = response.data.items || [];
+
         return {
-          available: response.data?.calendars?.primary?.busy?.length === 0,
+          date: formatISO(date, { representation: 'date' }),
+          eventsCount: events.length,
+          events: events.map((event) => ({
+            id: event.id,
+            summary: event.summary || 'No title',
+            description: event.description,
+            startTime: event.start?.dateTime || event.start?.date,
+            endTime: event.end?.dateTime || event.end?.date,
+            location: event.location,
+            attendees:
+              event.attendees?.map((attendee) => ({
+                email: attendee.email,
+                name: attendee.displayName,
+                responseStatus: attendee.responseStatus,
+              })) || [],
+            status: event.status,
+            htmlLink: event.htmlLink,
+          })),
         };
       } catch (error) {
         if (error instanceof GaxiosError) {
