From 84f1c80bb72f5095caf51935fbe2ec5e4c8ecf74 Mon Sep 17 00:00:00 2001
From: Nate Kelley <nate@buster.so>
Date: Sat, 19 Apr 2025 22:16:57 -0600
Subject: [PATCH] invalidate metric on new message

---
 web/src/api/asset_interfaces/chat/chatMessageInterfaces.ts | 1 -
 web/src/context/Chats/useChatStreamMessage.ts              | 6 ++++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/web/src/api/asset_interfaces/chat/chatMessageInterfaces.ts b/web/src/api/asset_interfaces/chat/chatMessageInterfaces.ts
index 7a0839da3..4a064f6c2 100644
--- a/web/src/api/asset_interfaces/chat/chatMessageInterfaces.ts
+++ b/web/src/api/asset_interfaces/chat/chatMessageInterfaces.ts
@@ -46,7 +46,6 @@ export type BusterChatResponseMessage_file = {
   file_type: FileType;
   file_name: string;
   version_number: number;
-  version_id: string;
   filter_version_id: string | null;
   metadata?: BusterChatResponseMessage_fileMetadata[];
 };
diff --git a/web/src/context/Chats/useChatStreamMessage.ts b/web/src/context/Chats/useChatStreamMessage.ts
index 71a71c16c..2fa30a8be 100644
--- a/web/src/context/Chats/useChatStreamMessage.ts
+++ b/web/src/context/Chats/useChatStreamMessage.ts
@@ -75,6 +75,11 @@ export const useChatStreamMessage = () => {
               { id: responseMessage.id, version_number: responseMessage.version_number },
               queryClient
             );
+            const options = queryKeys.metricsGetMetric(
+              responseMessage.id,
+              responseMessage.version_number
+            );
+            queryClient.invalidateQueries({ queryKey: options.queryKey });
           }
         });
       }
@@ -88,6 +93,7 @@ export const useChatStreamMessage = () => {
     normalizeChatMessage(iChatMessages);
     onUpdateChat(iChat);
     prefetchLastMessageMetricData(iChat, iChatMessages);
+
     queryClient.invalidateQueries({
       queryKey: [queryKeys.chatsGetList().queryKey, queryKeys.metricsGetList().queryKey]
     });