From 9a750fba74d4d4fe661af990c52d64017416bce4 Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Sat, 21 Mar 2026 01:44:54 +0530
Subject: [PATCH] feat: update Google Calendar tool actions in chat streaming

---
 surfsense_backend/app/tasks/chat/stream_new_chat.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/surfsense_backend/app/tasks/chat/stream_new_chat.py b/surfsense_backend/app/tasks/chat/stream_new_chat.py
index 0d72d8a2c..e33b67902 100644
--- a/surfsense_backend/app/tasks/chat/stream_new_chat.py
+++ b/surfsense_backend/app/tasks/chat/stream_new_chat.py
@@ -874,11 +874,12 @@ async def _stream_agent_events(
                 "create_google_drive_file",
                 "delete_google_drive_file",
                 "create_gmail_draft",
+                "update_gmail_draft",
                 "send_gmail_email",
                 "trash_gmail_email",
-                "create_google_calendar_event",
-                "update_google_calendar_event",
-                "delete_google_calendar_event",
+                "create_calendar_event",
+                "update_calendar_event",
+                "delete_calendar_event",
             ):
                 yield streaming_service.format_tool_output_available(
                     tool_call_id,