From 0ae859b2fdcfc7e39b60957ee3e563c1855434ad Mon Sep 17 00:00:00 2001
From: Freshcorn-Bryce <26725654+brycecf@users.noreply.github.com>
Date: Sat, 5 Oct 2024 23:51:05 -0400
Subject: [PATCH] Consolidated mock exception

---
 .../tests/test_llms_bedrock_converse.py                    | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/llama-index-integrations/llms/llama-index-llms-bedrock-converse/tests/test_llms_bedrock_converse.py b/llama-index-integrations/llms/llama-index-llms-bedrock-converse/tests/test_llms_bedrock_converse.py
index 6a49c9ce8994e..b47266ee5236d 100644
--- a/llama-index-integrations/llms/llama-index-llms-bedrock-converse/tests/test_llms_bedrock_converse.py
+++ b/llama-index-integrations/llms/llama-index-llms-bedrock-converse/tests/test_llms_bedrock_converse.py
@@ -20,12 +20,9 @@
 prompt = "Test"
 
 
-class MockThrottlingException(Exception):
-    pass
-
-
 class MockExceptions:
-    ThrottlingException = MockThrottlingException
+    class ThrottlingException(Exception):
+        pass
 
 
 class AsyncMockClient: