mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-11 03:38:38 +00:00
test(proxy): let the prompt cache observer finish on failure events
This commit is contained in:
parent
20a4df16d8
commit
670c1fb9ba
1 changed files with 3 additions and 0 deletions
|
|
@ -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 {
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue