From a957c5f053d2cd439f8bafb7e02123afdfcaf115 Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Mon, 28 Apr 2025 21:23:16 -0700
Subject: [PATCH] Update chat view decorators

---
 src/documents/views.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/documents/views.py b/src/documents/views.py
index fdd04d47c..5eec8c43c 100644
--- a/src/documents/views.py
+++ b/src/documents/views.py
@@ -1140,8 +1140,14 @@ class DocumentViewSet(
             )
 
 
-@method_decorator(ensure_csrf_cookie, name="dispatch")
-@method_decorator(login_required, name="dispatch")
+@method_decorator(
+    [
+        ensure_csrf_cookie,
+        login_required,
+        cache_control(no_cache=True),
+    ],
+    name="dispatch",
+)
 class ChatStreamingView(View):
     def post(self, request):
         request.compress_exempt = True
@@ -1177,7 +1183,6 @@ class ChatStreamingView(View):
             stream_chat_with_documents(query_str=question, documents=documents),
             content_type="text/event-stream",
         )
-        response["Cache-Control"] = "no-cache"
         return response