From 670c1fb9ba912f4052bbabadd536ad87004ac5f8 Mon Sep 17 00:00:00 2001
From: Deepanshu Pal <40927968+DeepanshuPal@users.noreply.github.com>
Date: Fri, 2 Oct 2026 12:45:31 +0530
Subject: [PATCH] test(proxy): let the prompt cache observer finish on failure
 events

---
 tests/unit/proxy/hooks/test_prompt_cache_observer.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/unit/proxy/hooks/test_prompt_cache_observer.py b/tests/unit/proxy/hooks/test_prompt_cache_observer.py
index 82af3e9a6ef..b5fcad3add4 100644
--- a/tests/unit/proxy/hooks/test_prompt_cache_observer.py
+++ b/tests/unit/proxy/hooks/test_prompt_cache_observer.py
@@ -204,6 +204,9 @@ class RecordingObserver(PromptCacheObserver):
         await super().async_log_success_event(kwargs, response_obj, start_time, end_time)
         self.finished.set()
 
+    async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time):
+        self.finished.set()
+
 
 def native_response():
     return {