From 04efdd66c089bfcd0159a07c4f3ea82bfc1f3b43 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 19 Aug 2024 18:58:31 -0700
Subject: [PATCH] fix test guardrails

---
 tests/otel_tests/test_guardrails.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/otel_tests/test_guardrails.py b/tests/otel_tests/test_guardrails.py
index b82bc57cc..c48a5ba79 100644
--- a/tests/otel_tests/test_guardrails.py
+++ b/tests/otel_tests/test_guardrails.py
@@ -41,7 +41,7 @@ async def chat_completion(
         print()
 
         if status != 200:
-            raise Exception(f"Request did not return a 200 status code: {status}")
+            return response_text
 
         # response headers
         response_headers = response.headers